diff options
Diffstat (limited to 'benchmark')
| -rw-r--r-- | benchmark/benchmark_test.go | 4 | ||||
| -rw-r--r-- | benchmark/postgres_test.go | 4 |
2 files changed, 4 insertions, 4 deletions
diff --git a/benchmark/benchmark_test.go b/benchmark/benchmark_test.go index 359f369..a3e32f7 100644 --- a/benchmark/benchmark_test.go +++ b/benchmark/benchmark_test.go @@ -98,12 +98,12 @@ func BenchmarkORMSelectMany(b *testing.B) { precomputedQuery := store.MustPrepareSelectQueryWhere(partialQuery) selectWithOblast := func(b *testing.B) { - records := must.Return(store.Select(noctx, db, query))(b) + records := must.Return(store.Select(noctx, db, query).Collect())(b) assert.Equal(b, len(records), batchSize) } selectWithOblastWhere := func(b *testing.B) { - records := must.Return(precomputedQuery.Select(noctx, db))(b) + records := must.Return(precomputedQuery.Select(noctx, db).Collect())(b) assert.Equal(b, len(records), batchSize) } diff --git a/benchmark/postgres_test.go b/benchmark/postgres_test.go index 778671b..a5957f7 100644 --- a/benchmark/postgres_test.go +++ b/benchmark/postgres_test.go @@ -92,14 +92,14 @@ func BenchmarkPostgresSelect(b *testing.B) { b.Run("driver=pq/strategy=oblast", func(b *testing.B) { for b.Loop() { - records := must.Return(store.Select(noctx, pqDB, query))(b) + records := must.Return(store.Select(noctx, pqDB, query).Collect())(b) assert.Equal(b, len(records), batchSize) } }) b.Run("driver=pgx/strategy=oblast", func(b *testing.B) { for b.Loop() { - records := must.Return(store.Select(noctx, pgxConnH, query))(b) + records := must.Return(store.Select(noctx, pgxConnH, query).Collect())(b) assert.Equal(b, len(records), batchSize) } }) |