mirror of
https://github.com/AlexxIT/go2rtc.git
synced 2025-09-27 04:36:12 +08:00
68 lines
1.2 KiB
Go
68 lines
1.2 KiB
Go
package api
|
|
|
|
import (
|
|
"github.com/AlexxIT/go2rtc/pkg/streamer"
|
|
"github.com/gorilla/websocket"
|
|
"net/http"
|
|
"sync"
|
|
)
|
|
|
|
type WSHandler func(ctx *Context, msg *streamer.Message)
|
|
|
|
var apiWsUp = websocket.Upgrader{
|
|
ReadBufferSize: 1024,
|
|
WriteBufferSize: 512000,
|
|
}
|
|
|
|
type Context struct {
|
|
Conn *websocket.Conn
|
|
Request *http.Request
|
|
Consumer interface{} // TODO: rewrite
|
|
|
|
onClose []func()
|
|
mu sync.Mutex
|
|
}
|
|
|
|
func (ctx *Context) Upgrade(w http.ResponseWriter, r *http.Request) (err error) {
|
|
ctx.Conn, err = apiWsUp.Upgrade(w, r, nil)
|
|
ctx.Request = r
|
|
return
|
|
}
|
|
|
|
func (ctx *Context) Close() {
|
|
for _, f := range ctx.onClose {
|
|
f()
|
|
}
|
|
_ = ctx.Conn.Close()
|
|
}
|
|
|
|
func (ctx *Context) Write(msg interface{}) {
|
|
ctx.mu.Lock()
|
|
defer ctx.mu.Unlock()
|
|
|
|
var err error
|
|
|
|
switch msg := msg.(type) {
|
|
case *streamer.Message:
|
|
err = ctx.Conn.WriteJSON(msg)
|
|
case []byte:
|
|
err = ctx.Conn.WriteMessage(websocket.BinaryMessage, msg)
|
|
default:
|
|
return
|
|
}
|
|
|
|
if err != nil {
|
|
//panic(err) // TODO: fix panic
|
|
}
|
|
}
|
|
|
|
func (ctx *Context) Error(err error) {
|
|
ctx.Write(&streamer.Message{
|
|
Type: "error", Value: err.Error(),
|
|
})
|
|
}
|
|
|
|
func (ctx *Context) OnClose(f func()) {
|
|
ctx.onClose = append(ctx.onClose, f)
|
|
}
|