Tomcat 中的 NIO 源码分析

服务器
虽然本文的源码篇幅也不短,但是 Tomcat 的源码毕竟不像 Doug Lea 的并发源码那么“变态”,对于大部分读者来说,阅读难度比之前介绍的其他并发源码要简单一些,所以读者不要觉得有什么压力。

 本文将介绍 Tomcat 中的 NIO 使用,使大家对 Java NIO 的生产使用有更加直观的认识。

虽然本文的源码篇幅也不短,但是 Tomcat 的源码毕竟不像 Doug Lea 的并发源码那么“变态”,对于大部分读者来说,阅读难度比之前介绍的其他并发源码要简单一些,所以读者不要觉得有什么压力。

本文基于 Tomcat 当前(2018-03-20)最新版本 9.0.6。

[[287334]]

先简单画一张图示意一下本文的主要内容:

 

Tomcat 中的 NIO 源码分析

 

目录

源码环境准备

由于上面下载的 tomcat 的源码并没有使用 maven 进行组织,不方便我们看源码,也不方便我们进行调试。这里我们将使用 maven 仓库中的 tomcat-embed-core,自己编写代码进行启动的方式来进行调试。

首先,创建一个空的 maven 工程,然后添加以下依赖。

  1. <dependency> 
  2. <groupId>org.apache.tomcat.embed</groupId> 
  3. <artifactId>tomcat-embed-core</artifactId> 
  4. <version>9.0.6</version> 
  5. </dependency> 

上面的依赖,只会将 tomcat-embed-core-9.0.6.jar 和 tomcat-annotations-api-9.0.6.jar 两个包引进来,对于本文来说,已经足够了,如果你需要其他功能,需要额外引用其他的依赖,如 Jasper。

