Function BenchmarkReceiveMetadata

benchmark_test.go:459–511

BenchmarkReceiveMetadata measures performance of receiving lots of metadata compare to data in rows

(b *testing.B)

// BenchmarkReceiveMetadata measures performance of receiving lots of metadata compare to data in rows
func BenchmarkReceiveMetadata(b *testing.B) {
	tb := (*TB)(b)

	// Create a table with 1000 integer fields
	var createTableQuery strings.Builder
	createTableQuery.WriteString("CREATE TABLE large_integer_table (")
	for i := range 1000 {
		createTableQuery.WriteString(fmt.Sprintf("col_%d INT", i))
		if i < 999 {
			createTableQuery.WriteString(", ")
		}
	}
	createTableQuery.WriteString(")")

	// Initialize database
	db := initDB(b, false,
		"DROP TABLE IF EXISTS large_integer_table",
		createTableQuery.String(),
		"INSERT INTO large_integer_table VALUES ("+
			strings.Repeat("0,", 999)+"0)", // Insert a row of zeros
	)
	defer db.Close()

	b.Run("query", func(b *testing.B) {
		db.SetMaxIdleConns(0)
		db.SetMaxIdleConns(1)

		// Create a slice to scan all columns
		values := make([]any, 1000)
		valuePtrs := make([]any, 1000)
		for j := range values {
			valuePtrs[j] = &values[j]
		}

		// Prepare a SELECT query to retrieve metadata
		stmt := tb.checkStmt(db.Prepare("SELECT * FROM large_integer_table LIMIT 1"))
		defer stmt.Close()

		// Benchmark metadata retrieval
		b.ReportAllocs()
		b.ResetTimer()
		for range b.N {
			rows := tb.checkRows(stmt.Query())

			rows.Next()
			// Scan the row
			err := rows.Scan(valuePtrs...)
			tb.check(err)

			rows.Close()
		}
	})
}