package main import ( "io" "net/http" "sync" "git.icomsys.co.kr/ljhwan026/joy4/av/avutil" "git.icomsys.co.kr/ljhwan026/joy4/av/pubsub" "git.icomsys.co.kr/ljhwan026/joy4/format" "git.icomsys.co.kr/ljhwan026/joy4/format/flv" "git.icomsys.co.kr/ljhwan026/joy4/format/rtmp" ) func init() { format.RegisterAll() } type writeFlusher struct { httpflusher http.Flusher io.Writer } func (self writeFlusher) Flush() error { self.httpflusher.Flush() return nil } func main() { server := &rtmp.Server{} l := &sync.RWMutex{} type Channel struct { que *pubsub.Queue } channels := map[string]*Channel{} server.HandlePlay = func(conn *rtmp.Conn) { l.RLock() ch := channels[conn.URL.Path] l.RUnlock() if ch != nil { cursor := ch.que.Latest() avutil.CopyFile(conn, cursor) } } server.HandlePublish = func(conn *rtmp.Conn) { streams, _ := conn.Streams() l.Lock() ch := channels[conn.URL.Path] if ch == nil { ch = &Channel{} ch.que = pubsub.NewQueue() ch.que.WriteHeader(streams) channels[conn.URL.Path] = ch } else { ch = nil } l.Unlock() if ch == nil { return } avutil.CopyPackets(ch.que, conn) l.Lock() delete(channels, conn.URL.Path) l.Unlock() ch.que.Close() } http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) { l.RLock() ch := channels[r.URL.Path] l.RUnlock() if ch != nil { w.Header().Set("Content-Type", "video/x-flv") w.Header().Set("Transfer-Encoding", "chunked") w.Header().Set("Access-Control-Allow-Origin", "*") w.WriteHeader(200) flusher := w.(http.Flusher) flusher.Flush() muxer := flv.NewMuxerWriteFlusher(writeFlusher{httpflusher: flusher, Writer: w}) cursor := ch.que.Latest() avutil.CopyFile(muxer, cursor) } else { http.NotFound(w, r) } }) go http.ListenAndServe(":8089", nil) server.ListenAndServe() // ffmpeg -re -i movie.flv -c copy -f flv rtmp://localhost/movie // ffmpeg -f avfoundation -i "0:0" .... -f flv rtmp://localhost/screen // ffplay http://localhost:8089/movie // ffplay http://localhost:8089/screen }