From db87f71f3f673b240de80d5a7592d45c0beae805 Mon Sep 17 00:00:00 2001 From: baishen Date: Sun, 12 Mar 2023 11:02:24 +0800 Subject: [PATCH] Made `num_values` public (#1431) --- src/io/parquet/write/mod.rs | 2 +- src/io/parquet/write/nested/rep.rs | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/src/io/parquet/write/mod.rs b/src/io/parquet/write/mod.rs index d2723a4808b..a0040a9a0d7 100644 --- a/src/io/parquet/write/mod.rs +++ b/src/io/parquet/write/mod.rs @@ -33,7 +33,7 @@ use crate::types::days_ms; use crate::types::i256; use crate::types::NativeType; -pub use nested::write_rep_and_def; +pub use nested::{num_values, write_rep_and_def}; pub use pages::{to_leaves, to_nested, to_parquet_leaves}; use parquet2::schema::types::PrimitiveType as ParquetPrimitiveType; pub use parquet2::{ diff --git a/src/io/parquet/write/nested/rep.rs b/src/io/parquet/write/nested/rep.rs index b1bca7bab33..5bec9f323dc 100644 --- a/src/io/parquet/write/nested/rep.rs +++ b/src/io/parquet/write/nested/rep.rs @@ -21,6 +21,7 @@ fn iter<'a>(nested: &'a [Nested]) -> Vec> { .collect() } +/// return number values of the nested pub fn num_values(nested: &[Nested]) -> usize { let pr = match nested.last().unwrap() { Nested::Primitive(_, _, len) => *len,