snapshot of https://github.com/livekit/livekit.git @ 08c5433449c106943231037d0b892895f6e92d3b (2026-09-15)

test/client/datachannel_reader.go

@@ -0,0 +1,31 @@
+package client
+
+import (
+	"time"
+
+	"github.com/livekit/livekit-server/pkg/sfu/datachannel"
+)
+
+type DataChannelReader struct {
+	bitrate *datachannel.BitrateCalculator
+	target  int
+}
+
+func NewDataChannelReader(bitrate int) *DataChannelReader {
+	return &DataChannelReader{
+		target:  bitrate,
+		bitrate: datachannel.NewBitrateCalculator(datachannel.BitrateDuration*5, datachannel.BitrateWindow, datachannel.BitrateModeWallClock),
+	}
+}
+
+func (d *DataChannelReader) Read(p []byte, sid string) {
+	for {
+		if bitrate, ok := d.bitrate.ForceBitrate(time.Now()); ok && bitrate > 0 && bitrate > d.target {
+			time.Sleep(10 * time.Millisecond)
+			d.bitrate.AddBytes(0, 0, time.Now())
+			continue
+		}
+		break
+	}
+	d.bitrate.AddBytes(len(p), 0, time.Now())
+}