多个SocketChannel注册Selector统一管理

虽然我们可以自己处理每一个Socket事件，比如读写数据，不过更常规的方式是注册一个选择器。这个选择器侦听着数据的变化事件。 每个注册的通道都有自己的SelectionKey,用这个可以区分到底是哪个通道产生了事件。

看代码

1. package net.java2000.nio;
3. import java.io.IOException;
4. import java.net.InetSocketAddress;
5. import java.nio.ByteBuffer;
6. import java.nio.channels.SelectionKey;
7. import java.nio.channels.Selector;
8. import java.nio.channels.SocketChannel;
9. import java.util.Iterator;
11. /**
12.  * 多个SocketChannel注册Selector。
13.  * 
14.  * @author 赵学庆，Java世纪网(java2000.net)
15.  * 
16.  */
17. public class SocketChannelSelector {
18.   public static SocketChannel createSocketChannel(String hostName, int port)
19.       throws IOException {
20.     SocketChannel sChannel = SocketChannel.open();
21.     sChannel.configureBlocking(false);
22.     sChannel.connect(new InetSocketAddress(hostName, port));
23.     return sChannel;
24.   }
26.   // 2个连接注册的选择器关键字
27.   static SelectionKey key1;
28.   static SelectionKey key2;
30.   public static void main(String[] args) {
31.     // 1个选择器，注册2个Socket 通道
32.     Selector selector = null;
33.     try {
34.       // 创建选择器
35.       selector = Selector.open();
37.       // 创建2个通道
38.       SocketChannel sChannel1 = createSocketChannel("163.net", 25);
39.       SocketChannel sChannel2 = createSocketChannel("mail.youkuaiyun.com", 25);
41.       // 注册选择器，侦听所有的事件
42.       key1 = sChannel1.register(selector, sChannel1.validOps());
43.       key2 = sChannel2.register(selector, sChannel1.validOps());
44.     } catch (IOException e) {
45.     }
47.     // 等待事件的循环
48.     while (true) {
49.       try {
50.         // 等待
51.         selector.select();
52.       } catch (IOException e) {
53.         break;
54.       }
56.       // 所有事件列表
57.       Iterator<SelectionKey> it = selector.selectedKeys().iterator();
59.       // 处理每一个事件
60.       while (it.hasNext()) {
61.         // 得到关键字
62.         SelectionKey selKey = it.next();
64.         // 删除已经处理的关键字
65.         it.remove();
67.         try {
68.           // 处理事件
69.           processSelectionKey(selKey);
70.         } catch (IOException e) {
71.           // 处理异常
72.           selKey.cancel();
73.         }
74.       }
75.     }
76.   }
78.   public static void processSelectionKey(SelectionKey selKey) throws IOException {
79.     ByteBuffer buf = ByteBuffer.allocateDirect(1024);
80.     // 确认连接正常
81.     if (selKey.isValid() && selKey.isConnectable()) {
82.       // 得到通道
83.       SocketChannel sChannel = (SocketChannel) selKey.channel();
85.       // 是否连接完毕？
86.       boolean success = sChannel.finishConnect();
87.       if (!success) {
88.         // 异常
89.         selKey.cancel();
90.       }
91.     }
93.     // 如果可以读取数据
94.     if (selKey.isValid() && selKey.isReadable()) {
95.       // 得到通道
96.       SocketChannel sChannel = (SocketChannel) selKey.channel();
98.       if (sChannel.read(buf) > 0) {
99.         // 转到最开始
100.         buf.flip();
102.         while (buf.remaining() > 0) {
103.           System.out.print((char) buf.get());
104.         }
106.         // 也可以转化为字符串，不过需要借助第三个变量了。
107.         // buf.get(buff, 0, numBytesRead);
108.         // System.out.println(new String(buff, 0, numBytesRead, "UTF-8"));
110.         // 复位，清空
111.         buf.clear();
112.       }
113.     }
115.     // 如果可以写入数据
116.     if (selKey.isValid() && selKey.isWritable()) {
117.       // 得到通道
118.       SocketChannel sChannel = (SocketChannel) selKey.channel();
119.       // 区分2个侦听器的关键字
120.       // 我这里只写一次数据。
121.       if (!s1 && key1.equals(selKey)) {
122.         System.out.println("channel1 write data..");
123.         buf.clear();
124.         buf.put("HELO localhost/n".getBytes());
125.         buf.flip();
126.         sChannel.write(buf);
127.         s1 = true;
128.       } else if (!s2 && key2.equals(selKey)) {
129.         System.out.println("channel2 write data..");
130.         buf.clear();
131.         buf.put("HELO localhost/n".getBytes());
132.         buf.flip();
133.         sChannel.write(buf);
134.         s2 = true;
135.       }
136.     }
137.   }
139.   // 判断已经写过数据的标志
140.   static boolean s1 = false;
141.   static boolean s2 = false;
142. }

一个运行的效果
channel1 write data..
220 Coremail SMTP(Anti Spam) System (163net[040302])
250 bjfee_app2
channel2 write data..
220 youkuaiyun.com ESMTP "can't tell you"
250 youkuaiyun.com


结论
  这样做，我们的多个数据通道就可以统一管理，分别处理器读写操作。