149 lines
2.6 KiB
Go
149 lines
2.6 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"strings"
|
|
"sync"
|
|
)
|
|
|
|
type Handler struct {
|
|
c *Config
|
|
managers map[string]*Manager
|
|
mutex sync.RWMutex
|
|
close chan string
|
|
}
|
|
|
|
func NewHandler(c *Config) *Handler {
|
|
h := &Handler{
|
|
c: c,
|
|
managers: make(map[string]*Manager),
|
|
close: make(chan string),
|
|
}
|
|
|
|
// Recreate tempdir
|
|
os.RemoveAll(c.tempdir)
|
|
os.MkdirAll(c.tempdir, 0755)
|
|
|
|
go h.watchClose()
|
|
return h
|
|
}
|
|
|
|
func (h *Handler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|
url := r.URL.Path
|
|
parts := make([]string, 0)
|
|
|
|
for _, part := range strings.Split(url, "/") {
|
|
if part != "" {
|
|
parts = append(parts, part)
|
|
}
|
|
}
|
|
|
|
if len(parts) < 3 {
|
|
log.Println("Invalid URL", url)
|
|
w.WriteHeader(http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
streamid := parts[0]
|
|
path := "/" + strings.Join(parts[1:len(parts)-1], "/")
|
|
chunk := parts[len(parts)-1]
|
|
|
|
// log.Println("Serving", path, streamid, chunk)
|
|
|
|
if streamid == "" || chunk == "" || path == "" {
|
|
w.WriteHeader(http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
manager := h.getManager(path, streamid)
|
|
if manager == nil {
|
|
manager = h.createManager(path, streamid)
|
|
}
|
|
|
|
if manager == nil {
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
return
|
|
}
|
|
manager.ServeHTTP(w, r, chunk)
|
|
}
|
|
|
|
func (h *Handler) getManager(path string, streamid string) *Manager {
|
|
h.mutex.RLock()
|
|
defer h.mutex.RUnlock()
|
|
|
|
m := h.managers[streamid]
|
|
if m == nil || m.path != path {
|
|
return nil
|
|
}
|
|
return m
|
|
}
|
|
|
|
func (h *Handler) createManager(path string, streamid string) *Manager {
|
|
manager, err := NewManager(h.c, path, streamid, h.close)
|
|
if err != nil {
|
|
log.Println("Error creating manager", err)
|
|
return nil
|
|
}
|
|
|
|
h.mutex.Lock()
|
|
defer h.mutex.Unlock()
|
|
|
|
old := h.managers[streamid]
|
|
if old != nil {
|
|
old.Destroy()
|
|
}
|
|
|
|
h.managers[streamid] = manager
|
|
return manager
|
|
}
|
|
|
|
func (h *Handler) removeManager(streamid string) {
|
|
h.mutex.Lock()
|
|
defer h.mutex.Unlock()
|
|
delete(h.managers, streamid)
|
|
}
|
|
|
|
func (h *Handler) watchClose() {
|
|
for {
|
|
id := <-h.close
|
|
if id == "" {
|
|
return
|
|
}
|
|
h.removeManager(id)
|
|
}
|
|
}
|
|
|
|
func (h *Handler) Close() {
|
|
h.close <- ""
|
|
}
|
|
|
|
func main() {
|
|
if len(os.Args) >= 2 && os.Args[1] == "test" {
|
|
fmt.Println("test successful")
|
|
return
|
|
}
|
|
|
|
log.Println("Starting VOD server")
|
|
|
|
h := NewHandler(&Config{
|
|
ffmpeg: "ffmpeg",
|
|
ffprobe: "ffprobe",
|
|
tempdir: "/tmp/go-vod",
|
|
chunkSize: 3,
|
|
lookBehind: 5,
|
|
goalBufferMin: 3,
|
|
goalBufferMax: 8,
|
|
streamIdleTime: 300,
|
|
managerIdleTime: 600,
|
|
})
|
|
|
|
http.Handle("/", h)
|
|
http.ListenAndServe(":47788", nil)
|
|
|
|
log.Println("Exiting VOD server")
|
|
h.Close()
|
|
}
|