如何在Golang中进行数据库性能基准测试

func BenchmarkInsertUser(b *testing.B) {
    db, err := sql.Open("pgx", "your-dsn")
    if err != nil {
        b.Fatal(err)
    }
    defer db.Close()

    stmt := "INSERT INTO users(name, email) VALUES($1, $2)"
    
    b.ResetTimer()
    for i := 0; i < b.N; i++ {
        _, err := db.Exec(stmt, fmt.Sprintf("user_%d", i), fmt.Sprintf("user%d@demo.com", i))
        if err != nil {
            b.Error(err)
        }
    }
}

func BenchmarkQueryUsers(b *testing.B) {
    db, _ := sql.Open("pgx", "your-dsn")
    defer db.Close()

    b.SetParallelism(4)
    b.RunParallel(func(pb *testing.PB) {
        for pb.Next() {
            var name string
            err := db.QueryRow("SELECT name FROM users WHERE id = $1", 1).Scan(&name)
            if err != nil && err != sql.ErrNoRows {
                b.Error(err)
            }
        }
    })
}

go test -bench=BenchmarkInsertUser -benchmem