LevelDB 完全解析(9):写操作

LevelDB 完全解析(9):写操作Put :插入/修改一条记录。 Delete :删除一条记录。 Write :原子地插入/修改/删除多条记录。 leveldb::WriteOptions 是写操作的控制参数,只有一个成员变量 sync 表示是否每次写完都要将日志 flush 到外存。 leveldb::Wri…

LevelDB 提供三个和写操作相关的接口:

  • Put :插入/修改一条记录。
  • Delete :删除一条记录。
  • Write :原子地插入/修改/删除多条记录。

Put 和 Delete 都是直接调用 Write 来实现的:

  • leveldb::DBImpl::Put => leveldb::DB::Put => leveldb::DBImpl::Write
  • leveldb::DBImpl::Delete => leveldb::DB::Delete => leveldb::DBImpl::Write

Write 接口

leveldb::DBImpl::Write 的函数声明如下:

virtual Status Write(const WriteOptions& options, WriteBatch* updates) = 0;
  • leveldb::WriteOptions 是写操作的控制参数,只有一个成员变量 sync 表示是否每次写完都要将日志 flush 到外存。
  • leveldb::WriteBatch 表示多个 key-value 数据的操作。

Write 的实现

具体代码是 leveldb::DBImpl::Write

  1. 通过传入的参数构造一个 Writer 对象来表示本次写操作。Writer 的定义如下:
struct DBImpl::Writer {
  explicit Writer(port::Mutex* mu) : batch(nullptr), sync(false), done(false), cv(mu) {}

  Status status;              // 执行结果
  WriteBatch* batch;   // 更新的数据(1~多个 key-value) 
  bool sync;                    // 是否 flush,WriteOptions.sync
  bool done;                  // 是否已经执行
  port::CondVar cv;    // 并发控制的条件变量
};
  1. 获取互斥锁,将自己放入写队列,然后等待条件变量的通知。有两种情况可以跳出等待:1)本次写操作已由其它线程代为写入;2)本次写操作成为写队列的队首。这里涉及 LevelDB 写操作的并发控制和性能优化:由于 MemTable 和 WAL 都不支持并发写入,所以只有写队列队首的 writer 会执行真正的写入。队首的 writer 会将队列中的多个请求合并成一个请求,然后执行批量写入,并更新各个 writer 的状态。

  2. 检查 writer.done,如果已经被其它线程完成写入了,直接返回结果。否则就是队首 writer 了,继续往下执行。

  3. 调用 MakeRoomForWrite 检查:level-0 的文件数量是否超过限制?MemTable 是否超过阈值需要切换?等等。(MakeRoomForWrite 提供一个 force 参数表示是否强制切换新 MemTable,并触发 Compaction。正常写流程 force 为 false。)MakeRoomForWrite 的详细逻辑如下:

    MakeRoomForWrite

  4. 调用 BuildBatchGroup 将从队首开始的连续多个符合条件的 writer 合并到 tmp_batch_。合并时主要考虑:

    1. 合并写入的数据大小,默认 max_size 是 1MB。如果第一个写请求的 size 比较小(小于128 KB, 128 << 10),则 max_size 为 size + 128 KB。这样做是为了避免数据小的请求被其它请求给拖慢了。
    2. 如果第一个写请求 sync == false,那么就不要加入 sync == true 的写请求。
  5. 设置写入数据的 sequence

  6. 释放互斥锁。这里代码保证同一时刻只有一个线程会执行写入操作。

  7. 写日志(WAL) 。

  8. 根据参数决定是否 sync 日志

  9. 更新 MemTable

  10. 获取互斥锁

  11. 如果 sync 失败,设置 bg_error_,后续所有写入都将失败。

  12. 清空临时合并的批量操作(tmp_batch_) 。

  13. 更新 LastSequence

  14. 通知所有数据已经被写入的线程

  15. 通知还在写队列排队的线程

小结

以上,便是 LevelDB 的写入流程。写入队列 + 合并写操作,逻辑和代码都十分简洁。比较不足的是,整个写入过程都是单线程的。

今天的文章LevelDB 完全解析(9):写操作分享到此就结束了,感谢您的阅读。

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 举报,一经查实,本站将立刻删除。
如需转载请保留出处:https://bianchenghao.cn/17188.html

(0)
编程小号编程小号

相关推荐

发表回复

您的电子邮箱地址不会被公开。 必填项已用*标注