initial bench and generate command structure

1 files changed, 32 insertions, 0 deletions

diff --git a/cmd/bolt/generate.go b/cmd/bolt/generate.go
new file mode 100644
index 0000000..ecd391e
--- /dev/null
+++ b/cmd/bolt/generate.go
@@ -0,0 +1,32 @@
+package main
+
+import (
+	"bufio"
+	"fmt"
+	"os"
+	"strings"
+)
+
+// Generate data for benchmarks.
+func Generate(numEvents int, destPath string) {
+	f, err := os.Create(destPath)
+	if err != nil {
+		fatal(err)
+	}
+	defer func() {
+		if err := f.Close(); err != nil {
+			fatal(err)
+		}
+	}()
+	w := bufio.NewWriter(f)
+
+	for i := 0; i < numEvents; i++ {
+		if _, err := w.Write([]byte(fmt.Sprintf("key%d:%s\n", i, strings.Repeat("0", 64)))); err != nil {
+			fatal(err)
+		}
+	}
+
+	if err = w.Flush(); err != nil {
+		fatal(err)
+	}
+}