From 38842f77e549fd0946e43a40c65abe0f196c3f04 Mon Sep 17 00:00:00 2001
From: Mohamed Bassem <me@mbassem.com>
Date: Mon, 24 Nov 2025 00:45:29 +0000
Subject: fix: support invocation cancellation while awaiting sempahore

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

(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 8618b8ab..06ed97f5 100644
--- a/packages/plugins/queue-restate/src/service.ts
+++ b/packages/plugins/queue-restate/src/service.ts
@@ -75,6 +75,9 @@ export function buildRestateService<T, R>(
           });
           await semaphore.release();
           if (res.error) {
+            if (res.error instanceof restate.CancelledError) {
+              throw res.error;
+            }
             lastError = res.error;
             // TODO: add backoff
             await ctx.sleep(1000);
-- 
cgit v1.2.3-70-g09d2