[FLINK-35992][runtime] Close the TieredStorageMasterClient when Netty…

…ShuffleMaster close Signed-off-by: Xu Huang <[email protected]>
codenohup · Aug 7, 2024 · 96bb5ce · 96bb5ce
1 parent 68d8174
commit 96bb5ce
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/flink-runtime/src/main/java/org/apache/flink/runtime/shuffle/NettyShuffleMaster.java b/flink-runtime/src/main/java/org/apache/flink/runtime/shuffle/NettyShuffleMaster.java
@@ -219,4 +219,11 @@ public void snapshotState(
     public void notifyPartitionRecoveryStarted(JobID jobId) {
         checkNotNull(jobShuffleContexts.get(jobId)).notifyPartitionRecoveryStarted();
     }
+
+    @Override
+    public void close() throws Exception {
+        if (tieredInternalShuffleMaster != null) {
+            tieredInternalShuffleMaster.close();
+        }
+    }
 }