mirror of
https://github.com/Monibuca/plugin-rtsp.git
synced 2025-09-26 19:51:14 +08:00
108 lines
3.0 KiB
Go
108 lines
3.0 KiB
Go
package rtsp
|
||
|
||
import (
|
||
"net/http"
|
||
"strconv"
|
||
"sync"
|
||
"time"
|
||
|
||
"github.com/bluenviron/gortsplib/v3"
|
||
"go.uber.org/zap"
|
||
. "m7s.live/engine/v4"
|
||
"m7s.live/engine/v4/config"
|
||
"m7s.live/engine/v4/util"
|
||
)
|
||
|
||
type RTSPConfig struct {
|
||
config.Publish
|
||
config.Subscribe
|
||
config.Pull
|
||
config.Push
|
||
ListenAddr string `default:":554"`
|
||
UDPAddr string `default:":8000"`
|
||
RTCPAddr string `default:":8001"`
|
||
ReadBufferCount int `default:"2048"`
|
||
WriteBufferCount int `default:"2048"`
|
||
PullProtocol string //tcp、udp、 auto（default）
|
||
sync.Map
|
||
}
|
||
|
||
func (conf *RTSPConfig) OnEvent(event any) {
|
||
switch v := event.(type) {
|
||
case FirstConfig:
|
||
s := &gortsplib.Server{
|
||
Handler: conf,
|
||
RTSPAddress: conf.ListenAddr,
|
||
UDPRTPAddress: conf.UDPAddr,
|
||
UDPRTCPAddress: conf.RTCPAddr,
|
||
MulticastIPRange: "224.1.0.0/16",
|
||
MulticastRTPPort: 8002,
|
||
MulticastRTCPPort: 8003,
|
||
ReadBufferCount: conf.ReadBufferCount,
|
||
WriteBufferCount: conf.WriteBufferCount,
|
||
}
|
||
if err := s.Start(); err != nil {
|
||
RTSPPlugin.Error("server start", zap.Error(err))
|
||
v["enable"] = false
|
||
}
|
||
for streamPath, url := range conf.PullOnStart {
|
||
if err := RTSPPlugin.Pull(streamPath, url, new(RTSPPuller), 0); err != nil {
|
||
RTSPPlugin.Error("pull", zap.String("streamPath", streamPath), zap.String("url", url), zap.Error(err))
|
||
}
|
||
}
|
||
case SEpublish:
|
||
for streamPath, url := range conf.PushList {
|
||
if streamPath == v.Target.Path {
|
||
if err := RTSPPlugin.Push(streamPath, url, new(RTSPPusher), false); err != nil {
|
||
RTSPPlugin.Error("push", zap.String("streamPath", streamPath), zap.String("url", url), zap.Error(err))
|
||
}
|
||
}
|
||
}
|
||
case *Stream: //按需拉流
|
||
for streamPath, url := range conf.PullOnSub {
|
||
if streamPath == v.Path {
|
||
if err := RTSPPlugin.Pull(streamPath, url, new(RTSPPuller), 0); err != nil {
|
||
RTSPPlugin.Error("pull", zap.String("streamPath", streamPath), zap.String("url", url), zap.Error(err))
|
||
}
|
||
break
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
var rtspConfig = &RTSPConfig{}
|
||
var RTSPPlugin = InstallPlugin(rtspConfig)
|
||
|
||
func filterStreams() (ss []*Stream) {
|
||
Streams.Range(func(key string, s *Stream) {
|
||
switch s.Publisher.(type) {
|
||
case *RTSPPublisher, *RTSPPuller:
|
||
ss = append(ss, s)
|
||
}
|
||
})
|
||
return
|
||
}
|
||
|
||
func (*RTSPConfig) API_list(w http.ResponseWriter, r *http.Request) {
|
||
util.ReturnJson(filterStreams, time.Second, w, r)
|
||
}
|
||
|
||
func (*RTSPConfig) API_Pull(rw http.ResponseWriter, r *http.Request) {
|
||
save, _ := strconv.Atoi(r.URL.Query().Get("save"))
|
||
err := RTSPPlugin.Pull(r.URL.Query().Get("streamPath"), r.URL.Query().Get("target"), new(RTSPPuller), save)
|
||
if err != nil {
|
||
http.Error(rw, err.Error(), http.StatusBadRequest)
|
||
} else {
|
||
rw.Write([]byte("ok"))
|
||
}
|
||
}
|
||
|
||
func (*RTSPConfig) API_Push(rw http.ResponseWriter, r *http.Request) {
|
||
err := RTSPPlugin.Push(r.URL.Query().Get("streamPath"), r.URL.Query().Get("target"), new(RTSPPusher), r.URL.Query().Has("save"))
|
||
if err != nil {
|
||
http.Error(rw, err.Error(), http.StatusBadRequest)
|
||
} else {
|
||
rw.Write([]byte("ok"))
|
||
}
|
||
}
|