forked from gitchain/gitchain
-
Notifications
You must be signed in to change notification settings - Fork 0
/
db.go
48 lines (43 loc) · 727 Bytes
/
db.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
package db
import (
"github.com/boltdb/bolt"
)
type T struct {
Path string
DB *bolt.DB
}
func NewDB(path string) (*T, error) {
db, err := bolt.Open(path, 0666)
return &T{Path: path, DB: db}, err
}
func writable(e *error, db *T, f func(*bolt.Tx) bool) error {
dbtx, err := db.DB.Begin(true)
success := false
if err != nil {
return err
}
defer func() {
if success {
dbtx.Commit()
} else {
dbtx.Rollback()
}
}()
success = f(dbtx)
if e != nil && *e == nil {
*e = err
}
return err
}
func readable(e *error, db *T, f func(*bolt.Tx)) error {
dbtx, err := db.DB.Begin(false)
if err != nil {
return err
}
defer dbtx.Rollback()
f(dbtx)
if e != nil && *e == nil {
*e = err
}
return err
}