From b5e218ee363f58c91054c4aad83747d3a661d63b Mon Sep 17 00:00:00 2001
From: Matthias Pohl <mpohl@confluent.io>
Date: Thu, 9 Jul 2026 16:38:01 +0200
Subject: [PATCH] [FLINK-40086][runtime] Adds error handling to newly added
 preCompletionAction

---
 .../org/apache/flink/runtime/executiongraph/Execution.java  | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/flink-runtime/src/main/java/org/apache/flink/runtime/executiongraph/Execution.java b/flink-runtime/src/main/java/org/apache/flink/runtime/executiongraph/Execution.java
index a704e3290fdb4..430cd9ee76da4 100644
--- a/flink-runtime/src/main/java/org/apache/flink/runtime/executiongraph/Execution.java
+++ b/flink-runtime/src/main/java/org/apache/flink/runtime/executiongraph/Execution.java
@@ -1708,7 +1708,11 @@ private boolean transitionState(
                 initializingOrRunningFuture.complete(null);
             } else if (targetState.isTerminal()) {
                 if (preCompletionAction != null) {
-                    preCompletionAction.run();
+                    try {
+                        preCompletionAction.run();
+                    } catch (Exception e) {
+                        LOG.error("Error while executing pre-completion action.", e);
+                    }
                 }
                 // complete the terminal state future
                 terminalStateFuture.complete(targetState);