如果思考上面的代碼,實際上是因為阻塞的IO操作(阻塞等待客戶端連接進來,和發送消息),所以需要用每一個線程來包裝每一個阻塞操作,使得「阻塞等待」在線程之間編程了非阻塞的非同步操作,然而每一個線程內部,還是阻塞等待的狀態.那麼如果將阻塞的I/O改成為非阻塞的I/O也就能避免創建過多線程的情況. 這個時候,我們有請NIO上台表演一下.

來,直接把寶貝掏出來給你們看

public static void main(String[] args) throws IOException {
Selector selector = Selector.open();//創建一個NIO的Selector
ServerSocketChannel serverSocket = ServerSocketChannel.open();//創建serverChannel，就是創建了一個NIO版的Socket
serverSocket.bind(new InetSocketAddress("localhost", 1111));//將該socket綁定到本地1111埠進行監聽
serverSocket.configureBlocking(false);//置通道為非阻塞模式
serverSocket.register(selector, SelectionKey.OP_ACCEPT);//註冊對 Accept操作的訂閱
ByteBuffer buffer = ByteBuffer.allocate(256);

while (true) {
selector.select();
Set<SelectionKey> selectionKeys = selector.selectedKeys();//得到已經I/O就緒的key
Iterator<SelectionKey> iter = selectionKeys.iterator();
while (iter.hasNext()) {
SelectionKey key = iter.next();
if (key.isAcceptable()) {//如果是客戶端新的連接
register(selector, serverSocket);
}
if (key.isReadable()) {//如果是已連接客戶端的新數據
answerWithEcho(buffer, key);
}
iter.remove();
}
}
}

public static void register(Selector selector, ServerSocketChannel serverSocket) throws IOException {
SocketChannel client = serverSocket.accept();//為客戶端新的連接請求創建socket
client.configureBlocking(false);
client.register(selector, SelectionKey.OP_READ);//為這個客戶端socket訂閱"數據就緒"（有新的數據發過來，就是一個"數據就緒"）
}

private static void answerWithEcho(ByteBuffer buffer, SelectionKey key) throws IOException {
SocketChannel client = (SocketChannel) key.channel();
client.read(buffer);
if ("DONE".equals(new String(buffer.array()).trim())) {
client.close();
System.out.println("Closed");
}
buffer.flip();
client.write(buffer);//將客戶端的數據原樣返回客戶端
buffer.clear();
}

上述代碼用一幅圖來描述的話,就是下面這個樣子

上述模型是使用Java的NIO的改造,對於大量的並發可以使用更少的線程,從而達到更好的性能.對於NIO不了解的同學,可以移步我的另一篇學習筆記 玩轉Java NIO

預計下次更新時間為本周末

推薦閱讀：