Add concurrent benchmarks

Author: julienschmidt

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()
 }