[FLINK-34227][runtime] Adds assertion to chained operation in JobMast…

…er#stopJobExecution
apache · Jan 31, 2025 · 04a3dca · 04a3dca
1 parent 67e2dc8
commit 04a3dca
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/flink-runtime/src/main/java/org/apache/flink/runtime/jobmaster/JobMaster.java b/flink-runtime/src/main/java/org/apache/flink/runtime/jobmaster/JobMaster.java
@@ -1197,6 +1197,8 @@ private CompletableFuture<Void> stopJobExecution(final Exception cause) {
         return FutureUtils.runAfterwards(
                 terminationFuture,
                 () -> {
+                    getMainThreadExecutor().assertRunningInMainThread();
+
                     shuffleMaster.unregisterJob(executionPlan.getJobID());
                     disconnectTaskManagerResourceManagerConnections(cause);
                     stopJobMasterServices();