leveldb 源码分析 20¶
本系列《leveldb 源码分析》共有 22 篇文章，这是第二十篇¶
12 DB 的打开¶
先分析 LevelDB 是如何打开 db 的，万物始于创建。在打开流程中有几个辅助函数：DBImpl()，DBImpl::Recover, DBImpl::DeleteObsoleteFiles, DBImpl::RecoverLogFile, DBImpl::MaybeScheduleCompaction。
12.1 DB::Open()¶
打开一个 db，进行 PUT、GET 操作，就是前面的静态函数 DB::Open 的工作。如果操作成功，它就返回一个 db 指针。前面说过 DB 就是一个接口类，其具体实现在 DBImp 类中，这是一个 DB 的子类。 函数声明为：
Status DB::Open(const Options& options, const std::string&dbname, DB** dbptr);
分解来看，Open() 函数主要有以下 5 个执行步骤。 S1 创建 DBImpl 对象，其后进入DBImpl::Recover()函数执行 S2 和 S3。 S2 从已存在的 db 文件恢复 db 数据，根据 CURRENT 记录的 MANIFEST 文件读取 db 元信息；这通过调用VersionSet::Recover()完成。 S3 然后过滤出那些最近的更新 log，前一个版本可能新加了这些 log，但并没有记录在 MANIFEST 中。然后依次根据时间顺序，调用DBImpl::RecoverLogFile()从旧到新回放这些操作 log。回放 log 时可能会修改 db 元信息，比如 dump 了新的 level 0 文件，因此它将返回一个 VersionEdit 对象，记录 db 元信息的变动。 S4 如果DBImpl::Recover()返回成功，就执行VersionSet::LogAndApply()应用 VersionEdit，并保存当前的 DB 信息到新的 MANIFEST 文件中。 S5 最后删除一些过期文件，并检查是否需要执行 compaction，如果需要，就启动后台线程执行。 下面就来具体分析 Open 函数的代码，在 Open 函数中涉及到上面的 3 个流程。 S1 首先创建 DBImpl 对象，锁定并试图做 Recover 操作。Recover 操作用来处理创建 flag，比如存在就返回失败等等，尝试从已存在的 sstable 文件恢复 db。并返回 db 元信息的变动信息，一个 VersionEdit 对象。
1DBImpl* impl = newDBImpl(options, dbname);
2impl->mutex_.Lock(); // 锁db
3VersionEdit edit;
4Status s =impl->Recover(&edit); // 处理flag&恢复：create_if_missing,error_if_exists
S2 如果 Recover 返回成功，则调用 VersionSet 取得新的 log 文件编号——实际上是在当前基础上 +1，准备新的 log 文件。如果 log 文件创建成功，则根据 log 文件创建 log::Writer。然后执行 VersionSet::LogAndApply，根据 edit 记录的增量变动生成新的 current version，并写入 MANIFEST 文件。
函数 NewFileNumber(){returnnext_file_number_++;}，直接返回 next_file_number_。
1uint64_t new_log_number = impl->versions_->NewFileNumber();
2WritableFile* lfile;
3s = options.env->NewWritableFile(LogFileName(dbname, new_log_number), &lfile);
4if (s.ok()) {
5 edit.SetLogNumber(new_log_number);
6 impl->logfile_ = lfile;
7 impl->logfile_number_ = new_log_number;
8 impl->log_ = newlog::Writer(lfile);
9 s = impl->versions_->LogAndApply(&edit, &impl->mutex_);
10}
S3 如果 VersionSet::LogAndApply 返回成功，则删除过期文件，检查是否需要执行 compaction，最终返回创建的 DBImpl 对象。
1if (s.ok()) {
2 impl->DeleteObsoleteFiles();
3 impl->MaybeScheduleCompaction();
4}
5impl->mutex_.Unlock();
6if (s.ok()) *dbptr = impl;
7return s;
以上就是 DB::Open 的主题逻辑。
12.2 DBImpl::DBImpl()
构造函数做的都是初始化操作，
DBImpl::DBImpl(const Options& options, const std::string&dbname)
首先是初始化列表中，直接根据参数赋值，或者直接初始化。Comparator 和 filter policy 都是参数传入的。在传递 option 时会首先将 option 中的参数合法化，logfile_number_ 初始化为 0，指针初始化为 NULL。 创建 MemTable，并增加引用计数，创建 WriteBatch。
1mem_(newMemTable(internal_comparator_)),
2tmp_batch_(new WriteBatch),
3mem_->Ref();
4// 然后在函数体中，创建TableCache和VersionSet。
5// 为其他预留10个文件，其余的都给TableCache.
6const int table_cache_size = options.max_open_files - 10;
7table_cache_ = newTableCache(dbname_, &options_, table_cache_size);
8versions_ = newVersionSet(dbname_, &options_, table_cache_, &internal_comparator_);
¶
12.3 DBImp::NewDB()¶
当外部在调用 DB::Open() 时设置了 option 指定如果 db 不存在就创建，如果 db 不存在 leveldb 就会调用函数创建新的 db。判断 db 是否存在的依据是
1// S1首先生产DB元信息，设置comparator名，以及log文件编号、文件编号，以及seq no。
2VersionEdit new_db;
3new_db.SetComparatorName(user_comparator()->Name());
4new_db.SetLogNumber(0);
5new_db.SetNextFile(2);
6new_db.SetLastSequence(0);
7// S2 生产MANIFEST文件，将db元信息写入MANIFEST文件。
8const std::string manifest = DescriptorFileName(dbname_, 1);
9WritableFile* file;
10Status s = env_->NewWritableFile(manifest, &file);
11if (!s.ok()) return s;
12{
13 log::Writer log(file);
14 std::string record;
15 new_db.EncodeTo(&record);
16 s = log.AddRecord(record);
17 if (s.ok()) s = file->Close();
18}
19delete file;
20// S3 如果成功，就把MANIFEST文件名写入到CURRENT文件中
21if (s.ok()) s = SetCurrentFile(env_, dbname_, 1);
22elseenv_->DeleteFile(manifest);
23return s;
这就是创建新 DB 的逻辑，很简单。
12.4 DBImpl::Recover()
函数声明为：
StatusDBImpl::Recover(VersionEdit* edit)
如果调用成功则设置 VersionEdit。Recover 的基本功能是：首先是处理创建 flag，比如存在就返回失败等等；然后是尝试从已存在的 sstable 文件恢复 db；最后如果发现有大于原信息记录的 log 编号的 log 文件，则需要回放 log，更新 db 数据。回放期间 db 可能会 dump 新的 level 0 文件，因此需要把 db 元信息的变动记录到 edit 中返回。函数逻辑如下：
S1 创建目录，目录以 db name 命名，忽略任何创建错误，然后尝试获取 db name/LOCK 文件锁，失败则返回。
1env_->CreateDir(dbname_);
2Status s = env_->LockFile(LockFileName(dbname_), &db_lock_);
3if (!s.ok()) return s;
S2 根据 CURRENT 文件是否存在，以及 option 参数执行检查。 如果文件不存在&create_is_missing=true，则调用函数 NewDB() 创建；否则报错。 如果文件存在& error_if_exists=true，则报错。 S3 调用 VersionSet 的Recover()函数，就是从文件中恢复数据。如果出错则打开失败，成功则向下执行 S4。
s = versions_->Recover();
S4 尝试从所有比 manifest 文件中记录的 log 要新的 log 文件中恢复（前一个版本可能会添加新的 log 文件，却没有记录在 manifest 中）。另外，函数 PrevLogNumber() 已经不再用了，仅为了兼容老版本。
1// S4.1 这里先找出所有满足条件的log文件：比manifest文件记录的log编号更新。
2SequenceNumber max_sequence(0);
3const uint64_t min_log = versions_->LogNumber();
4const uint64_t prev_log = versions_->PrevLogNumber();
5std::vector<std::string>filenames;
6s = env_->GetChildren(dbname_, &filenames); // 列出目录内的所有文件
7uint64_t number;
8FileType type;
9std::vector<uint64_t>logs;
10for (size_t i = 0; i < filenames.size(); i++) { // 检查log文件是否比min log更新
11 if (ParseFileName(filenames[i], &number, &type) && type == kLogFile
12 && ((number >= min_log) || (number == prev_log))) {
13 logs.push_back(number);
14 }
15}
16// S4.2 找到log文件后，首先排序，保证按照生成顺序，依次回放log。并把DB元信息的变动（sstable文件的变动）追加到edit中返回。
17std::sort(logs.begin(), logs.end());
18for (size_t i = 0; i < logs.size(); i++) {
19 s = RecoverLogFile(logs[i], edit, &max_sequence);
20 // 前一版可能在生成该log编号后没有记录在MANIFEST中，
21 //所以这里我们手动更新VersionSet中的文件编号计数器
22 versions_->MarkFileNumberUsed(logs[i]);
23}
24// S4.3 更新VersionSet的sequence
25if (s.ok()) {
26 if (versions_->LastSequence() < max_sequence)
27 versions_->SetLastSequence(max_sequence);
28}
上面就是 Recover 的执行流程。
12.5 DBImpl::DeleteObsoleteFiles()¶
这个是垃圾回收函数，如前所述，每次 compaction 和 recovery 之后都会有文件被废弃。DeleteObsoleteFiles 就是删除这些垃圾文件的，它在每次 compaction 和 recovery 完成之后被调用。 其调用点包括：DBImpl::CompactMemTable,DBImpl::BackgroundCompaction, 以及 DB::Open 的 recovery 步骤之后。 它会删除所有过期的 log 文件，没有被任何 level 引用到、或不是正在执行的 compaction 的 output 的 sstable 文件。 该函数没有参数，其代码逻辑也很直观，就是列出 db 的所有文件，对不同类型的文件分别判断，如果是过期文件，就删除之，如下：
1// S1 首先，确保不会删除pending文件，将versionset正在使用的所有文件加入到live中。
2std::set<uint64_t> live = pending_outputs_;
3versions_->AddLiveFiles(&live); //该函数其后分析
4 // S2 列举db的所有文件
5std::vector<std::string>filenames;
6env_->GetChildren(dbname_, &filenames);
7// S3 遍历所有列举的文件，根据文件类型，分别处理；
8uint64_t number;
9FileType type;
10for (size_t i = 0; i < filenames.size(); i++) {
11 if (ParseFileName(filenames[i], &number, &type)) {
12 bool keep = true; //false表明是过期文件
13 // S3.1 kLogFile，log文件，根据log编号判断是否过期
14 keep = ((number >= versions_->LogNumber()) ||
15 (number == versions_->PrevLogNumber()));
16 // S3.2 kDescriptorFile，MANIFEST文件，根据versionset记录的编号判断
17 keep = (number >= versions_->ManifestFileNumber());
18 // S3.3 kTableFile，sstable文件，只要在live中就不能删除
19 // S3.4 kTempFile，如果是正在写的文件，只要在live中就不能删除
20 keep = (live.find(number) != live.end());
21 // S3.5 kCurrentFile,kDBLockFile, kInfoLogFile，不能删除
22 keep = true;
23 // S3.6 如果keep为false，表明需要删除文件，如果是table还要从cache中删除
24 if (!keep) {
25 if (type == kTableFile) table_cache_->Evict(number);
26 Log(options_.info_log, "Delete type=%d #%lld\n", type, number);
27 env_->DeleteFile(dbname_ + "/" + filenames[i]);
28 }
29 }
30}
这就是删除过期文件的逻辑，其中调用到了 VersionSet::AddLiveFiles 函数，保证不会删除 active 的文件。
函数 DbImpl::MaybeScheduleCompaction() 放在 Compaction 一节分析，基本逻辑就是如果需要 compaction，就启动后台线程执行 compaction 操作。
12.6 DBImpl::RecoverLogFile()¶
函数声明：
StatusRecoverLogFile(uint64_t log_number, VersionEdit* edit,SequenceNumber* max_sequence)
参数说明： @log_number 是指定的 log 文件编号 @edit 记录 db 元信息的变化——sstable 文件变动 @max_sequence 返回 max{log 记录的最大序号, *max_sequence} 该函数打开指定的 log 文件，回放日志。期间可能会执行 compaction，生产新的 level 0sstable 文件，记录文件变动到 edit 中。 它声明了一个局部类 LogReporter 以打印错误日志，没什么好说的，下面来看代码逻辑。
1// S1 打开log文件返回SequentialFile*file，出错就返回，否则向下执行S2。
2// S2 根据log文件句柄file创建log::Reader，准备读取log。
3log::Reader reader(file, &reporter, true/*checksum*/, 0/*initial_offset*/);
4// S3 依次读取所有的log记录，并插入到新生成的memtable中。这里使用到了批量更新接口WriteBatch，具体后面再分析。
5std::string scratch;
6Slice record;
7WriteBatch batch;
8MemTable* mem = NULL;
9while (reader.ReadRecord(&record, &scratch) && status.ok()) { // 读取全部log
10 if (record.size() < 12) { // log数据错误，不满足最小长度12
11 reporter.Corruption(record.size(), Status::Corruption("log recordtoo small"));
12 continue;
13 }
14 WriteBatchInternal::SetContents(&batch, record); // log内容设置到WriteBatch中
15 if (mem == NULL) { // 创建memtable
16 mem = new MemTable(internal_comparator_);
17 mem->Ref();
18 }
19 status = WriteBatchInternal::InsertInto(&batch, mem); // 插入到memtable中
20 MaybeIgnoreError(&status);
21 if (!status.ok()) break;
22 const SequenceNumber last_seq =
23 WriteBatchInternal::Sequence(&batch) + WriteBatchInternal::Count(&batch) - 1;
24 if (last_seq > *max_sequence) *max_sequence = last_seq; // 更新max sequence
25 // 如果mem的内存超过设置值，则执行compaction，如果compaction出错，
26 // 立刻返回错误，DB::Open失败
27 if (mem->ApproximateMemoryUsage() > options_.write_buffer_size) {
28 status = WriteLevel0Table(mem, edit, NULL);
29 if (!status.ok()) break;
30 mem->Unref(); // 释放当前memtable
31 mem = NULL;
32 }
33}
34// S4 扫尾工作，如果mem != NULL，说明还需要dump到新的sstable文件中。
35if (status.ok() && mem != NULL) {// 如果compaction出错，立刻返回错误
36 status = WriteLevel0Table(mem, edit, NULL);
37}
38if (mem != NULL)mem->Unref();
39delete file;
40return status;
把 MemTabledump 到 sstable 是函数 WriteLevel0Table 的工作，其实这是 compaction 的一部分，准备放在 compaction 一节来分析。
12.7 小结
如上 DB 打开的逻辑就已经分析完了，打开逻辑参见 DB::Open() 中描述的 5 个步骤。此外还有两个东东：把 Memtable dump 到 sstable 的 WriteLevel0Table() 函数，以及批量修改 WriteBatch。第一个放在后面的 compaction 一节，第二个放在 DB 更新操作中。接下来就是 db 的关闭。
¶
13 DB 的关闭&销毁¶
13.1 DB 关闭¶
外部调用者通过 DB::Open() 获取一个 DB对象，如果要关闭打开的 DBdb 对象，则直接 delete db 即可，这会调用到 DBImpl 的析构函数。
析构依次执行如下的 5 个逻辑：
S1 等待后台 compaction 任务结束
S2 释放 db 文件锁，
13.2 DB 销毁¶
函数声明：
StatusDestroyDB(const std::string& dbname, const Options& options)
该函数会删除掉 db 的数据内容，要谨慎使用。函数逻辑为：
S1 获取 dbname 目录的文件列表到 filenames 中，如果为空则直接返回，否则进入 S2。
S2 锁文件