Heartbeat every 10 seconds

neoforged · Jan 22, 2025 · 933fb0a · 933fb0a
1 parent e7da6ee
commit 933fb0a
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 2 deletions.
diff --git a/action-runner/dist/index.js b/action-runner/dist/index.js
diff --git a/action-runner/src/main.ts b/action-runner/src/main.ts
@@ -7,7 +7,7 @@ import * as fs from 'fs/promises'
 
 import process from 'process'
 
-import WebSocket from 'ws'
+import WebSocket, {OPEN} from 'ws'
 
 let workspace: string;
 let currentCommand: Promise<ExecOutput> | null = null;
@@ -90,6 +90,7 @@ export async function setupWs(url: string, msg: (ws: WebSocket, message: any) =>
 
   ws.on('open', () => {
     console.error(`Connection opened... awaiting command`)
+    heartbeat(ws)
   })
   ws.on('close', (code, reason) => {
     console.error(`Connection closed with code ${code}: ${reason.toString()}`)
@@ -98,6 +99,13 @@ export async function setupWs(url: string, msg: (ws: WebSocket, message: any) =>
   return ws
 }
 
+function heartbeat(ws: WebSocket) {
+  if (ws.readyState == OPEN) {
+    ws.ping()
+    setTimeout(() => heartbeat(ws), 10 * 1000)
+  }
+}
+
 export function getRunURL(): string {
   return `${process.env['GITHUB_SERVER_URL']}/${process.env['GITHUB_REPOSITORY']}/actions/runs/${process.env['GITHUB_RUN_ID']}`
 }