diff options
Diffstat (limited to 'benchmark/postgres_test.go')
| -rw-r--r-- | benchmark/postgres_test.go | 4 |
1 files changed, 2 insertions, 2 deletions
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) } }) |