Merge pull request #1738 from qieqieplus/rocksdb

fix rocksdb crash when list directory
This commit is contained in:
Chris Lu 2021-01-07 02:26:41 -08:00 committed by GitHub
commit 07f4703bfc
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -181,18 +181,12 @@ func enumerate(iter *gorocksdb.Iterator, prefix, lastKey []byte, includeLastKey
iter.Seek(prefix)
} else {
iter.Seek(lastKey)
if !includeLastKey {
key := iter.Key().Data()
if !bytes.HasPrefix(key, prefix) {
return nil
if iter.Valid() {
if bytes.Equal(iter.Key().Data(), lastKey) {
iter.Next()
}
}
if bytes.Equal(key, lastKey) {
iter.Next()
}
}
}
@ -250,10 +244,6 @@ func (store *RocksDBStore) ListDirectoryPrefixedEntries(ctx context.Context, ful
if fileName == "" {
return true
}
limit--
if limit < 0 {
return false
}
entry := &filer.Entry{
FullPath: weed_util.NewFullPath(string(fullpath), fileName),
}