@@ -558,10 +558,7 @@ impl Reader for DuckDBReader {
558558 let chunk_size = std:: cmp:: min ( MAX_ARROW_BATCH_ROWS , total_rows - offset) ;
559559 let chunk = df. slice ( offset as i64 , chunk_size) ;
560560 let params = dataframe_to_arrow_params ( chunk) ?;
561- let insert_sql = format ! (
562- "INSERT INTO \" {}\" SELECT * FROM arrow(?, ?)" ,
563- name
564- ) ;
561+ let insert_sql = format ! ( "INSERT INTO \" {}\" SELECT * FROM arrow(?, ?)" , name) ;
565562 self . conn . execute ( & insert_sql, params) . map_err ( |e| {
566563 GgsqlError :: ReaderError ( format ! (
567564 "Failed to insert chunk into table '{}': {}" ,
@@ -859,9 +856,18 @@ mod tests {
859856 let result = reader
860857 . execute_sql ( "SELECT id, name FROM large_table ORDER BY id LIMIT 1" )
861858 . unwrap ( ) ;
862- assert_eq ! ( result. column( "id" ) . unwrap( ) . i32 ( ) . unwrap( ) . get( 0 ) . unwrap( ) , 0 ) ;
863859 assert_eq ! (
864- result. column( "name" ) . unwrap( ) . str ( ) . unwrap( ) . get( 0 ) . unwrap( ) ,
860+ result. column( "id" ) . unwrap( ) . i32 ( ) . unwrap( ) . get( 0 ) . unwrap( ) ,
861+ 0
862+ ) ;
863+ assert_eq ! (
864+ result
865+ . column( "name" )
866+ . unwrap( )
867+ . str ( )
868+ . unwrap( )
869+ . get( 0 )
870+ . unwrap( ) ,
865871 "item_0"
866872 ) ;
867873
@@ -873,7 +879,13 @@ mod tests {
873879 ( n - 1 ) as i32
874880 ) ;
875881 assert_eq ! (
876- result. column( "name" ) . unwrap( ) . str ( ) . unwrap( ) . get( 0 ) . unwrap( ) ,
882+ result
883+ . column( "name" )
884+ . unwrap( )
885+ . str ( )
886+ . unwrap( )
887+ . get( 0 )
888+ . unwrap( ) ,
877889 format!( "item_{}" , n - 1 )
878890 ) ;
879891 }
0 commit comments