memories/main.go

122 lines
2.1 KiB
Go

package main
import (
"log"
"net/http"
"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),
}
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(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(streamid string) *Manager {
h.mutex.RLock()
defer h.mutex.RUnlock()
return h.managers[streamid]
}
func (h *Handler) createManager(path string, streamid string) *Manager {
h.mutex.Lock()
defer h.mutex.Unlock()
manager, err := NewManager(h.c, path, streamid, h.close)
if err != nil {
log.Println("Error creating manager", err)
return nil
}
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
}
log.Println("Closing stream", id)
h.removeManager(id)
}
}
func (h *Handler) Close() {
h.close <- ""
}
func main() {
log.Println("Starting VOD server")
h := NewHandler(&Config{
ffmpeg: "ffmpeg",
ffprobe: "ffprobe",
chunkSize: 4.0,
})
http.Handle("/", h)
http.ListenAndServe(":47788", nil)
log.Println("Exiting VOD server")
h.Close()
}