go benchmarking: setup and cleanup

dbq_setup.go

var db  *sql.DB

// Add DB credentials here
var (
	user   string = ""
	pword  string = ""
	host   string = ""
	port   string = ""
	dbname string = ""
)

func init() {
	db, _ = sql.Open("mysql", fmt.Sprintf("%s:%s@tcp(%s:%s)/%s", user, pword, host, port, dbname))
	db.SetMaxOpenConns(1)
	err := db.Ping()
	if err != nil {
		panic(err)
	}
}

func setup() {
	// Create table
	createQ := `
	CREATE TABLE tests (
		id int(11) unsigned NOT NULL AUTO_INCREMENT,
		name varchar(50) NOT NULL DEFAULT '',
		email varchar(150) NOT NULL DEFAULT '',
		PRIMARY KEY (id)
	)`

	_, err := db.Exec(createQ)
	if err != nil {
		panic(err)
	}

	// Add 10,000 fake entries
	entries := []interface{}{}
	for i := 0; i < 10000; i++ {
		entry := []interface{}{
			i + 1,
			gofakeit.Name(), // Fake name
			gofakeit.Email(), // Fake email
		}
		entries = append(entries, entry)
	}
	stmt := dbq.INSERTStmt("tests", []string{"id", "name", "email"}, len(entries))
	_, err = dbq.E(ctx, db, stmt, nil, entries)
	if err != nil {
		panic(err)
	}
}

func cleanup() {
	// Delete table
	_, err := db.Exec(`DROP TABLE tests`)
	if err != nil {
		panic(err)
	}
}

func Benchmark(b *testing.B) {
	setup()
	defer cleanup()
	
	// Benchmark goes in here
	
}