137 lines
4.0 KiB
Go
137 lines
4.0 KiB
Go
package tests
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"net/url"
|
|
"testing"
|
|
)
|
|
|
|
var strResult string
|
|
|
|
func BenchmarkServer_Query_Count_1(b *testing.B) { benchmarkServerQueryCount(b, 1) }
|
|
func BenchmarkServer_Query_Count_1K(b *testing.B) { benchmarkServerQueryCount(b, 1000) }
|
|
func BenchmarkServer_Query_Count_100K(b *testing.B) { benchmarkServerQueryCount(b, 100000) }
|
|
func BenchmarkServer_Query_Count_1M(b *testing.B) { benchmarkServerQueryCount(b, 1000000) }
|
|
|
|
func benchmarkServerQueryCount(b *testing.B, pointN int) {
|
|
if _, err := benchServer.Query(`DROP MEASUREMENT cpu`); err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
|
|
// Write data into server.
|
|
var buf bytes.Buffer
|
|
for i := 0; i < pointN; i++ {
|
|
fmt.Fprintf(&buf, `cpu value=100 %d`, i+1)
|
|
if i != pointN-1 {
|
|
fmt.Fprint(&buf, "\n")
|
|
}
|
|
}
|
|
benchServer.MustWrite("db0", "rp0", buf.String(), nil)
|
|
|
|
// Query simple count from server.
|
|
b.ResetTimer()
|
|
b.ReportAllocs()
|
|
var err error
|
|
for i := 0; i < b.N; i++ {
|
|
if strResult, err = benchServer.Query(`SELECT count(value) FROM db0.rp0.cpu`); err != nil {
|
|
b.Fatal(err)
|
|
} else if strResult != fmt.Sprintf(`{"results":[{"statement_id":0,"series":[{"name":"cpu","columns":["time","count"],"values":[["1970-01-01T00:00:00Z",%d]]}]}]}`, pointN) {
|
|
b.Fatalf("unexpected result: %s", strResult)
|
|
}
|
|
}
|
|
}
|
|
|
|
func BenchmarkServer_Query_Count_Where_500(b *testing.B) {
|
|
benchmarkServerQueryCountWhere(b, false, 500)
|
|
}
|
|
func BenchmarkServer_Query_Count_Where_1K(b *testing.B) {
|
|
benchmarkServerQueryCountWhere(b, false, 1000)
|
|
}
|
|
func BenchmarkServer_Query_Count_Where_10K(b *testing.B) {
|
|
benchmarkServerQueryCountWhere(b, false, 10000)
|
|
}
|
|
func BenchmarkServer_Query_Count_Where_100K(b *testing.B) {
|
|
benchmarkServerQueryCountWhere(b, false, 100000)
|
|
}
|
|
|
|
func BenchmarkServer_Query_Count_Where_Regex_500(b *testing.B) {
|
|
benchmarkServerQueryCountWhere(b, true, 500)
|
|
}
|
|
func BenchmarkServer_Query_Count_Where_Regex_1K(b *testing.B) {
|
|
benchmarkServerQueryCountWhere(b, true, 1000)
|
|
}
|
|
func BenchmarkServer_Query_Count_Where_Regex_10K(b *testing.B) {
|
|
benchmarkServerQueryCountWhere(b, true, 10000)
|
|
}
|
|
func BenchmarkServer_Query_Count_Where_Regex_100K(b *testing.B) {
|
|
benchmarkServerQueryCountWhere(b, true, 100000)
|
|
}
|
|
|
|
func benchmarkServerQueryCountWhere(b *testing.B, useRegex bool, pointN int) {
|
|
if _, err := benchServer.Query(`DROP MEASUREMENT cpu`); err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
|
|
// Write data into server.
|
|
var buf bytes.Buffer
|
|
for i := 0; i < pointN; i++ {
|
|
fmt.Fprintf(&buf, `cpu,host=server-%d value=100 %d`, i, i)
|
|
if i != pointN-1 {
|
|
fmt.Fprint(&buf, "\n")
|
|
}
|
|
}
|
|
benchServer.MustWrite("db0", "rp0", buf.String(), nil)
|
|
|
|
// Query count from server with WHERE
|
|
var (
|
|
err error
|
|
condition = `host = 'server-487'`
|
|
)
|
|
|
|
if useRegex {
|
|
condition = `host =~ /^server-487$/`
|
|
}
|
|
|
|
b.ResetTimer()
|
|
b.ReportAllocs()
|
|
for i := 0; i < b.N; i++ {
|
|
if strResult, err = benchServer.Query(fmt.Sprintf(`SELECT count(value) FROM db0.rp0.cpu WHERE %s`, condition)); err != nil {
|
|
b.Fatal(err)
|
|
} else if strResult == `{"results":[{}]}` {
|
|
b.Fatal("no results")
|
|
}
|
|
}
|
|
}
|
|
|
|
func BenchmarkServer_ShowSeries_1(b *testing.B) { benchmarkServerShowSeries(b, 1) }
|
|
func BenchmarkServer_ShowSeries_1K(b *testing.B) { benchmarkServerShowSeries(b, 1000) }
|
|
func BenchmarkServer_ShowSeries_100K(b *testing.B) { benchmarkServerShowSeries(b, 100000) }
|
|
func BenchmarkServer_ShowSeries_1M(b *testing.B) { benchmarkServerShowSeries(b, 1000000) }
|
|
|
|
func benchmarkServerShowSeries(b *testing.B, pointN int) {
|
|
if _, err := benchServer.Query(`DROP MEASUREMENT cpu`); err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
|
|
// Write data into server.
|
|
var buf bytes.Buffer
|
|
for i := 0; i < pointN; i++ {
|
|
fmt.Fprintf(&buf, `cpu,host=server%d value=100 %d`, i, i+1)
|
|
if i != pointN-1 {
|
|
fmt.Fprint(&buf, "\n")
|
|
}
|
|
}
|
|
benchServer.MustWrite("db0", "rp0", buf.String(), nil)
|
|
|
|
// Query simple count from server.
|
|
b.ResetTimer()
|
|
b.ReportAllocs()
|
|
var err error
|
|
for i := 0; i < b.N; i++ {
|
|
if strResult, err = benchServer.QueryWithParams(`SHOW SERIES`, url.Values{"db": {"db0"}}); err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
}
|
|
}
|