plot: Reuse decoder helper in plot

tsenart · Aug 18, 2018 · b22f4d4 · b22f4d4
1 parent 5892206
commit b22f4d4
Showing 1 changed file with 4 additions and 9 deletions.
diff --git a/plot.go b/plot.go
@@ -60,16 +60,11 @@ func plotCmd() command {
 }
 
 func plotRun(files []string, threshold int, title, output string) error {
-	srcs := make([]vegeta.Decoder, len(files))
-	for i, f := range files {
-		in, err := file(f, false)
-		if err != nil {
-			return err
-		}
-		defer in.Close()
-		srcs[i] = vegeta.NewDecoder(in)
+	dec, mc, err := decoder(files)
+	defer mc.Close()
+	if err != nil {
+		return err
 	}
-	dec := vegeta.NewRoundRobinDecoder(srcs...)
 
 	out, err := file(output, true)
 	if err != nil {