brian
/
boltrest
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

I forget what I did

This commit is contained in:
Brian Buller 2016-02-22 07:32:07 -06:00
parent eefa309913
commit 32f6cf6e1d
1 changed files with 67 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

87
boltrest.go
View File

@ -37,6 +37,39 @@ func Open(filename string) (*DB, error) {
return &b, nil return &b, nil
} }
// MkBucketPath builds all buckets in the string slice
func (b *DB) MkBucketPath(path []string) error {
// TODO: Make sure local db is fresh (or offline)
err := b.localDB.Update(func(tx *bolt.Tx) error {
var err error
bkt := tx.Bucket([]byte(path[0]))
if bkt == nil {
// Create it
bkt, err = tx.CreateBucket([]byte(path[0]))
if err != nil {
// error creating
return err
}
}
if len(path) > 1 {
path = path[1:]
for i := range path {
nextBkt := bkt.Bucket([]byte(path[i]))
if nextBkt == nil {
// Create it
nextBkt, err = bkt.CreateBucket([]byte(path[i]))
if err != nil {
return err
}
}
bkt = nextBkt
}
}
return err
})
return err
}
// GetValue returns the value at path // GetValue returns the value at path
// path is a '/' separated list of tokens // path is a '/' separated list of tokens
// the last token is a key, all others are buckets // the last token is a key, all others are buckets
@ -67,7 +100,10 @@ func (b *DB) GetValue(path []string, key string) (string, error) {
// SetValue sets the value at path to val // SetValue sets the value at path to val
// path is a slice of tokens // path is a slice of tokens
func (b *DB) SetValue(path []string, key, val string) error { func (b *DB) SetValue(path []string, key, val string) error {
var err error err := b.MkBucketPath(path)
if err != nil {
return err
}
b.localDB.Update(func(tx *bolt.Tx) error { b.localDB.Update(func(tx *bolt.Tx) error {
newBkt := tx.Bucket([]byte(path[0])) newBkt := tx.Bucket([]byte(path[0]))
if newBkt == nil { if newBkt == nil {
@ -154,14 +190,19 @@ func (b *DB) GetBucketList(path []string) ([]string, error) {
return fmt.Errorf("Couldn't find bucket " + path[0]) return fmt.Errorf("Couldn't find bucket " + path[0])
} }
var newBkt *bolt.Bucket var newBkt *bolt.Bucket
for idx := 1; idx < len(path); idx++ { var berr error
newBkt = bkt.Bucket([]byte(path[idx])) if len(path) > 1 {
if newBkt == nil { for idx := 1; idx < len(path); idx++ {
return fmt.Errorf("Couldn't find bucket " + strings.Join(path[:idx], "/")) newBkt = bkt.Bucket([]byte(path[idx]))
if newBkt == nil {
return fmt.Errorf("Couldn't find bucket " + strings.Join(path[:idx], "/"))
}
} }
bkt = newBkt
} }
// newBkt should have the last bucket in the path // newBkt should have the last bucket in the path
berr := newBkt.ForEach(func(k, v []byte) error { berr = bkt.ForEach(func(k, v []byte) error {
if v == nil { if v == nil {
// Must be a bucket // Must be a bucket
ret = append(ret, string(k)) ret = append(ret, string(k))
@ -184,15 +225,18 @@ func (b *DB) GetKeyList(path []string) ([]string, error) {
if bkt == nil { if bkt == nil {
return fmt.Errorf("Couldn't find bucket " + path[0]) return fmt.Errorf("Couldn't find bucket " + path[0])
} }
var newBkt *bolt.Bucket if len(path) > 1 {
for idx := 1; idx < len(path); idx++ { var newBkt *bolt.Bucket
newBkt = bkt.Bucket([]byte(path[idx])) for idx := 1; idx < len(path); idx++ {
if newBkt == nil { newBkt = bkt.Bucket([]byte(path[idx]))
return fmt.Errorf("Couldn't find bucket " + strings.Join(path[:idx], "/")) if newBkt == nil {
return fmt.Errorf("Couldn't find bucket " + strings.Join(path[:idx], "/"))
}
} }
bkt = newBkt
} }
// newBkt should have the last bucket in the path // newBkt should have the last bucket in the path
berr := newBkt.ForEach(func(k, v []byte) error { berr := bkt.ForEach(func(k, v []byte) error {
if v != nil { if v != nil {
// Not a bucket // Not a bucket
ret = append(ret, string(k)) ret = append(ret, string(k))
@ -214,17 +258,20 @@ func (b *DB) DeletePair(path []string, key string) error {
if bkt == nil { if bkt == nil {
return fmt.Errorf("Couldn't find bucket " + path[0]) return fmt.Errorf("Couldn't find bucket " + path[0])
} }
var newBkt *bolt.Bucket if len(path) > 1 {
for idx := 1; idx < len(path); idx++ { var newBkt *bolt.Bucket
newBkt = bkt.Bucket([]byte(path[idx])) for idx := 1; idx < len(path); idx++ {
if newBkt == nil { newBkt = bkt.Bucket([]byte(path[idx]))
return fmt.Errorf("Couldn't find bucket " + strings.Join(path[:idx], "/")) if newBkt == nil {
return fmt.Errorf("Couldn't find bucket " + strings.Join(path[:idx], "/"))
}
} }
bkt = newBkt
} }
// newBkt should have the last bucket in the path // bkt should have the last bucket in the path
// Test to make sure that key is a pair, if so, delete it // Test to make sure that key is a pair, if so, delete it
if tst := newBkt.Bucket([]byte(key)); tst == nil { if tst := bkt.Bucket([]byte(key)); tst == nil {
return newBkt.Delete([]byte(key)) return bkt.Delete([]byte(key))
} }
return nil return nil
}) })