func dump(dumper, inputs, output string) error {
dump, ok := dumpers[dumper]
if !ok {
return fmt.Errorf("unsupported dumper: %s", dumper)
}
files := strings.Split(inputs, ",")
srcs := make([]io.Reader, len(files))
for i, f := range files {
in, err := file(f, false)
if err != nil {
return err
}
defer in.Close()
srcs[i] = in
}
out, err := file(output, true)
if err != nil {
return err
}
defer out.Close()
sig := make(chan os.Signal, 1)
signal.Notify(sig, os.Interrupt)
res, errs := vegeta.Collect(srcs...)
for {
select {
case _ = <-sig:
return nil
case r, ok := <-res:
if !ok {
return nil
}
dmp, err := dump.Dump(r)
if err != nil {
return err
} else if _, err = out.Write(dmp); err != nil {
return err
}
case err, ok := <-errs:
if !ok {
return nil
}
return err
}
}
}
// report validates the report arguments, sets up the required resources
// and writes the report
func report(reporter, inputs, output string) error {
if len(reporter) < 4 {
return fmt.Errorf("bad reporter: %s", reporter)
}
var rep vegeta.Reporter
switch reporter[:4] {
case "text":
rep = vegeta.ReportText
case "json":
rep = vegeta.ReportJSON
case "plot":
rep = vegeta.ReportPlot
case "hist":
if len(reporter) < 6 {
return fmt.Errorf("bad buckets: '%s'", reporter[4:])
}
var hist vegeta.HistogramReporter
if err := hist.Set(reporter[4:]); err != nil {
return err
}
rep = hist
}
files := strings.Split(inputs, ",")
srcs := make([]io.Reader, len(files))
for i, f := range files {
in, err := file(f, false)
if err != nil {
return err
}
defer in.Close()
srcs[i] = in
}
out, err := file(output, true)
if err != nil {
return err
}
defer out.Close()
var results vegeta.Results
res, errs := vegeta.Collect(srcs...)
sig := make(chan os.Signal, 1)
signal.Notify(sig, os.Interrupt)
outer:
for {
select {
case _ = <-sig:
break outer
case r, ok := <-res:
if !ok {
break outer
}
results = append(results, r)
case err, ok := <-errs:
if !ok {
break outer
}
return err
}
}
sort.Sort(results)
data, err := rep.Report(results)
if err != nil {
return err
}
_, err = out.Write(data)
return err
}