接上篇,接下来练习通过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下才能编译通过)
运行时会打印出如下信息:
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]
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下才能编译通过)
- public class AIOEchoServer {
- private AsynchronousServerSocketChannel server;
- public static void main(String[] args) throws IOException {
- AIOEchoServer aioServer = new AIOEchoServer();
- aioServer.init("localhost", 6025);
- }
- private void init(String host, int port) throws IOException {
- //ChannelGroup用来管理共享资源
- AsynchronousChannelGroup group = AsynchronousChannelGroup.withCachedThreadPool(Executors.newCachedThreadPool(), 10);
- server = AsynchronousServerSocketChannel.open(group);
- //通过setOption配置Socket
- server.setOption(StandardSocketOptions.SO_REUSEADDR, true);
- server.setOption(StandardSocketOptions.SO_RCVBUF, 16 * 1024);
- //绑定到指定的主机,端口
- server.bind(new InetSocketAddress(host, port));
- System.out.println("Listening on " + host + ":" + port);
- //输出provider
- System.out.println("Channel Provider : " + server.provider());
- //等待连接,并注册CompletionHandler处理内核完成后的操作。
- server.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 {
- //把socket中的数据读取到buffer中
- result.read(buffer).get();
- buffer.flip();
- System.out.println("Echo " + new String(buffer.array()).trim() + " to " + result);
- //把收到的直接返回给客户端
- result.write(buffer);
- buffer.flip();
- } catch (InterruptedException e) {
- e.printStackTrace();
- } catch (ExecutionException e) {
- e.printStackTrace();
- } finally {
- try {
- //关闭处理完的socket,并重新调用accept等待新的连接
- result.close();
- server.accept(null, this);
- } catch (IOException e) {
- e.printStackTrace();
- }
- }
- }
- @Override
- public void failed(Throwable exc, Object attachment) {
- System.out.print("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]