AIO——AsynchronousServerSocketChannel

2019独角兽企业重金招聘Python工程师标准>>>

  接上篇，接下来练习通过AIO实现一个简单的TCP server。熟悉NIO的朋友应该都了解ServerSocket，在AIO中与之对应的就是AsynchronousServerSocketChannel，查看其API文档，介绍如下”An asynchronous channel for stream-oriented listening sockets. “，就是一个异步的数据流监听Socket。其主要函数有如下几个： 

accept()： 接受一个连接，返回一个Future，可通过Future获取到Socket的状态，和数据。 
accept(A attachment, CompletionHandler<AsynchronousSocketChannel,? super A> handler)：接受连接，并为连接绑定一个CompletionHandler处理Socket连接 

bind(SocketAddress local)：把ServerSocket绑定到本地端口上，等待连接。 
bind(SocketAddress local, int backlog)：功能和上面一个方法一样，添加了backlog参数指定队列中挂起的连接的最大个数 

open()：开启一个异步Socket通道， 
open(AsynchronousChannelGroup group)：开启一个异步Socket通道，并把通道加入指定的组做资源管理 

provider()：返回这个Channel的创建者 

setOption(SocketOption<T> name, T value)：配置Socket参数的方法。 


下面就是通过AsynchronousServerSocketChannel实现的一个简单的EchoServer，服务器会打印出收到的客户端输入，并把输入写回客户端。（注：代码只有在JDK1.7上才能编译通过）

package cn.bugsee.aio;

import java.io.IOException;
import java.net.InetSocketAddress;
import java.net.StandardSocketOptions;
import java.nio.ByteBuffer;
import java.nio.channels.AsynchronousChannelGroup;
import java.nio.channels.AsynchronousServerSocketChannel;
import java.nio.channels.AsynchronousSocketChannel;
import java.nio.channels.CompletionHandler;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.Executors;

/**
 * @USER echo
 * @TIME 2017-08-15 17:20
 * @DESC AIOEchoServer
 **/
public class AIOEchoServer {
    private AsynchronousServerSocketChannel serverSocketChannel;
    public static void main(String[] args) throws IOException {
        AIOEchoServer echoServer = new AIOEchoServer();
        echoServer.init("localhost",8090);
    }

    private void init(String host, int port) throws IOException {
        //ChannelGroup用来管理共享资源
        AsynchronousChannelGroup asynchronousChannelGroup = AsynchronousChannelGroup.withCachedThreadPool(Executors.newCachedThreadPool(),10);
        serverSocketChannel = AsynchronousServerSocketChannel.open(asynchronousChannelGroup);
        //通过setOption配置Socket
        serverSocketChannel.setOption(StandardSocketOptions.SO_REUSEADDR, true);
        serverSocketChannel.setOption(StandardSocketOptions.SO_RCVBUF, 16 * 1024);
        //绑定到指定的主机，端口
        serverSocketChannel.bind(new InetSocketAddress(host, port));
        System.out.println("Listening on " + host + ":" + port);
        //输出provider
        System.out.println("Channel Provider : " + serverSocketChannel.provider());
        //等待连接，并注册CompletionHandler处理内核完成后的操作。
        serverSocketChannel.accept(null, new CompletionHandler<AsynchronousSocketChannel, Object>() {
            final ByteBuffer buffer = ByteBuffer.allocate(1024);
            @Override
            public void completed(AsynchronousSocketChannel result, Object attachment) {
                System.out.println("waiting....");
                buffer.clear();
                try {
                    result.read(buffer).get();
                    buffer.flip();
                    System.out.println("Echo "+ new String(buffer.array()).toString() + " 0 " + result);
                    result.write(buffer);
                    buffer.flip();
                } catch (InterruptedException e) {
                    e.printStackTrace();
                } catch (ExecutionException e) {
                    e.printStackTrace();
                } finally {

                    try {
                        //关闭处理完的socket，并重新调用accept等待新的连接
                        result.close();
                        serverSocketChannel.accept(null, this);
                    } catch (IOException e) {
                        e.printStackTrace();
                    }
                }
            }

            @Override
            public void failed(Throwable exc, Object attachment) {
                System.out.println("Server failed... " + exc.getCause());
            }
        });
        //因为AIO不会阻塞调用进程，因此必须在主进程阻塞，才能保持进程存活。
        try {
            Thread.sleep(Integer.MAX_VALUE);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }
}

运行时会打印出如下信息： 
Listening on localhost:6025 
Channel Provider : sun.nio.ch.BsdAsynchronousChannelProvider@625cb0bb 


在客户端，可以通过telnet localhost 6025验证服务 
Trying ::1... 
telnet: connect to address ::1: Connection refused 
Trying 127.0.0.1... 
Connected to localhost. 
Escape character is '^]'. 

输入 hello，回车，客户端会打印 
hello 
hello 
Connection closed by foreign host. 

服务器端会有打印 
waiting.... 
Echo hello to sun.nio.ch.UnixAsynchronousSocketChannelImpl[connected local=/127.0.0.1:6025 remote=/127.0.0.1:54611] 

转载于:https://my.oschina.net/u/3613710/blog/1510640