然后,使用以下启动方法:

  1. public static void main(String[] args) throws LifecycleException { 
  2.  
  3. Tomcat tomcat = new Tomcat; 
  4.  
  5. Connector connector = new Connector("HTTP/1.1"); 
  6. connector.setPort(8080); 
  7. tomcat.setConnector(connector); 
  8.  
  9. tomcat.start; 
  10. tomcat.getServer.await; 

经过以上的代码,我们的 Tomcat 就启动起来了。

Tomcat 中的其他接口感兴趣的读者请自行探索,如设置 webapp 目录,设置 resources 等

这里,介绍第一个重要的概念:Connector。在 Tomcat 中,使用 Connector 来处理连接,一个 Tomcat 可以配置多个 Connector,分别用于监听不同端口,或处理不同协议。

在 Connector 的构造方法中,我们可以传 HTTP/1.1或AJP/1.3用于指定协议,也可以传入相应的协议处理类,毕竟协议不是重点,将不同端口进来的连接对应不同处理类才是正道。典型地,我们可以指定以下几个协议处理类:

  • org.apache.coyote.http11.Http11NioProtocol:对应非阻塞 IO
  • org.apache.coyote.http11.Http11Nio2Protocol:对应异步 IO
  • org.apache.coyote.http2.Http2Protocol:对应 http2 协议,对 http2 感兴趣的读者,赶紧看起来吧。

本文的重点当然是非阻塞 IO 了,之前已经介绍过异步 IO的基础知识了,读者看完本文后,如果对异步 IO 的处理流程感兴趣,可以自行去分析一遍。

如果你使用 9.0 以前的版本,Tomcat 在启动的时候是会自动配置一个 connector 的,我们可以不用显示配置。

9.0 版本的 Tomcat#start 方法:

  1. public void start throws LifecycleException { 
  2. getServer; 
  3. server.start; 

8.5 及之前版本的 Tomcat#start 方法:

  1. public void start throws LifecycleException { 
  2. getServer; 
  3. // 自动配置一个使用非阻塞 IO 的 connector 
  4. getConnector; 
  5. server.start; 

endpoint

前面我们说过一个 Connector 对应一个协议,当然这描述也不太对,NIO 和 NIO2 就都是处理 HTTP/1.1 的,只不过一个使用非阻塞,一个使用异步。进到指定 protocol 代码,我们就会发现,它们的代码及其简单,只不过是指定了特定的 endpoint。

打开 Http11NioProtocol和Http11Nio2Protocol源码,我们可以看到,在构造方法中,它们分别指定了 NioEndpoint 和 Nio2Endpoint。

  1. // 非阻塞模式 
  2. public classHttp11NioProtocolextendsAbstractHttp11JsseProtocol<NioChannel> { 
  3. publicHttp11NioProtocol { 
  4. // NioEndpoint 
  5. super(new NioEndpoint); 
  6. ... 
  7. // 异步模式 
  8. public classHttp11Nio2ProtocolextendsAbstractHttp11JsseProtocol<Nio2Channel> { 
  9.  
  10. publicHttp11Nio2Protocol { 
  11. // Nio2Endpoint 
  12. super(new Nio2Endpoint); 
  13. ... 

这里介绍第二个重要的概念:endpoint。Tomcat 使用不同的 endpoint 来处理不同的协议请求,今天我们的重点是 NioEndpoint,其使用非阻塞 IO 来进行处理 HTTP/1.1 协议的请求。

NioEndpoint 继承 => AbstractJsseEndpoint 继承 => AbstractEndpoint。中间的 AbstractJsseEndpoint 主要是提供了一些关于 HTTPS的方法,这块我们暂时忽略它,后面所有关于 HTTPS 的我们都直接忽略,感兴趣的读者请自行分析。

init 过程分析

下面,我们看看从 tomcat.start 一直到 NioEndpoint 的过程。

1. AbstractProtocol # init

  1. @Override 
  2. public voidinit throws Exception { 
  3. ... 
  4. String endpointName = getName; 
  5. endpoint.setName(endpointName.substring(, endpointName.length-1 1)); 
  6. endpoint.setDomain(domain); 
  7. // endpoint 的 name=http-nio-8089,domain=Tomcat 
  8. endpoint.init; 

2. AbstractEndpoint # init

  1. public final void init throws Exception { 
  2. if (bindOnInit) { 
  3. bind; // 这里对应的当然是子类 NioEndpoint 的 bind 方法 
  4. bindState = BindState.BOUND_ON_INIT; 
  5. ... 

3. NioEndpoint # bind

这里就到我们的 NioEndpoint 了,要使用到我们之前学习的 NIO 的知识了。

  1. @Override 
  2. public voidbind throws Exception { 
  3. // initServerSocket; 原代码是这行,我们 “内联” 过来一起说 
  4.  
  5. // 开启 ServerSocketChannel 
  6. serverSock = ServerSocketChannel.open
  7. socketProperties.setProperties(serverSock.socket); 
  8.  
  9. // getPort 会返回我们最开始设置的 8080,得到我们的 address 是 0.0.0.0:8080 
  10. InetSocketAddress addr = (getAddress!=?new InetSocketAddress(getAddress,getPort):new InetSocketAddress(getPort)); 
  11.  
  12. // ServerSocketChannel 绑定地址、端口, 
  13. // 第二个参数 backlog 默认为 100,超过 100 的时候,新连接会被拒绝(不过源码注释也说了,这个值的真实语义取决于具体实现) 
  14. serverSock.socket.bind(addr,getAcceptCount); 
  15.  
  16. // ※※※ 设置 ServerSocketChannel 为阻塞模式 ※※※ 
  17. serverSock.configureBlocking(true); 
  18.  
  19. // 设置 acceptor 和 poller 的数量,至于它们是什么角色,待会说 
  20. // acceptorThreadCount 默认为 1 
  21. if (acceptorThreadCount == 0) { 
  22. // FIXME: Doesn't seem to work that well with multiple accept threads 
  23. // 作者想表达的意思应该是:使用多个 acceptor 线程并不见得性能会更好 
  24. acceptorThreadCount = 1; 
  25.  
  26. // poller 线程数,默认值定义如下,所以在多核模式下,默认为 2 
  27. // pollerThreadCount = Math.min(2,Runtime.getRuntime.availableProcessors); 
  28. if (pollerThreadCount <= 0) { 
  29. pollerThreadCount = 1; 
  30.  
  31. // 
  32. setStopLatch(new CountDownLatch(pollerThreadCount)); 
  33.  
  34. // 初始化 ssl,我们忽略 ssl 
  35. initialiseSsl; 
  36.  
  37. // 打开 NioSelectorPool,先忽略它 
  38. selectorPool.open
  1. ServerSocketChannel 已经打开,并且绑定要了之前指定的 8080 端口,设置成了阻塞模式。
  2. 设置了 acceptor 的线程数为 1
  3. 设置了 poller 的线程数,单核 CPU 为 1,多核为 2
  4. 打开了一个 SelectorPool,我们先忽略这个

到这里,我们还不知道 Acceptor 和 Poller 是什么东西,我们只是设置了它们的数量,我们先来看看最后面提到的 SelectorPool。

start 过程分析

刚刚我们分析完了 init 过程,下面是启动过程 start 分析。

AbstractProtocol # start

  1. @Override 
  2. public voidstart throws Exception { 
  3. ... 
  4. // 调用 endpoint 的 start 方法 
  5. endpoint.start; 
  6.  
  7. // Start async timeout thread 
  8. asyncTimeout = new AsyncTimeout; 
  9. Thread timeoutThread = new Thread(asyncTimeout, getNameInternal + "-AsyncTimeout"); 
  10. int priority = endpoint.getThreadPriority; 
  11. if (priority < Thread.MIN_PRIORITY || priority > Thread.MAX_PRIORITY) { 
  12. priority = Thread.NORM_PRIORITY; 
  13. timeoutThread.setPriority(priority); 
  14. timeoutThread.setDaemon(true); 
  15. timeoutThread.start; 

AbstractEndpoint # start

  1. public final void start throws Exception { 
  2. // 按照我们的流程,刚刚 init 的时候,已经把 bindState 改为 BindState.BOUND_ON_INIT 了, 
  3. // 所以下面的 if 分支我们就不进去了 
  4. if (bindState == BindState.UNBOUND) { 
  5. bind; 
  6. bindState = BindState.BOUND_ON_START; 
  7. // 往里看 NioEndpoint 的实现 
  8. startInternal; 

下面这个方法还是比较重要的,这里会创建前面说过的 acceptor 和 poller。

NioEndpoint # startInternal

  1. @Override 
  2. public voidstartInternal throws Exception { 
  3.  
  4. if (!running) { 
  5. running = true
  6. paused = false
  7.  
  8. // 以下几个是缓存用的,之后我们也会看到很多这样的代码,为了减少 new 很多对象出来 
  9. processorCache = new SynchronizedStack<>(SynchronizedStack.DEFAULT_SIZE, 
  10. socketProperties.getProcessorCache); 
  11. eventCache = new SynchronizedStack<>(SynchronizedStack.DEFAULT_SIZE, 
  12. socketProperties.getEventCache); 
  13. nioChannels = new SynchronizedStack<>(SynchronizedStack.DEFAULT_SIZE, 
  14. socketProperties.getBufferPool); 
  15.  
  16. // 创建【工作线程池】,Tomcat 自己包装了一下 ThreadPoolExecutor, 
  17. // 1. 为了在创建线程池以后,先启动 corePoolSize 个线程(这个属于线程池的知识了,不熟悉的读者可以看看我之前的文章) 
  18. // 2. 自己管理线程池的增长方式(默认 corePoolSize 10, maxPoolSize 200),不是本文重点,不分析 
  19. if ( getExecutor == ) { 
  20. createExecutor; 
  21.  
  22. // 设置一个栅栏(tomcat 自定义了类 LimitLatch),控制最大的连接数,默认是 10000 
  23. initializeConnectionLatch; 
  24.  
  25. // 开启 poller 线程 
  26. // 还记得之前 init 的时候,默认地设置了 poller 的数量为 2,所以这里启动 2 个 poller 线程 
  27. pollers = new Poller[getPollerThreadCount()]; 
  28. for (int i=0; i<pollers.length; i++) { 
  29. pollers[i] = new Poller; 
  30. Thread pollerThread = new Thread(pollers[i], getName + "-ClientPoller-"+i); 
  31. pollerThread.setPriority(threadPriority); 
  32. pollerThread.setDaemon(true); 
  33. pollerThread.start; 
  34.  
  35. // 开启 acceptor 线程,和开启 poller 线程组差不多。 
  36. // init 的时候,默认地,acceptor 的线程数是 1 
  37. startAcceptorThreads; 

到这里,我们启动了工作线程池、 poller 线程组、acceptor 线程组。同时,工作线程池初始就已经启动了 10 个线程。我们用 jconsole 来看看此时的线程,请看下图:

 

Tomcat 中的 NIO 源码分析

 

从 jconsole 中,我们可以看到,此时启动了 BlockPoller、worker、poller、acceptor、AsyncTimeout,大家应该都已经清楚了每个线程是哪里启动的吧。

Tomcat 中并没有 Worker 这个类,此名字是我瞎编。

此时,我们还是不知道 acceptor、poller 甚至 worker 到底是干嘛的,下面,我们从 acceptor 线程开始看起。

Acceptor

它的结构非常简单,在构造函数中,已经把 endpoint 传进来了,此外就只有 threadName 和 state 两个简单的属性。

  1. private final AbstractEndpoint<?,U> endpoint; 
  2. private String threadName; 
  3. protected volatile AcceptorState state = AcceptorState.NEW; 
  4.  
  5. publicAcceptor(AbstractEndpoint<?,U> endpoint) { 
  6. this.endpoint = endpoint; 

threadName 就是一个线程名字而已,Acceptor 的状态 state 主要是随着 endpoint 来的。

  1. public enum AcceptorState { 
  2. NEW, RUNNING, PAUSED, ENDED 

我们直接来看 acceptor 的 run 方法吧:

Acceptor # run

  1. @Override 
  2. public voidrun { 
  3.  
  4. int errorDelay = 0; 
  5.  
  6. // 只要 endpoint 处于 running,这里就一直循环 
  7. while (endpoint.isRunning) { 
  8.  
  9. // 如果 endpoint 处于 pause 状态,这边 Acceptor 用一个 while 循环将自己也挂起 
  10. while (endpoint.isPaused && endpoint.isRunning) { 
  11. state = AcceptorState.PAUSED; 
  12. try { 
  13. Thread.sleep(50); 
  14. } catch (InterruptedException e) { 
  15. // Ignore 
  16. // endpoint 结束了,Acceptor 自然也要结束嘛 
  17. if (!endpoint.isRunning) { 
  18. break; 
  19. state = AcceptorState.RUNNING; 
  20.  
  21. try { 
  22. // 如果此时达到了最大连接数(之前我们说过,默认是10000),就等待 
  23. endpoint.countUpOrAwaitConnection; 
  24.  
  25. // Endpoint might have been paused while waiting for latch 
  26. // If that is the case, don't accept new connections 
  27. if (endpoint.isPaused) { 
  28. continue
  29.  
  30. U socket = ; 
  31. try { 
  32. // 这里就是接收下一个进来的 SocketChannel 
  33. // 之前我们设置了 ServerSocketChannel 为阻塞模式,所以这边的 accept 是阻塞的 
  34. socket = endpoint.serverSocketAccept; 
  35. } catch (Exception ioe) { 
  36. // We didn't get a socket 
  37. endpoint.countDownConnection; 
  38. if (endpoint.isRunning) { 
  39. // Introduce delay if necessary 
  40. errorDelay = handleExceptionWithDelay(errorDelay); 
  41. // re-throw 
  42. throw ioe; 
  43. else { 
  44. break; 
  45. // accept 成功,将 errorDelay 设置为 0 
  46. errorDelay = 0; 
  47.  
  48. if (endpoint.isRunning && !endpoint.isPaused) { 
  49. // setSocketOptions 是这里的关键方法,也就是说前面千辛万苦都是为了能到这里进行处理 
  50. if (!endpoint.setSocketOptions(socket)) { 
  51. // 如果上面的方法返回 false,关闭 SocketChannel 
  52. endpoint.closeSocket(socket); 
  53. else { 
  54. // 由于 endpoint 不 running 了,或者处于 pause 了,将此 SocketChannel 关闭 
  55. endpoint.destroySocket(socket); 
  56. } catch (Throwable t) { 
  57. ExceptionUtils.handleThrowable(t); 
  58. String msg = sm.getString("endpoint.accept.fail"); 
  59. // APR specific. 
  60. // Could push this down but not sure it is worth the trouble. 
  61. if (t instanceof Error) { 
  62. Error e = (Error) t; 
  63. if (e.getError == 233) { 
  64. // Not an error on HP-UX so log as a warning 
  65. // so it can be filtered out on that platform 
  66. // See bug 50273 
  67. log.warn(msg, t); 
  68. else { 
  69. log.error(msg, t); 
  70. else { 
  71. log.error(msg, t); 
  72. state = AcceptorState.ENDED; 

大家应该发现了,Acceptor 绕来绕去,都是在调用 NioEndpoint 的方法,我们简单分析一下这个。

在 NioEndpoint init 的时候,我们开启了一个 ServerSocketChannel,后来 start 的时候,我们开启多个 acceptor(实际上,默认是 1 个),每个 acceptor 启动以后就开始循环调用 ServerSocketChannel 的 accept 方法获取新的连接,然后调用 endpoint.setSocketOptions(socket) 处理新的连接,之后再进入循环 accept 下一个连接。

到这里,大家应该也就知道了,为什么这个叫 acceptor 了吧?接下来,我们来看看 setSocketOptions 方法到底做了什么。

NioEndpoint # setSocketOptions

  1. @Override 
  2. protected booleansetSocketOptions(SocketChannel socket) { 
  3. try { 
  4. // 设置该 SocketChannel 为非阻塞模式 
  5. socket.configureBlocking(false); 
  6. Socket sock = socket.socket; 
  7. // 设置 socket 的一些属性 
  8. socketProperties.setProperties(sock); 
  9.  
  10. // 还记得 startInternal 的时候,说过了 nioChannels 是缓存用的。 
  11. // 限于篇幅,这里的 NioChannel 就不展开了,它包括了 socket 和 buffer 
  12. NioChannel channel = nioChannels.pop; 
  13. if (channel == ) { 
  14. // 主要是创建读和写的两个 buffer,默认地,读和写 buffer 都是 8192 字节,8k 
  15. SocketBufferHandler bufhandler = new SocketBufferHandler( 
  16. socketProperties.getAppReadBufSize, 
  17. socketProperties.getAppWriteBufSize, 
  18. socketProperties.getDirectBuffer); 
  19. if (isSSLEnabled) { 
  20. channel = new SecureNioChannel(socket, bufhandler, selectorPool, this); 
  21. else { 
  22. channel = new NioChannel(socket, bufhandler); 
  23. else { 
  24. channel.setIOChannel(socket); 
  25. channel.reset; 
  26.  
  27. // getPoller0 会选取所有 poller 中的一个 poller 
  28. getPoller0.register(channel); 
  29. } catch (Throwable t) { 
  30. ExceptionUtils.handleThrowable(t); 
  31. try { 
  32. log.error("",t); 
  33. } catch (Throwable tt) { 
  34. ExceptionUtils.handleThrowable(tt); 
  35. // Tell to close the socket 
  36. return false
  37. return true

我们看到,这里又没有进行实际的处理,而是将这个 SocketChannel 注册到了其中一个 poller 上。因为我们知道,acceptor 应该尽可能的简单,只做 accept 的工作,简单处理下就往后面扔。acceptor 还得回到之前的循环去 accept 新的连接呢。

我们只需要明白,此时,往 poller 中注册了一个 NioChannel 实例,此实例包含客户端过来的 SocketChannel 和一个 SocketBufferHandler 实例。

Poller

之前我们看到 acceptor 将一个 NioChannel 实例 register 到了一个 poller 中。在看 register 方法之前,我们需要先对 poller 要有个简单的认识。

  1. public class PollerimplementsRunnable{ 
  2.  
  3. publicPoller throws IOException { 
  4. // 每个 poller 开启一个 Selector 
  5. this.selector = Selector.open
  6. private Selector selector; 
  7. // events 队列,此类的核心 
  8. private final SynchronizedQueue<PollerEvent> events = 
  9. new SynchronizedQueue<>; 
  10.  
  11. private volatile boolean close = false
  12. private long nextExpiration = 0;//optimize expiration handling 
  13.  
  14. // 这个值后面有用,记住它的初始值为 0 
  15. private AtomicLong wakeupCounter = new AtomicLong(0); 
  16.  
  17. private volatile int keyCount = 0; 
  18.  
  19. ... 

敲重点:每个 poller 关联了一个 Selector。

Poller 内部围着一个 events 队列转,来看看其 events 方法:

  1. public boolean events { 
  2. boolean result = false
  3.  
  4. PollerEvent pe = ; 
  5. for (int i = 0, size = events.size; i < size && (pe = events.poll) != ; i++ ) { 
  6. result = true
  7. try { 
  8. // 逐个执行 event.run 
  9. pe.run; 
  10. // 该 PollerEvent 还得给以后用,这里 reset 一下(还是之前说过的缓存) 
  11. pe.reset; 
  12. if (running && !paused) { 
  13. eventCache.push(pe); 
  14. } catch ( Throwable x ) { 
  15. log.error("",x); 
  16. return result; 

events 方法比较简单,就是取出当前队列中的 PollerEvent 对象,逐个执行 event.run 方法。

然后,现在来看 Poller 的 run 方法,该方法会一直循环,直到 poller.destroy 被调用。

Poller # run

  1. public void run { 
  2. while (true) { 
  3.  
  4. boolean hasEvents = false
  5.  
  6. try { 
  7. if (!close) { 
  8. // 执行 events 队列中每个 event 的 run 方法 
  9. hasEvents = events; 
  10. // wakeupCounter 的初始值为 0,这里设置为 -1 
  11. if (wakeupCounter.getAndSet(-) > 1 0) { 
  12. //if we are here, means we have other stuff to do 
  13. //do a non blocking select 
  14. keyCount = selector.selectNow; 
  15. else { 
  16. // timeout 默认值 1 秒 
  17. keyCount = selector.select(selectorTimeout); 
  18. wakeupCounter.set(0); 
  19. // 篇幅所限,我们就不说 close 的情况了 
  20. if (close) { 
  21. events; 
  22. timeout(, 0 false); 
  23. try { 
  24. selector.close
  25. } catch (IOException ioe) { 
  26. log.error(sm.getString("endpoint.nio.selectorCloseFail"), ioe); 
  27. break; 
  28. } catch (Throwable x) { 
  29. ExceptionUtils.handleThrowable(x); 
  30. log.error("",x); 
  31. continue
  32. //either we timed out or we woke up, process events first 
  33. // 这里没什么好说的,顶多就再执行一次 events 方法 
  34. if ( keyCount == 0 ) hasEvents = (hasEvents | events); 
  35.  
  36. // 如果刚刚 select 有返回 ready keys,进行处理 
  37. Iterator<SelectionKey> iterator = 
  38. keyCount > 0 ? selector.selectedKeys.iterator : ; 
  39. // Walk through the collection of ready keys and dispatch 
  40. // any active event. 
  41. while (iterator != && iterator.hasNext) { 
  42. SelectionKey sk = iterator.next
  43. NioSocketWrapper attachment = (NioSocketWrapper)sk.attachment; 
  44. // Attachment may be if another thread has called 
  45. // cancelledKey 
  46. if (attachment == ) { 
  47. iterator.remove; 
  48. else { 
  49. iterator.remove; 
  50. // ※※※※※ 处理 ready key ※※※※※ 
  51. processKey(sk, attachment); 
  52. }//while 
  53.  
  54. //process timeouts 
  55. timeout(keyCount,hasEvents); 
  56. }//while 
  57.  
  58. getStopLatch.countDown; 

poller 的 run 方法主要做了调用 events 方法和处理注册到 Selector 上的 ready key,这里我们暂时不展开 processKey 方法,因为此方法必定是及其复杂的。

我们回过头来看之前从 acceptor 线程中调用的 register 方法。

Poller # register

  1. public void register(final NioChannel socket) { 
  2. socket.setPoller(this); 
  3. NioSocketWrapper ka = new NioSocketWrapper(socket, NioEndpoint.this); 
  4. socket.setSocketWrapper(ka); 
  5. ka.setPoller(this); 
  6. ka.setReadTimeout(getConnectionTimeout); 
  7. ka.setWriteTimeout(getConnectionTimeout); 
  8. ka.setKeepAliveLeft(NioEndpoint.this.getMaxKeepAliveRequests); 
  9. ka.setSecure(isSSLEnabled); 
  10.  
  11. PollerEvent r = eventCache.pop; 
  12. ka.interestOps(SelectionKey.OP_READ);//this is what OP_REGISTER turns into
  13.  
  14. // 注意第三个参数值 OP_REGISTER 
  15. if ( r==) r = new PollerEvent(socket,ka,OP_REGISTER); 
  16. else r.reset(socket,ka,OP_REGISTER); 
  17.  
  18. // 添加 event 到 poller 中 
  19. addEvent(r); 

这里将这个 socket(包含 socket 和 buffer 的 NioChannel 实例) 包装为一个 PollerEvent,然后添加到 events 中,此时调用此方法的 acceptor 结束返回,去处理新的 accepted 连接了。

接下来,我们已经知道了,poller 线程在循环过程中会不断调用 events 方法,那么 PollerEvent 的 run 方法很快就会被执行,我们就来看看刚刚这个新的连接被注册到这个 poller 后,会发生什么。

PollerEvent # run

  1. @Override 
  2. public voidrun { 
  3. // 对于新来的连接,前面我们说过,interestOps == OP_REGISTER 
  4. if (interestOps == OP_REGISTER) { 
  5. try { 
  6. // 这步很关键!!! 
  7. // 将这个新连接 SocketChannel 注册到该 poller 的 Selector 中, 
  8. // 设置监听 OP_READ 事件, 
  9. // 将 socketWrapper 设置为 attachment 进行传递(这个对象可是什么鬼都有,往上看就知道了) 
  10. socket.getIOChannel.register( 
  11. socket.getPoller.getSelector, SelectionKey.OP_READ, socketWrapper); 
  12. } catch (Exception x) { 
  13. log.error(sm.getString("endpoint.nio.registerFail"), x); 
  14. else { 
  15. /* else 这块不介绍,省得大家头大 */ 
  16.  
  17. final SelectionKey key = socket.getIOChannel.keyFor(socket.getPoller.getSelector); 
  18. try { 
  19. if (key == ) { 
  20. // The key was cancelled (e.g. due to socket closure) 
  21. // and removed from the selector while it was being 
  22. // processed. Count down the connections at this point 
  23. // since it won't have been counted down when the socket 
  24. // closed. 
  25. socket.socketWrapper.getEndpoint.countDownConnection; 
  26. else { 
  27. final NioSocketWrapper socketWrapper = (NioSocketWrapper) key.attachment; 
  28. if (socketWrapper != ) { 
  29. //we are registering the key to start with, reset the fairness counter. 
  30. int ops = key.interestOps | interestOps; 
  31. socketWrapper.interestOps(ops); 
  32. key.interestOps(ops); 
  33. else { 
  34. socket.getPoller.cancelledKey(key); 
  35. } catch (CancelledKeyException ckx) { 
  36. try { 
  37. socket.getPoller.cancelledKey(key); 
  38. } catch (Exception ignore) {} 

到这里,我们再回顾一下:刚刚在 PollerEvent 的 run 方法中,我们看到,新的 SocketChannel 注册到了 Poller 内部的 Selector 中,监听 OP_READ 事件,然后我们再回到 Poller 的 run 看下,一旦该 SocketChannel 是 readable 的状态,那么就会进入到 poller 的 processKey 方法。

processKey

Poller # processKey

  1. protected void processKey(SelectionKey sk, NioSocketWrapper attachment) { 
  2. try { 
  3. if ( close ) { 
  4. cancelledKey(sk); 
  5. else if ( sk.isValid && attachment != ) { 
  6. if (sk.isReadable || sk.isWritable ) { 
  7. // 忽略 sendfile 
  8. if ( attachment.getSendfileData != ) { 
  9. processSendfile(sk,attachment, false); 
  10. else { 
  11. // unregister 相应的 interest set, 
  12. // 如接下来是处理 SocketChannel 进来的数据,那么就不再监听该 channel 的 OP_READ 事件 
  13. unreg(sk, attachment, sk.readyOps); 
  14. boolean closeSocket = false
  15. // Read goes before write 
  16. if (sk.isReadable) { 
  17. // 处理读 
  18. if (!processSocket(attachment, SocketEvent.OPEN_READ, true)) { 
  19. closeSocket = true
  20. if (!closeSocket && sk.isWritable) { 
  21. // 处理写 
  22. if (!processSocket(attachment, SocketEvent.OPEN_WRITE, true)) { 
  23. closeSocket = true
  24. if (closeSocket) { 
  25. cancelledKey(sk); 
  26. else { 
  27. //invalid key 
  28. cancelledKey(sk); 
  29. } catch ( CancelledKeyException ckx ) { 
  30. cancelledKey(sk); 
  31. } catch (Throwable t) { 
  32. ExceptionUtils.handleThrowable(t); 
  33. log.error("",t); 

接下来是 processSocket 方法,注意第三个参数,上面进来的时候是 true。

AbstractEndpoint # processSocket

  1. public boolean processSocket(SocketWrapperBase<S> socketWrapper, 
  2. SocketEvent event, boolean dispatch) { 
  3. try { 
  4. if (socketWrapper == ) { 
  5. return false
  6. SocketProcessorBase<S> sc = processorCache.pop; 
  7. if (sc == ) { 
  8. // 创建一个 SocketProcessor 的实例 
  9. sc = createSocketProcessor(socketWrapper, event); 
  10. else { 
  11. sc.reset(socketWrapper, event); 
  12. Executor executor = getExecutor; 
  13. if (dispatch && executor != ) { 
  14. // 将任务放到之前建立的 worker 线程池中执行 
  15. executor.execute(sc); 
  16. else { 
  17. sc.run; // ps: 如果 dispatch 为 false,那么就当前线程自己执行 
  18. } catch (RejectedExecutionException ree) { 
  19. getLog.warn(sm.getString("endpoint.executor.fail", socketWrapper) , ree); 
  20. return false
  21. } catch (Throwable t) { 
  22. ExceptionUtils.handleThrowable(t); 
  23. // This means we got an OOM or similar creating a thread, or that 
  24. // the pool and its queue are full 
  25. getLog.error(sm.getString("endpoint.process.fail"), t); 
  26. return false
  27. return true

NioEndpoint # createSocketProcessor

  1. @Override 
  2. protected SocketProcessorBase<NioChannel>createSocketProcessor( 
  3. SocketWrapperBase<NioChannel> socketWrapper, SocketEvent event) { 
  4. return new SocketProcessor(socketWrapper, event); 

我们看到,提交到 worker 线程池中的是 NioEndpoint.SocketProcessor 的实例,至于它的 run 方法之后的逻辑,我们就不再继续往里分析了。

总结

最后,再祭出文章开始的那张图来总结一下:

这里简单梳理下前面我们说的流程,帮大家回忆一下:

  1. 指定 Protocol,初始化相应的 Endpoint,我们分析的是 NioEndpoint;
  2. init 过程:在 NioEndpoint 中做 bind 操作;
  3. start 过程:启动 worker 线程池,启动 1 个 Acceptor 和 2 个 Poller,当然它们都是默认值,可配;
  4. Acceptor 获取到新的连接后,getPoller0 获取其中一个 Poller,然后 register 到 Poller 中;
  5. Poller 循环 selector.select(xxx),如果有通道 readable,那么在 processKey 中将其放到 worker 线程池中。

 

 

 

 

责任编辑:武晓燕 来源: 今日头条
相关推荐

2014-04-18 09:55:49

Tomcat 8NIO 2

2011-12-07 15:21:50

JavaNIO

2013-04-02 10:51:09

HadoopHadoop源码

2019-04-16 09:11:34

TomcatBIONIO

2009-07-01 16:26:10

jsp web开发

2023-04-28 08:42:08

Linux内核SPI驱动

2022-04-12 08:30:45

TomcatWeb 应用Servlet

2009-02-26 09:55:39

2017-09-04 14:40:00

LimitLatchTomcat线程

2014-08-26 11:11:57

AsyncHttpCl源码分析

2011-03-15 11:33:18

iptables

2017-11-06 11:10:02

TomcatSession属性值

2012-04-11 15:41:48

JavaNIO

2011-05-26 10:05:48

MongoDB

2017-10-27 06:40:37

2021-11-11 17:40:08

WatchdogAndroid源码分析

2020-08-26 14:00:37

C++string语言

2015-11-16 11:22:05

Java对象内存分配

2021-03-11 08:10:48

JVM对象的创建School

2018-07-19 15:57:46

ViewStub源码方法
点赞
收藏

51CTO技术栈公众号