diff --git a/src/datatypes/src/prelude.rs b/src/datatypes/src/prelude.rs index 1aa4775354..de9b056a39 100644 --- a/src/datatypes/src/prelude.rs +++ b/src/datatypes/src/prelude.rs @@ -1,3 +1,5 @@ pub use crate::data_type::{DataType, DataTypeRef}; +pub use crate::scalars::{ScalarVector, ScalarVectorBuilder}; pub use crate::type_id::LogicalTypeId; pub use crate::value::Value; +pub use crate::vectors::{Vector, VectorRef}; diff --git a/src/datatypes/src/types.rs b/src/datatypes/src/types.rs index 02fa6bc718..61982e99f9 100644 --- a/src/datatypes/src/types.rs +++ b/src/datatypes/src/types.rs @@ -1,3 +1,7 @@ -pub mod binary_type; -pub mod primitive_traits; -pub mod primitive_type; +mod binary_type; +mod primitive_traits; +mod primitive_type; + +pub use binary_type::BinaryType; +pub use primitive_traits::Primitive; +pub use primitive_type::{DataTypeBuilder, PrimitiveType}; diff --git a/src/datatypes/src/vectors.rs b/src/datatypes/src/vectors.rs index c90106c2ea..1ea36c32a2 100644 --- a/src/datatypes/src/vectors.rs +++ b/src/datatypes/src/vectors.rs @@ -5,6 +5,8 @@ use std::any::Any; use std::sync::Arc; use arrow2::array::ArrayRef; +pub use binary::*; +pub use primitive::*; use crate::data_type::DataTypeRef; diff --git a/src/datatypes/src/vectors/binary.rs b/src/datatypes/src/vectors/binary.rs index bb6d5dc775..60b49dd827 100644 --- a/src/datatypes/src/vectors/binary.rs +++ b/src/datatypes/src/vectors/binary.rs @@ -7,7 +7,7 @@ use arrow2::bitmap::utils::ZipValidity; use crate::data_type::DataTypeRef; use crate::scalars::{ScalarVector, ScalarVectorBuilder}; -use crate::types::binary_type::BinaryType; +use crate::types::BinaryType; use crate::vectors::Vector; use crate::{LargeBinaryArray, MutableLargeBinaryArray}; diff --git a/src/datatypes/src/vectors/primitive.rs b/src/datatypes/src/vectors/primitive.rs index ac1e99447d..95d1266843 100644 --- a/src/datatypes/src/vectors/primitive.rs +++ b/src/datatypes/src/vectors/primitive.rs @@ -2,14 +2,12 @@ use std::any::Any; use std::slice::Iter; use std::sync::Arc; -use arrow2::array::ArrayRef; -use arrow2::array::{MutablePrimitiveArray, PrimitiveArray}; +use arrow2::array::{ArrayRef, MutablePrimitiveArray, PrimitiveArray}; use arrow2::bitmap::utils::ZipValidity; use crate::data_type::DataTypeRef; use crate::scalars::{ScalarVector, ScalarVectorBuilder}; -use crate::types::primitive_traits::Primitive; -use crate::types::primitive_type::DataTypeBuilder; +use crate::types::{DataTypeBuilder, Primitive}; use crate::vectors::Vector; /// Vector for primitive data types.