一、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