Database: return list of keys by prefix.

This commit is contained in:
Andrey Smirnov
2014-02-11 17:21:35 +04:00
parent aea8ae9a96
commit 8d72f1a959
2 changed files with 24 additions and 1 deletions
+19
View File
@@ -19,6 +19,7 @@ type Storage interface {
Get(key []byte) ([]byte, error)
Put(key []byte, value []byte) error
Delete(key []byte) error
KeysByPrefix(prefix []byte) [][]byte
FetchByPrefix(prefix []byte) [][]byte
Close() error
}
@@ -65,6 +66,24 @@ func (l *levelDB) Delete(key []byte) error {
return l.db.Delete(key, nil)
}
func (l *levelDB) KeysByPrefix(prefix []byte) [][]byte {
result := make([][]byte, 0, 20)
iterator := l.db.NewIterator(nil)
if iterator.Seek(prefix) {
for bytes.HasPrefix(iterator.Key(), prefix) {
key := iterator.Key()
keyc := make([]byte, len(key))
copy(keyc, key)
result = append(result, keyc)
if !iterator.Next() {
break
}
}
}
return result
}
func (l *levelDB) FetchByPrefix(prefix []byte) [][]byte {
result := make([][]byte, 0, 20)