Skip to content

Commit 387ae16

Browse files
fix(bench): add missing TryStreamExt import for stream iteration
1 parent 8983702 commit 387ae16

1 file changed

Lines changed: 7 additions & 4 deletions

File tree

benches/sqlite/simple.rs

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
1-
use criterion::{criterion_group, criterion_main, BenchmarkId, Criterion};
1+
use criterion::{criterion_group, criterion_main, Criterion};
2+
use futures_util::TryStreamExt;
23
use sqlx::sqlite::{SqlitePool, SqlitePoolOptions};
34
use sqlx::Executor;
45
use sqlx::Row;
@@ -29,7 +30,7 @@ async fn setup_pool() -> SqlitePool {
2930
.unwrap();
3031

3132
// Seed 10,000 rows for large result benchmarks.
32-
// Using a single INSERT with a recursive CTE to avoid 10K round-trips.
33+
// Using a recursive CTE to avoid 10K round-trips.
3334
pool.execute(
3435
r#"
3536
WITH RECURSIVE cnt(id) AS (
@@ -93,8 +94,10 @@ async fn bench_query_small(pool: &SqlitePool) {
9394

9495
/// #5: Small query, huge results (10K rows, 5+ columns with String and BLOB).
9596
async fn bench_query_large(pool: &SqlitePool) {
96-
let mut rows = sqlx::query("SELECT id, i32_val, i64_val, f64_val, text_val, blob_val FROM bench")
97-
.fetch(pool);
97+
let mut rows = sqlx::query(
98+
"SELECT id, i32_val, i64_val, f64_val, text_val, blob_val FROM bench",
99+
)
100+
.fetch(pool);
98101

99102
let mut count = 0u64;
100103
while let Some(row) = rows.try_next().await.unwrap() {

0 commit comments

Comments
 (0)