1 files changed, 0 insertions, 425 deletions
diff --git a/cmd/bolt/bench.go b/cmd/bolt/bench.go
deleted file mode 100644
index 3ade8b8..0000000
--- a/cmd/bolt/bench.go
+++ /dev/null
@@ -1,425 +0,0 @@
-package main
-
-import (
-	"encoding/binary"
-	"encoding/json"
-	"errors"
-	"fmt"
-	"io/ioutil"
-	"math/rand"
-	"os"
-	"runtime"
-	"runtime/pprof"
-	"time"
-
-	"github.com/boltdb/bolt"
-)
-
-// File handlers for the various profiles.
-var cpuprofile, memprofile, blockprofile *os.File
-
-var benchBucketName = []byte("bench")
-
-// Bench executes a customizable, synthetic benchmark against Bolt.
-func Bench(options *BenchOptions) {
-	var results BenchResults
-
-	// Validate options.
-	if options.BatchSize == 0 {
-		options.BatchSize = options.Iterations
-	} else if options.Iterations%options.BatchSize != 0 {
-		fatal("number of iterations must be divisible by the batch size")
-	}
-
-	// Generate temp path if one is not passed in.
-	path := options.Path
-	if path == "" {
-		path = tempfile()
-	}
-
-	if options.Clean {
-		defer os.Remove(path)
-	} else {
-		println("work:", path)
-	}
-
-	// Create database.
-	db, err := bolt.Open(path, 0600, nil)
-	if err != nil {
-		fatal(err)
-		return
-	}
-	db.NoSync = options.NoSync
-	defer db.Close()
-
-	// Enable streaming stats.
-	if options.StatsInterval > 0 {
-		go printStats(db, options.StatsInterval)
-	}
-
-	// Start profiling for writes.
-	if options.ProfileMode == "rw" || options.ProfileMode == "w" {
-		benchStartProfiling(options)
-	}
-
-	// Write to the database.
-	if err := benchWrite(db, options, &results); err != nil {
-		fatal("bench: write: ", err)
-	}
-
-	// Stop profiling for writes only.
-	if options.ProfileMode == "w" {
-		benchStopProfiling()
-	}
-
-	// Start profiling for reads.
-	if options.ProfileMode == "r" {
-		benchStartProfiling(options)
-	}
-
-	// Read from the database.
-	if err := benchRead(db, options, &results); err != nil {
-		fatal("bench: read: ", err)
-	}
-
-	// Stop profiling for writes only.
-	if options.ProfileMode == "rw" || options.ProfileMode == "r" {
-		benchStopProfiling()
-	}
-
-	// Print results.
-	fmt.Fprintf(os.Stderr, "# Write\t%v\t(%v/op)\t(%v op/sec)\n", results.WriteDuration, results.WriteOpDuration(), results.WriteOpsPerSecond())
-	fmt.Fprintf(os.Stderr, "# Read\t%v\t(%v/op)\t(%v op/sec)\n", results.ReadDuration, results.ReadOpDuration(), results.ReadOpsPerSecond())
-	fmt.Fprintln(os.Stderr, "")
-}
-
-// Writes to the database.
-func benchWrite(db *bolt.DB, options *BenchOptions, results *BenchResults) error {
-	var err error
-	var t = time.Now()
-
-	switch options.WriteMode {
-	case "seq":
-		err = benchWriteSequential(db, options, results)
-	case "rnd":
-		err = benchWriteRandom(db, options, results)
-	case "seq-nest":
-		err = benchWriteSequentialNested(db, options, results)
-	case "rnd-nest":
-		err = benchWriteRandomNested(db, options, results)
-	default:
-		return fmt.Errorf("invalid write mode: %s", options.WriteMode)
-	}
-
-	results.WriteDuration = time.Since(t)
-
-	return err
-}
-
-func benchWriteSequential(db *bolt.DB, options *BenchOptions, results *BenchResults) error {
-	var i = uint32(0)
-	return benchWriteWithSource(db, options, results, func() uint32 { i++; return i })
-}
-
-func benchWriteRandom(db *bolt.DB, options *BenchOptions, results *BenchResults) error {
-	r := rand.New(rand.NewSource(time.Now().UnixNano()))
-	return benchWriteWithSource(db, options, results, func() uint32 { return r.Uint32() })
-}
-
-func benchWriteSequentialNested(db *bolt.DB, options *BenchOptions, results *BenchResults) error {
-	var i = uint32(0)
-	return benchWriteNestedWithSource(db, options, results, func() uint32 { i++; return i })
-}
-
-func benchWriteRandomNested(db *bolt.DB, options *BenchOptions, results *BenchResults) error {
-	r := rand.New(rand.NewSource(time.Now().UnixNano()))
-	return benchWriteNestedWithSource(db, options, results, func() uint32 { return r.Uint32() })
-}
-
-func benchWriteWithSource(db *bolt.DB, options *BenchOptions, results *BenchResults, keySource func() uint32) error {
-	results.WriteOps = options.Iterations
-
-	for i := 0; i < options.Iterations; i += options.BatchSize {
-		err := db.Update(func(tx *bolt.Tx) error {
-			b, _ := tx.CreateBucketIfNotExists(benchBucketName)
-			b.FillPercent = options.FillPercent
-
-			for j := 0; j < options.BatchSize; j++ {
-				var key = make([]byte, options.KeySize)
-				var value = make([]byte, options.ValueSize)
-				binary.BigEndian.PutUint32(key, keySource())
-				if err := b.Put(key, value); err != nil {
-					return err
-				}
-			}
-
-			return nil
-		})
-		if err != nil {
-			return err
-		}
-	}
-	return nil
-}
-
-func benchWriteNestedWithSource(db *bolt.DB, options *BenchOptions, results *BenchResults, keySource func() uint32) error {
-	results.WriteOps = options.Iterations
-
-	for i := 0; i < options.Iterations; i += options.BatchSize {
-		err := db.Update(func(tx *bolt.Tx) error {
-			top, _ := tx.CreateBucketIfNotExists(benchBucketName)
-			top.FillPercent = options.FillPercent
-
-			var name = make([]byte, options.KeySize)
-			binary.BigEndian.PutUint32(name, keySource())
-			b, _ := top.CreateBucketIfNotExists(name)
-			b.FillPercent = options.FillPercent
-
-			for j := 0; j < options.BatchSize; j++ {
-				var key = make([]byte, options.KeySize)
-				var value = make([]byte, options.ValueSize)
-				binary.BigEndian.PutUint32(key, keySource())
-				if err := b.Put(key, value); err != nil {
-					return err
-				}
-			}
-
-			return nil
-		})
-		if err != nil {
-			return err
-		}
-	}
-	return nil
-}
-
-// Reads from the database.
-func benchRead(db *bolt.DB, options *BenchOptions, results *BenchResults) error {
-	var err error
-	var t = time.Now()
-
-	switch options.ReadMode {
-	case "seq":
-		if options.WriteMode == "seq-nest" || options.WriteMode == "rnd-nest" {
-			err = benchReadSequentialNested(db, options, results)
-		} else {
-			err = benchReadSequential(db, options, results)
-		}
-	default:
-		return fmt.Errorf("invalid read mode: %s", options.ReadMode)
-	}
-
-	results.ReadDuration = time.Since(t)
-
-	return err
-}
-
-func benchReadSequential(db *bolt.DB, options *BenchOptions, results *BenchResults) error {
-	return db.View(func(tx *bolt.Tx) error {
-		var t = time.Now()
-
-		for {
-			c := tx.Bucket(benchBucketName).Cursor()
-			var count int
-			for k, v := c.First(); k != nil; k, v = c.Next() {
-				if v == nil {
-					return errors.New("invalid value")
-				}
-				count++
-			}
-
-			if options.WriteMode == "seq" && count != options.Iterations {
-				return fmt.Errorf("read seq: iter mismatch: expected %d, got %d", options.Iterations, count)
-			}
-
-			results.ReadOps += count
-
-			// Make sure we do this for at least a second.
-			if time.Since(t) >= time.Second {
-				break
-			}
-		}
-
-		return nil
-	})
-}
-
-func benchReadSequentialNested(db *bolt.DB, options *BenchOptions, results *BenchResults) error {
-	return db.View(func(tx *bolt.Tx) error {
-		var t = time.Now()
-
-		for {
-			var count int
-			var top = tx.Bucket(benchBucketName)
-			top.ForEach(func(name, _ []byte) error {
-				c := top.Bucket(name).Cursor()
-				for k, v := c.First(); k != nil; k, v = c.Next() {
-					if v == nil {
-						return errors.New("invalid value")
-					}
-					count++
-				}
-				return nil
-			})
-
-			if options.WriteMode == "seq-nest" && count != options.Iterations {
-				return fmt.Errorf("read seq-nest: iter mismatch: expected %d, got %d", options.Iterations, count)
-			}
-
-			results.ReadOps += count
-
-			// Make sure we do this for at least a second.
-			if time.Since(t) >= time.Second {
-				break
-			}
-		}
-
-		return nil
-	})
-}
-
-// Starts all profiles set on the options.
-func benchStartProfiling(options *BenchOptions) {
-	var err error
-
-	// Start CPU profiling.
-	if options.CPUProfile != "" {
-		cpuprofile, err = os.Create(options.CPUProfile)
-		if err != nil {
-			fatalf("bench: could not create cpu profile %q: %v", options.CPUProfile, err)
-		}
-		pprof.StartCPUProfile(cpuprofile)
-	}
-
-	// Start memory profiling.
-	if options.MemProfile != "" {
-		memprofile, err = os.Create(options.MemProfile)
-		if err != nil {
-			fatalf("bench: could not create memory profile %q: %v", options.MemProfile, err)
-		}
-		runtime.MemProfileRate = 4096
-	}
-
-	// Start fatal profiling.
-	if options.BlockProfile != "" {
-		blockprofile, err = os.Create(options.BlockProfile)
-		if err != nil {
-			fatalf("bench: could not create block profile %q: %v", options.BlockProfile, err)
-		}
-		runtime.SetBlockProfileRate(1)
-	}
-}
-
-// Stops all profiles.
-func benchStopProfiling() {
-	if cpuprofile != nil {
-		pprof.StopCPUProfile()
-		cpuprofile.Close()
-		cpuprofile = nil
-	}
-
-	if memprofile != nil {
-		pprof.Lookup("heap").WriteTo(memprofile, 0)
-		memprofile.Close()
-		memprofile = nil
-	}
-
-	if blockprofile != nil {
-		pprof.Lookup("block").WriteTo(blockprofile, 0)
-		blockprofile.Close()
-		blockprofile = nil
-		runtime.SetBlockProfileRate(0)
-	}
-}
-
-// Continuously prints stats on the database at given intervals.
-func printStats(db *bolt.DB, interval time.Duration) {
-	var prevStats = db.Stats()
-	var encoder = json.NewEncoder(os.Stdout)
-
-	for {
-		// Wait for the stats interval.
-		time.Sleep(interval)
-
-		// Retrieve new stats and find difference from previous iteration.
-		var stats = db.Stats()
-		var diff = stats.Sub(&prevStats)
-
-		// Print as JSON to STDOUT.
-		if err := encoder.Encode(diff); err != nil {
-			fatal(err)
-		}
-
-		// Save stats for next iteration.
-		prevStats = stats
-	}
-}
-
-// BenchOptions represents the set of options that can be passed to Bench().
-type BenchOptions struct {
-	ProfileMode   string
-	WriteMode     string
-	ReadMode      string
-	Iterations    int
-	BatchSize     int
-	KeySize       int
-	ValueSize     int
-	CPUProfile    string
-	MemProfile    string
-	BlockProfile  string
-	StatsInterval time.Duration
-	FillPercent   float64
-	NoSync        bool
-	Clean         bool
-	Path          string
-}
-
-// BenchResults represents the performance results of the benchmark.
-type BenchResults struct {
-	WriteOps      int
-	WriteDuration time.Duration
-	ReadOps       int
-	ReadDuration  time.Duration
-}
-
-// Returns the duration for a single write operation.
-func (r *BenchResults) WriteOpDuration() time.Duration {
-	if r.WriteOps == 0 {
-		return 0
-	}
-	return r.WriteDuration / time.Duration(r.WriteOps)
-}
-
-// Returns average number of write operations that can be performed per second.
-func (r *BenchResults) WriteOpsPerSecond() int {
-	var op = r.WriteOpDuration()
-	if op == 0 {
-		return 0
-	}
-	return int(time.Second) / int(op)
-}
-
-// Returns the duration for a single read operation.
-func (r *BenchResults) ReadOpDuration() time.Duration {
-	if r.ReadOps == 0 {
-		return 0
-	}
-	return r.ReadDuration / time.Duration(r.ReadOps)
-}
-
-// Returns average number of read operations that can be performed per second.
-func (r *BenchResults) ReadOpsPerSecond() int {
-	var op = r.ReadOpDuration()
-	if op == 0 {
-		return 0
-	}
-	return int(time.Second) / int(op)
-}
-
-// tempfile returns a temporary file path.
-func tempfile() string {
-	f, _ := ioutil.TempFile("", "bolt-bench-")
-	f.Close()
-	os.Remove(f.Name())
-	return f.Name()
-}