hadoop的DFSOutputStream

当我们用命令:

  1. hadoop fs -copyFromLocal localfile hdfs://... 
hadoop fs -copyFromLocal localfile hdfs://...

将本地文件复制到HDFS时,其背后的复制过程是怎样的?本地文件通过什么方式传输到datanode上的呢?

这里面很显然的是:
1、文件在多个电脑之间进行了传输(至少有2台电脑:本地电脑和一个datanode节点)。
2、如果文件超过一个block的大小(默认是64M),那么将一个文件分割成多个block是在哪里发生的?

带着这些疑问,我们来解读一下源代码。

一、找到“幕后英雄”

通过简单的跟踪,就会发现这一功能是由FileSystem类的copyFromLocalFile方法完成的。

继续跟踪,会发现拷贝其实是在2个文件系统中进行的,下面是FileUtil类的一段代码:

  1. /** Copy files between FileSystems. */ 
  2. public static boolean copy(FileSystem srcFS, Path src,  
  3.                            FileSystem dstFS, Path dst,  
  4.                            boolean deleteSource, 
  5.                            boolean overwrite, 
  6.                            Configuration conf) throws IOException { 
  7.     ... // 为了突出重要代码,这里省略了部分代码 
  8.     InputStream in=null
  9.     OutputStream out = null
  10.     try
  11.       in = srcFS.open(src); 
  12.       out = dstFS.create(dst, overwrite); 
  13.       IOUtils.copyBytes(in, out, conf, true); 
  14.     } catch (IOException e) { 
  15.       IOUtils.closeStream(out); 
  16.       IOUtils.closeStream(in); 
  17.       throw e; 
  18.     } 
  19.     ...  
/** Copy files between FileSystems. */
  public static boolean copy(FileSystem srcFS, Path src, 
                             FileSystem dstFS, Path dst, 
                             boolean deleteSource,
                             boolean overwrite,
                             Configuration conf) throws IOException {
      ... // 为了突出重要代码,这里省略了部分代码
      InputStream in=null;
      OutputStream out = null;
      try {
        in = srcFS.open(src);
        out = dstFS.create(dst, overwrite);
        IOUtils.copyBytes(in, out, conf, true);
      } catch (IOException e) {
        IOUtils.closeStream(out);
        IOUtils.closeStream(in);
        throw e;
      }
      ... 
  }

copyFromLocalFile的实质是将文件从LocalFileSystem复制到DistributedFileSystem

从上面的代码可以看出,复制的关键是:
1、获得本地文件系统的输入流(用来读取本地文件)
2、获得HDFS的输出流(用来向HDFS写入数据)
3、从第一流读取数据,写入第二个流。

从LocalFileSystem获取输入流很简单。问题是,如何获取DistributedFileSystem的输出流呢?

继续读代码,会发现:
DistributedFileSystem借助了DFSClient类,来实现客户端与HDFS之间文件的传输任务。

在DFSClient类中,创建流的是这一句:
  1. OutputStream result = new DFSOutputStream(src, masked, 
  2.         overwrite, replication, blockSize, progress, buffersize, 
  3.         conf.getInt("io.bytes.per.checksum", 512)); 
OutputStream result = new DFSOutputStream(src, masked,
        overwrite, replication, blockSize, progress, buffersize,
        conf.getInt("io.bytes.per.checksum", 512));

所以,所有的奥秘就应该在类DFSOutputStream中了。


二、解读DFSOutputStream

DFSOutputStream负责将数据传输到HDFS中,既然数据是在本地读取的,又要保存在另外一台机器(datanode)上,所以这里面一定会涉及到Socket。

通过阅读DFSOutputStream源码,果然发现了DFSOutputStream对底层的socket通信进行的包装的细节,先说说DFSOutputStream中的几个变量:
  1. private Socket s;   // 与datanode之间建立的socket连接 
  2. private DataOutputStream blockStream;   // socket的输出流(client->datanode),用于将数据传输给datanode 
  3. private DataInputStream blockReplyStream; // socket的输入流(datanode->client),用户收到datanode的确认包 
	private Socket s;	// 与datanode之间建立的socket连接
	private DataOutputStream blockStream;	// socket的输出流(client->datanode),用于将数据传输给datanode
	private DataInputStream blockReplyStream; // socket的输入流(datanode->client),用户收到datanode的确认包

