From bdbce10ecd57b2547a490707231b0931e3a5def6 Mon Sep 17 00:00:00 2001
From: Yang Gao <6821729+criyle@users.noreply.github.com>
Date: Thu, 25 Jan 2024 11:26:08 +0800
Subject: [PATCH] core: websocket: use ping frame rather than ping message
 (#717)

---
 packages/hydrooj/src/service/server.ts | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/packages/hydrooj/src/service/server.ts b/packages/hydrooj/src/service/server.ts
index 4e628d23d..10d69c76e 100644
--- a/packages/hydrooj/src/service/server.ts
+++ b/packages/hydrooj/src/service/server.ts
@@ -438,8 +438,11 @@ export function Connection(
                     clean();
                     conn.terminate();
                 }
-                if (Date.now() - lastHeartbeat > 30000) conn.send('ping');
+                if (Date.now() - lastHeartbeat > 30000) conn.ping();
             }, 40000);
+            conn.on('pong', () => {
+                lastHeartbeat = Date.now();
+            });
             conn.onmessage = (e) => {
                 lastHeartbeat = Date.now();
                 if (e.data === 'pong') return;