Only send PLIs for video tracks in examples

pion · Aug 15, 2024 · d54f787 · d54f787
1 parent ddb9219
commit d54f787
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 11 deletions.
diff --git a/examples/simulcast/main.go b/examples/simulcast/main.go
@@ -111,16 +111,18 @@ func main() {
 
 		// Start reading from all the streams and sending them to the related output track
 		rid := track.RID()
-		go func() {
-			ticker := time.NewTicker(3 * time.Second)
-			defer ticker.Stop()
-			for range ticker.C {
-				fmt.Printf("Sending pli for stream with rid: %q, ssrc: %d\n", track.RID(), track.SSRC())
-				if writeErr := peerConnection.WriteRTCP([]rtcp.Packet{&rtcp.PictureLossIndication{MediaSSRC: uint32(track.SSRC())}}); writeErr != nil {
-					fmt.Println(writeErr)
+		if track.Kind() == webrtc.RTPCodecTypeVideo {
+			go func() {
+				ticker := time.NewTicker(3 * time.Second)
+				defer ticker.Stop()
+				for range ticker.C {
+					fmt.Printf("Sending pli for stream with rid: %q, ssrc: %d\n", track.RID(), track.SSRC())
+					if writeErr := peerConnection.WriteRTCP([]rtcp.Packet{&rtcp.PictureLossIndication{MediaSSRC: uint32(track.SSRC())}}); writeErr != nil {
+						fmt.Println(writeErr)
+					}
 				}
-			}
-		}()
+			}()
+		}
 		for {
 			// Read RTP packets being sent to Pion
 			packet, _, readErr := track.ReadRTP()

diff --git a/examples/swap-tracks/main.go b/examples/swap-tracks/main.go
@@ -119,8 +119,10 @@ func main() { // nolint:gocognit
 				// If just switched to this track, send PLI to get picture refresh
 				if !isCurrTrack {
 					isCurrTrack = true
-					if writeErr := peerConnection.WriteRTCP([]rtcp.Packet{&rtcp.PictureLossIndication{MediaSSRC: uint32(track.SSRC())}}); writeErr != nil {
-						fmt.Println(writeErr)
+					if track.Kind() == webrtc.RTPCodecTypeVideo {
+						if writeErr := peerConnection.WriteRTCP([]rtcp.Packet{&rtcp.PictureLossIndication{MediaSSRC: uint32(track.SSRC())}}); writeErr != nil {
+							fmt.Println(writeErr)
+						}
 					}
 				}
 				packets <- rtp