diff --git a/src/mito2/src/read.rs b/src/mito2/src/read.rs index d8ac5ce46b62..c4de103f1000 100644 --- a/src/mito2/src/read.rs +++ b/src/mito2/src/read.rs @@ -861,6 +861,18 @@ impl BatchBuilder { } } +impl From 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 ae51a0d37c29..3dd53ba645f2 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;