除了socket和流以外,DFSOutputStream还有2个队列和2个线程:
  1. private LinkedList<Packet> dataQueue = new LinkedList<Packet>(); 
  2. // dataQueue是数据队列,用于保存等待发送给datanode的数据包 
  3. private LinkedList<Packet> ackQueue = new LinkedList<Packet>(); 
  4. // ackQueue是确认队列,保存还没有被datanode确认接收的数据包 
  5. ... 
  6. private DataStreamer streamer = new DataStreamer();; 
  7. // streamer线程,不停的从dataQueue中取出数据包,发送给datanode 
  8. private ResponseProcessor response = null
  9. // response线程,用于接收从datanode返回的反馈信息 
private LinkedList<Packet> dataQueue = new LinkedList<Packet>();
    // dataQueue是数据队列,用于保存等待发送给datanode的数据包
    private LinkedList<Packet> ackQueue = new LinkedList<Packet>();
    // ackQueue是确认队列,保存还没有被datanode确认接收的数据包
    ...
    private DataStreamer streamer = new DataStreamer();;
    // streamer线程,不停的从dataQueue中取出数据包,发送给datanode
    private ResponseProcessor response = null;
    // response线程,用于接收从datanode返回的反馈信息

所以,在向DFSOutputStream中,写入数据(通常是byte数组)的时候,实际的传输过程是:
1、byte[]被封装成64KB的Packet,然后扔进dataQueue中
2、DataStreamer线程不断的从dataQueue中取出Packet,通过socket发送给datanode(向blockStream写数据)
    发送前,将当前的Packet从dataQueue中移除,并addLast进ackQueue
3、ResponseProcessor线程从blockReplyStream中读出从datanode的反馈信息
       反馈信息很简单,就是一个seqno,再加上每个datanode返回的标志(成功标志为DataTransferProtocol.OP_STATUS_SUCCESS)
       通过判断seqno(序列号,每个Packet有一个序列号),判断datanode是否接收到正确的包。
       只有收到反馈包中的seqno与ackQueue.getFirst()的包seqno相同时,说明正确。否则可能出现了丢包的情况。

如果一切OK,则从ackQueue中移出:ackQueue.removeFirst(); 说明这个Packet被datanode成功接收了。


三、datanode端是怎么接收数据的?

上面分析的代码都位于客户端,那么datanode端的代码又如何呢?

答案是:

在DataNode端,有一个Daemon线程:dataXceiverServer,它有一个用于数据传输的ServerSocket一直开在那里。

每当有client连接到datanode时,datanode会new一个DataXceiver

DataXceiver负责数据的传输工作。

如果是写操作(client->datanode),则调用writeBlock方法:
  1. case DataTransferProtocol.OP_WRITE_BLOCK: 
  2.         writeBlock( in ); 
	case DataTransferProtocol.OP_WRITE_BLOCK:
        	writeBlock( in );

writeBlock方法负责:将数据写入本地磁盘,并负责将数据传输给其他datanode,保证数据的拷贝数目(可以通过dfs.replication设置)。

