From 356730208b8b909a9959272e99b1a79c0246cb63 Mon Sep 17 00:00:00 2001 From: Stefan Majewsky Date: Fri, 17 Jul 2026 19:03:48 +0200 Subject: add type Selection --- benchmark/benchmark_test.go | 4 ++-- benchmark/postgres_test.go | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) (limited to 'benchmark') 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) } }) -- cgit v1.3.1