HDFS源码分析之编辑日志编辑相关双缓冲区EditsDoubleBuffer

EditsDoubleBuffer是为edits准备的双缓冲区。新的编辑被写入第一个缓冲区，同时第二个缓冲区可以被flush。为edits准备的双缓冲区。新的编辑被写入第一个缓冲区，同时第二个缓冲区可以被flush。在其内部，有两个重要的缓冲区成员变量，如下：

 

// 当前被写入的缓冲区bufCurrent
private TxnBuffer bufCurrent; // current buffer for writing

// 正在进行flush的缓冲区bufReady
private TxnBuffer bufReady; // buffer ready for flushing

// 初始化缓冲区大小initBufferSize
private final int initBufferSize;

        其中，bufCurrent是当前被写入的缓冲区，当前被写入的缓冲区是正在进行flush的缓冲区，而initBufferSize则是初始化缓冲区大小。我们再看下EditsDoubleBuffer的构造函数，如下：

 

 

 // 构造函数
 public EditsDoubleBuffer(int defaultBufferSize) {
   // 根据入参赋值initBufferSizeinitBufferSize
initBufferSize = defaultBufferSize;

// 创建当前被写入的缓冲区bufCurrent
bufCurrent = new TxnBuffer(initBufferSize);

// 创建正在进行flush的缓冲区bufReady
   bufReady = new TxnBuffer(initBufferSize);

 }

        根据入参赋值initBufferSizeinitBufferSize，然后分别创建上述两个缓冲区：创建当前被写入的缓冲区bufCurrent、创建正在进行flush的缓冲区bufReady。

 

        而EditsDoubleBuffer最基本的写入功能有两个，一个是用于写入操作符的writeOp()方法，另外一个就是用于写入事务的writeRaw()方法，代码分别如下：

 

// 写入操作符至bufCurrent
public void writeOp(FSEditLogOp op) throws IOException {
  bufCurrent.writeOp(op);
}

// 写入事务至bufCurrent
public void writeRaw(byte[] bytes, int offset, int length) throws IOException {
  bufCurrent.write(bytes, offset, length);
}

        均是将操作符或事物写入bufCurrent缓冲区。而在准备flush前，需要先调用setReadyToFlush()方法，设置缓冲区可以进行flush，代码如下：

 

 

 // 设置双缓冲区为可以进行flsuh
 public void setReadyToFlush() {

// 确保之前的数据已经被flush完毕，调用isFlushed()方法判断bufReady的大小是否为0即可
   assert isFlushed() : "previous data not flushed yet";

   // 交换bufReady、bufCurrent
   TxnBuffer tmp = bufReady;
   bufReady = bufCurrent;
   bufCurrent = tmp;
 }

        它首先会确保之前的数据已经被flush完毕，调用isFlushed()方法判断bufReady的大小是否为0即可，然后交换bufReady、bufCurrent。

 

        接着，我们需要调用flushTo()方法，将bufReady的内容写入指定输出流，并清空bufReady。此时不交换任何缓冲区，代码如下：

 

/**
 * Writes the content of the "ready" buffer to the given output stream,
 * and resets it. Does not swap any buffers.
 * 将bufReady的内容写入指定输出流，并清空bufReady。此时不交换任何缓冲区。
 */
public void flushTo(OutputStream out) throws IOException {
  bufReady.writeTo(out); // write data to file
  bufReady.reset(); // erase all data in the buffer
}

        而bufCurrent、bufReady都是一个TxnBuffer类型的缓冲区，这个TxnBuffer是对DataOutputBuffer的一个封装，保存了第一个事务艾迪firstTxId、事务数量numTxns、写入者writer等变量，它主要的两个方法，一个是写入操作符的writeOp()方法，实现如下：

 

 

// 写入操作符
public void writeOp(FSEditLogOp op) throws IOException {

  // 首次事务艾迪firstTxId被赋值为操作符的事务ID
  if (firstTxId == HdfsConstants.INVALID_TXID) {
    firstTxId = op.txid;
  } else {

    // 之后确保操作符的事务ID永远大于首次事务ID
    assert op.txid > firstTxId;
  }

  // 调用writer写入操作符
  writer.writeOp(op);

  // 事务数量numTxns累加
  numTxns++;
}

        首次事务艾迪firstTxId被赋值为操作符的事务ID，之后确保操作符的事务ID永远大于首次事务ID，然后调用writer写入操作符，并将事务数量numTxns累加。