Kafka源码剖析 —— 网络I/O篇 —— 浅析KafkaChannel、NetworkReceive、Send

一、SocketChannel和KafkaChannel有什么区别？

上篇文章说道KafkaSelector在创建一个连接的时候和普通的nioSelector并没有什么不同，它是基于nioSelector的封装。我们知道创建连接的一系列操作都是由Channel去完成，而KafkaChannel实际上就是对它的进一步封装：
　　
　　KafkaChannel不仅封装了SocketChannel，还封装了Kafka自己的认证器Authenticator，和读写相关的NetworkReceive、Send。NetworkReceive和Send的底层都是通过ByteBuffer来实现的。

二、KafkaChannel的创建

实际上基本等同于KafkaSelector的创建：

按照普通的方式创建完通道后，将其注册到NioSelector上，并关注OP_CONNECT，再以节点Id，SelectionKey来创建KafkaChannel，这里先不详细说明KafkaChannel，它是对通道的进一步封装。在创建完KafkaChannel后，将KafkaChannel与SelectionKey、节点ID做进一步绑定。

        SelectionKey key = socketChannel.register(nioSelector, SelectionKey.OP_CONNECT);// 将当前这个socketChannel注册到nioSelector上，并关注OP_CONNECT事件
        KafkaChannel channel = channelBuilder.buildChannel(id, key, maxReceiveSize);// 创建KafkaChannel
        key.attach(channel);// 将channel绑定到key上
        this.channels.put(id, channel);// 将 nodeId 和 Channel绑定

这样有一个好处，首先KafkaChannel中包含了节点ID与SelectionKey，而我们也可以根据节点ID来拿到KafkaChannel，同样可以根据SelectionKey来拿到KafkaChannel，这就意味着，我们只要拿到了KafkaChannel、SelectionKey、节点ID中的任意一个，都可以通过这些引用关系拿到彼此，从而进行相关操作。

三、预发送

实际上就是将要发送的ByteBuffer扔进KafkaChannel，此时并未进行IO操作，这里的Send对象，实际上就是对ByteBuffer的进一步封装，它主要包含了将要发往的节点ID、ByteBuffer大小、是否发送完毕等信息。我们这里根据节点ID，从我们刚才的channels中，取出KafkaChannel。

public void send(Send send) {
        // 看看send要发的这个nodeId在不在
        KafkaChannel channel = channelOrFail(send.destination());
        try {
            // 把数据扔进KafkaChannel中（只能放一个，放多个会报错），并关注write事件
            channel.setSend(send);
        } catch (CancelledKeyException e) {

            // 失败了加一条node_id的失败记录
            this.failedSends.add(send.destination());
            close(channel);
        }
    }

这个KafkaChannel的setSend方法实际上非常简单，就是将要发送的send对象的引用交给KafkaChannel中的send。并且使这个channel的SelectionKey去关注OP_WRITE事件。

this.send = send;
this.transportLayer.addInterestOps(SelectionKey.OP_WRITE);

四、nio中的io操作

在上篇文章里，我们知道KafkaSelector也是通过轮询器去进行IO操作，看一下原始的nioSelector是如何进行io操作的：

public class NioEchoServer {
   
    private static final int BUF_SIZE = 256;
    private static