-
Notifications
You must be signed in to change notification settings - Fork 8
/
persistence.go
119 lines (96 loc) · 2.41 KB
/
persistence.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
package bcache
import (
"time"
"github.com/boltdb/bolt"
"github.com/uol/gobol"
)
func newBolt(path string) (*persistence, gobol.Error) {
var err error
db, err := bolt.Open(path, 0600, &bolt.Options{Timeout: 5 * time.Second})
if err != nil {
return nil, errPersist("New", err)
}
tx, err := db.Begin(true)
if err != nil {
return nil, errPersist("New", err)
}
defer tx.Rollback()
if _, err := tx.CreateBucketIfNotExists([]byte("keyspace")); err != nil {
return nil, errPersist("New", err)
}
if _, err := tx.CreateBucketIfNotExists([]byte("number")); err != nil {
return nil, errPersist("New", err)
}
if _, err := tx.CreateBucketIfNotExists([]byte("text")); err != nil {
return nil, errPersist("New", err)
}
err = tx.Commit()
if err != nil {
return nil, errPersist("New", err)
}
return &persistence{
db: db,
}, nil
}
type persistence struct {
db *bolt.DB
}
func (persist *persistence) Get(buckName, key []byte) ([]byte, gobol.Error) {
start := time.Now()
tx, err := persist.db.Begin(false)
if err != nil {
statsError("begin", buckName)
return nil, errPersist("Get", err)
}
defer tx.Rollback()
bucket := tx.Bucket(buckName)
val := bucket.Get(key)
if val == nil {
statsNotFound(buckName)
return nil, nil
}
statsSuccess("get", buckName, time.Since(start))
return append([]byte{}, val...), nil
}
func (persist *persistence) Put(buckName, key, value []byte) gobol.Error {
start := time.Now()
tx, err := persist.db.Begin(true)
if err != nil {
statsError("begin", buckName)
return errPersist("Put", err)
}
defer tx.Rollback()
bucket := tx.Bucket(buckName)
if err := bucket.Put(key, value); err != nil {
statsError("put", buckName)
return errPersist("Put", err)
}
err = tx.Commit()
if err != nil {
statsError("put", buckName)
return errPersist("Put", err)
}
statsSuccess("put", buckName, time.Since(start))
return nil
}
func (persist *persistence) Delete(buckName, key []byte) gobol.Error {
start := time.Now()
tx, err := persist.db.Begin(true)
if err != nil {
statsError("begin", buckName)
return errPersist("Delete", err)
}
defer tx.Rollback()
bucket := tx.Bucket(buckName)
if err := bucket.Delete(key); err != nil {
statsError("delete", buckName)
return errPersist("delete", err)
}
err = tx.Commit()
if err != nil {
statsError("delete", buckName)
return errPersist("delete", err)
}
statsSuccess("delete", buckName, time.Since(start))
return nil
}