From fa645c506b9f7fa90aaee44f9310e7c4483b5224 Mon Sep 17 00:00:00 2001
From: Sam Whittle <samuelw@google.com>
Date: Thu, 18 Jun 2026 15:53:57 +0200
Subject: [PATCH] cancel user code with interruption after timeout

---
 .../src/main/java/org/apache/beam/io/requestresponse/Call.java   | 1 +
 1 file changed, 1 insertion(+)

diff --git a/sdks/java/io/rrio/src/main/java/org/apache/beam/io/requestresponse/Call.java b/sdks/java/io/rrio/src/main/java/org/apache/beam/io/requestresponse/Call.java
index b318cac17379..616a178d1c39 100644
--- a/sdks/java/io/rrio/src/main/java/org/apache/beam/io/requestresponse/Call.java
+++ b/sdks/java/io/rrio/src/main/java/org/apache/beam/io/requestresponse/Call.java
@@ -535,6 +535,7 @@ public ResponseT call(RequestT request) throws UserCodeExecutionException {
       try {
         return future.get(timeout.getMillis(), TimeUnit.MILLISECONDS);
       } catch (TimeoutException | InterruptedException e) {
+        future.cancel(true);
         throw new UserCodeTimeoutException(e);
       } catch (ExecutionException e) {
         parseAndThrow(future, e);