Ping the signaling server from the client

internal/signaling/handler.go

@@ -155,12 +155,16 @@ func Handler(ctx context.Context, store stores.Store, cloudflare *cloudflare.Cre
 				}
 				go metrics.RecordEvent(ctx, params)
 
-			case "pong":
+			case "ping", "pong":
 				// ignore, ping/pong is just for the tcp keepalive.
 
 			default:
 				if err := peer.HandlePacket(ctx, base.Type, raw); err != nil {
-					util.ErrorAndDisconnect(ctx, conn, err)
+					if err == ErrUnknownPacketType {
+						logger.Warn("unknown packet type received", zap.String("type", base.Type), zap.String("peer", peer.ID), zap.String("game", peer.Game), zap.String("origin", r.Header.Get("Origin")))
+					} else {
+						util.ErrorAndDisconnect(ctx, conn, err)
+					}
 				}
 			}
 		}

internal/signaling/peer.go

@@ -17,6 +17,8 @@ import (
 
 const DefaultMaxPlayers = 4
 
+var ErrUnknownPacketType = fmt.Errorf("unknown packet type")
+
 type Peer struct {
 	store stores.Store
 	conn  *websocket.Conn
@@ -167,7 +169,7 @@ func (p *Peer) HandlePacket(ctx context.Context, typ string, raw []byte) error {
 		}
 
 	default:
-		logger.Warn("unknown packet type received", zap.String("type", typ))
+		return ErrUnknownPacketType
 	}
 
 	return nil

lib/signaling.ts

@@ -33,6 +33,12 @@ export default class Signaling extends EventEmitter<SignalingListeners> {
     this.replayQueue = new Map()
 
     this.ws = this.connect()
+
+    // Send a ping every 5 seconds to keep the connection alive,
+    // and to detect when the connection is lost.
+    setInterval(() => {
+      this.ping()
+    }, 5000)
   }
 
   private connect (): WebSocket {
@@ -141,6 +147,19 @@ export default class Signaling extends EventEmitter<SignalingListeners> {
     }
   }
 
+  ping (): void {
+    // Send a ping to the server to keep the connection alive,
+    // and to detect when the connection is lost.
+    // Sending is enough, we don't need to listen for a pong.
+    // Don't use this.send() as we don't need every ping to be logged.
+    if (this.ws.readyState === WebSocket.OPEN) {
+      const data = JSON.stringify({
+        type: 'ping'
+      })
+      this.ws.send(data)
+    }
+  }
+
   private async handleSignalingMessage (data: string): Promise<void> {
     try {
       const packet = JSON.parse(data) as SignalingPacketTypes