blob: 5dca6a331c276f727afb05e0f3d9daf73c2f5201 (
plain) (
blame)
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
|
package storage
import bolt "go.etcd.io/bbolt"
var _ CHR = &Bolt{}
// Redis storage engine
type Bolt bolt.DB
var bname = []byte("brpaste")
func OpenBolt(db *bolt.DB) (*Bolt, error) {
err := db.Update(func(tx *bolt.Tx) error {
_, err := tx.CreateBucketIfNotExists(bname)
return err
})
return (*Bolt)(db), err
}
// Create an entry in redis
func (db *Bolt) Create(key, value string, checkcollision bool) error {
err := (*bolt.DB)(db).Update(func(tx *bolt.Tx) error {
b := tx.Bucket(bname)
if b == nil {
return Unhealthy
}
k := []byte(key)
if checkcollision {
if b.Get(k) != nil {
return Collision
}
}
return b.Put(k, []byte(value))
})
if err != nil {
return err
}
return nil
}
func (db *Bolt) Read(key string) (string, error) {
var out string
err := (*bolt.DB)(db).View(func(tx *bolt.Tx) error {
b := tx.Bucket(bname)
if b == nil {
return Unhealthy
}
v := b.Get([]byte(key))
if v == nil {
return NotFound
}
out = string(v)
return nil
})
if err != nil {
return "", err
}
return out, nil
}
// Healthy TODO: destub ?
func (db *Bolt) Healthy() bool {
return db != nil
}
|
