WAL主要的功能是当RocksDB异常退出后,能够恢复出错前的内存中(memtable)数据,因此RocksDB默认是每次用户写都会刷新数据到WAL. 每次当当前WAL对应的内存数据(memtable)刷新到磁盘之后,都会新建一个WAL.

所有的WAL文件都是保存在WAL目录(options.wal_dir),为了保证数据的状态,所有的WAL文件的名字都是按照顺序的(log_number).

WAL文件由一堆变长的record组成,而每个record是由kBlockSize(32k)来分组,比如某一个record大于kBlockSize的话,他就会被切分为多个record(通过type来判断).

record的格式如下:

  1. |CRC (4B) | Size (2B) | Type (1B) | Payload |
  2. +---------+-----------+-----------+--- ... ---+
  3. CRC = 32bit hash computed over the payload using CRC
  4. Size = Length of the payload data
  5. Type = Type of record
  6. (kZeroType, kFullType, kFirstType, kLastType, kMiddleType )
  7. The type is used to group a bunch of records together to represent
  8. blocks that are larger than kBlockSize
  9. Payload = Byte stream as long as specified by the payload size
  1. // WriteBatch::rep_ :=
  2. // sequence: fixed64
  3. // count: fixed32
  4. // data: record[count]
  5. // record :=
  6. // kTypeValue varstring varstring
  7. // kTypeDeletion varstring
  8. // kTypeSingleDeletion varstring
  9. // kTypeMerge varstring varstring
  10. // kTypeColumnFamilyValue varint32 varstring varstring
  11. // kTypeColumnFamilyDeletion varint32 varstring varstring
  12. // kTypeColumnFamilySingleDeletion varint32 varstring varstring
  13. // kTypeColumnFamilyMerge varint32 varstring varstring
  14. // kTypeBeginPrepareXID varstring
  15. // kTypeEndPrepareXID
  16. // kTypeCommitXID varstring
  17. // kTypeRollbackXID varstring
  18. // varstring :=
  19. // len: varint32
  20. // data: uint8[len]

上面的格式中可以看到有一个sequence的值,这个值主要用来表示WAL中操作的时序,这里要注意每次sequence的更新是按照WriteBatch来更新的.

首先是一个新的DB被打开的时候会创建一个WAL;

  1. const std::vector<ColumnFamilyDescriptor>& column_families,
  2. std::vector<ColumnFamilyHandle*>* handles, DB** dbptr) {
  3. ......................................................................
  4. s = impl->Recover(column_families);
  5. if (s.ok()) {
  6. uint64_t new_log_number = impl->versions_->NewFileNumber();
  7. .............................................
  8. s = NewWritableFile(
  9. impl->immutable_db_options_.env,
  10. LogFileName(impl->immutable_db_options_.wal_dir, new_log_number),
  11. &lfile, opt_env_options);
  12. ................................................

第二个情况是当一个CF(column family)被刷新到磁盘之后,也会创建新的WAL,这种情况下创建WAL是用过SwitchMemtable函数. 这个函数主要是用来切换memtable,也就是做flush之前的切换(生成新的memtable,然后把老的刷新到磁盘)

  1. Status DBImpl::SwitchMemtable(ColumnFamilyData* cfd, WriteContext* context) {
  2. ..................................................
  3. {
  4. if (creating_new_log) {
  5. ...............................................
  6. } else {
  7. s = NewWritableFile(
  8. env_, LogFileName(immutable_db_options_.wal_dir, new_log_number),
  9. &lfile, opt_env_opt);
  10. }
  11. .................................
  12. }
  13. ...............................................
  14. return s;
  15. }

通过上面的两个函数我们可以看到每次新建WAL都会有一个new_log_number,这个值就是对应的WAL的文件名前缀,可以看到每次生成新的log_number, 基本都会调用NewFileNumber函数.这里注意如果option设置了recycle_log_file_num的话,是有可能重用老的log_number的。我们先来看下NewFileNumber函数:

WAL的删除只有当包含在此WAL中的所有的数据都已经被持久化为SST之后(也有可能会延迟删除,因为有时候需要master发送transcation Log到slave来回放). 先来看DBImpl::FIndObsoleteFiles函数,这个函数很长,我们只关注对应的WAL部分,这里逻辑很简单,就是遍历所有的WAL,然后找出log_number小于当前min_log_number的文件然后加入到对应的结构(log_delete_files).

  1. if (!alive_log_files_.empty() && !logs_.empty()) {
  2. size_t num_alive_log_files = alive_log_files_.size();
  3. // find newly obsoleted log files
  4. auto& earliest = *alive_log_files_.begin();
  5. if (immutable_db_options_.recycle_log_file_num >
  6. log_recycle_files.size()) {
  7. ROCKS_LOG_INFO(immutable_db_options_.info_log,
  8. "adding log %" PRIu64 " to recycle list\n",
  9. earliest.number);
  10. log_recycle_files.push_back(earliest.number);
  11. } else {
  12. job_context->log_delete_files.push_back(earliest.number);
  13. }
  14. .....................................................................
  15. }
  16. while (!logs_.empty() && logs_.front().number < min_log_number) {
  17. auto& log = logs_.front();
  18. if (log.getting_synced) {
  19. log_sync_cv_.Wait();
  20. // logs_ could have changed while we were waiting.
  21. continue;
  22. }
  23. logs_to_free_.push_back(log.ReleaseWriter());
  24. {
  25. InstrumentedMutexLock wl(&log_write_mutex_);
  26. logs_.pop_front();
  27. }
  28. }
  29. // Current log cannot be obsolete.
  30. assert(!logs_.empty());
  31. }

这里可以看到有两个核心的数据结构alive_log_files和logs_,他们的区别就是前一个表示有写入的WAL,而后一个则是包括了所有的WAL(比如open一个DB,而没有写入数据,此时也会生成WAL).

最终删除WAL的操作是在DBImpl::DeleteObsoleteFileImpl这个函数,而WAL删除不会单独触发,而是和temp/sst这类文件一起被删除的(PurgeObsoleteFiles).

我们可以使用RocksDB自带的ldb工具来查看对应的WAL内容

  1. pagefault@god ~/tools/rocksdb/data/.rocksdb $ ../../bin/ldb dump_wal --walfile=./000285.log --header
  2. 1255,1,110,0,PUT(1) : 0x00000006000000000000013C