-
Notifications
You must be signed in to change notification settings - Fork 1
/
main.go
112 lines (100 loc) · 1.83 KB
/
main.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
package multidbfile
import (
"strconv"
"time"
"github.com/boltdb/bolt"
)
type top struct {
dbs []*bolt.DB
sequence int
}
const bucketkey = "blocks"
func dbname(sequence int) string {
return "test" + strconv.Itoa(sequence) + ".db"
}
func (t *top) createDB() error {
t.sequence++
db, err := bolt.Open(dbname(t.sequence), 0600, &bolt.Options{Timeout: 1 * time.Second})
if err != nil {
t.sequence--
panic(err)
}
t.dbs = append(t.dbs, db)
return nil
}
func (t *top) find(key []byte) []byte {
var value []byte
var err error
for _, db := range t.dbs {
err = db.View(func(tx *bolt.Tx) error {
b := tx.Bucket([]byte(bucketkey))
value = b.Get(key)
return nil
})
if value != nil {
break
}
}
if err != nil {
return nil
}
return value
}
func (t *top) exist(key []byte) bool {
return t.find(key) != nil
}
func (t *top) store(key []byte, value []byte) error {
db := t.dbs[len(t.dbs)-1]
err := db.Update(func(tx *bolt.Tx) error {
b, _ := tx.CreateBucketIfNotExists([]byte(bucketkey))
err := b.Put(key, value)
return err
})
return err
}
func (t *top) update(key []byte, value []byte) error {
var found bool
var err error
for _, db := range t.dbs {
err = db.Update(func(tx *bolt.Tx) error {
b := tx.Bucket([]byte(bucketkey))
v := b.Get(key)
if v != nil {
err := b.Put(key, value)
found = true
return err
}
return nil
})
if found {
break
}
}
return err
}
func (t *top) del(key []byte) error {
var found bool
var err error
for _, db := range t.dbs {
err = db.Update(func(tx *bolt.Tx) error {
b := tx.Bucket([]byte(bucketkey))
v := b.Get(key)
if v != nil {
err := b.Delete(key)
found = true
return err
}
return nil
})
if found {
break
}
}
return err
}
func newTop() *top {
return &top{
dbs: make([]*bolt.DB, 0),
sequence: 0,
}
}