Java网络编程- IO多路复用(多Reactor)(主从式Reactor)

1. 介绍

本篇文章将IO多路复用(多线程)中的单Reactor+多线程网络模型做了改进，

把Reactor拆成两个角色Main Reactor及Sub Reactor，以提升效能与资源利用率​​。

Main Reactor：负责监听外部的连线请求，并派发给Acceptor处理。故Main Reactor中的selector只有注册OP_ACCEPT事件，也只能监听OP_ACCEPT事件。

Acceptor接受连线后会给client绑定一个Handler并注册IO事件到Sub Reactor上监听，

对于有多个Sub Reactor的情况下，IO事件选择注册给哪个Sub Reactor则是采用Round-robin的机制来分配。

Sub Reactor：負責監聽IO事件，並派發IO事件給Handler處理。Sub Reactor線程的數量可以設置為CPU核心數。

2. 网络模型

3. 源代码

Main Reactor

[TCPReactor.Java]

[java]  view plain   copy

 

1. // Reactor線程  
2. package server;  
3.   
4. import java.io.IOException;  
5. import java.net.InetSocketAddress;  
6. import java.nio.channels.SelectionKey;  
7. import java.nio.channels.Selector;  
8. import java.nio.channels.ServerSocketChannel;  
9. import java.util.Iterator;  
10. import java.util.Set;  
11.   
12. public class TCPReactor implements Runnable {  
13.   
14.     private final ServerSocketChannel ssc;  
15.     private final Selector selector; // mainReactor用的selector  
16.   
17.     public TCPReactor(int port) throws IOException {  
18.         selector = Selector.open();  
19.         ssc = ServerSocketChannel.open();  
20.         InetSocketAddress addr = new InetSocketAddress(port);  
21.         ssc.socket().bind(addr); // 在ServerSocketChannel綁定監聽端口  
22.         ssc.configureBlocking(false); // 設置ServerSocketChannel為非阻塞  
23.         SelectionKey sk = ssc.register(selector, SelectionKey.OP_ACCEPT); // ServerSocketChannel向selector註冊一個OP_ACCEPT事件，然後返回該通道的key  
24.         sk.attach(new Acceptor(ssc)); // 給定key一個附加的Acceptor對象  
25.     }  
26.   
27.     @Override  
28.     public void run() {  
29.         while (!Thread.interrupted()) { // 在線程被中斷前持續運行  
30.             System.out.println("mainReactor waiting for new event on port: "  
31.                     + ssc.socket().getLocalPort() + "...");  
32.             try {  
33.                 if (selector.select() == 0) // 若沒有事件就緒則不往下執行  
34.                     continue;  
35.             } catch (IOException e) {  
36.                 e.printStackTrace();  
37.             }  
38.             Set<SelectionKey> selectedKeys = selector.selectedKeys(); // 取得所有已就緒事件的key集合  
39.             Iterator<SelectionKey> it = selectedKeys.iterator();  
40.             while (it.hasNext()) {  
41.                 dispatch((SelectionKey) (it.next())); // 根據事件的key進行調度  
42.                 it.remove();  
43.             }  
44.         }  
45.     }  
46.   
47.     /* 
48.      * name: dispatch(SelectionKey key) 
49.      * description: 調度方法，根據事件綁定的對象開新線程 
50.      */  
51.     private void dispatch(SelectionKey key) {  
52.         Runnable r = (Runnable) (key.attachment()); // 根據事件之key綁定的對象開新線程  
53.         if (r != null)  
54.             r.run();  
55.     }  
56.   
57. }  

Acceptor

[Acceptor.java]

[java]  view plain   copy

 

1. // 接受連線請求線程  
2. package server;  
3.   
4. import java.io.IOException;  
5. import java.nio.channels.SelectionKey;  
6. import java.nio.channels.Selector;  
7. import java.nio.channels.ServerSocketChannel;  
8. import java.nio.channels.SocketChannel;  
9.   
10. public class Acceptor implements Runnable {  
11.   
12.     private final ServerSocketChannel ssc; // mainReactor監聽的socket通道  
13.     private final int cores = Runtime.getRuntime().availableProcessors(); // 取得CPU核心數  
14.     private final Selector[] selectors