直播：PLI 关键帧请求、观众 RTCP 读取、断线自动重连

Made-with: Cursor

server/pkg/weblive/ws.go

@@ -5,12 +5,14 @@ import (
 	"log"
 	"net/http"
 	"sync"
+	"time"
 
 	"yh_web/server/handlers"
 
 	"github.com/gin-gonic/gin"
 	"github.com/google/uuid"
 	"github.com/gorilla/websocket"
+	"github.com/pion/rtcp"
 	"github.com/pion/webrtc/v3"
 )
 
@@ -117,6 +119,20 @@ func handlePublisherWS(c *gin.Context, h *Hub) {
 	pc.OnTrack(func(track *webrtc.TrackRemote, _ *webrtc.RTPReceiver) {
 		log.Printf("weblive: publisher track kind=%s", track.Kind().String())
 		h.onPublisherTrack(track)
+		if track.Kind() == webrtc.RTPCodecTypeVideo {
+			go func(ssrc uint32) {
+				_ = pc.WriteRTCP([]rtcp.Packet{&rtcp.PictureLossIndication{MediaSSRC: ssrc}})
+				ticker := time.NewTicker(2 * time.Second)
+				defer ticker.Stop()
+				for range ticker.C {
+					st := pc.ConnectionState()
+					if st == webrtc.PeerConnectionStateClosed || st == webrtc.PeerConnectionStateFailed {
+						return
+					}
+					_ = pc.WriteRTCP([]rtcp.Packet{&rtcp.PictureLossIndication{MediaSSRC: ssrc}})
+				}
+			}(uint32(track.SSRC()))
+		}
 	})
 
 	pc.OnConnectionStateChange(func(s webrtc.PeerConnectionState) {