add bench command

also adds a dedicated simulation test

Signed-off-by: Thomas Jungblut <tjungblu@redhat.com>

Author: tjungblu

cmd/bbolt/command/command_bench.go

@@ -41,6 +41,7 @@ type benchOptions struct {
 	pageSize        int
 	initialMmapSize int
 	deleteFraction  float64 // Fraction of keys of last tx to delete during writes. works only with "seq-del" write mode.
+	compression     bool
 }
 
 func (o *benchOptions) AddFlags(fs *pflag.FlagSet) {
@@ -61,6 +62,7 @@ func (o *benchOptions) AddFlags(fs *pflag.FlagSet) {
 	fs.BoolVar(&o.goBenchOutput, "gobench-output", false, "")
 	fs.IntVar(&o.pageSize, "page-size", common.DefaultPageSize, "Set page size in bytes.")
 	fs.IntVar(&o.initialMmapSize, "initial-mmap-size", 0, "Set initial mmap size in bytes for database file.")
+	fs.BoolVar(&o.compression, "compression", false, "Enables compression.")
 }
 
 // Returns an error if `bench` options are not valid.
@@ -145,6 +147,7 @@ func benchFunc(cmd *cobra.Command, options *benchOptions) error {
 	dbOptions := *bolt.DefaultOptions
 	dbOptions.PageSize = options.pageSize
 	dbOptions.InitialMmapSize = options.initialMmapSize
+	dbOptions.Compression = options.compression
 	db, err := bolt.Open(options.path, 0600, &dbOptions)
 	if err != nil {
 		return err

cmd/bbolt/command/command_bench_test.go

@@ -38,8 +38,9 @@ func TestBenchCommand_Run(t *testing.T) {
 	tests := map[string]struct {
 		args []string
 	}{
-		"no-args":    {},
-		"100k count": {[]string{"--count", "100000"}},
+		"no-args":     {},
+		"100k count":  {[]string{"--count", "100000"}},
+		"compression": {[]string{"--compression"}},
 	}
 
 	for name, test := range tests {

simulation_compression_test.go

@@ -0,0 +1,47 @@
+package bbolt_test
+
+import (
+	"testing"
+
+	bolt "go.etcd.io/bbolt"
+)
+
+func TestSimulateCompression_1op_1p(t *testing.T) {
+	testSimulate(t, &bolt.Options{Compression: true}, 8, 1, 1)
+}
+func TestSimulateCompression_10op_1p(t *testing.T) {
+	testSimulate(t, &bolt.Options{Compression: true}, 8, 10, 1)
+}
+func TestSimulateCompression_100op_1p(t *testing.T) {
+	testSimulate(t, &bolt.Options{Compression: true}, 8, 100, 1)
+}
+func TestSimulateCompression_1000op_1p(t *testing.T) {
+	testSimulate(t, &bolt.Options{Compression: true}, 8, 1000, 1)
+}
+func TestSimulateCompression_10000op_1p(t *testing.T) {
+	testSimulate(t, &bolt.Options{Compression: true}, 8, 10000, 1)
+}
+func TestSimulateCompression_10op_10p(t *testing.T) {
+	testSimulate(t, &bolt.Options{Compression: true}, 8, 10, 10)
+}
+func TestSimulateCompression_100op_10p(t *testing.T) {
+	testSimulate(t, &bolt.Options{Compression: true}, 8, 100, 10)
+}
+func TestSimulateCompression_1000op_10p(t *testing.T) {
+	testSimulate(t, &bolt.Options{Compression: true}, 8, 1000, 10)
+}
+func TestSimulateCompression_10000op_10p(t *testing.T) {
+	testSimulate(t, &bolt.Options{Compression: true}, 8, 10000, 10)
+}
+func TestSimulateCompression_100op_100p(t *testing.T) {
+	testSimulate(t, &bolt.Options{Compression: true}, 8, 100, 100)
+}
+func TestSimulateCompression_1000op_100p(t *testing.T) {
+	testSimulate(t, &bolt.Options{Compression: true}, 8, 1000, 100)
+}
+func TestSimulateCompression_10000op_100p(t *testing.T) {
+	testSimulate(t, &bolt.Options{Compression: true}, 8, 10000, 100)
+}
+func TestSimulateCompression_10000op_1000p(t *testing.T) {
+	testSimulate(t, &bolt.Options{Compression: true}, 8, 10000, 1000)
+}