From fabf5bd63c4be0c49516fa538cfc58ed6430889c Mon Sep 17 00:00:00 2001
From: ZENOTME <ZENOTME@users.noreply.github.com>
Date: Mon, 2 Nov 2026 11:36:51 +0800
Subject: [PATCH] feat(c): expose TableWrite close for explicit cleanup

---
 bindings/c/src/tests.rs |  3 +++
 bindings/c/src/write.rs | 22 ++++++++++++++++++++++
 2 files changed, 25 insertions(+)

diff --git a/bindings/c/src/tests.rs b/bindings/c/src/tests.rs
index 91a0f8e69..3b3925526 100644
--- a/bindings/c/src/tests.rs
+++ b/bindings/c/src/tests.rs
@@ -1932,6 +1932,9 @@ fn test_write_commit_read_roundtrip() {
         assert!(pc_result.error.is_null());
         assert!(!pc_result.messages.is_null());
 
+        // Closing must preserve the commit messages already handed to the caller.
+        assert!(paimon_table_write_close(tw).is_null());
+
         let tc_result = paimon_write_builder_new_commit(wb);
         assert!(tc_result.error.is_null());
         let tc = tc_result.commit;
diff --git a/bindings/c/src/write.rs b/bindings/c/src/write.rs
index 9294bb4bc..f6bad6c1a 100644
--- a/bindings/c/src/write.rs
+++ b/bindings/c/src/write.rs
@@ -571,6 +571,26 @@ pub unsafe extern "C" fn paimon_postpone_fixed_bucket_write_builder_new_write(
     }
 }
 
+/// Close a standard TableWrite, discarding outstanding output without preparing a commit.
+///
+/// Previously prepared commit messages remain owned by the caller. This does not free
+/// the handle; call `paimon_table_write_free` after closing it.
+///
+/// # Safety
+/// `tw` must be a live standard TableWrite handle, or null (returns an error).
+/// The handle must not be used concurrently.
+#[no_mangle]
+pub unsafe extern "C" fn paimon_table_write_close(
+    tw: *mut paimon_table_write,
+) -> *mut paimon_error {
+    if let Err(error) = check_non_null(tw, "tw") {
+        return error;
+    }
+    let state = &mut *((*tw).inner as *mut TableWriteState);
+    runtime().block_on(state.write.close());
+    ptr::null_mut()
+}
+
 /// Free a standard TableWrite.
 ///
 /// # Safety
@@ -1510,6 +1530,8 @@ const _: unsafe extern "C" fn(
     *mut c_void,
     *mut c_void,
 ) -> *mut paimon_error = paimon_postpone_fixed_bucket_write_builder_with_bucket_plan;
+const _: unsafe extern "C" fn(*mut paimon_table_write) -> *mut paimon_error =
+    paimon_table_write_close;
 const _: unsafe extern "C" fn(*mut paimon_table_write) -> paimon_result_prepare_commit =
     paimon_table_write_prepare_commit;
 const _: extern "C" fn() -> i32 = paimon_commit_messages_serializer_version;