From 32d1d68441bdea30fd667d394e214b46eb209b91 Mon Sep 17 00:00:00 2001
From: Yingwen <realevenyag@gmail.com>
Date: Sat, 23 Sep 2023 18:05:34 +0800
Subject: [PATCH] fix(mito): reset is_sorted to true after the merger finishing
 one series (#2474)

fix: reset is_sorted flag to true after the merger finishing one series
---
 src/mito2/src/read/merge.rs | 48 +++++++++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)

diff --git a/src/mito2/src/read/merge.rs b/src/mito2/src/read/merge.rs
index e5b8651fa7..797e2f9f70 100644
--- a/src/mito2/src/read/merge.rs
+++ b/src/mito2/src/read/merge.rs
@@ -223,6 +223,9 @@ impl BatchMerger {
         // rows by sequence for simplicity and performance reason.
         batch.filter_deleted()?;
 
+        // Reset merger.
+        self.is_sorted = true;
+
         Ok(Some(batch))
     }
 }
@@ -480,4 +483,49 @@ mod tests {
         )
         .await;
     }
+
+    #[test]
+    fn test_batch_merger_empty() {
+        let mut merger = BatchMerger::new();
+        assert!(merger.merge_batches().unwrap().is_none());
+    }
+
+    #[test]
+    fn test_batch_merger_unsorted() {
+        let mut merger = BatchMerger::new();
+        merger.push(new_batch(
+            b"k1",
+            &[1, 3, 5],
+            &[10, 10, 10],
+            &[OpType::Put, OpType::Put, OpType::Put],
+            &[21, 23, 25],
+        ));
+        assert!(merger.is_sorted);
+        merger.push(new_batch(
+            b"k1",
+            &[2, 4],
+            &[11, 11],
+            &[OpType::Put, OpType::Put],
+            &[22, 24],
+        ));
+        assert!(!merger.is_sorted);
+        let batch = merger.merge_batches().unwrap().unwrap();
+        assert_eq!(
+            batch,
+            new_batch(
+                b"k1",
+                &[1, 2, 3, 4, 5],
+                &[10, 11, 10, 11, 10],
+                &[
+                    OpType::Put,
+                    OpType::Put,
+                    OpType::Put,
+                    OpType::Put,
+                    OpType::Put
+                ],
+                &[21, 22, 23, 24, 25]
+            )
+        );
+        assert!(merger.is_sorted);
+    }
 }