Netty源码分析第二章: NioEventLoop
** **
第七节:处理IO事件
上一小节我们了解了执行select()操作的相关逻辑, 这一小节我们继续学习select()之后, 轮询到io事件的相关逻辑:
回到NioEventLoop的run()方法:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44 1protected void run() {
2 for (;;) {
3 try {
4 switch (selectStrategy.calculateStrategy(selectNowSupplier, hasTasks())) {
5 case SelectStrategy.CONTINUE:
6 continue;
7 case SelectStrategy.SELECT:
8 //轮询io事件(1)
9 select(wakenUp.getAndSet(false));
10 if (wakenUp.get()) {
11 selector.wakeup();
12 }
13 default:
14 }
15 cancelledKeys = 0;
16 needsToSelectAgain = false;
17 //默认是50
18 final int ioRatio = this.ioRatio;
19 if (ioRatio == 100) {
20 try {
21 processSelectedKeys();
22 } finally {
23 runAllTasks();
24 }
25 } else {
26 //记录下开始时间
27 final long ioStartTime = System.nanoTime();
28 try {
29 //处理轮询到的key(2)
30 processSelectedKeys();
31 } finally {
32 //计算耗时
33 final long ioTime = System.nanoTime() - ioStartTime;
34 //执行task(3)
35 runAllTasks(ioTime * (100 - ioRatio) / ioRatio);
36 }
37 }
38 } catch (Throwable t) {
39 handleLoopException(t);
40 }
41 //代码省略
42 }
43}
44
我们首先看
if (ioRatio == 100) 这个判断, ioRatio主要是用来控制processSelectedKeys()方法执行时间和任务队列执行时间的比例, 其中ioRatio默认是50, 所以会走到下一步else
首先通过
final
long ioStartTime = System.nanoTime() 记录下开始时间, 再通过processSelectedKeys()方法处理轮询到的key
我们跟到processSelectedKeys()方法中:
1
2
3
4
5
6
7
8
9 1private void processSelectedKeys() {
2 if (selectedKeys != null) {
3 //flip()方法会直接返回key的数组
4 processSelectedKeysOptimized(selectedKeys.flip());
5 } else {
6 processSelectedKeysPlain(selector.selectedKeys());
7 }
8}
9
我们知道selector通过netty优化之后, 会初始化 selectedKeys这个属性, 所以这个属性不为空就会走到
processSelectedKeysOptimized(selectedKeys.flip()) 方法, 这个方法就是对应优化过的selector进行操作的
如果是非优化的selector, 则会进入
processSelectedKeysPlain(selector.selectedKeys()) 方法
selectedKeys.flip()为selectedKey中绑定的数组, 我们之前小节讲过selectedKeys其实是通过数组存储的, 所以经过select()操作如果监听到事件selectedKeys的数组就会有值
跟进到processSelectedKeysOptimized(selectedKeys.flip())方法中:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25 1private void processSelectedKeysOptimized(SelectionKey[] selectedKeys) {
2 //通过for循环遍历数组
3 for (int i = 0;; i ++) {
4 //拿到当前的selectionKey
5 final SelectionKey k = selectedKeys[i];
6 if (k == null) {
7 break;
8 }
9 //将当前引用设置为null
10 selectedKeys[i] = null;
11 //获取channel(NioSeverSocketChannel)
12 final Object a = k.attachment();
13 //如果是AbstractNioChannel, 则调用processSelectedKey()方法处理io事件
14 if (a instanceof AbstractNioChannel) {
15 processSelectedKey(k, (AbstractNioChannel) a);
16 } else {
17 @SuppressWarnings("unchecked")
18 NioTask<SelectableChannel> task = (NioTask<SelectableChannel>) a;
19 processSelectedKey(k, task);
20 }
21
22 //代码省略
23 }
24}
25
首先通过for循环遍历数组中的每一个key, 获得key之后首先将数组中对应的下标清空, 因为selector不会自动清空, 这与我们使用原生selector时候, 通过遍历selector.selectedKeys()的set的时候, 拿到key之后要执行remove()是一个意思
之后获取注册在key上的channel, 判断channel是不是AbstractNioChannel, 通常情况都是AbstractNioChannel, 所以这里会执行
processSelectedKey(k, (AbstractNioChannel) a)
跟到processSelectedKey(k, (AbstractNioChannel) a)方法中:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35 1private void processSelectedKey(SelectionKey k, AbstractNioChannel ch) {
2 //获取到channel中的unsafe
3 final AbstractNioChannel.NioUnsafe unsafe = ch.unsafe();
4 //如果这个key不是合法的, 说明这个channel可能有问题
5 if (!k.isValid()) {
6 //代码省略
7 }
8 try {
9 //如果是合法的, 拿到key的io事件
10 int readyOps = k.readyOps();
11 //链接事件
12 if ((readyOps & SelectionKey.OP_CONNECT) != 0) {
13 int ops = k.interestOps();
14 ops &= ~SelectionKey.OP_CONNECT;
15 k.interestOps(ops);
16 unsafe.finishConnect();
17 }
18 //写事件
19 if ((readyOps & SelectionKey.OP_WRITE) != 0) {
20 ch.unsafe().forceFlush();
21 }
22 //读事件和接受链接事件
23 //如果当前NioEventLoop是work线程的话, 这里就是op_read事件
24 //如果是当前NioEventLoop是boss线程的话, 这里就是op_accept事件
25 if ((readyOps & (SelectionKey.OP_READ | SelectionKey.OP_ACCEPT)) != 0 || readyOps == 0) {
26 unsafe.read();
27 if (!ch.isOpen()) {
28 return;
29 }
30 }
31 } catch (CancelledKeyException ignored) {
32 unsafe.close(unsafe.voidPromise());
33 }
34}
35
我们首先获取和channel绑定的unsafe, 之后拿到channel注册的事件
我们关注
if ((readyOps & (SelectionKey.OP_READ | SelectionKey.OP_ACCEPT)) != 0 || readyOps == 0) 这个判断, 这个判断相信注释上写的很明白, 如果当前NioEventLoop是work线程的话, 这里就是op_read事件, 如果是当前NioEventLoop是boss线程的话, 这里就是op_accept事件
然后会通过channel绑定的unsafe对象执行read()方法用于处理链接或者读写事件
以上就是NioEventLoop对io事件的处理过程, 有关read()方法执行逻辑, 会在以后的章节中详细剖析
上一节: 执行select操作
下一节: 执行任务队列