Add a context parameter to the sample provider NextSample interface (#…

…343)
livekit · Nov 13, 2023 · 7e90bc6 · 7e90bc6
1 parent a3b0f9a
commit 7e90bc6
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/localsampletrack.go b/localsampletrack.go
@@ -429,7 +429,7 @@ func (s *LocalSampleTrack) writeWorker(provider SampleProvider, onComplete func(
 	defer ticker.Stop()
 
 	for {
-		sample, err := provider.NextSample()
+		sample, err := provider.NextSample(ctx)
 		if err == io.EOF {
 			return
 		}

diff --git a/readersampleprovider.go b/readersampleprovider.go
@@ -15,6 +15,7 @@
 package lksdk
 
 import (
+	"context"
 	"io"
 	"os"
 	"path/filepath"
@@ -213,7 +214,7 @@ func (p *ReaderSampleProvider) CurrentAudioLevel() uint8 {
 	return p.AudioLevel
 }
 
-func (p *ReaderSampleProvider) NextSample() (media.Sample, error) {
+func (p *ReaderSampleProvider) NextSample(ctx context.Context) (media.Sample, error) {
 	sample := media.Sample{}
 	switch p.Mime {
 	case webrtc.MimeTypeH264:

diff --git a/sampleprovider.go b/sampleprovider.go
@@ -15,13 +15,14 @@
 package lksdk
 
 import (
+	"context"
 	"time"
 
 	"github.com/pion/webrtc/v3/pkg/media"
 )
 
 type SampleProvider interface {
-	NextSample() (media.Sample, error)
+	NextSample(context.Context) (media.Sample, error)
 	OnBind() error
 	OnUnbind() error
 	Close() error
@@ -62,7 +63,7 @@ func NewNullSampleProvider(bitrate uint32) *NullSampleProvider {
 	}
 }
 
-func (p *NullSampleProvider) NextSample() (media.Sample, error) {
+func (p *NullSampleProvider) NextSample(ctx context.Context) (media.Sample, error) {
 	return media.Sample{
 		Data:     make([]byte, p.BytesPerSample),
 		Duration: p.SampleDuration,