现在的位置: 首页 > 综合 > 正文

NIO的写操作分析

2017年12月19日 ⁄ 综合 ⁄ 共 1422字 ⁄ 字号 评论关闭
很多人对NIO的写操作理解的不深,不知道为什么要注册写事件,何时注册写事件,为什么写完之后要取消注册写事件。
Selector.isWriteable()方法监控的是内核的写缓冲器是否可写,所以除非内核的写缓冲区满了,否则一旦一个SocketChannel注册了写事件,每次selector都会选中这个SocketChannel,进入到isWriteable()分支中去。

实际上服务器端写消息时完全可以不注册写事件,直接调用SocketChannel.write(ByteBuffer),也能把数据写到缓冲区并发送出去。但是这种情况下selector不会选择到isWriteable()分支。在写缓冲区满的情况下处理也不灵活,需要while(ByteBuffer.hasRemain())来检查写的状态,会导致CPU空转while.

更好的做法是注册写事件。注册写事件也有两种做法,一种是SocketChannel.register(selector, SelectionKey.OP_WRITE),直接注册到SocketChannel。第二种是用SelectionKey.interestOps(SelectionKey.interestOps() | SelectionKey.OP_WRITE),用SelectionKey来注册。每个SelectionKey只会关联一个SocketChannel,相当于注册到了SocketChannel上。

所以NIO写方法的一般做法是:
1. 注册写事件
2. 写数据
3. 取消注册写事件

注意第3步,每次写完之后要取消注册写事件,SelectionKey.interestOps(SelectionKey.interestOps() & ~SelectionKey.OP_WRITE),否则只要内核写缓冲区可写,这个SocketChannel就会一直被selector选中。

由于有两种注册写事件的方式,所以导致写数据也有两种做法,如果用SocketChannel来注册写事件,那么直接用SocketChannel来写ByteBuffer。
更好的一种做法是第二种,将业务处理和底层Socket操作隔离,采用SelectionKey的Attachment,把业务数据用Attachment的方式和SocketChannel绑定。在SocketChannel注册事件时可以添加Attachment,比如SocketChannel.register(selector, SelectionKey.OP_READ, ByteBuffer.allocate(1024));

业务方法只需要操作Attachment,往Attachment里面写,然后在Selector.isWriteable()分支里面取出写好的Attachment,然后调用SocketChannel.Write(ByteBuffer) 方法来把数据写到Channel。

大部分的框架都采用第二种方式,比如Thrift的TNonblockingServer处理写事件时,采用一个业务类FrameBuffer来封装业务数据的读写,把FrameBuffer作为Attachment绑定到SocketChannel,并用FrameBuffer记录读写的状态。这样业务功能只需要处理FrameBuffer,
TNonblockingServer来处理SocketChannel实际的写操作。

抱歉!评论已关闭.