leveldb深度剖析-压缩流程(3)

最新推荐文章于 2023-02-27 08:22:12 发布

xxb249

最新推荐文章于 2023-02-27 08:22:12 发布

阅读量721

点赞数

分类专栏：存储开源软件文章标签： leveldb压缩流程 DoCompactionWork函数解析

本文链接：https://blog.csdn.net/xxb249/article/details/96022422

版权

开源软件同时被 2 个专栏收录

53 篇文章 5 订阅

订阅专栏

存储

22 篇文章 15 订阅

订阅专栏

本篇是压缩流程中最后一篇，介绍DoCompactionWork方法。压缩主要工作是什么呢？剔除无效数据。什么是无效数据呢？就是被标记为删除的数据，我们要在这个流程中从磁盘中删除掉。下面来看一下具体是如何进行删除的。

一、迭代器

该函数比较长，这里分段介绍该函数。

/**
 * 执行压缩
 * @param compact 压缩信息
 */
Status DBImpl::DoCompactionWork(CompactionState* compact) {
  const uint64_t start_micros = env_->NowMicros();
  int64_t imm_micros = 0;  // Micros spent doing imm_ compactions

  Log(options_.info_log,  "Compacting %d@%d + %d@%d files",
      compact->compaction->num_input_files(0),
      compact->compaction->level(),
      compact->compaction->num_input_files(1),
      compact->compaction->level() + 1);

  assert(versions_->NumLevelFiles(compact->compaction->level()) > 0);
  assert(compact->builder == NULL);
  assert(compact->outfile == NULL);

  if (snapshots_.empty()) {//如果没有快照，则重复的旧k/v数据都可以删掉
    compact->smallest_snapshot = versions_->LastSequence();
  } else {//如果有快照，则只有sequenceNumber小于最老的快照的sequenceNumber的旧k/v数据才可以删掉
    compact->smallest_snapshot = snapshots_.oldest()->number_;
  }

  // Release mutex while we're actually doing the compaction work
  mutex_.Unlock();

  //创建迭代器 这个创建迭代器的过程非常重要  为下面判断数据是否有效 起到至关重要的作用
  Iterator* input = versions_->MakeInputIterator(compact->compaction);
  input->SeekToFirst();
  Status status;
  ParsedInternalKey ikey;
  std::string current_user_key;//当前记录user key
  bool has_current_user_key = false;//是否有当前user key 和current_user_key成对出现
  SequenceNumber last_sequence_for_key = kMaxSequenceNumber;

说明：

1）这部分最重要的内容就是创建了一个迭代器Iterator，这个迭代器非常的关键，是将所有的key按照从小到大的顺序进行排序。切记：创建完迭代器后就已经按照key的顺序进行排序了。

2）上面几个变了是处理记录是否真正删除主要控制逻辑。

二、循环遍历key并删除数据

for (; input->Valid() && !shutting_down_.Acquire_Load(); ) {//循环遍历
    // Prioritize immutable compaction work
    if (has_imm_.NoBarrier_Load() != NULL) {
      const uint64_t imm_start = env_->NowMicros();
      mutex_.Lock();
      if (imm_ != NULL) {
        CompactMemTable();//优先压缩immutable memtable
        bg_cv_.SignalAll();  // Wakeup MakeRoomForWrite() if necessary
      }
      mutex_.Unlock();
      imm_micros += (env_->NowMicros() - imm_start);
    }

    Slice key = input->key();//InternalKey
    if (compact->compaction->ShouldStopBefore(key) &&
        compact->builder != NULL) {
      status = FinishCompactionOutputFile(compact, input);
      if (!status.ok()) {
        break;
      }
    }
    ...
  }//end for

这部分逻辑比较简单，就是从迭代器中获取一条数据，这里的input->key()为InternalKey对象，并非是user key。下面这部分是关键


    // Handle key/value, add to state, etc.
    bool drop = false;//代表记录是否要被删除 true 要从数据库中删除
    if (!ParseInternalKey(key, &ikey)) {//解析key 失败
      // Do not hide error keys
      current_user_key.clear();
      has_current_user_key = false;
      last_sequence_for_key = kMaxSequenceNumber;
    } else {//解析key 成功
      /**
       * 经过排序之后 相同的user key被放到迭代器 组织到一起了(相同user key放到一起)
       * current_user_key 保存上次user key
       * ikey.user_key是本次循环中 处理的user key
       * 如果compare返回非0 表示这两个user key不一样，返回为0则说明两个key相同，
       * 那么相同说明了什么呢？ 说明上一条是标记为Delete的数据，而这条记录是原始添加数据，所以
       * 这条记录是需要删除的。
       */
      if (!has_current_user_key ||
          user_comparator()->Compare(ikey.user_key,
                                     Slice(current_user_key)) != 0) {
        // First occurrence of this user key
        current_user_key.assign(ikey.user_key.data(), ikey.user_key.size());
        has_current_user_key = true;//表示第一次出现 设置为current user key
        last_sequence_for_key = kMaxSequenceNumber; //设置为最大序列号
      }

      if (last_sequence_for_key <= compact->smallest_snapshot) {
        // Hidden by an newer entry for same user key
        drop = true;    // (A)
      } else if (ikey.type == kTypeDeletion &&
                 ikey.sequence <= compact->smallest_snapshot &&
                 compact->compaction->IsBaseLevelForKey(ikey.user_key)) {
        // IsBaseLevelForKey函数返回true表示user_key不存在于高层中   false 表示存在于高层中
        // For this user key 对于当前user key来说:
        // (1) there is no data in higher levels 它不存在高层ldb文件中
        // (2) data in lower levels will have larger sequence numbers
        // (3) data in layers that are being compacted here and have
        //     smaller sequence numbers will be dropped in the next
        //     few iterations of this loop (by rule (A) above).
        // Therefore this deletion marker is obsolete and can be dropped.
        //要删除该记录 && 序列号比之前数据库中还要小 && 当前记录不存在于高层次文件中 所以标记为删除
        drop = true;
      }

      last_sequence_for_key = ikey.sequence;
    }
#if 0
    Log(options_.info_log,
        "  Compact: %s, seq %d, type: %d %d, drop: %d, is_base: %d, "
        "%d smallest_snapshot: %d",
        ikey.user_key.ToString().c_str(),
        (int)ikey.sequence, ikey.type, kTypeValue, drop,
        compact->compaction->IsBaseLevelForKey(ikey.user_key),
        (int)last_sequence_for_key, (int)compact->smallest_snapshot);
#endif

    if (!drop) {//不是删除操作 写入文件
      // Open output file if necessary
      if (compact->builder == NULL) {
        status = OpenCompactionOutputFile(compact);
        if (!status.ok()) {
          break;
        }
      }
      if (compact->builder->NumEntries() == 0) {
        compact->current_output()->smallest.DecodeFrom(key);
      }
      compact->current_output()->largest.DecodeFrom(key);
      compact->builder->Add(key, input->value());

      // Close output file if it is big enough
      if (compact->builder->FileSize() >=
          compact->compaction->MaxOutputFileSize()) {
        status = FinishCompactionOutputFile(compact, input);
        if (!status.ok()) {
          break;
        }
      }
    }

    input->Next(); //获取下一条记录

说明：

1）局部变量drop为true表示记录从物理文件中删除，反之不需要删除掉。

2） ParseInternalKey方法是用于解析InternalKey对象，正常场景会进入else分支。

3）上面提到了创建的迭代器会对InternalKey进行排序，那么相同的user-key就会相邻，那么带有删除标记的一定在前面（删除标记为0，添加标记为1）。所以在处理是删除标记时直接删除掉了，但是原始数据应该怎么删除呢？

4）代码注释中有这样的内容：

/**
* 经过排序之后相同的user key被放到迭代器组织到一起了(相同user key放到一起)
* current_user_key 保存上次user key
* ikey.user_key是本次循环中处理的user key
* 如果compare返回非0 表示这两个user key不一样，返回为0则说明两个key相同，
* 那么相同说明了什么呢？说明上一条是标记为Delete的数据，而这条记录是原始添加数据，所以
* 这条记录是需要删除的。
*/
if (!has_current_user_key || user_comparator()->Compare(ikey.user_key, Slice(current_user_key)) != 0)

5）写入文件这里不在介绍了。

三、记录统计信息


  if (status.ok() && shutting_down_.Acquire_Load()) {
    status = Status::IOError("Deleting DB during compaction");
  }
  if (status.ok() && compact->builder != NULL) {
    status = FinishCompactionOutputFile(compact, input);
  }
  if (status.ok()) {
    status = input->status();
  }
  delete input;
  input = NULL;

  // 统计信息
  CompactionStats stats;
  stats.micros = env_->NowMicros() - start_micros - imm_micros;
  for (int which = 0; which < 2; which++) {
    for (int i = 0; i < compact->compaction->num_input_files(which); i++) {
      stats.bytes_read += compact->compaction->input(which, i)->file_size;
    }
  }
  for (size_t i = 0; i < compact->outputs.size(); i++) {
    stats.bytes_written += compact->outputs[i].file_size;
  }

  mutex_.Lock();
  stats_[compact->compaction->level() + 1].Add(stats);

  if (status.ok()) {
    status = InstallCompactionResults(compact);
  }
  if (!status.ok()) {
    RecordBackgroundError(status);
  }
  VersionSet::LevelSummaryStorage tmp;
  Log(options_.info_log,
      "compacted to: %s", versions_->LevelSummary(&tmp));
  return status;
}// end function

这里部分就是统计信息，这里不在深入展开了，大家可自行阅读相关源码