bcd707446a
In order to report results from multiple attack runs, the report sub command now accepts multiple input files. For you to run a distributed attack, use a tool like pdsh to run vegeta attack on different machines and then aggregate the results into one machine where vegeta report would consume them.
64 lines
1.3 KiB
Go
64 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
vegeta "github.com/tsenart/vegeta/lib"
|
|
"log"
|
|
"strings"
|
|
)
|
|
|
|
func reportCmd(args []string) command {
|
|
fs := flag.NewFlagSet("report", flag.ExitOnError)
|
|
reporter := fs.String("reporter", "text", "Reporter [text, json, plot:timings]")
|
|
input := fs.String("input", "stdin", "Input files (comma separated)")
|
|
output := fs.String("output", "stdout", "Output file")
|
|
fs.Parse(args)
|
|
|
|
return func() error {
|
|
return report(*reporter, *input, *output)
|
|
}
|
|
}
|
|
|
|
// report validates the report arguments, sets up the required resources
|
|
// and writes the report
|
|
func report(reporter, input, output string) error {
|
|
var rep vegeta.Reporter
|
|
switch reporter {
|
|
case "text":
|
|
rep = vegeta.ReportText
|
|
case "json":
|
|
rep = vegeta.ReportJSON
|
|
case "plot:timings":
|
|
rep = vegeta.ReportTimingsPlot
|
|
default:
|
|
log.Println("Reporter provided is not supported. Using text")
|
|
rep = vegeta.ReportText
|
|
}
|
|
|
|
all := vegeta.Results{}
|
|
for _, input := range strings.Split(input, ",") {
|
|
in, err := file(input, false)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer in.Close()
|
|
results := vegeta.Results{}
|
|
if err := results.ReadFrom(in); err != nil {
|
|
return err
|
|
}
|
|
all = append(all, results...)
|
|
}
|
|
|
|
out, err := file(output, true)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer out.Close()
|
|
|
|
if err := rep(all.Sort(), out); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|