go-sqlite/cgo_test.go

// Copyright 2017 The Sqlite Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

//+build cgo,cgobench

package sqlite // import "modernc.org/sqlite"

import (
	"database/sql"
	"fmt"
	"os"
	"path/filepath"
	"testing"

	_ "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 reading1Memory(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) {
					reading1Memory(b, driver, filename, n)
					if !memory {
						err := os.Remove(filename)
						if err != nil {
							b.Fatal(err)
						}
					}
				})
			}
		}
	}
}