File tree 2 files changed +16
-2
lines changed
src/datasource/physical_plan/parquet
2 files changed +16
-2
lines changed Original file line number Diff line number Diff line change @@ -701,6 +701,20 @@ macro_rules! get_data_page_statistics {
701
701
TimeUnit :: Nanosecond => Arc :: new( TimestampNanosecondArray :: from_iter( iter) . with_timezone_opt( timezone. clone( ) ) ) ,
702
702
} )
703
703
} ,
704
+ Some ( DataType :: Date32 ) => Ok ( Arc :: new( Date32Array :: from_iter( [ <$stat_type_prefix Int32DataPageStatsIterator >] :: new( $iterator) . flatten( ) ) ) ) ,
705
+ Some ( DataType :: Date64 ) => Ok (
706
+ Arc :: new(
707
+ Date64Array :: from( [ <$stat_type_prefix Int32DataPageStatsIterator >] :: new( $iterator)
708
+ . map( |x| {
709
+ x. into_iter( )
710
+ . filter_map( |x| {
711
+ x. and_then( |x| i64 :: try_from( x) . ok( ) )
712
+ } )
713
+ . map( |x| x * 24 * 60 * 60 * 1000 )
714
+ } ) . flatten( ) . collect:: <Vec <_>>( )
715
+ )
716
+ )
717
+ ) ,
704
718
_ => unimplemented!( )
705
719
}
706
720
}
Original file line number Diff line number Diff line change @@ -1181,7 +1181,7 @@ async fn test_dates_32_diff_rg_sizes() {
1181
1181
// row counts are [13, 7]
1182
1182
expected_row_counts : Some ( UInt64Array :: from ( vec ! [ 13 , 7 ] ) ) ,
1183
1183
column_name : "date32" ,
1184
- check : Check :: RowGroup ,
1184
+ check : Check :: Both ,
1185
1185
}
1186
1186
. run ( ) ;
1187
1187
}
@@ -1324,7 +1324,7 @@ async fn test_dates_64_diff_rg_sizes() {
1324
1324
expected_null_counts : UInt64Array :: from ( vec ! [ 2 , 2 ] ) ,
1325
1325
expected_row_counts : Some ( UInt64Array :: from ( vec ! [ 13 , 7 ] ) ) ,
1326
1326
column_name : "date64" ,
1327
- check : Check :: RowGroup ,
1327
+ check : Check :: Both ,
1328
1328
}
1329
1329
. run ( ) ;
1330
1330
}
You can’t perform that action at this time.
0 commit comments