1
1
mod data_chunk_adaptor;
2
2
3
- use duckdb:: core:: { DataChunkHandle , FlatVector } ;
3
+ use duckdb:: core:: DataChunkHandle ;
4
4
use duckdb:: vtab:: arrow:: {
5
5
WritableVector , flat_vector_to_arrow_array, write_arrow_array_to_vector,
6
6
} ;
@@ -9,10 +9,11 @@ use vortex_array::arrow::{FromArrowArray, IntoArrowArray};
9
9
use vortex_array:: validity:: Validity ;
10
10
use vortex_array:: variants:: StructArrayTrait ;
11
11
use vortex_array:: { Array , ArrayRef } ;
12
- use vortex_dtype:: FieldNames ;
13
12
use vortex_error:: { VortexResult , vortex_err} ;
14
13
15
- use crate :: convert:: array:: data_chunk_adaptor:: DataChunkHandleSlice ;
14
+ use crate :: convert:: array:: data_chunk_adaptor:: {
15
+ DataChunkHandleSlice , NamedDataChunk , SizedFlatVector ,
16
+ } ;
16
17
17
18
pub trait ToDuckDB {
18
19
fn to_duckdb ( & self , chunk : & mut dyn WritableVector ) -> VortexResult < ( ) > ;
@@ -68,7 +69,7 @@ impl<'a> FromDuckDB<&'a NamedDataChunk<'a>> for ArrayRef {
68
69
names
69
70
. as_ref ( )
70
71
. map ( |names| names[ i] . clone ( ) )
71
- . unwrap_or ( i. to_string ( ) . into ( ) ) ,
72
+ . unwrap_or_else ( || i. to_string ( ) . into ( ) ) ,
72
73
array,
73
74
) )
74
75
} )
@@ -87,10 +88,7 @@ impl FromDuckDB<SizedFlatVector> for ArrayRef {
87
88
fn from_duckdb ( mut sized_vector : SizedFlatVector ) -> VortexResult < ArrayRef > {
88
89
let len = sized_vector. len ;
89
90
let arrow_arr = flat_vector_to_arrow_array ( & mut sized_vector. vector , len) . map_err ( |e| {
90
- println ! (
91
- "Failed to convert duckdb duckdb array vortex: {}" ,
92
- e. to_string( )
93
- ) ;
91
+ println ! ( "Failed to convert duckdb duckdb array vortex: {}" , e) ;
94
92
95
93
vortex_err ! ( "Failed to convert duckdb array to vortex: {}" , e)
96
94
} ) ?;
@@ -108,7 +106,8 @@ mod tests {
108
106
use vortex_array:: { Array , ArrayRef , ToCanonical } ;
109
107
use vortex_dtype:: { DType , FieldNames , Nullability } ;
110
108
111
- use crate :: convert:: array:: { NamedDataChunk , to_duckdb_chunk} ;
109
+ use crate :: convert:: array:: data_chunk_adaptor:: NamedDataChunk ;
110
+ use crate :: convert:: array:: to_duckdb_chunk;
112
111
use crate :: { FromDuckDB , ToDuckDBType } ;
113
112
114
113
fn data ( ) -> ArrayRef {
0 commit comments