Julien Riou
c43270a1d4
- Add debug messages - Use mutex to compute statistics - Add UUID to identify database connections - Fix connection handling for MySQL Signed-off-by: Julien Riou <julien@riou.xyz>
95 lines
2.9 KiB
Go
95 lines
2.9 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
"time"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
|
|
_ "github.com/go-sql-driver/mysql"
|
|
_ "github.com/lib/pq"
|
|
)
|
|
|
|
// AppName to store application name
|
|
var AppName string = "benchito"
|
|
|
|
// AppVersion to set version at compilation time
|
|
var AppVersion string = "9999"
|
|
|
|
// GitCommit to set git commit at compilation time (can be empty)
|
|
var GitCommit string
|
|
|
|
// GoVersion to set Go version at compilation time
|
|
var GoVersion string
|
|
|
|
func init() {
|
|
log.SetOutput(os.Stdout)
|
|
}
|
|
|
|
func main() {
|
|
|
|
config := NewConfig()
|
|
|
|
version := flag.Bool("version", false, "Print version and exit")
|
|
quiet := flag.Bool("quiet", false, "Log errors only")
|
|
verbose := flag.Bool("verbose", false, "Print more logs")
|
|
debug := flag.Bool("debug", false, "Print even more logs")
|
|
configFile := flag.String("config", "", "Configuration file")
|
|
flag.StringVar(&config.Driver, "driver", "postgres", "Database driver (postgres or mysql)")
|
|
flag.IntVar(&config.Connections, "connections", 1, "Number of concurrent connections to the database")
|
|
flag.StringVar(&config.Query, "query", "SELECT /* "+AppName+" */ NOW();", "Query to execute for the benchmark")
|
|
flag.DurationVar(&config.Duration, "duration", 1*time.Second, "Duration of the benchmark")
|
|
flag.BoolVar(&config.Reconnect, "reconnect", false, "Force database reconnection between each queries")
|
|
flag.StringVar(&config.DSN, "dsn", "", "Database cpnnection string")
|
|
flag.StringVar(&config.Host, "host", "", "Host address of the database")
|
|
flag.IntVar(&config.Port, "port", 0, "Port of the database")
|
|
flag.StringVar(&config.User, "user", "", "Username of the database")
|
|
flag.StringVar(&config.Password, "password", "", "Password of the database")
|
|
flag.StringVar(&config.Database, "database", "", "Database name")
|
|
flag.StringVar(&config.TLS, "tls", "", "TLS configuration")
|
|
flag.Parse()
|
|
|
|
if *version {
|
|
showVersion()
|
|
return
|
|
}
|
|
|
|
log.SetLevel(log.WarnLevel)
|
|
if *debug {
|
|
log.SetLevel(log.DebugLevel)
|
|
}
|
|
if *verbose {
|
|
log.SetLevel(log.InfoLevel)
|
|
}
|
|
if *quiet {
|
|
log.SetLevel(log.ErrorLevel)
|
|
}
|
|
|
|
if *configFile != "" {
|
|
err := config.Read(*configFile)
|
|
if err != nil {
|
|
log.Fatalf("Failed to read configuration file: %v", err)
|
|
}
|
|
}
|
|
|
|
config.ParseDSN()
|
|
|
|
benchmark, err := NewBenchmark(config.Connections, config.Duration, config.Driver, config.DSN, config.Query, config.Reconnect)
|
|
if err != nil {
|
|
log.Fatalf("Cannot perform benchmark: %v", err)
|
|
}
|
|
benchmark.Run()
|
|
fmt.Printf("Queries: %.0f\n", benchmark.Queries())
|
|
fmt.Printf("Queries per second: %.0f\n", benchmark.QueriesPerSecond())
|
|
fmt.Printf("Average query time: %s\n", benchmark.AverageQueryTime())
|
|
fmt.Printf("Average query time (ms): %d\n", benchmark.AverageQueryTime().Milliseconds())
|
|
}
|
|
|
|
func showVersion() {
|
|
if GitCommit != "" {
|
|
AppVersion = fmt.Sprintf("%s-%s", AppVersion, GitCommit)
|
|
}
|
|
fmt.Printf("%s version %s (compiled with %s)\n", AppName, AppVersion, GoVersion)
|
|
}
|