ws ping

deepch · Jul 14, 2022 · 4937383 · 4937383
1 parent 3947209
commit 4937383
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/apiHTTPMSE.go b/apiHTTPMSE.go
@@ -102,8 +102,20 @@ func HTTPAPIServerStreamMSE(ws *websocket.Conn) {
 		}
 	}()
 	noVideo := time.NewTimer(10 * time.Second)
+	pingTicker := time.NewTicker(500 * time.Millisecond)
+	defer pingTicker.Stop()
 	for {
 		select {
+		case <-pingTicker.C:
+			ws.PayloadType = websocket.PingFrame
+			err = ws.SetWriteDeadline(time.Now().Add(3 * time.Second))
+			if err != nil {
+				return
+			}
+			_, err = ws.Write(nil)
+			if err != nil {
+				return
+			}
 		case <-controlExit:
 			requestLogger.WithFields(logrus.Fields{
 				"call": "controlExit",

diff --git a/streamCore.go b/streamCore.go
@@ -261,7 +261,6 @@ func StreamServerRunStreamRTMP(streamID string, channelID string, opt *ChannelST
 		case <-Signals:
 			return 0, ErrorStreamStopRTSPSignal
 		case packetAV := <-OutgoingPacketQueue:
-
 			if preDur[packetAV.Idx] != 0 {
 				packetAV.Duration = packetAV.Time - preDur[packetAV.Idx]
 			}