diff --git a/src/mito2/src/memtable.rs b/src/mito2/src/memtable.rs index ea3875ac7a..e95d5e348f 100644 --- a/src/mito2/src/memtable.rs +++ b/src/mito2/src/memtable.rs @@ -57,6 +57,10 @@ pub(crate) mod version; #[cfg(any(test, feature = "test"))] pub use bulk::part::BulkPart; +pub use bulk::part::{ + BulkPartEncoder, BulkPartMeta, UnorderedPart, record_batch_estimated_size, + sort_primary_key_record_batch, +}; #[cfg(any(test, feature = "test"))] pub use time_partition::filter_record_batch; diff --git a/src/mito2/src/memtable/bulk/part.rs b/src/mito2/src/memtable/bulk/part.rs index aa2278cb78..6aedf9a5d7 100644 --- a/src/mito2/src/memtable/bulk/part.rs +++ b/src/mito2/src/memtable/bulk/part.rs @@ -464,7 +464,7 @@ impl UnorderedPart { } /// More accurate estimation of the size of a record batch. -pub(crate) fn record_batch_estimated_size(batch: &RecordBatch) -> usize { +pub fn record_batch_estimated_size(batch: &RecordBatch) -> usize { batch .columns() .iter() @@ -715,7 +715,7 @@ fn new_primary_key_column_builders( } /// Sorts the record batch with primary key format. -fn sort_primary_key_record_batch(batch: &RecordBatch) -> Result { +pub fn sort_primary_key_record_batch(batch: &RecordBatch) -> Result { let total_columns = batch.num_columns(); let sort_columns = vec![ // Primary key column (ascending)