add buntdb

Author: Vadim Kilibaba

cmd/pogreb-bench/buntdb.go

@@ -0,0 +1,46 @@
+package main
+
+import (
+	"github.com/tidwall/buntdb"
+)
+
+type buntEngine struct {
+	db *buntdb.DB
+}
+
+func newBunt(path string) (kvEngine, error) {
+	db, err := buntdb.Open(path)
+
+	if err != nil {
+		return nil, err
+	}
+
+	return &buntEngine{db: db}, err
+}
+
+func (db *buntEngine) Put(key []byte, value []byte) error {
+	return db.db.Update(func(tx *buntdb.Tx) error {
+		_, _, err := tx.Set(string(key), string(value), nil)
+		return err
+	})
+}
+
+func (db *buntEngine) Get(key []byte) ([]byte, error) {
+	var val []byte
+	err := db.db.View(func(tx *buntdb.Tx) error {
+		valstr, err := tx.Get(string(key))
+		if err != nil {
+			return err
+		}
+		val = []byte(valstr)
+		return nil
+	})
+	return val, err
+}
+
+func (db *buntEngine) Close() error {
+	return db.db.Close()
+}
+func (db *buntEngine) FileSize() (int64, error) {
+	return 0, nil
+}

cmd/pogreb-bench/engine.go

@@ -22,6 +22,7 @@ var engines = map[string]engineCtr{
 	"badgerdb":  newBadgerdb,
 	"slowpoke":  newSlowpoke,
 	"pudge":     newPudge,
+	"buntdb":    newBunt,
 }
 
 func getEngineCtr(name string) (engineCtr, error) {

cmd/pogreb-bench/main.go

@@ -7,7 +7,7 @@ import (
 )
 
 var (
-	engine       = flag.String("e", "pogreb", "database engine name. pogreb, goleveldb, bolt, badgerdb, pudge and slowpoke")
+	engine       = flag.String("e", "pogreb", "database engine name. pogreb, goleveldb, bolt, badgerdb, pudge and slowpoke, buntdb")
 	numKeys      = flag.Int("n", 100000, "number of keys")
 	minKeySize   = flag.Int("mink", 16, "minimum key size")
 	maxKeySize   = flag.Int("maxk", 64, "maximum key size")