diff --git a/sdk-core/src/main/java/dev/restate/sdk/core/ExecutorSwitchingHandlerContextImpl.java b/sdk-core/src/main/java/dev/restate/sdk/core/ExecutorSwitchingHandlerContextImpl.java index b5a36c3e..83a3dacb 100644 --- a/sdk-core/src/main/java/dev/restate/sdk/core/ExecutorSwitchingHandlerContextImpl.java +++ b/sdk-core/src/main/java/dev/restate/sdk/core/ExecutorSwitchingHandlerContextImpl.java @@ -162,6 +162,27 @@ public CompletableFuture> rejectPromise(String key, TerminalEx .thenCompose(Function.identity()); } + @Override + public CompletableFuture> signal(String name) { + return CompletableFuture.supplyAsync(() -> super.signal(name), coreExecutor) + .thenCompose(Function.identity()); + } + + @Override + public CompletableFuture resolveSignal(String invocationId, String name, Slice payload) { + return CompletableFuture.supplyAsync( + () -> super.resolveSignal(invocationId, name, payload), coreExecutor) + .thenCompose(Function.identity()); + } + + @Override + public CompletableFuture rejectSignal( + String invocationId, String name, TerminalException reason) { + return CompletableFuture.supplyAsync( + () -> super.rejectSignal(invocationId, name, reason), coreExecutor) + .thenCompose(Function.identity()); + } + @Override public void proposeRunSuccess(int runHandle, Slice toWrite) { coreExecutor.execute(() -> super.proposeRunSuccess(runHandle, toWrite));