Function BenchmarkStorage

xenomint/sqlite/sqlite_test.go:453–662

(b *testing.B)

}

func BenchmarkStorage(b *testing.B) {
	var (
		st, q, dm, e, src = setupBenchmarkStorage(b)

		tx   *sql.Tx
		dest = dm()
		read = func(b *testing.B, conn *sql.DB, dest []interface{}) {
			var err error
			if err = conn.QueryRow(q, rrkg.next()).Scan(dest...); err != nil {
				b.Fatalf("failed to query values: %v", err)
			}
		}
		readTx = func(b *testing.B, i int, conn *sql.DB, dest []interface{}) {
			var err error
			if i%benchmarkQueriesPerTx == 0 {
				if tx, err = conn.Begin(); err != nil {
					b.Fatalf("failed to begin transaction: %v", err)
				}
			}
			// Query in [n, 2n-1] key space
			if err = tx.QueryRow(q, nrkg.next()).Scan(dest...); err != nil && err != sql.ErrNoRows {
				b.Fatalf("failed to query values: %v", err)
			}
			if (i+1)%benchmarkQueriesPerTx == 0 || i == b.N-1 {
				if err = tx.Rollback(); err != nil {
					b.Fatalf("failed to close transaction: %v", err)
				}
			}
		}
		write = func(b *testing.B, conn *sql.DB) {
			var err error
			if _, err = conn.Exec(e, src[ipkg.next()]...); err != nil {
				b.Errorf("failed to execute: %v", err)
			}
		}
		writeTx = func(b *testing.B, i int, conn *sql.DB) {
			var err error
			if i%benchmarkQueriesPerTx == 0 {
				if tx, err = st.Writer().Begin(); err != nil {
					b.Errorf("failed to begin transaction: %v", err)
				}
			}
			if _, err = tx.Exec(e, src[ipkg.next()]...); err != nil {
				b.Errorf("failed to execute: %v", err)
			}
			if (i+1)%benchmarkQueriesPerTx == 0 || i == b.N-1 {
				if err = tx.Commit(); err != nil {
					b.Errorf("failed to commit transaction: %v", err)
				}
			}
		}
		mixRW = func(b *testing.B, rconn, wconn *sql.DB, dest []interface{}) {
			if rand.Int()%2 == 0 {
				read(b, rconn, dest)
			} else {
				write(b, wconn)
			}
		}