LevelDB:实现迭代器以按键前缀枚举

时间:2012-05-07 17:06:51

标签: iterator c++-cli leveldb

我正在寻找一种有效的方法来实现一个密钥枚举器,使用leveldb迭代一个键前缀。关键是字节数组(并且db使用默认字节数组比较器,因此所有具有特定前缀的键按顺序存储/检索),并且我希望我的迭代器能够获取键前缀并仅返回具有该键的键的数据前缀。

我是否必须使用或继承默认的db迭代器,寻找范围中的第一个键(当然我需要知道它是什么),然后验证并返回以前缀开头的每个切片(通过覆盖movenext或什么)?或者有更有效的方法来实现这个吗?

如果有人已经解决了这个问题并且可以分享代码或一般想法,请告诉我。我是从C ++ / CLI尝试这个,但是用任何语言实现都会有所帮助。

感谢。 -raj。

2 个答案:

答案 0 :(得分:2)

比较器用于确定键是否不同,因此重载它无济于事,因为当您扫描数据库时,您必须能够比较完整键(而不仅仅是前缀)。不需要重载迭代器:键在leveldb中排序,你会知道如果你遇到一个具有不同前缀的键,它将已经超出范围。您可以像往常一样使用迭代器,只要您的密钥得到正确评估,您就应该得到正确的结果:

void ScanRecordRange(const leveldb::Slice& startSlice, const leveldb::Slice& endSlice)
{
    // Get a database iterator
    shared_ptr<leveldb::Iterator> dbIter(_database->NewIterator(leveldb::ReadOptions()));

    // Possible optimization suggested by Google engineers 
    // for critical loops. Reduces memory thrash.
    for(dbIter->Seek(startSlice); dbIter->Valid() && _options.comparator->Compare(dbIter->key(), endSlice)<=0 ;dbIter->Next())
    {                
        // Read the record
        if( !dbIter->value().empty() )
        {
            leveldb::Slice keySlice(dbIter->key());
            leveldb::Slice dataSlice(dbIter->data());
            // TODO: process the key/data

        }
    }
}

答案 1 :(得分:0)

我在my LevelDB wrapper中有一个前缀迭代器 它在startsWith方法返回的范围内使用:

int count = 0;
for (auto& en: ldb.startsWith (std::string ("prefix"))) {++count;}