Ready-to-use SRT / WebRTC / RTSP / RTMP / LL-HLS media server and media proxy that allows to read, publish, proxy, record and playback video and audio streams.
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

66 lines
1.4 KiB

package core
import (
"github.com/aler9/gortsplib/v2"
"github.com/aler9/gortsplib/v2/pkg/format"
"github.com/aler9/gortsplib/v2/pkg/media"
"github.com/aler9/rtsp-simple-server/internal/formatprocessor"
)
type stream struct {
bytesReceived *uint64
rtspStream *gortsplib.ServerStream
smedias map[*media.Media]*streamMedia
}
func newStream(
medias media.Medias,
generateRTPPackets bool,
bytesReceived *uint64,
) (*stream, error) {
s := &stream{
bytesReceived: bytesReceived,
rtspStream: gortsplib.NewServerStream(medias),
}
s.smedias = make(map[*media.Media]*streamMedia)
for _, media := range s.rtspStream.Medias() {
var err error
s.smedias[media], err = newStreamMedia(media, generateRTPPackets)
if err != nil {
return nil, err
}
}
return s, nil
}
func (s *stream) close() {
s.rtspStream.Close()
}
func (s *stream) medias() media.Medias {
return s.rtspStream.Medias()
}
func (s *stream) readerAdd(r reader, medi *media.Media, forma format.Format, cb func(formatprocessor.Unit)) {
sm := s.smedias[medi]
sf := sm.formats[forma]
sf.readerAdd(r, cb)
}
func (s *stream) readerRemove(r reader) {
for _, sm := range s.smedias {
for _, sf := range sm.formats {
sf.readerRemove(r)
}
}
}
func (s *stream) writeData(medi *media.Media, forma format.Format, data formatprocessor.Unit) error {
sm := s.smedias[medi]
sf := sm.formats[forma]
return sf.writeData(s, medi, data)
}