From a0aad147e0c537e768b8fed75aa50b5f0ba0e7eb Mon Sep 17 00:00:00 2001
From: jaffee <jaffee@umbel.com>
Date: Tue, 3 Jan 2017 16:54:02 -0600
Subject: [PATCH] re-arrange bagent output

---
 bench/bench.go        | 19 +++++++++++--------
 cmd/pilosactl/main.go |  2 +-
 2 files changed, 12 insertions(+), 9 deletions(-)

diff --git a/bench/bench.go b/bench/bench.go
index 7949ae062..23e49fb3e 100644
--- a/bench/bench.go
+++ b/bench/bench.go
@@ -115,18 +115,21 @@ func (sb *serialBenchmark) Init(hosts []string, agentNum int) error {
 // top level keys are the indices of each benchmark in the list of benchmarks,
 // and the values are the results of each benchmark's Run method.
 func (sb *serialBenchmark) Run(ctx context.Context, agentNum int) map[string]interface{} {
-	results := make(map[string]interface{}, len(sb.benchmarkers))
-	runtimes := make(map[string]interface{})
+	benchmarks := make([]map[string]interface{}, len(sb.benchmarkers))
+	results := map[string]interface{}{"benchmarks": benchmarks}
+
 	total_start := time.Now()
 	for i, b := range sb.benchmarkers {
 		start := time.Now()
-		ret := b.Run(ctx, agentNum)
-		end := time.Now()
-		results[strconv.Itoa(i)] = ret
-		runtimes[strconv.Itoa(i)] = end.Sub(start)
+		output := b.Run(ctx, agentNum)
+		if _, ok := output["runtime"]; ok {
+			panic(fmt.Sprintf("Benchmark %v added 'runtime' to its results", b))
+		}
+		output["runtime"] = time.Now().Sub(start)
+		ret := map[string]interface{}{"output": output, "metadata": b}
+		benchmarks[i] = ret
 	}
-	runtimes["total"] = time.Now().Sub(total_start)
-	results["runtimes"] = runtimes
+	results["total_runtime"] = time.Now().Sub(total_start)
 	return results
 }
 
diff --git a/cmd/pilosactl/main.go b/cmd/pilosactl/main.go
index e606ab9c6..90d3915fa 100644
--- a/cmd/pilosactl/main.go
+++ b/cmd/pilosactl/main.go
@@ -1293,7 +1293,7 @@ func (cmd *BagentCommand) Run(ctx context.Context) error {
 	}
 
 	res := sbm.Run(ctx, cmd.AgentNum)
-	res["metadata"] = cmd
+	res["agent-num"] = cmd.AgentNum
 	if cmd.RunUUID != "" {
 		res["run-uuid"] = cmd.RunUUID
 	}