open key-database deterministic

server/database/helpers.go

@@ -3,10 +3,9 @@ package database
 import (
 	"bytes"
 	"encoding/gob"
-	"github.com/akrylysov/pogreb"
 )
 
-func PogrebPut(db *pogreb.DB, name []byte, obj interface{}) {
+func PogrebPut(db KeyDB, name []byte, obj interface{}) {
 	var resGob bytes.Buffer
 	resEnc := gob.NewEncoder(&resGob)
 	err := resEnc.Encode(obj)
@@ -19,7 +18,7 @@ func PogrebPut(db *pogreb.DB, name []byte, obj interface{}) {
 	}
 }
 
-func PogrebGet(db *pogreb.DB, name []byte, obj interface{}) bool {
+func PogrebGet(db KeyDB, name []byte, obj interface{}) bool {
 	resBytes, err := db.Get(name)
 	if err != nil {
 		panic(err)

server/database/interface.go

@@ -0,0 +1,12 @@
+package database
+
+import "github.com/akrylysov/pogreb"
+
+type KeyDB interface {
+	Sync() error
+	Put(key []byte, value []byte) error
+	Get(key []byte) ([]byte, error)
+	Delete(key []byte) error
+	Compact() (pogreb.CompactionResult, error)
+	Items() *pogreb.ItemIterator
+}

server/database/setup.go

@@ -0,0 +1,19 @@
+package database
+
+import (
+	"fmt"
+	"github.com/akrylysov/pogreb"
+	"github.com/akrylysov/pogreb/fs"
+	"time"
+)
+
+func New(path string) (KeyDB, error) {
+	if path == "" {
+		return nil, fmt.Errorf("path not set")
+	}
+	return pogreb.Open(path, &pogreb.Options{
+		BackgroundSyncInterval:       30 * time.Second,
+		BackgroundCompactionInterval: 6 * time.Hour,
+		FileSystem:                   fs.OSMMap,
+	})
+}