From 18e8c45384e506cdea9a3c4bddc010fca1e7f10d Mon Sep 17 00:00:00 2001
From: LFC <990479+MichaelScofield@users.noreply.github.com>
Date: Wed, 18 Dec 2024 10:42:33 +0800
Subject: [PATCH] refactor: produce BatchBuilder from a Batch to modify it
 again (#5186)

chore: pub some mods
---
 src/mito2/src/read.rs        | 12 ++++++++++++
 src/mito2/src/sst/parquet.rs |  2 +-
 2 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/src/mito2/src/read.rs b/src/mito2/src/read.rs
index d8ac5ce46b..c4de103f10 100644
--- a/src/mito2/src/read.rs
+++ b/src/mito2/src/read.rs
@@ -861,6 +861,18 @@ impl BatchBuilder {
     }
 }
 
+impl From<Batch> for BatchBuilder {
+    fn from(batch: Batch) -> Self {
+        Self {
+            primary_key: batch.primary_key,
+            timestamps: Some(batch.timestamps),
+            sequences: Some(batch.sequences),
+            op_types: Some(batch.op_types),
+            fields: batch.fields,
+        }
+    }
+}
+
 /// Async [Batch] reader and iterator wrapper.
 ///
 /// This is the data source for SST writers or internal readers.
diff --git a/src/mito2/src/sst/parquet.rs b/src/mito2/src/sst/parquet.rs
index ae51a0d37c..3dd53ba645 100644
--- a/src/mito2/src/sst/parquet.rs
+++ b/src/mito2/src/sst/parquet.rs
@@ -24,7 +24,7 @@ use crate::sst::index::IndexOutput;
 use crate::sst::DEFAULT_WRITE_BUFFER_SIZE;
 
 pub(crate) mod file_range;
-pub(crate) mod format;
+pub mod format;
 pub(crate) mod helper;
 pub(crate) mod metadata;
 mod page_reader;