具体负责数据接收的是这一行:

  1. blockReceiver.receiveBlock(mirrorOut, mirrorIn, replyOut, 
  2.                                 mirrorAddr, null, targets.length); 
  3.  
  4. 几个参数的含义: 
  5.           DataOutputStream mirrOut, // output to next datanode 
  6.                     // 下一个datanode的输出流 
  7.           DataInputStream mirrIn,   // input from next datanode 
  8.                     // 下一个datanode的输入流 
  9.           DataOutputStream replyOut,  // output to previous datanode 
  10.                     // 数据来源节点(可能是最初的client)的输出流 
  11.                     // 用来发送反馈通知包 
  12.           String mirrAddr, BlockTransferThrottler throttlerArg, 
  13.           int numTargets) throws IOException { 
	blockReceiver.receiveBlock(mirrorOut, mirrorIn, replyOut,
                                 mirrorAddr, null, targets.length);

	几个参数的含义:
		      DataOutputStream mirrOut, // output to next datanode
						// 下一个datanode的输出流
		      DataInputStream mirrIn,   // input from next datanode
						// 下一个datanode的输入流
		      DataOutputStream replyOut,  // output to previous datanode
						// 数据来源节点(可能是最初的client)的输出流
						// 用来发送反馈通知包
		      String mirrAddr, BlockTransferThrottler throttlerArg,
		      int numTargets) throws IOException {

在receiveBlock方法中,循环接收数据:
  
  1. /*
  2.      * Receive until packet length is zero.
  3.      */ 
  4.     while (receivePacket() > 0) {} 
  /* 
       * Receive until packet length is zero.
       */
      while (receivePacket() > 0) {}

在receivePacket方法中:
  1. 不断地从输入流中读取Packet数据: 
  2.     int payloadLen = readNextPacket(); 
  3.  
  4. 并将数据传输至下一个datanode节点: 
  5.     mirrorOut.write(buf.array(), buf.position(), buf.remaining()); 
  6.     mirrorOut.flush(); 
  7.  
  8.  
  9. 写入磁盘: 
  10.     out.write(pktBuf, dataOff, len); 
	不断地从输入流中读取Packet数据:
		int payloadLen = readNextPacket();

	并将数据传输至下一个datanode节点:
		mirrorOut.write(buf.array(), buf.position(), buf.remaining());
		mirrorOut.flush();


	写入磁盘:
		out.write(pktBuf, dataOff, len);



四、如果一个文件超过1个block大小,怎么重定向到新的datanode的?在哪里分割的(file分割成blocks)?

答案是:在DFSOutputStream类的writeChunk方法里。

  1. line 3043
  2.     if (bytesCurBlock == blockSize) {  // 问题是:它们能正好相等吗?万一bytesCurBlock > blockSize了怎么办? 
  3.             currentPacket.lastPacketInBlock = true
  4.             bytesCurBlock = 0
  5.             lastFlushOffset = -1
  6.     } 
line 3043:
	if (bytesCurBlock == blockSize) {  // 问题是:它们能正好相等吗?万一bytesCurBlock > blockSize了怎么办?
            currentPacket.lastPacketInBlock = true;
            bytesCurBlock = 0;
            lastFlushOffset = -1;
	}
再往下几行:
  1. int psize = Math.min((int)(blockSize-bytesCurBlock), writePacketSize); 
  2.         computePacketChunkSize(psize, bytesPerChecksum); 
int psize = Math.min((int)(blockSize-bytesCurBlock), writePacketSize);
        computePacketChunkSize(psize, bytesPerChecksum);

就是说,在new每个新的Packet之前,都会重新计算一下新的Packet的大小,
以保证新的Packet大小不会超过Block的剩余大小
如果block还有不到一个Packet的大小(比如还剩3kb的空间),则最后一个Packet的大小就是:
blockSize-bytesCurBlock,也就是3kb


  1. line 2285
  2.               // get new block from namenode. 
  3.               if (blockStream == null) { 
  4.                 LOG.debug("Allocating new block"); 
  5.                 nodes = nextBlockOutputStream(src);  
  6.                 this.setName("DataStreamer for file " + src + 
  7.                              " block " + block); 
  8.                 response = new ResponseProcessor(nodes); 
  9.                 response.start(); 
  10.               } 
line 2285:
              // get new block from namenode.
              if (blockStream == null) {
                LOG.debug("Allocating new block");
                nodes = nextBlockOutputStream(src); 
                this.setName("DataStreamer for file " + src +
                             " block " + block);
                response = new ResponseProcessor(nodes);
                response.start();
              }


在DataStreamer中,如果遇到one.lastPacketInBlock==true,则将blockStream设为null,之后会重新写入新的block。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值