Merge pull request pingcap#1022 from pingcap/coocood/select-bench

tidb: add some simple select benchmark.

bench_test.go

@@ -1,30 +1,172 @@
 package tidb
 
 import (
+	"fmt"
 	"testing"
 
 	"github.com/ngaut/log"
+	"github.com/pingcap/tidb/ast"
 )
 
-func BenchmarkBasic(b *testing.B) {
+var smallCount = 100
+
+func prepareBenchSession() Session {
 	store, err := NewStore("memory://bench")
 	if err != nil {
-		b.Fatal(err)
+		log.Fatal(err)
 	}
 	log.SetLevel(log.LOG_LEVEL_ERROR)
 	se, err := CreateSession(store)
 	if err != nil {
-		b.Fatal(err)
+		log.Fatal(err)
+	}
+	mustExecute(se, "use test")
+	return se
+}
+
+func prepareBenchData(se Session, colType string, valueFormat string, valueCount int) {
+	mustExecute(se, "drop table if exists t")
+	mustExecute(se, fmt.Sprintf("create table t (pk int primary key auto_increment, col %s, index idx (col))", colType))
+	mustExecute(se, "begin")
+	for i := 0; i < valueCount; i++ {
+		mustExecute(se, "insert t (col) values ("+fmt.Sprintf(valueFormat, i)+")")
 	}
+	mustExecute(se, "commit")
+}
+
+func readResult(rs ast.RecordSet, count int) {
+	for count > 0 {
+		x, err := rs.Next()
+		if err != nil {
+			log.Fatal(err)
+		}
+		if x == nil {
+			log.Fatal(count)
+		}
+		count--
+	}
+	rs.Close()
+}
+
+func BenchmarkBasic(b *testing.B) {
+	se := prepareBenchSession()
 	for i := 0; i < b.N; i++ {
 		rs, err := se.Execute("select 1")
 		if err != nil {
 			b.Fatal(err)
 		}
-		row, err := rs[0].Next()
-		if err != nil || row == nil {
+		readResult(rs[0], 1)
+	}
+}
+
+func BenchmarkTableScan(b *testing.B) {
+	b.StopTimer()
+	se := prepareBenchSession()
+	prepareBenchData(se, "int", "%v", smallCount)
+	b.StartTimer()
+	for i := 0; i < b.N; i++ {
+		rs, err := se.Execute("select * from t")
+		if err != nil {
+			b.Fatal(err)
+		}
+		readResult(rs[0], smallCount)
+	}
+}
+
+func BenchmarkTableLookup(b *testing.B) {
+	b.StopTimer()
+	se := prepareBenchSession()
+	prepareBenchData(se, "int", "%d", smallCount)
+	b.StartTimer()
+	for i := 0; i < b.N; i++ {
+		rs, err := se.Execute("select * from t where pk = 64")
+		if err != nil {
+			b.Fatal(err)
+		}
+		readResult(rs[0], 1)
+	}
+}
+
+func BenchmarkStringIndexScan(b *testing.B) {
+	b.StopTimer()
+	se := prepareBenchSession()
+	prepareBenchData(se, "varchar(255)", "'hello %d'", smallCount)
+	b.StartTimer()
+	for i := 0; i < b.N; i++ {
+		rs, err := se.Execute("select * from t where col > 'hello'")
+		if err != nil {
+			b.Fatal(err)
+		}
+		readResult(rs[0], smallCount)
+	}
+}
+
+func BenchmarkStringIndexLookup(b *testing.B) {
+	b.StopTimer()
+	se := prepareBenchSession()
+	prepareBenchData(se, "varchar(255)", "'hello %d'", smallCount)
+	b.StartTimer()
+	for i := 0; i < b.N; i++ {
+		rs, err := se.Execute("select * from t where col = 'hello 64'")
+		if err != nil {
+			b.Fatal(err)
+		}
+		readResult(rs[0], 1)
+	}
+}
+
+func BenchmarkIntegerIndexScan(b *testing.B) {
+	b.StopTimer()
+	se := prepareBenchSession()
+	prepareBenchData(se, "int", "%v", smallCount)
+	b.StartTimer()
+	for i := 0; i < b.N; i++ {
+		rs, err := se.Execute("select * from t where col >= 0")
+		if err != nil {
+			b.Fatal(err)
+		}
+		readResult(rs[0], smallCount)
+	}
+}
+
+func BenchmarkIntegerIndexLookup(b *testing.B) {
+	b.StopTimer()
+	se := prepareBenchSession()
+	prepareBenchData(se, "int", "%v", smallCount)
+	b.StartTimer()
+	for i := 0; i < b.N; i++ {
+		rs, err := se.Execute("select * from t where col = 64")
+		if err != nil {
+			b.Fatal(err)
+		}
+		readResult(rs[0], 1)
+	}
+}
+
+func BenchmarkDecimalIndexScan(b *testing.B) {
+	b.StopTimer()
+	se := prepareBenchSession()
+	prepareBenchData(se, "decimal(32,6)", "%v.1234", smallCount)
+	b.StartTimer()
+	for i := 0; i < b.N; i++ {
+		rs, err := se.Execute("select * from t where col >= 0")
+		if err != nil {
+			b.Fatal(err)
+		}
+		readResult(rs[0], smallCount)
+	}
+}
+
+func BenchmarkDecimalIndexLookup(b *testing.B) {
+	b.StopTimer()
+	se := prepareBenchSession()
+	prepareBenchData(se, "decimal(32,6)", "%v.1234", smallCount)
+	b.StartTimer()
+	for i := 0; i < b.N; i++ {
+		rs, err := se.Execute("select * from t where col = 64.1234")
+		if err != nil {
 			b.Fatal(err)
 		}
-		rs[0].Close()
+		readResult(rs[0], 1)
 	}
 }