LevelDB 完全解析(8):读操作之 Get
LevelDB 支持的读操作分为两种:
- 点查询(Point Query):读一个 key 的数据。
- 范围查询(Range Query):有序读一段 key 范围的数据。
本文主要介绍点查询的实现。
Get 接口
LevelDB 通过 leveldb::DB::Get 接口对外提供点查询的能力,具体的实现是 leveldb::DBImpl::Get。接口声明如下:
virtual Status Get(const ReadOptions& options, const Slice& key, std::string* value) = 0;
key 和 value 都很简单:key 是本次查询的目标;如果找到对应的 key,数据会保存到 *value 中。
leveldb::ReadOptions 是读操作的控制参数:
- verify_checksums - 是否检查 crc32 校验和,默认 false。
- fill_cache - 从 SSTable 读取的数据是否要放到 block cache,默认 true。
- snapshot - 表示本次读操作要从哪个快照读取。默认是 nullptr,会从当前最新的快照读取数据。
快照
先来看看 leveldb::Snapshot 的实现。leveldb::Snapshot 是个空壳,具体实现是在 leveldb::SnapshotImpl ,也相当简单,主要就维护一个 sequence_number_—— 只读取小于等于 sequence_number_ 的数据。
如果一个快照有被外部使用(GetSnapshot),这个快照对应的 SnapshotImpl 对象会被放在 SnapshotList 中。Compaction 的时候,遇到可以清理的数据,还需要判断是否会影响这些快照。
Get 的实现
我们来看看 leveldb::DBImpl::Get 的实现:
- 获取互斥锁。
- 获取本次读操作的 Sequence Number:如果 ReadOptions 参数的 snapshot 不为空,则使用这个 snapshot 的 sequence number;否则,默认使用 LastSequence(LastSequence 会在每次写操作后更新)。
- MemTable, Immutable Memtable 和 Current Version(SSTable) 增加引用计数,避免在读取过程中被后台线程进行 compaction 时“垃圾回收”了。
- 释放互斥锁 。所以,下面 5 和 6 两步是没有持有锁的,不同线程的请求可以并发执行。
- 构造 LookupKey 。
- 执行查找:
- 从 MemTable 查找 。
- 从 Immutable Memtable 查找。
- 从 SSTable 查找。
- 获取互斥锁
- 更新 SSTable 的读统计信息,根据统计结果决定是否调度后台 Compaction。=> 极少遇到有读触发 compaction 的场景,这一步的似乎意义不大。
- MemTable, Immutable Memtable 和 Current Version 减少引用计数,返回结果。
主要逻辑在于读 MemTable(包括 Immutable MemTable) 和读 SSTable。
LookupKey
LevelDB 通过 user_key 和 sequence 构造 leveldb::LookupKey ,用于 LevelDB 内部接口的查找。其格式为:
- klength 的类型是 varint32,存储 userkey + tag 的长度。
- userkey 就是 Get 接口传入的 key 参数。
- tag 是 7 字节 sequence number 和 1 字节 value type。
- 一个 varint32 最多需要 5 个字节,tag 需要 8 字节。所以一个 LookupKey 最多需要分配 usize + 13 字节内存(usize 是 userkey 的 size)。
读 MemTable
MemTable 底层的实现是 leveldb::SkipList ,具体可以参考本系列的一篇文章《LevelDB 完全解析(1):MemTable》。
leveldb::SkipList 支持无锁一写多读,具体来讲:
- 写写并发由上层进行同步,保证同一时刻最多只有一个线程会写 SkipList。
- 读写并发不需要外部同步,只要保证 SkipList 不会被垃圾回收就好——这个通过引用计数保证。
具体实现查询操作的是 FindGreaterOrEqual 函数,返回值是一个指针,指向第一个大于等于 key 的结点(如果不存大于等于 key 的结点,则是 nullptr)。
在 MemTable 中,同一个 userkey 的多个版本是按照 sequence number 降序排序的,也就说“sequence number 小的在排序中比较大,sequence number 大的在排序中比较小”。所以,如果使用一个旧的 snapshot,只能查到比这个 snapshot 旧(或一样旧)的数据。
举个例子,假设 userkey 是 hello,当前有 3 个版本—— sequence number 分别是 5、10、20。那么它们在 MemTable 中的排序(从小到大)是:
... | hello-20 | hello-10 | hello-5 | ...
假设我们通过一个 sequence number 为 15 的快照进行查找,拼装 LookupKey 为 hello-15,查找发现第一个大于等于 hello-15 的 key 是 hello-10。
如果通过一个 sequence number 大于等于 20 的快照对 hello 这个 key 进行查找,那么 FindGreaterOrEqual 返回的就是执行 hello-20 的指针。
如果我们通过一个 sequence number 为 1 的快照进行查找呢?hello 的几个版本都不符合要求,FindGreaterOrEuqal 返回的不是 hello 这个 user key 的数据。所以,MemTable 对 FindGreaterOrEqual 返回的 key 会进行检查,再返回最终结果。
读 SSTTable
LevelDB 中将 SSTable 的管理实现成 leveldb::Version ,同时通过 leveldb::VersionSet 实现多版本管理。
Version::Get
SSTable 的点查询是从 leveldb::Version::Get 开始的:
-
根据每个 SSTable 的 key 范围 [smallest, largest] 收集可能需要查找的 SSTable,按照从新到旧维护
- level-0 的每个 SSTable 的 key 范围可能相交,每一个 SSTable 都需要判断。
- 非 level-0 的每一层内,SSTable 的 key 范围是不相交的——SSTable 是根据 key 范围有序排序的,可以通过二分查找优化查找效率。
-
根据上一步收集到的 SSTable,按照从新到旧开始查找。如果找到了,就可以直接返回结果。每个 SSTable 的具体查找逻辑是在 leveldb::TableCache::Get。
TableCache::Get
- FindTable - 从 table cache 中找到对应的 table 对象。FindTable 实现了缓存 table 对象的逻辑。
- Table::InternalGet - 执行查找。
Table::InternalGet
Table::InternalGet 实现了从一个 SSTable 中查找一个 key 的逻辑。
- 从 index block 查找对应的 data block。如果没有对应的 data block,说明 key 不存在,返回。
- 通过 bloom filter 判断 key 是否存在。
- 读取 data block 进行查找。
- Table::BlockReader 封装了 block cache 和 SSTable 读取 data block 的逻辑。
- Data block 中的数据查找逻辑与其格式强相关,可以参考本系列的一篇文章《LevelDB 完全解析(3):SSTable》。
小结
最后,用这张简图来总结一下 LevelDB Get 操作的逻辑:这是一个从上到下的过程,这个过程可能产生多次 I/O。