better heartbeat

seamapi · Aug 27, 2023 · a1dda74 · a1dda74
1 parent 12efa21
commit a1dda74
Show file tree

Hide file tree

Showing 5 changed files with 42 additions and 5 deletions.
diff --git a/dist/worker-wrapper.js b/dist/worker-wrapper.js
diff --git a/dist/worker-wrapper.js.map b/dist/worker-wrapper.js.map
diff --git a/dist/worker-wrapper.mjs b/dist/worker-wrapper.mjs
@@ -253,7 +253,19 @@ var Worker = class {
           type: "ava-postgres",
           message: "running postgres heartbeat..."
         });
-        const { rows } = await postgresClient.query("SELECT 1");
+        const { rows } = await Promise.race([
+          postgresClient.query("SELECT 1"),
+          new Promise(
+            (_, reject) => setTimeout(
+              () => reject(
+                new Error(
+                  "postgres heartbeat timed out after 6 seconds (query took too long)"
+                )
+              ),
+              6e3
+            )
+          )
+        ]);
         parentPort.postMessage({
           type: "ava-postgres",
           message: "postgres heartbeat success",

diff --git a/dist/worker-wrapper.mjs.map b/dist/worker-wrapper.mjs.map
diff --git a/src/worker.ts b/src/worker.ts
@@ -343,7 +343,20 @@ export class Worker {
           message: "running postgres heartbeat...",
         })
 
-        const { rows } = await postgresClient.query("SELECT 1")
+        const { rows } = await Promise.race([
+          postgresClient.query("SELECT 1"),
+          new Promise<never>((_, reject) =>
+            setTimeout(
+              () =>
+                reject(
+                  new Error(
+                    "postgres heartbeat timed out after 6 seconds (query took too long)"
+                  )
+                ),
+              6000
+            )
+          ),
+        ])
 
         parentPort!.postMessage({
           type: "ava-postgres",