func dump(dumper, inputs, output string) error {
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
}
dec := vegeta.NewDecoder(srcs...)
out, err := file(output, true)
if err != nil {
return err
}
defer out.Close()
var enc vegeta.Encoder
switch dumper {
case "csv":
enc = vegeta.NewCSVEncoder(out)
case "json":
enc = vegeta.NewJSONEncoder(out)
default:
return fmt.Errorf("unsupported dumper: %s", dumper)
}
for {
var r vegeta.Result
if err = dec.Decode(&r); err != nil {
if err == io.EOF {
break
}
return err
} else if err = enc.Encode(&r); err != nil {
return err
}
}
return nil
}
// 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)
}
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
}
dec := vegeta.NewDecoder(srcs...)
out, err := file(output, true)
if err != nil {
return err
}
defer out.Close()
var (
rep vegeta.Reporter
report vegeta.Report
)
switch reporter[:4] {
case "text":
var m vegeta.Metrics
rep, report = vegeta.NewTextReporter(&m), &m
case "json":
var m vegeta.Metrics
rep, report = vegeta.NewJSONReporter(&m), &m
case "plot":
var rs vegeta.Results
rep, report = vegeta.NewPlotReporter(&rs), &rs
case "hist":
if len(reporter) < 6 {
return fmt.Errorf("bad buckets: '%s'", reporter[4:])
}
var hist vegeta.Histogram
if err := hist.Buckets.UnmarshalText([]byte(reporter[4:])); err != nil {
return err
}
rep, report = vegeta.NewHistogramReporter(&hist), &hist
default:
return fmt.Errorf("unknown reporter: %q", reporter)
}
sigch := make(chan os.Signal, 1)
signal.Notify(sigch, os.Interrupt)
decode:
for {
select {
case <-sigch:
break decode
default:
var r vegeta.Result
if err = dec.Decode(&r); err != nil {
if err == io.EOF {
break decode
}
return err
}
report.Add(&r)
}
}
if c, ok := report.(vegeta.Closer); ok {
c.Close()
}
return rep.Report(out)
}