Alexey Khit
58 lines
1.1 KiB
Go
58 lines
1.1 KiB
Go
package mp4
|
|
|
|
import (
|
|
"github.com/AlexxIT/go2rtc/cmd/api"
|
|
"github.com/AlexxIT/go2rtc/cmd/streams"
|
|
"github.com/AlexxIT/go2rtc/pkg/mp4"
|
|
"github.com/AlexxIT/go2rtc/pkg/streamer"
|
|
)
|
|
|
|
const MsgTypeMSE = "mse" // fMP4
|
|
|
|
const packetSize = 8192
|
|
|
|
func handlerWS(ctx *api.Context, msg *streamer.Message) {
|
|
src := ctx.Request.URL.Query().Get("src")
|
|
stream := streams.GetOrNew(src)
|
|
if stream == nil {
|
|
return
|
|
}
|
|
|
|
cons := &mp4.Consumer{}
|
|
cons.UserAgent = ctx.Request.UserAgent()
|
|
cons.RemoteAddr = ctx.Request.RemoteAddr
|
|
|
|
cons.Listen(func(msg interface{}) {
|
|
if data, ok := msg.([]byte); ok {
|
|
for len(data) > packetSize {
|
|
ctx.Write(data[:packetSize])
|
|
data = data[packetSize:]
|
|
}
|
|
ctx.Write(data)
|
|
}
|
|
})
|
|
|
|
if err := stream.AddConsumer(cons); err != nil {
|
|
log.Warn().Err(err).Caller().Send()
|
|
ctx.Error(err)
|
|
return
|
|
}
|
|
|
|
ctx.OnClose(func() {
|
|
stream.RemoveConsumer(cons)
|
|
})
|
|
|
|
ctx.Write(&streamer.Message{Type: MsgTypeMSE, Value: cons.MimeType()})
|
|
|
|
data, err := cons.Init()
|
|
if err != nil {
|
|
log.Warn().Err(err).Caller().Send()
|
|
ctx.Error(err)
|
|
return
|
|
}
|
|
|
|
ctx.Write(data)
|
|
|
|
cons.Start()
|
|
}
|