add gobench output option

Signed-off-by: Thomas Jungblut <[email protected]>

cmd/bbolt/main.go

@@ -16,6 +16,7 @@ import (
 	"strconv"
 	"strings"
 	"sync/atomic"
+	"testing"
 	"time"
 	"unicode"
 	"unicode/utf8"
@@ -1050,9 +1051,23 @@ func (cmd *benchCommand) Run(args ...string) error {
 	}
 
 	// Print results.
-	fmt.Fprintf(cmd.Stderr, "# Write\t%v(ops)\t%v\t(%v/op)\t(%v op/sec)\n", writeResults.CompletedOps(), writeResults.Duration(), writeResults.OpDuration(), writeResults.OpsPerSecond())
-	fmt.Fprintf(cmd.Stderr, "# Read\t%v(ops)\t%v\t(%v/op)\t(%v op/sec)\n", readResults.CompletedOps(), readResults.Duration(), readResults.OpDuration(), readResults.OpsPerSecond())
+	if options.GoBenchOutput {
+		// below replicates the output of testing.B benchmarks, e.g. for external tooling
+		maxLen := 5 // the length of "Write"
+		gobenchResult := testing.BenchmarkResult{}
+		gobenchResult.T = writeResults.Duration()
+		gobenchResult.N = int(writeResults.CompletedOps())
+		fmt.Fprintf(cmd.Stderr, "%-*s\t%s\n", maxLen, "Write", gobenchResult.String())
+		gobenchResult = testing.BenchmarkResult{}
+		gobenchResult.T = readResults.Duration()
+		gobenchResult.N = int(readResults.CompletedOps())
+		fmt.Fprintf(cmd.Stderr, "%-*s\t%s\n", maxLen, "Read", gobenchResult.String())
+	} else {
+		fmt.Fprintf(cmd.Stderr, "# Write\t%v(ops)\t%v\t(%v/op)\t(%v op/sec)\n", writeResults.CompletedOps(), writeResults.Duration(), writeResults.OpDuration(), writeResults.OpsPerSecond())
+		fmt.Fprintf(cmd.Stderr, "# Read\t%v(ops)\t%v\t(%v/op)\t(%v op/sec)\n", readResults.CompletedOps(), readResults.Duration(), readResults.OpDuration(), readResults.OpsPerSecond())
+	}
 	fmt.Fprintln(cmd.Stderr, "")
+
 	return nil
 }
 
@@ -1076,6 +1091,7 @@ func (cmd *benchCommand) ParseFlags(args []string) (*BenchOptions, error) {
 	fs.BoolVar(&options.NoSync, "no-sync", false, "")
 	fs.BoolVar(&options.Work, "work", false, "")
 	fs.StringVar(&options.Path, "path", "", "")
+	fs.BoolVar(&options.GoBenchOutput, "gobench-output", false, "")
 	fs.SetOutput(cmd.Stderr)
 	if err := fs.Parse(args); err != nil {
 		return nil, err
@@ -1555,6 +1571,7 @@ type BenchOptions struct {
 	NoSync        bool
 	Work          bool
 	Path          string
+	GoBenchOutput bool
 }
 
 // BenchResults represents the performance results of the benchmark and is thread-safe.