mirror of
https://github.com/aler9/gortsplib
synced 2025-09-27 03:25:52 +08:00
111 lines
2.5 KiB
Go
111 lines
2.5 KiB
Go
package main
|
|
|
|
import (
|
|
"log"
|
|
"sync"
|
|
|
|
"github.com/bluenviron/gortsplib/v4"
|
|
"github.com/bluenviron/gortsplib/v4/pkg/base"
|
|
"github.com/bluenviron/gortsplib/v4/pkg/description"
|
|
)
|
|
|
|
type server struct {
|
|
s *gortsplib.Server
|
|
mutex sync.Mutex
|
|
stream *gortsplib.ServerStream
|
|
}
|
|
|
|
func (s *server) initialize() {
|
|
// configure the server
|
|
s.s = &gortsplib.Server{
|
|
Handler: s,
|
|
RTSPAddress: ":8554",
|
|
UDPRTPAddress: ":8000",
|
|
UDPRTCPAddress: ":8001",
|
|
MulticastIPRange: "224.1.0.0/16",
|
|
MulticastRTPPort: 8002,
|
|
MulticastRTCPPort: 8003,
|
|
}
|
|
}
|
|
|
|
// called when a connection is opened.
|
|
func (s *server) OnConnOpen(ctx *gortsplib.ServerHandlerOnConnOpenCtx) {
|
|
log.Printf("conn opened")
|
|
}
|
|
|
|
// called when a connection is closed.
|
|
func (s *server) OnConnClose(ctx *gortsplib.ServerHandlerOnConnCloseCtx) {
|
|
log.Printf("conn closed (%v)", ctx.Error)
|
|
}
|
|
|
|
// called when a session is opened.
|
|
func (s *server) OnSessionOpen(ctx *gortsplib.ServerHandlerOnSessionOpenCtx) {
|
|
log.Printf("session opened")
|
|
}
|
|
|
|
// called when a session is closed.
|
|
func (s *server) OnSessionClose(ctx *gortsplib.ServerHandlerOnSessionCloseCtx) {
|
|
log.Printf("session closed")
|
|
}
|
|
|
|
// called when receiving a DESCRIBE request.
|
|
func (s *server) OnDescribe(ctx *gortsplib.ServerHandlerOnDescribeCtx) (*base.Response, *gortsplib.ServerStream, error) {
|
|
log.Printf("describe request")
|
|
|
|
s.mutex.Lock()
|
|
defer s.mutex.Unlock()
|
|
|
|
// stream is not available yet
|
|
if s.stream == nil {
|
|
return &base.Response{
|
|
StatusCode: base.StatusNotFound,
|
|
}, nil, nil
|
|
}
|
|
|
|
return &base.Response{
|
|
StatusCode: base.StatusOK,
|
|
}, s.stream, nil
|
|
}
|
|
|
|
// called when receiving a SETUP request.
|
|
func (s *server) OnSetup(ctx *gortsplib.ServerHandlerOnSetupCtx) (*base.Response, *gortsplib.ServerStream, error) {
|
|
log.Printf("setup request")
|
|
|
|
s.mutex.Lock()
|
|
defer s.mutex.Unlock()
|
|
|
|
// stream is not available yet
|
|
if s.stream == nil {
|
|
return &base.Response{
|
|
StatusCode: base.StatusNotFound,
|
|
}, nil, nil
|
|
}
|
|
|
|
return &base.Response{
|
|
StatusCode: base.StatusOK,
|
|
}, s.stream, nil
|
|
}
|
|
|
|
// called when receiving a PLAY request.
|
|
func (s *server) OnPlay(ctx *gortsplib.ServerHandlerOnPlayCtx) (*base.Response, error) {
|
|
log.Printf("play request")
|
|
|
|
return &base.Response{
|
|
StatusCode: base.StatusOK,
|
|
}, nil
|
|
}
|
|
|
|
func (s *server) setStreamReady(desc *description.Session) *gortsplib.ServerStream {
|
|
s.mutex.Lock()
|
|
defer s.mutex.Unlock()
|
|
s.stream = gortsplib.NewServerStream(s.s, desc)
|
|
return s.stream
|
|
}
|
|
|
|
func (s *server) setStreamUnready() {
|
|
s.mutex.Lock()
|
|
defer s.mutex.Unlock()
|
|
s.stream.Close()
|
|
s.stream = nil
|
|
}
|