54 lines
1.3 KiB
Go
54 lines
1.3 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"sync"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Benchmark to store benchmark state
|
||
|
|
type Benchmark struct {
|
||
|
|
duration time.Duration
|
||
|
|
connections int
|
||
|
|
databases []*Database
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewBenchmark connects to the database then creates a Benchmark struct
|
||
|
|
func NewBenchmark(connections int, duration time.Duration, driver string, dsn string, query string, reconnect bool) (*Benchmark, error) {
|
||
|
|
var databases []*Database
|
||
|
|
for i := 0; i < connections; i++ {
|
||
|
|
database, err := NewDatabase(driver, dsn, query, reconnect)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
databases = append(databases, database)
|
||
|
|
}
|
||
|
|
return &Benchmark{
|
||
|
|
duration: duration,
|
||
|
|
connections: connections,
|
||
|
|
databases: databases,
|
||
|
|
}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// Run performs the benchmark by runing queries for a duration
|
||
|
|
func (b *Benchmark) Run() {
|
||
|
|
wg := new(sync.WaitGroup)
|
||
|
|
wg.Add(b.connections)
|
||
|
|
for _, database := range b.databases {
|
||
|
|
go database.Run(b.duration, wg)
|
||
|
|
}
|
||
|
|
wg.Wait()
|
||
|
|
}
|
||
|
|
|
||
|
|
// Queries returns the number of executed queries during the benchmark
|
||
|
|
func (b *Benchmark) Queries() (queries float64) {
|
||
|
|
for _, database := range b.databases {
|
||
|
|
queries = queries + database.Queries()
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
// QueriesPerSecond returns the number of executed queries per second during the benchmark
|
||
|
|
func (b *Benchmark) QueriesPerSecond() float64 {
|
||
|
|
return b.Queries() / b.duration.Seconds()
|
||
|
|
}
|