1 files changed, 24 insertions, 0 deletions
diff --git a/cmd/bolt/bench/generate.go b/cmd/bolt/bench/generate.go
new file mode 100644
index 0000000..8c5554d
--- /dev/null
+++ b/cmd/bolt/bench/generate.go
@@ -0,0 +1,24 @@
+package bench
+
+import (
+	"fmt"
+	"strings"
+
+	"github.com/boltdb/bolt"
+)
+
+// Generate and write data to specified number of buckets/items.
+func GenerateDB(db *bolt.DB, numBuckets, numItemsPerBucket int) error {
+	return db.Update(func(tx *bolt.Tx) error {
+		for bucketIndex := 0; bucketIndex < numBuckets; bucketIndex++ {
+			bucketName := fmt.Sprintf("bucket%08d")
+			tx.CreateBucket([]byte(bucketName))
+			bucket := tx.Bucket([]byte(bucketName))
+			for i := 0; i < numItemsPerBucket; i++ {
+				value := []byte(strings.Repeat("0", 100))
+				bucket.Put([]byte(fmt.Sprintf("key%08d", i)), value)
+			}
+		}
+		return nil
+	})
+}