From 6db14ac492cd5d9e26d0d986513771f14faa7fd0 Mon Sep 17 00:00:00 2001
From: Mohamed Bassem <me@mbassem.com>
Date: Mon, 15 Dec 2025 00:02:24 +0000
Subject: fix: fix restate service to return control to restate service on
 timeout

---
 packages/plugins/queue-restate/src/service.ts | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

(limited to 'packages/plugins/queue-restate/src/service.ts')

diff --git a/packages/plugins/queue-restate/src/service.ts b/packages/plugins/queue-restate/src/service.ts
index 5ba7d1df..fb7e08c2 100644
--- a/packages/plugins/queue-restate/src/service.ts
+++ b/packages/plugins/queue-restate/src/service.ts
@@ -121,7 +121,14 @@ async function runWorkerLogic<T, R>(
     ctx.run(
       `main logic`,
       async () => {
-        return await run(data);
+        return await Promise.race([
+          run(data),
+          new Promise<R>((_, reject) => {
+            data.abortSignal.addEventListener("abort", () => {
+              reject(new Error(`Job didn't complete within timeout`));
+            });
+          }),
+        ]);
       },
       {
         maxRetryAttempts: 1,
-- 
cgit v1.2.3-70-g09d2