Java Socket编程(四) 异步服务器

辛苦堆砌，转载请注明出处，谢谢！

        今天给大家介绍最后一种服务器模式，如果开发异步服务器，需要使用Java的NIO才可以，所以，会发现代码中很多使用的类，与之前的不同了，下面给出代码。

package com.yjp.server;import java.io.IOException;import java.net.InetSocketAddress;import java.nio.ByteBuffer;import java.nio.channels.SelectionKey;import java.nio.channels.Selector;import java.nio.channels.ServerSocketChannel;import java.nio.channels.SocketChannel;import java.util.Iterator;public class ToUpperTCPNonBlockServer {//服务器IPpublic static final String SERVER_IP = "127.0.0.1";//服务器端口号public static final int SERVER_PORT = 10005;//请求终结字符串public static final char REQUEST_END_CHAR = '#';public void startServer(String serverIP, int serverPort) throws IOException {//使用NIO需要用到ServerSocketChannel//其中包含一个ServerSocket对象ServerSocketChannel serverChannel = ServerSocketChannel.open();//创建地址对象InetSocketAddress localAddr = new InetSocketAddress(serverIP, serverPort);//服务器绑定地址serverChannel.bind(localAddr);//设置为非阻塞serverChannel.configureBlocking(false);//注册到selector，会调用ServerSocket的accept//我们用selector监听accept能否返回//当调用accept可以返回时，会得到通知//注意，是可以返回，还需要调用acceptSelector selector = Selector.open();serverChannel.register(selector, SelectionKey.OP_ACCEPT);while (true) {//调用select，阻塞在这里，直到有注册的channel满足条件selector.select();//如果走到这里，有符合条件的channel//可以通过selector.selectedKeys().iterator()拿到符合条件的迭代器Iterator<SelectionKey> keys = selector.selectedKeys().iterator();//处理满足条件的keyswhile (keys.hasNext()) {//取出一个key并移除SelectionKey key = keys.next();keys.remove();try {if (key.isAcceptable()) {//有accept可以返回//取得可以操作的channelServerSocketChannel server = (ServerSocketChannel) key.channel();//调用accept完成三次握手，返回与客户端可以通信的channelSocketChannel channel = server.accept();//将该channel置非阻塞channel.configureBlocking(false);//注册进selector，当可读或可写时将得到通知，select返回channel.register(selector, SelectionKey.OP_READ);} else if (key.isReadable()) {//有channel可读,取出可读的channelSocketChannel channel = (SocketChannel) key.channel();//创建读取缓冲区,一次读取1024字节ByteBuffer buffer = ByteBuffer.allocate(1024);channel.read(buffer);//锁住缓冲区，缓冲区使用的大小将固定buffer.flip();//附加上buffer，供写出使用key.attach(buffer);key.interestOps(SelectionKey.OP_WRITE);} else if (key.isWritable()) {//有channel可写,取出可写的channelSocketChannel channel = (SocketChannel) key.channel();//取出可读时设置的缓冲区ByteBuffer buffer = (ByteBuffer) key.attachment();//将缓冲区指针移动到缓冲区开始位置buffer.rewind();//读取为StringString recv = new String(buffer.array());//清空缓冲区buffer.clear();buffer.flip();//写回数据byte[] sendBytes = recv.toUpperCase().getBytes();channel.write(ByteBuffer.wrap(sendBytes));}} catch (IOException e) {//当客户端Socket关闭时，会走到这里，清理资源key.cancel();try {key.channel().close();} catch (IOException e1) {e1.printStackTrace();}}}}}public static void main(String[] args) {ToUpperTCPNonBlockServer server = new ToUpperTCPNonBlockServer();try {server.startServer(SERVER_IP, SERVER_PORT);} catch (IOException e) {e.printStackTrace();}}}

可以看到，新的服务器使用了ServerSocketChannel以及SocketChannel，而不再是之前ServerSocket以及Socket，异步服务器的好处在于，服务器没有工作可做的时候，会等在select调用上，不会占用系统资源，而当不同的条件满足时，又可以第一时间被唤醒，执行相应的操作，所以无论从资源的利用上，还是从响应的及时性上都优于前两种。另外，如果write和read的时间比较长，处理也可以放到线程中处理，这样就结合了并发服务器的优势。