From b7c61e133b11380e8327fffebcc4768f736481b3 Mon Sep 17 00:00:00 2001
From: undefined <i@undefined.moe>
Date: Tue, 28 Mar 2023 22:32:40 +0800
Subject: [PATCH] core: websocket: allow any message as heartbeat

---
 packages/hydrooj/src/service/server.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/packages/hydrooj/src/service/server.ts b/packages/hydrooj/src/service/server.ts
index d0f5d9e3c..de1fd03ff 100644
--- a/packages/hydrooj/src/service/server.ts
+++ b/packages/hydrooj/src/service/server.ts
@@ -388,9 +388,9 @@ export function Connection(
                 if (Date.now() - lastHeartbeat > 40000) conn.close(4000, 'Heartbeat timeout');
             }, 40000);
             conn.onmessage = (e) => {
+                lastHeartbeat = Date.now();
                 if (e.data === 'ping') {
                     conn.send('pong');
-                    lastHeartbeat = Date.now();
                     return;
                 }
                 h.message?.(JSON.parse(e.data.toString()));