-
Notifications
You must be signed in to change notification settings - Fork 0
/
bolt.go
127 lines (114 loc) · 2.48 KB
/
bolt.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
package cachestore
import (
"log"
bolt "github.com/boltdb/bolt"
)
// Bolt implements ds.Datastore
// TODO: use buckets to represent the heirarchy of the ds.Keys
type Bolt struct {
db *bolt.DB
bucketName []byte
Debug bool
}
func NewBolt(dbFile, bucket string) (*Bolt, error) {
db, err := bolt.Open(dbFile, 0600, nil)
if err != nil {
return nil, err
}
// TODO: need to do db.Close() sometime...
err = db.Update(func(tx *bolt.Tx) error {
_, err := tx.CreateBucketIfNotExists([]byte(bucket))
return err
})
if err != nil {
return nil, err
}
return &Bolt{
db: db,
bucketName: []byte(bucket),
}, nil
}
func (bd *Bolt) Close() error {
return bd.db.Close()
}
func (bd *Bolt) Del(key string) error {
return bd.db.Update(func(tx *bolt.Tx) error {
return tx.Bucket(bd.bucketName).Delete([]byte(key))
})
}
func (bd *Bolt) Get(key string) (interface{}, error) {
var out []byte
err := bd.db.View(func(tx *bolt.Tx) error {
mmval := tx.Bucket(bd.bucketName).Get([]byte(key))
if mmval == nil {
return nil
}
out = make([]byte, len(mmval))
copy(out, mmval)
return nil
})
if err != nil {
return nil, err
}
if out == nil {
return nil, err
}
var v interface{}
err = Decode(out, &v)
if err != nil {
if bd.Debug {
log.Println("[Bolt]DecodeErr: ", err, "Key:", key)
}
return nil, err
}
return v, err
}
func (bd *Bolt) ConsumeValue(key string, f func([]byte) error) error {
return bd.db.View(func(tx *bolt.Tx) error {
mmval := tx.Bucket(bd.bucketName).Get([]byte(key))
if mmval == nil {
return nil
}
return f(mmval)
})
}
func (bd *Bolt) Has(key string) (bool, error) {
var found bool
err := bd.db.View(func(tx *bolt.Tx) error {
val := tx.Bucket(bd.bucketName).Get([]byte(key))
found = (val != nil)
return nil
})
return found, err
}
func (bd *Bolt) Put(key string, val interface{}) error {
bval, err := Encode(val)
if err != nil {
if bd.Debug {
log.Println("[Bolt]EncodeErr: ", err, "Key:", key)
}
return err
}
return bd.db.Update(func(tx *bolt.Tx) error {
return tx.Bucket(bd.bucketName).Put([]byte(key), bval)
})
}
func (bd *Bolt) PutMany(data map[string]interface{}) error {
return bd.db.Update(func(tx *bolt.Tx) error {
buck := tx.Bucket(bd.bucketName)
for k, v := range data {
bval, err := Encode(v)
if err != nil {
if bd.Debug {
log.Println("[Bolt]EncodeErr: ", err, "Key:", k)
}
return err
}
err = buck.Put([]byte(k), bval)
if err != nil {
return err
}
}
return nil
})
}