Skip to content

Commit c22a981

Browse files
committed
Run cargo fmt
1 parent dbb6508 commit c22a981

1 file changed

Lines changed: 19 additions & 7 deletions

File tree

src/reader/duckdb.rs

Lines changed: 19 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)