[HUDI-2959] Fix the thread leak of cleaning service (#4252)

2021-12-11 12:08:47 +08:00
parent 9797fdfbb2
commit 9bdcee00c0
5 changed files with 31 additions and 38 deletions
--- a/hudi-flink/src/main/java/org/apache/hudi/sink/CleanFunction.java
+++ b/hudi-flink/src/main/java/org/apache/hudi/sink/CleanFunction.java
@@ -98,4 +98,11 @@ public class CleanFunction<T> extends AbstractRichFunction
  public void initializeState(FunctionInitializationContext context) throws Exception {
    // no operation
  }
+
+  @Override
+  public void close() throws Exception {
+    if (this.writeClient != null) {
+      this.writeClient.close();
+    }
+  }
 }