Add concurrent benchmarks

Author: Best-Selling Produts

benchmarks.go

@@ -2,6 +2,8 @@ package main
 
 import (
  "database/sql"
+ "sync"
+ "sync/atomic"
 )
 
 func bmSimpleExec(db *sql.DB, n int) error {
@@ -222,3 +224,99 @@ func bmSelectPreparedLargeRaw(db *sql.DB, n int) error {
  }
  return nil
 }
+
+func runPreparedExecConcurrent(db *sql.DB, n int, co int) error {
+ stmt, err := db.Prepare("DO 1")
+ if err != nil {
+ return err
+ }
+
+ remain := int64(n)
+ var wg sync.WaitGroup
+ wg.Add(co)
+
+ for i := 0; i < co; i++ {
+ go func() {
+ for {
+ if atomic.AddInt64(&remain, -1) < 0 {
+ wg.Done()
+ return
+ }
+
+ if _, err1 := stmt.Exec(); err1 != nil {
+ wg.Done()
+ err = err1
+ return
+ }
+ }
+ }()
+ }
+ wg.Wait()
+ stmt.Close()
+ return err
+}
+
+func bmPreparedExecConcurrent2(db *sql.DB, n int) error {
+ return runPreparedExecConcurrent(db, n, 2)
+}
+
+func bmPreparedExecConcurrent4(db *sql.DB, n int) error {
+ return runPreparedExecConcurrent(db, n, 4)
+}
+
+func bmPreparedExecConcurrent8(db *sql.DB, n int) error {
+ return runPreparedExecConcurrent(db, n, 8)
+}
+
+func bmPreparedExecConcurrent16(db *sql.DB, n int) error {
+ return runPreparedExecConcurrent(db, n, 16)
+}
+
+func runPreparedQueryConcurrent(db *sql.DB, n int, co int) error {
+ stmt, err := db.Prepare("SELECT ?, \"foobar\"")
+ if err != nil {
+ return err
+ }
+
+ remain := int64(n)
+ var wg sync.WaitGroup
+ wg.Add(co)
+
+ for i := 0; i < co; i++ {
+ go func() {
+ var id int
+ var str string
+ for {
+ if atomic.AddInt64(&remain, -1) < 0 {
+ wg.Done()
+ return
+ }
+
+ if err1 := stmt.QueryRow(i).Scan(&id, &str); err1 != nil {
+ wg.Done()
+ err = err1
+ return
+ }
+ }
+ }()
+ }
+ wg.Wait()
+ stmt.Close()
+ return err
+}
+
+func bmPreparedQueryConcurrent2(db *sql.DB, n int) error {
+ return runPreparedQueryConcurrent(db, n, 2)
+}
+
+func bmPreparedQueryConcurrent4(db *sql.DB, n int) error {
+ return runPreparedQueryConcurrent(db, n, 4)
+}
+
+func bmPreparedQueryConcurrent8(db *sql.DB, n int) error {
+ return runPreparedQueryConcurrent(db, n, 8)
+}
+
+func bmPreparedQueryConcurrent16(db *sql.DB, n int) error {
+ return runPreparedQueryConcurrent(db, n, 16)
+}

framework/suite.go

@@ -130,7 +130,7 @@ func (bs *BenchmarkSuite) Run() {
  } else {
  fmt.Println(
  res.Duration.String(), " \t",
- res.QueriesPerSecond(), "queries/sec \t",
+ int(res.QueriesPerSecond()+0.5), "queries/sec \t",
  res.AllocsPerQuery(), "allocs/query \t",
  res.BytesPerQuery(), "B/query",
  )

main.go

@@ -29,11 +29,11 @@ func main() {
  WarmUp: warmup,
  }
 
- if err = bs.AddDriver("mymysql godrv", "mymysql", "gotest/root/root"); err != nil {
+ if err = bs.AddDriver("Go-MySQL-Driver", "mysql", "root:root@/gotest"); err != nil {
  fmt.Println(err)
  return
  }
- if err = bs.AddDriver("Go-MySQL-Driver", "mysql", "root:root@/gotest"); err != nil {
+ if err = bs.AddDriver("mymysql godrv", "mymysql", "gotest/root/root"); err != nil {
  fmt.Println(err)
  return
  }
@@ -50,6 +50,14 @@ func main() {
  bs.AddBenchmark("SelectPreparedLargeBytes", 100000, bmSelectPreparedLargeBytes)
  bs.AddBenchmark("SelectLargeRaw", 100000, bmSelectLargeRaw)
  bs.AddBenchmark("SelectPreparedLargeRaw", 100000, bmSelectPreparedLargeRaw)
+ bs.AddBenchmark("PreparedExecConcurrent2", 500000, bmPreparedExecConcurrent2)
+ bs.AddBenchmark("PreparedExecConcurrent4", 500000, bmPreparedExecConcurrent4)
+ bs.AddBenchmark("PreparedExecConcurrent8", 500000, bmPreparedExecConcurrent8)
+ bs.AddBenchmark("PreparedExecConcurrent16", 500000, bmPreparedExecConcurrent16)
+ bs.AddBenchmark("PreparedQueryConcurrent2", 500000, bmPreparedQueryConcurrent2)
+ bs.AddBenchmark("PreparedQueryConcurrent4", 500000, bmPreparedQueryConcurrent4)
+ bs.AddBenchmark("PreparedQueryConcurrent8", 500000, bmPreparedQueryConcurrent8)
+ bs.AddBenchmark("PreparedQueryConcurrent16", 500000, bmPreparedQueryConcurrent16)
 
  bs.Run()
 }