103 lines
1.8 KiB
Go
103 lines
1.8 KiB
Go
package main
|
|
|
|
import (
|
|
"log"
|
|
"net/http"
|
|
"strings"
|
|
"sync"
|
|
)
|
|
|
|
type Handler struct {
|
|
managers map[string]*Manager
|
|
mutex sync.RWMutex
|
|
close chan string
|
|
}
|
|
|
|
func NewHandler() *Handler {
|
|
h := &Handler{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)-2], "/")
|
|
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)
|
|
}
|
|
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 := NewManager(path, streamid, h.close)
|
|
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()
|
|
|
|
http.Handle("/", h)
|
|
http.ListenAndServe(":47788", nil)
|
|
|
|
log.Println("Exiting VOD server")
|
|
h.Close()
|
|
}
|