(flink) branch master updated: [FLINK-35992][runtime] Close the TieredStorageMasterClient when NettyShuffleMaster close

[guoweijie]

This is an automated email from the ASF dual-hosted git repository.

guoweijie pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/flink.git


The following commit(s) were added to refs/heads/master by this push:
     new 7f0e0e81bac [FLINK-35992][runtime] Close the TieredStorageMasterClient 
when NettyShuffleMaster close
7f0e0e81bac is described below

commit 7f0e0e81bac2cc512d0669b25995ca3b13e262f9
Author: Xu Huang <zuosi...@alibaba-inc.com>
AuthorDate: Wed Aug 7 14:24:18 2024 +0800

    [FLINK-35992][runtime] Close the TieredStorageMasterClient when 
NettyShuffleMaster close
    
    Signed-off-by: Xu Huang <zuosi...@alibaba-inc.com>
---
 .../java/org/apache/flink/runtime/shuffle/NettyShuffleMaster.java  | 7 +++++++
 1 file changed, 7 insertions(+)

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
index 5235da964b8..a9a116a9e34 100644
--- 
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 class NettyShuffleMaster implements 
ShuffleMaster<NettyShuffleDescriptor>
     public void notifyPartitionRecoveryStarted(JobID jobId) {
         
checkNotNull(jobShuffleContexts.get(jobId)).notifyPartitionRecoveryStarted();
     }
+
+    @Override
+    public void close() throws Exception {
+        if (tieredInternalShuffleMaster != null) {
+            tieredInternalShuffleMaster.close();
+        }
+    }
 }