Fix 'GetKeyList' function

This commit is contained in:
Brian Buller 2017-07-06 07:17:26 -05:00
parent 50a16b9056
commit 69e69463f4

View File

@ -242,7 +242,7 @@ func (b *DB) GetBucketList(path []string) ([]string, error) {
} }
} }
// newBkt should have the last bucket in the path // bkt should have the last bucket in the path
berr = bkt.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
@ -271,20 +271,20 @@ 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 berr error
if len(path) > 1 { if len(path) > 1 {
var newBkt *bolt.Bucket
for idx := 1; idx < len(path); idx++ { for idx := 1; idx < len(path); idx++ {
newBkt = bkt.Bucket([]byte(path[idx])) bkt = bkt.Bucket([]byte(path[idx]))
if newBkt == nil { if bkt == nil {
return fmt.Errorf("Couldn't find bucket " + strings.Join(path[:idx], "/")) return fmt.Errorf("Couldn't find bucket " + strings.Join(path[:idx], " / "))
} }
} }
bkt = newBkt
} }
// newBkt should have the last bucket in the path
berr := bkt.ForEach(func(k, v []byte) error { // bkt should have the last bucket in the path
berr = bkt.ForEach(func(k, v []byte) error {
if v != nil { if v != nil {
// Not a bucket // Must be a key
ret = append(ret, string(k)) ret = append(ret, string(k))
} }
return nil return nil