7. 彤哥说netty系列之Java NIO核心组件之Selector (2)

当Channel触发了某个事件,通常也叫作那个事件就绪了。比如,数据准备好可以读取了就叫作读就绪了,同样地,还有写就绪、连接就绪、接受就绪,当然后面两个不常听到。

在Java中,这四种监听事件是定义在SelectionKey中的:

SelectionKey.OP_READ,值为 1 << 0 = 0000 0001

SelectionKey.OP_WRITE,值 为 1 << 2 = 0000 0100

SelectionKey.OP_CONNECT,值为 1 << 3 = 0000 1000

SelectionKey.OP_ACCEPT,值为 1 << 4 = 0001 0000

所以,也可以通过位或命令监听多个感兴趣的事件:

int interestSet = SelectionKey.OP_READ | SelectionKey.OP_WRITE; SelectionKey

正如上面所看到的,Channel注册到Selector后返回的是一个SelectionKey,所以SelectionKey又可以看作是Channel和Selector之间的一座桥梁,把两者绑定在了一起。

SelectionKey具有以下几个重要属性:

interest set,感兴趣的事件集

ready set,就绪的事件集

保存着的Channel

保存着的Selector

attached object,附件

interest set

里面保存了注册Channel到Selector时传入的第二个参数,即感兴趣的事件集。

int interestSet = selectionKey.interestOps(); boolean isInterestedInAccept = interestSet & SelectionKey.OP_ACCEPT; boolean isInterestedInConnect = interestSet & SelectionKey.OP_CONNECT; boolean isInterestedInRead = interestSet & SelectionKey.OP_READ; boolean isInterestedInWrite = interestSet & SelectionKey.OP_WRITE;

可以通过位与运算查看是否注册了相应的事件。

ready set

里面保存了就绪了的事件集。

int readySet = selectionKey.readyOps(); selectionKey.isAcceptable(); selectionKey.isConnectable(); selectionKey.isReadable(); selectionKey.isWritable();

可以通过readyOps()方法获取所有就绪了的事件,也可以通过isXxxable()方法检查某个事件是否就绪。

保存的Channel和Selector Channel channel = selectionKey.channel(); Selector selector = selectionKey.selector();

通过channel()和selector()方法可以获取绑定的Channel和Selector。

attachment

可以调用attach(obj)方法绑定一个对象到SelectionKey上,并在后面需要用到的时候通过attachment()方法取出绑定的对象,也可以翻译为附件,它可以看作是数据传递的一种媒介,跟ThreadLocal有点类似,在前面绑定数据,在后面使用。

selectionKey.attach(theObject); Object attachedObj = selectionKey.attachment();

当然,也可以在注册Channel到Selector的时候就绑定附件:

SelectionKey key = channel.register(selector, SelectionKey.OP_READ, theObject); Selector.select()

一旦将一个或多个Channel注册到Selector上了,我们就可以调用它的select()方法了,它会返回注册时感兴趣的事件中就绪的事件,本文来源于工从号彤哥读源码。

select()方法有三种变体:

select(),无参数,阻塞直到某个Channel有就绪的事件了才返回(当然是我们注册的感兴趣的事件)

select(timeout),带超时,阻塞直到某个Channel有就绪的事件了,或者超时了才返回

selectNow(),立即返回,不会阻塞,不管有没有就绪的Channel都立即返回

select()的返回值为int类型,表示两次select()之间就绪的Channel,即使上一次调用select()时返回的就绪Channel没有被处理,下一次调用select()也不会再返回上一次就绪的Channel。比如,第一次调用select()返回了一个就绪的Channel,但是没有处理它,第二次调用select()时又有一个Channel就绪了,那也只会返回1,而不是2。

Selector.selectedKeys()

一旦调用select()方法返回了有就绪的Channel,我们就可以使用selectedKeys()方法来获取就绪的Channel了。

Set<SelectionKey> selectedKeys = selector.selectedKeys();

然后,就可以遍历这些SelectionKey来查看感兴趣的事件是否就绪了:

Set<SelectionKey> selectedKeys = selector.selectedKeys(); Iterator<SelectionKey> keyIterator = selectedKeys.iterator(); while(keyIterator.hasNext()) { SelectionKey key = keyIterator.next(); if(key.isAcceptable()) { // a connection was accepted by a ServerSocketChannel. } else if (key.isConnectable()) { // a connection was established with a remote server. } else if (key.isReadable()) { // a channel is ready for reading } else if (key.isWritable()) { // a channel is ready for writing } keyIterator.remove(); }

最后,一定要记得调用keyIterator.remove();移除已经处理的SelectionKey。

Selector.wakeup()

前面我们说了调用select()方法时,调用者线程会进入阻塞状态,直到有就绪的Channel才会返回。其实也不一定,wakeup()就是用来破坏规则的,可以在另外一个线程调用wakeup()方法强行唤醒这个阻塞的线程,这样select()方法也会立即返回。

内容版权声明:除非注明,否则皆为本站原创文章。

转载注明出处:https://www.heiqu.com/wpssgz.html