package main
import (
"bufio"
"bytes"
"flag"
htemplate "html/template"
"os"
"sort"
"strconv"
ttemplate "text/template"
)
func main() {
outputCSV := flag.Bool("csv", false, "output CSV instead of HTML")
flag.Parse()
var data = make(map[int][]int)
for _, fname := range flag.Args() {
f, _ := os.Open(fname)
b := bufio.NewReader(f)
scanner := bufio.NewScanner(b)
// skip first line with the header
scanner.Scan()
for scanner.Scan() {
line := scanner.Bytes()
parts := bytes.Split(line, []byte{','})
epochms, reqms := parts[3], parts[4]
epoch, _ := strconv.Atoi(string(bytes.TrimSpace(epochms)))
epoch /= 1000 * 60
latency, _ := strconv.Atoi(string(bytes.TrimSpace(reqms)))
data[epoch] = append(data[epoch], latency)
}
}
var epochs []int
for k, v := range data {
epochs = append(epochs, k)
sort.Ints(v)
}
sort.Ints(epochs)
var graphdata [][]int
for _, epoch := range epochs {
l := len(data[epoch])
graphdata = append(graphdata, []int{epoch * 1000 * 60, data[epoch][(50*l)/100], data[epoch][(75*l)/100], data[epoch][(95*l)/100], data[epoch][(99*l)/100], l})
}
if *outputCSV {
csvTmpl.Execute(os.Stdout, graphdata)
return
}
reportTmpl.Execute(os.Stdout, graphdata)
}
func joinInts(ints []int) string {
b := make([]byte, 0, len(ints)*21)
needComma := false
for _, v := range ints {
if needComma {
b = append(b, ',')
}
b = strconv.AppendInt(b, int64(v), 10)
needComma = true
}
return string(b)
}
var csvTmpl = ttemplate.Must(ttemplate.New("report").Funcs(ttemplate.FuncMap{"joinInts": joinInts}).Parse("" +
`epoch,p50,p75,p95,p99,requests
{{ range . }}{{ joinInts . }}
{{ end }}`))
var reportTmpl = htemplate.Must(htemplate.New("report").Parse(`
`))