zishuo/go-sqlite
1
0
Fork 0
mirror of https://github.com/glebarez/go-sqlite.git synced 2025-10-06 16:26:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

Go vs CGo benchmarks refactored (not TPCH)

Browse Source
This commit is contained in:
glebarez
2021-12-16 12:50:41 +00:00
parent 019ed373da
commit abe4fa449b
15 changed files with 1473 additions and 189 deletions
Download Patch File Download Diff File Expand all files Collapse all files

191
cgo_test.go
View File

@@ -2,14 +2,12 @@
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
//go:build (cgo && cgobench) || ignore || (cgo && cgotest)
// +build cgo,cgobench ignore cgo,cgotest
//go:build ignore || (cgo && cgotest)
// +build ignore cgo,cgotest
package sqlite // import "modernc.org/sqlite"
import (
"database/sql"
"fmt"
"io/ioutil"
"os"
"path/filepath"
@@ -18,191 +16,6 @@ import (
_ "github.com/mattn/go-sqlite3"
)
const gcoDriver = "sqlite3"
var drivers = []string{
driverName,
gcoDriver,
}
var inMemory = []bool{
true,
false,
}
func makename(inMemory bool, driver string, e int) string {
name := driver
if inMemory {
name += "InMemory"
} else {
name += "OnDisk"
}
return fmt.Sprintf("%s1e%d", name, e)
}
func benchmarkRead(b *testing.B, drivername, file string, n int) {
os.Remove(file)
db, err := sql.Open(drivername, file)
if err != nil {
b.Fatal(err)
}
defer func() {
db.Close()
}()
if _, err := db.Exec(`
create table t(i int);
begin;
`); err != nil {
b.Fatal(err)
}
s, err := db.Prepare("insert into t values(?)")
if err != nil {
b.Fatal(err)
}
defer s.Close()
for i := 0; i < n; i++ {
if _, err := s.Exec(int64(i)); err != nil {
b.Fatal(err)
}
}
if _, err := db.Exec("commit"); err != nil {
b.Fatal(err)
}
dst := 0
b.ReportAllocs()
b.ResetTimer()
for i := 0; i < b.N; i++ {
b.StopTimer()
r, err := db.Query("select * from t")
if err != nil {
b.Fatal(err)
}
b.StartTimer()
for i := 0; i < n; i++ {
if !r.Next() {
b.Fatal(r.Err())
}
err = r.Scan(&dst)
if err != nil {
b.Fatal(err)
}
}
b.StopTimer()
r.Close()
}
b.StopTimer()
if *oRecsPerSec {
b.SetBytes(1e6 * int64(n))
}
}
func BenchmarkReading1(b *testing.B) {
dir := b.TempDir()
for _, memory := range inMemory {
filename := "file::memory:"
if !memory {
filename = filepath.Join(dir, "test.db")
}
for _, driver := range drivers {
for i, n := range []int{1e1, 1e2, 1e3, 1e4, 1e5, 1e6} {
b.Run(makename(memory, driver, i+1), func(b *testing.B) {
benchmarkRead(b, driver, filename, n)
if !memory {
err := os.Remove(filename)
if err != nil {
b.Fatal(err)
}
}
})
}
}
}
}
func benchmarkInsertComparative(b *testing.B, drivername, file string, n int) {
os.Remove(file)
db, err := sql.Open(drivername, file)
if err != nil {
b.Fatal(err)
}
defer func() {
db.Close()
}()
if _, err := db.Exec(`
create table t(i int);
`); err != nil {
b.Fatal(err)
}
s, err := db.Prepare("insert into t values(?)")
if err != nil {
b.Fatal(err)
}
defer s.Close()
b.ReportAllocs()
b.ResetTimer()
for i := 0; i < b.N; i++ {
b.StopTimer()
if _, err := db.Exec("begin"); err != nil {
b.Fatal(err)
}
if err, _ := db.Exec("delete * from t"); err != nil {
b.Fatal(err)
}
b.StartTimer()
for i := 0; i < n; i++ {
if _, err := s.Exec(int64(i)); err != nil {
b.Fatal(err)
}
}
if _, err := db.Exec("commit"); err != nil {
b.Fatal(err)
}
b.StopTimer()
}
b.StopTimer()
if *oRecsPerSec {
b.SetBytes(1e6 * int64(n))
}
}
// https://gitlab.com/cznic/sqlite/-/issues/39
func BenchmarkInsertComparative(b *testing.B) {
dir := b.TempDir()
for _, memory := range inMemory {
filename := "file::memory:"
if !memory {
filename = filepath.Join(dir, "test.db")
}
for _, driver := range drivers {
for i, n := range []int{1e1, 1e2, 1e3, 1e4, 1e5, 1e6} {
b.Run(makename(memory, driver, i+1), func(b *testing.B) {
benchmarkInsertComparative(b, driver, filename, n)
if !memory {
err := os.Remove(filename)
if err != nil {
b.Fatal(err)
}
}
})
}
}
}
}
// https://gitlab.com/cznic/sqlite/-/issues/65
func TestIssue65CGo(t *testing.T) {
tempDir, err := ioutil.TempDir("", "")