Merge pull request #31 from make-software/fix-ping-reconnect

fix: Fix ping reconnect for event handler
make-software · Nov 21, 2024 · 77d1481 · 77d1481
2 parents eea7f18 + d1f5286
commit 77d1481
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/server/src/config.ts b/server/src/config.ts
@@ -11,6 +11,7 @@ interface Config {
   lotteryContractPackageHash: string;
   dbURI: string;
   clientURL: string[];
+  pingCheckIntervalInMilliseconds: number;
 }
 
 export const config: Config = {
@@ -21,4 +22,5 @@ export const config: Config = {
   lotteryContractPackageHash: process.env.LOTTERY_CONTRACT_PACKAGE_HASH as string,
   dbURI: process.env.DB_URI as string,
   clientURL: process.env.CLIENT_URL ? (process.env.CLIENT_URL as string).split(',') : [],
+  pingCheckIntervalInMilliseconds: 60000,
 };
diff --git a/server/src/event-handler.ts b/server/src/event-handler.ts
@@ -25,9 +25,22 @@ async function main() {
     console.log(`Connected to streaming API: ${config.csprCloudStreamingUrl}`);
   })
 
+  let lastPingTimestamp = new Date();
+
+  setInterval(() => {
+    const now = new Date();
+    if (now.getTime() - lastPingTimestamp.getTime() > config.pingCheckIntervalInMilliseconds) {
+      console.log(`No ping events from Streaming API for ${config.pingCheckIntervalInMilliseconds/1000} seconds, closing ws connection...`);
+      ws.close();
+      process.exit(1);
+    }
+  }, config.pingCheckIntervalInMilliseconds);
+
   ws.on('message', async (data: Buffer) => {
     const rawData = data.toString();
+
     if (rawData === 'Ping') {
+      lastPingTimestamp = new Date();
       return;
     }