本文出自组内小伙伴
RocksDB iterator 允许用户遍历整个数据库,本文主要介绍 iterator 整体结构,如何实现seek,如何实现next等。介绍的顺序如下:
- RocksDB iterator 类层次关系,iterator 架构。用于建立整体认识。
- seek、next的实现。seek用于进行等值查询或寻找区间的上下限,next用于获取某个范围内的下一条数据;
Iterator 类之间的关系
介绍RocksDB 如何实现遍历整个数据库之前,有必要讲清楚实现迭代器的类,提供一个清晰的iterator 类之间的关系图便于后面的seek、next实现的分析,因为RocksDB iterator实现的逻辑是很复杂的。
通常使用 iterator 的方式如下:
rocksdb::Iterator* it = db->NewIterator(rocksdb::ReadOptions());
for (it->SeekToFirst(); it->Valid(); it->Next()) {
cout << it->key().ToString() << ": " << it->value().ToString() << endl;
}
assert(it->status().ok()); // Check for any errors found during the scan
delete it;
NewIterator()函数会创建一个迭代器,我会先讲一下该函数的实现,目的是引出后面seek或者next可能会用到的各种iterator,最后给出一个iterator的框架图,有了整体的结构,后面分析seek或者next的实现会容易很多。
上面的代码是准备创建 iterator的代码,可以看到有传入snapshot,实际创建迭代器的工作在NewIteratorImpl中完成,代码如下:
函数最后返回的是ArenaWrappedDBIter 类型的迭代器,根据类名可以判断出这只是一个包装类,其内部包含了DBIter和Arena:
DBIter 其实是 ite