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)
}