mirror of
https://github.com/AlexxIT/go2rtc.git
synced 2026-04-23 08:00:11 +08:00
add webcodecs support
This commit is contained in:
@@ -0,0 +1,45 @@
|
||||
package webcodecs
|
||||
|
||||
import (
|
||||
"errors"
|
||||
|
||||
"github.com/AlexxIT/go2rtc/internal/api"
|
||||
"github.com/AlexxIT/go2rtc/internal/api/ws"
|
||||
"github.com/AlexxIT/go2rtc/internal/app"
|
||||
"github.com/AlexxIT/go2rtc/internal/streams"
|
||||
"github.com/AlexxIT/go2rtc/pkg/webcodecs"
|
||||
"github.com/rs/zerolog"
|
||||
)
|
||||
|
||||
func Init() {
|
||||
log = app.GetLogger("webcodecs")
|
||||
|
||||
ws.HandleFunc("webcodecs", handlerWS)
|
||||
}
|
||||
|
||||
var log zerolog.Logger
|
||||
|
||||
func handlerWS(tr *ws.Transport, msg *ws.Message) error {
|
||||
stream, _ := streams.GetOrPatch(tr.Request.URL.Query())
|
||||
if stream == nil {
|
||||
return errors.New(api.StreamNotFound)
|
||||
}
|
||||
|
||||
cons := webcodecs.NewConsumer(nil)
|
||||
cons.WithRequest(tr.Request)
|
||||
|
||||
if err := stream.AddConsumer(cons); err != nil {
|
||||
log.Debug().Err(err).Msg("[webcodecs] add consumer")
|
||||
return err
|
||||
}
|
||||
|
||||
tr.Write(&ws.Message{Type: "webcodecs", Value: cons.GetInitInfo()})
|
||||
|
||||
go cons.WriteTo(tr.Writer())
|
||||
|
||||
tr.OnClose(func() {
|
||||
stream.RemoveConsumer(cons)
|
||||
})
|
||||
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user