65 lines
1.5 KiB
Go
65 lines
1.5 KiB
Go
package repository
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
)
|
|
|
|
type latencyHistogramBucket struct {
|
|
upperMs int
|
|
label string
|
|
}
|
|
|
|
var latencyHistogramBuckets = []latencyHistogramBucket{
|
|
{upperMs: 100, label: "0-100ms"},
|
|
{upperMs: 200, label: "100-200ms"},
|
|
{upperMs: 500, label: "200-500ms"},
|
|
{upperMs: 1000, label: "500-1000ms"},
|
|
{upperMs: 2000, label: "1000-2000ms"},
|
|
{upperMs: 0, label: "2000ms+"}, // default bucket
|
|
}
|
|
|
|
var latencyHistogramOrderedRanges = func() []string {
|
|
out := make([]string, 0, len(latencyHistogramBuckets))
|
|
for _, b := range latencyHistogramBuckets {
|
|
out = append(out, b.label)
|
|
}
|
|
return out
|
|
}()
|
|
|
|
func latencyHistogramRangeCaseExpr(column string) string {
|
|
var sb strings.Builder
|
|
_, _ = sb.WriteString("CASE\n")
|
|
|
|
for _, b := range latencyHistogramBuckets {
|
|
if b.upperMs <= 0 {
|
|
continue
|
|
}
|
|
_, _ = sb.WriteString(fmt.Sprintf("\tWHEN %s < %d THEN '%s'\n", column, b.upperMs, b.label))
|
|
}
|
|
|
|
// Default bucket.
|
|
last := latencyHistogramBuckets[len(latencyHistogramBuckets)-1]
|
|
_, _ = sb.WriteString(fmt.Sprintf("\tELSE '%s'\n", last.label))
|
|
_, _ = sb.WriteString("END")
|
|
return sb.String()
|
|
}
|
|
|
|
func latencyHistogramRangeOrderCaseExpr(column string) string {
|
|
var sb strings.Builder
|
|
_, _ = sb.WriteString("CASE\n")
|
|
|
|
order := 1
|
|
for _, b := range latencyHistogramBuckets {
|
|
if b.upperMs <= 0 {
|
|
continue
|
|
}
|
|
_, _ = sb.WriteString(fmt.Sprintf("\tWHEN %s < %d THEN %d\n", column, b.upperMs, order))
|
|
order++
|
|
}
|
|
|
|
_, _ = sb.WriteString(fmt.Sprintf("\tELSE %d\n", order))
|
|
_, _ = sb.WriteString("END")
|
|
return sb.String()
|
|
}
|