leveldb源码分析--插入删除流程

Status DB::Put(const WriteOptions& opt, const Slice& key, const Slice& value) {
  WriteBatch batch;        //leveldb中不管单个插入还是多个插入都是以WriteBatch的方式进行的
  batch.Put(key, value);
  return Write(opt, &batch);
}

Status DBImpl::Write(const WriteOptions& options, WriteBatch* my_batch) {
  Writer w(&mutex_);
  w.batch = my_batch;
  w.sync = options.sync;
  w.done = false;
    /* 产生一个Writer对象，然后保存必要的锁、batch、和同步写的相关信息 */
  MutexLock l(&mutex_);
  writers_.push_back(&w);   // 上锁，然后放入待写的队列中
  while (!w.done && &w != writers_.front()) {

/* 这里设计比较特别，需要跟后面的 BuildBatchGroup结合起来看，这里大致的意思是

   一直等待到这次写完成或者这次写被放在队列的最前面，BuildBatchGroup会将队列  

   里所有sync设置相同的写请求组成一个WriteBatch进行写入，这里的写请求有可能在

   别的线程完成写操作了，而是否在队列首的判断是有可能此刻没有其他线程在写循环中，

   或者本次写请求和前面的写请求的同步设置不一致，那么这种情况就需要自己进入该线

   程完成写的操作。

*/
    w.cv.Wait();
  }
  if (w.done) {
    return w.status;
  }

  // 这个函数的主要作用是清理内存表和外存（磁盘）的表使内存表腾出空间插入新的数据

// 这里的设计比较复杂设计到leveldb 的很多核心设计，我们这里先大致了解其功能
  Status status = MakeRoomForWrite(my_batch == NULL);
  uint64_t last_sequence = versions_->LastSequence();
  Writer* last_writer = &w;
  if (status.ok() && my_batch != NULL) {  // NULL batch is for compactions
    WriteBatch* updates = BuildBatchGroup(&last_writer);
    WriteBatchInternal::SetSequence(updates, last_sequence + 1);
    last_sequence += WriteBatchInternal::Count(updates);

     {
      mutex_.Unlock();BuildBatchGroup

   // 这里讲组装好的batch内容写入log，并根据同步设置判断是否同步到磁盘
      status = log_->AddRecord(WriteBatchInternal::Contents(updates));
      bool sync_error = false;
      if (status.ok() && options.sync) {
        status = logfile_->Sync();
        if (!status.ok()) {
          sync_error = true;
        }
      }
      if (status.ok()) {

    // 写入内存表，这里采用了一个遍历WriteBatch完成插入的方式，稍后分析
        status = WriteBatchInternal::InsertInto(updates, mem_);
      }
      mutex_.Lock();
      if (sync_error) {
        // 如果同步错误则记录相应错误信息.
        RecordBackgroundError(status);
      }
    }

// 删除在BuildBatch里面设置的零时Batch的内容 

if (updates == tmp_batch_) tmp_batch_->Clear();

    versions_->SetLastSequence(last_sequence);
  }

  while (true) {

// 唤醒所有等待写入的线程
    Writer* ready = writers_.front();
    writers_.pop_front();
    if (ready != &w) {
      ready->status = status;
      ready->done = true;
      ready->cv.Signal();
    }
    if (ready == last_writer) break;
  }

  // Notify new head of write queue，因为可能请求时不在队首而进入了等待状态，

// 这样唤醒他使其成为新的队首写线程，进行MakeRoomForWrite等一系列操作
  if (!writers_.empty()) {
    writers_.front()->cv.Signal();
  }

  return status;
}