Alexey Khit
48 lines
944 B
Go
48 lines
944 B
Go
package fake
|
|
|
|
import (
|
|
"github.com/AlexxIT/go2rtc/pkg/streamer"
|
|
"github.com/pion/rtp"
|
|
"time"
|
|
)
|
|
|
|
type Consumer struct {
|
|
streamer.Element
|
|
Medias []*streamer.Media
|
|
Tracks []*streamer.Track
|
|
|
|
RecvPackets int
|
|
SendPackets int
|
|
}
|
|
|
|
func (c *Consumer) GetMedias() []*streamer.Media {
|
|
return c.Medias
|
|
}
|
|
|
|
func (c *Consumer) AddTrack(media *streamer.Media, track *streamer.Track) *streamer.Track {
|
|
switch track.Direction {
|
|
case streamer.DirectionSendonly:
|
|
track = track.Bind(func(packet *rtp.Packet) error {
|
|
if track.Codec.PayloadType != packet.PayloadType {
|
|
panic("wrong payload type")
|
|
}
|
|
c.RecvPackets++
|
|
return nil
|
|
})
|
|
case streamer.DirectionRecvonly:
|
|
go func() {
|
|
for {
|
|
pkt := &rtp.Packet{}
|
|
pkt.PayloadType = track.Codec.PayloadType
|
|
if err := track.WriteRTP(pkt); err != nil {
|
|
return
|
|
}
|
|
c.SendPackets++
|
|
time.Sleep(time.Second)
|
|
}
|
|
}()
|
|
}
|
|
c.Tracks = append(c.Tracks, track)
|
|
return track
|
|
}
|