Initial commit

pkg/fake/consumer.go

@@ -0,0 +1,47 @@
+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
+}