Add multi-row select benchmarks

stdlib/bench_test.go

@@ -0,0 +1,109 @@
+package stdlib_test
+
+import (
+	"database/sql"
+	"fmt"
+	"os"
+	"strconv"
+	"strings"
+	"testing"
+	"time"
+)
+
+func getSelectRowsCounts(b *testing.B) []int64 {
+	var rowCounts []int64
+	{
+		s := os.Getenv("PGX_BENCH_SELECT_ROWS_COUNTS")
+		if s != "" {
+			for _, p := range strings.Split(s, " ") {
+				n, err := strconv.ParseInt(p, 10, 64)
+				if err != nil {
+					b.Fatalf("Bad PGX_BENCH_SELECT_ROWS_COUNTS value: %v", err)
+				}
+				rowCounts = append(rowCounts, n)
+			}
+		}
+	}
+
+	if len(rowCounts) == 0 {
+		rowCounts = []int64{1, 10, 100, 1000}
+	}
+
+	return rowCounts
+}
+
+type BenchRowSimple struct {
+	ID         int32
+	FirstName  string
+	LastName   string
+	Sex        string
+	BirthDate  time.Time
+	Weight     int32
+	Height     int32
+	UpdateTime time.Time
+}
+
+func BenchmarkSelectRowsScanSimple(b *testing.B) {
+	db := openDB(b)
+	defer closeDB(b, db)
+
+	rowCounts := getSelectRowsCounts(b)
+
+	for _, rowCount := range rowCounts {
+		b.Run(fmt.Sprintf("%d rows", rowCount), func(b *testing.B) {
+			br := &BenchRowSimple{}
+			for i := 0; i < b.N; i++ {
+				rows, err := db.Query("select n, 'Adam', 'Smith ' || n, 'male', '1952-06-16'::date, 258, 72, now() from generate_series(1, $1) n", rowCount)
+				if err != nil {
+					b.Fatal(err)
+				}
+
+				for rows.Next() {
+					rows.Scan(&br.ID, &br.FirstName, &br.LastName, &br.Sex, &br.BirthDate, &br.Weight, &br.Height, &br.UpdateTime)
+				}
+
+				if rows.Err() != nil {
+					b.Fatal(rows.Err())
+				}
+			}
+		})
+	}
+}
+
+type BenchRowNull struct {
+	ID         sql.NullInt32
+	FirstName  sql.NullString
+	LastName   sql.NullString
+	Sex        sql.NullString
+	BirthDate  sql.NullTime
+	Weight     sql.NullInt32
+	Height     sql.NullInt32
+	UpdateTime sql.NullTime
+}
+
+func BenchmarkSelectRowsScanNull(b *testing.B) {
+	db := openDB(b)
+	defer closeDB(b, db)
+
+	rowCounts := getSelectRowsCounts(b)
+
+	for _, rowCount := range rowCounts {
+		b.Run(fmt.Sprintf("%d rows", rowCount), func(b *testing.B) {
+			br := &BenchRowSimple{}
+			for i := 0; i < b.N; i++ {
+				rows, err := db.Query("select n, 'Adam', 'Smith ' || n, 'male', '1952-06-16'::date, 258, 72, now() from generate_series(1, $1) n", rowCount)
+				if err != nil {
+					b.Fatal(err)
+				}
+
+				for rows.Next() {
+					rows.Scan(&br.ID, &br.FirstName, &br.LastName, &br.Sex, &br.BirthDate, &br.Weight, &br.Height, &br.UpdateTime)
+				}
+
+				if rows.Err() != nil {
+					b.Fatal(rows.Err())
+				}
+			}
+		})
+	}
+}

stdlib/sql_test.go

@@ -19,7 +19,7 @@ import (
 	"github.com/stretchr/testify/require"
 )
 
-func closeDB(t *testing.T, db *sql.DB) {
+func closeDB(t testing.TB, db *sql.DB) {
 	err := db.Close()
 	if err != nil {
 		t.Fatalf("db.Close unexpectedly failed: %v", err)
@@ -27,7 +27,7 @@ func closeDB(t *testing.T, db *sql.DB) {
 }
 
 // Do a simple query to ensure the connection is still usable
-func ensureConnValid(t *testing.T, db *sql.DB) {
+func ensureConnValid(t testing.TB, db *sql.DB) {
 	var sum, rowCount int32
 
 	rows, err := db.Query("select generate_series(1,$1)", 10)

stdlib/stdlibutil110_test.go

@@ -11,7 +11,7 @@ import (
 	"github.com/jackc/pgx/v4/stdlib"
 )
 
-func openDB(t *testing.T) *sql.DB {
+func openDB(t testing.TB) *sql.DB {
 	config, err := pgx.ParseConfig(os.Getenv("PGX_TEST_DATABASE"))
 	if err != nil {
 		t.Fatalf("pgx.ParseConnectionString failed: %v", err)