From 71e91e24b064c1c8e33d92b7f324b6cd57d07f87 Mon Sep 17 00:00:00 2001 From: Steven Normore Date: Mon, 14 Apr 2014 16:25:04 +0000 Subject: move bench package to bench/cmd/bolt/bench --- cmd/bolt/bench/generate.go | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) create mode 100644 cmd/bolt/bench/generate.go (limited to 'cmd/bolt/bench/generate.go') 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 + }) +} -- cgit v1.2.3