fix

Author: zjw1111

src/paimon/core/mergetree/merge_tree_writer.cpp

@@ -226,7 +226,7 @@ Status MergeTreeWriter::Flush(bool wait_for_latest_compaction, bool forced_full_
         }
         // 1. flush write buffer to get in-memory readers
         PAIMON_ASSIGN_OR_RAISE(std::vector<std::unique_ptr<KeyValueRecordReader>> readers,
-                               write_buffer_->Flush(last_sequence_number_));
+                               write_buffer_->Flush(&last_sequence_number_));
         // 2. prepare loser tree sort merge reader
         auto sort_merge_reader = std::make_unique<SortMergeReaderWithLoserTree>(
             std::move(readers), key_comparator_, user_defined_seq_comparator_,

src/paimon/core/mergetree/write_buffer.cpp

@@ -1,5 +1,5 @@
 /*
- * Copyright 2024-present Alibaba Inc.
+ * Copyright 2026-present Alibaba Inc.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -67,16 +67,16 @@ Status WriteBuffer::Write(std::unique_ptr<RecordBatch>&& moved_batch) {
 }
 
 Result<std::vector<std::unique_ptr<KeyValueRecordReader>>> WriteBuffer::Flush(
-    int64_t& last_sequence_number) {
+    int64_t* last_sequence_number) {
     std::vector<std::unique_ptr<KeyValueRecordReader>> readers;
     if (batch_vec_.empty()) {
         return readers;
     }
 
     readers.reserve(batch_vec_.size());
     for (size_t i = 0; i < batch_vec_.size(); ++i) {
-        int64_t sequence_number = last_sequence_number;
-        last_sequence_number += batch_vec_[i]->length();
+        int64_t sequence_number = *last_sequence_number;
+        *last_sequence_number += batch_vec_[i]->length();
         auto in_memory_reader = std::make_unique<KeyValueInMemoryRecordReader>(
             sequence_number, std::move(batch_vec_[i]), std::move(row_kinds_vec_[i]),
             trimmed_primary_keys_, user_defined_sequence_fields_, key_comparator_,

src/paimon/core/mergetree/write_buffer.h

@@ -1,5 +1,5 @@
 /*
- * Copyright 2024-present Alibaba Inc.
+ * Copyright 2026-present Alibaba Inc.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -61,7 +61,7 @@ class WriteBuffer {
     /// Flush all buffered batches into KeyValueInMemoryRecordReaders and clear the buffer.
     /// @param[in,out] last_sequence_number current sequence number, updated after flush
     /// @return list of KeyValueRecordReaders built from buffered data
-    Result<std::vector<std::unique_ptr<KeyValueRecordReader>>> Flush(int64_t& last_sequence_number);
+    Result<std::vector<std::unique_ptr<KeyValueRecordReader>>> Flush(int64_t* last_sequence_number);
 
     /// Return current memory usage in bytes.
     int64_t GetMemoryUsage() const {

src/paimon/core/mergetree/write_buffer_test.cpp

@@ -1,5 +1,5 @@
 /*
- * Copyright 2024-present Alibaba Inc.
+ * Copyright 2026-present Alibaba Inc.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -100,7 +100,7 @@ TEST_F(WriteBufferTest, TestFlushResetsStateAndAdvancesSequenceNumber) {
     ASSERT_GT(write_buffer.GetMemoryUsage(), 0);
 
     int64_t last_sequence_number = 10;
-    ASSERT_OK_AND_ASSIGN(auto readers, write_buffer.Flush(last_sequence_number));
+    ASSERT_OK_AND_ASSIGN(auto readers, write_buffer.Flush(&last_sequence_number));
 
     ASSERT_EQ(readers.size(), 2);
     ASSERT_TRUE(write_buffer.IsEmpty());
@@ -150,7 +150,7 @@ TEST_F(WriteBufferTest, TestFlushPreservesRowKinds) {
     ASSERT_OK(write_buffer.Write(CreateBatch(array, row_kinds)));
 
     int64_t last_sequence_number = 0;
-    ASSERT_OK_AND_ASSIGN(auto readers, write_buffer.Flush(last_sequence_number));
+    ASSERT_OK_AND_ASSIGN(auto readers, write_buffer.Flush(&last_sequence_number));
     ASSERT_EQ(readers.size(), 1);
     ASSERT_EQ(last_sequence_number, 4);