From eacae5a8453dbb86709e6c3d0d1294220587e058 Mon Sep 17 00:00:00 2001
From: Zhenqiu Huang <huangzhenqiu0825@gmail.com>
Date: Thu, 14 May 2026 21:42:01 -0700
Subject: [PATCH] fix(flink): add close function in AbstractStreamWriteFunction

---
 .../hudi/sink/common/AbstractStreamWriteFunction.java     | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/hudi-flink-datasource/hudi-flink/src/main/java/org/apache/hudi/sink/common/AbstractStreamWriteFunction.java b/hudi-flink-datasource/hudi-flink/src/main/java/org/apache/hudi/sink/common/AbstractStreamWriteFunction.java
index 3bd19fa0699ed..ec02097e9de6e 100644
--- a/hudi-flink-datasource/hudi-flink/src/main/java/org/apache/hudi/sink/common/AbstractStreamWriteFunction.java
+++ b/hudi-flink-datasource/hudi-flink/src/main/java/org/apache/hudi/sink/common/AbstractStreamWriteFunction.java
@@ -176,6 +176,14 @@ public void endInput() {
     this.inputEnded = true;
   }
 
+  @Override
+  public void close() throws Exception {
+    if (this.writeClient != null) {
+      this.writeClient.close();
+    }
+    super.close();
+  }
+
   // -------------------------------------------------------------------------
   //  Getter/Setter
   // -------------------------------------------------------------------------