feat(qs2) add qs2 framework

src/kvstore/boltdbpvd/provider.go

@@ -0,0 +1,225 @@
+package boltdbpvd
+
+import (
+	"encoding/binary"
+	"fmt"
+	"math"
+	"path"
+	"time"
+
+	"github.com/boltdb/bolt"
+
+	"github.com/ihexxa/quickshare/src/kvstore"
+)
+
+type BoltPvd struct {
+	dbPath    string
+	db        *bolt.DB
+	maxStrLen int
+}
+
+func New(dbPath string, maxStrLen int) *BoltPvd {
+	boltPath := path.Join(path.Clean(dbPath), "quickshare.db")
+	db, err := bolt.Open(boltPath, 0600, &bolt.Options{Timeout: 1 * time.Second})
+	if err != nil {
+		panic(err)
+	}
+
+	buckets := []string{"bools", "ints", "int64s", "floats", "strings", "locks"}
+	for _, bucketName := range buckets {
+		db.Update(func(tx *bolt.Tx) error {
+			b := tx.Bucket([]byte(bucketName))
+			if b != nil {
+				return nil
+			}
+
+			_, err := tx.CreateBucket([]byte(bucketName))
+			if err != nil {
+				panic(err)
+			}
+			return nil
+		})
+	}
+
+	return &BoltPvd{
+		dbPath:    dbPath,
+		db:        db,
+		maxStrLen: maxStrLen,
+	}
+}
+
+func (bp *BoltPvd) Close() error {
+	return bp.db.Close()
+}
+
+func (bp *BoltPvd) GetBool(key string) (bool, bool) {
+	buf, ok := make([]byte, 1), false
+
+	bp.db.View(func(tx *bolt.Tx) error {
+		b := tx.Bucket([]byte("bools"))
+		v := b.Get([]byte(key))
+		copy(buf, v)
+		ok = v != nil
+		return nil
+	})
+
+	// 1 means true, 0 means false
+	return buf[0] == 1, ok
+}
+
+func (bp *BoltPvd) SetBool(key string, val bool) error {
+	var bVal byte = 0
+	if val {
+		bVal = 1
+	}
+	return bp.db.Update(func(tx *bolt.Tx) error {
+		b := tx.Bucket([]byte("bools"))
+		return b.Put([]byte(key), []byte{bVal})
+	})
+}
+
+func (bp *BoltPvd) DelBool(key string) error {
+	return bp.db.Update(func(tx *bolt.Tx) error {
+		b := tx.Bucket([]byte("bools"))
+		return b.Delete([]byte(key))
+	})
+}
+
+func (bp *BoltPvd) GetInt(key string) (int, bool) {
+	x, ok := bp.GetInt64(key)
+	return int(x), ok
+}
+
+func (bp *BoltPvd) SetInt(key string, val int) error {
+	return bp.SetInt64(key, int64(val))
+}
+
+func (bp *BoltPvd) DelInt(key string) error {
+	return bp.DelInt64(key)
+}
+
+func (bp *BoltPvd) GetInt64(key string) (int64, bool) {
+	buf, ok := make([]byte, binary.MaxVarintLen64), false
+
+	bp.db.View(func(tx *bolt.Tx) error {
+		b := tx.Bucket([]byte("int64s"))
+		v := b.Get([]byte(key))
+		copy(buf, v)
+		ok = v != nil
+		return nil
+	})
+
+	if !ok {
+		return 0, false
+	}
+	x, n := binary.Varint(buf)
+	if n < 0 {
+		return 0, false
+	}
+	return x, true
+}
+
+func (bp *BoltPvd) SetInt64(key string, val int64) error {
+	buf := make([]byte, binary.MaxVarintLen64)
+	n := binary.PutVarint(buf, val)
+
+	return bp.db.Update(func(tx *bolt.Tx) error {
+		b := tx.Bucket([]byte("int64s"))
+		return b.Put([]byte(key), buf[:n])
+	})
+}
+func (bp *BoltPvd) DelInt64(key string) error {
+	return bp.db.Update(func(tx *bolt.Tx) error {
+		b := tx.Bucket([]byte("int64s"))
+		return b.Delete([]byte(key))
+	})
+}
+
+func float64ToBytes(num float64) []byte {
+	buf := make([]byte, 64)
+	binary.PutUvarint(buf, math.Float64bits(num))
+	return buf
+}
+
+func bytesToFloat64(buf []byte) float64 {
+	uintVal, _ := binary.Uvarint(buf[:64])
+	return math.Float64frombits(uintVal)
+}
+
+func (bp *BoltPvd) GetFloat(key string) (float64, bool) {
+	buf, ok := make([]byte, 64), false
+	bp.db.View(func(tx *bolt.Tx) error {
+		b := tx.Bucket([]byte("floats"))
+		v := b.Get([]byte(key))
+		copy(buf, v)
+		ok = v != nil
+		return nil
+	})
+	if !ok {
+		return 0.0, false
+	}
+	return bytesToFloat64(buf), true
+}
+
+func (bp *BoltPvd) SetFloat(key string, val float64) error {
+	return bp.db.Update(func(tx *bolt.Tx) error {
+		b := tx.Bucket([]byte("floats"))
+		return b.Put([]byte(key), float64ToBytes(val))
+	})
+}
+func (bp *BoltPvd) DelFloat(key string) error {
+	return bp.db.Update(func(tx *bolt.Tx) error {
+		b := tx.Bucket([]byte("floats"))
+		return b.Delete([]byte(key))
+	})
+}
+
+func (bp *BoltPvd) GetString(key string) (string, bool) {
+	buf, ok, length := make([]byte, bp.maxStrLen), false, bp.maxStrLen
+	bp.db.View(func(tx *bolt.Tx) error {
+		b := tx.Bucket([]byte("strings"))
+		v := b.Get([]byte(key))
+		length = copy(buf, v)
+		ok = v != nil
+		return nil
+	})
+	return string(buf[:length]), ok
+}
+
+func (bp *BoltPvd) SetString(key string, val string) error {
+	if len(val) > bp.maxStrLen {
+		return fmt.Errorf("can not set string value longer than %d", bp.maxStrLen)
+	}
+
+	return bp.db.Update(func(tx *bolt.Tx) error {
+		b := tx.Bucket([]byte("strings"))
+		return b.Put([]byte(key), []byte(val))
+	})
+}
+
+func (bp *BoltPvd) DelString(key string) error {
+	return bp.db.Update(func(tx *bolt.Tx) error {
+		b := tx.Bucket([]byte("strings"))
+		return b.Delete([]byte(key))
+	})
+}
+
+func (bp *BoltPvd) TryLock(key string) error {
+	return bp.db.Update(func(tx *bolt.Tx) error {
+		b := tx.Bucket([]byte("locks"))
+		if b.Get([]byte(key)) != nil {
+			return kvstore.ErrLocked
+		}
+		return b.Put([]byte(key), []byte{})
+	})
+}
+
+func (bp *BoltPvd) Unlock(key string) error {
+	return bp.db.Update(func(tx *bolt.Tx) error {
+		b := tx.Bucket([]byte("locks"))
+		if b.Get([]byte(key)) != nil {
+			return b.Delete([]byte(key))
+		}
+		return kvstore.ErrNoLock
+	})
+}