65 lines
2.0 KiB
Go
65 lines
2.0 KiB
Go
package main
|
|
|
|
/*
|
|
* ----------------------------------------------------------------------------
|
|
* "THE BEER-WARE LICENSE" (Revision 42):
|
|
* <mitch@nerdfortress.dev> wrote this file. As long as you retain this notice you
|
|
* can do whatever you want with this stuff. If we meet some day, and you think
|
|
* this stuff is worth it, you can buy me a beer in return
|
|
* ----------------------------------------------------------------------------
|
|
*/
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"time"
|
|
)
|
|
|
|
func main() {
|
|
hostPtr := flag.String("host", "", "Redis Server FQDN:port")
|
|
usernamePtr := flag.String("username", "", "Redis Server username")
|
|
passwordPtr := flag.String("password", "", "Redis user password")
|
|
//dbPtr := flag.Int("db", 0, "Redis db, default 0")
|
|
initializeDBPtr := flag.Bool("initialize", false, "Boolean initialize db, default false")
|
|
ratePtr := flag.Int("rate", 50, "Test rate limit, default 30/sec")
|
|
burstPtr := flag.Bool("burst", true, "Boolean burst test default true")
|
|
//todo this should be off that same csv rather than an input
|
|
maxEntriesPtr := flag.Int("dbEntries", 50000, "Test rate limit, default 50/sec")
|
|
testDurationPtr := flag.Int("duration", 50, "Duration of each test")
|
|
|
|
flag.Parse()
|
|
host := *hostPtr
|
|
username := *usernamePtr
|
|
password := *passwordPtr
|
|
//db := *dbPtr
|
|
initializeDB := *initializeDBPtr
|
|
rate := *ratePtr
|
|
burst := *burstPtr
|
|
keyMaxValue := *maxEntriesPtr
|
|
durationInt := *testDurationPtr
|
|
duration := time.Second * time.Duration(durationInt)
|
|
|
|
pool := newPool(host, username, password)
|
|
client := pool.Get()
|
|
defer client.Close()
|
|
|
|
if initializeDB {
|
|
buildTestData(&client, 50000)
|
|
}
|
|
client.Close()
|
|
//refreshing state for next test
|
|
pool = newPool(host, username, password)
|
|
client = pool.Get()
|
|
defer client.Close()
|
|
getRateTest(&client, rate, keyMaxValue, duration)
|
|
client.Close()
|
|
//refreshing state for next test
|
|
pool = newPool(host, username, password)
|
|
client = pool.Get()
|
|
defer client.Close()
|
|
if burst {
|
|
getBurstTest(&client, rate, keyMaxValue, duration)
|
|
}
|
|
fmt.Println("Tests completed...")
|
|
}
|