Java AIO 入门实例

原文出处：http://tigerlchen.iteye.com/blog/1747221

Java7 AIO入门实例，首先是服务端实现：

服务端代码

SimpleServer:

Java代码

1. public class SimpleServer {
3. public SimpleServer(int port) throws IOException {
4. final AsynchronousServerSocketChannel listener = AsynchronousServerSocketChannel.open().bind(new InetSocketAddress(port));
6. listener.accept(null, new CompletionHandler<AsynchronousSocketChannel, Void>() {
7. public void completed(AsynchronousSocketChannel ch, Void att) {
8. // 接受下一个连接
9. listener.accept(null, this);
11. // 处理当前连接
12. handle(ch);
13. }
15. public void failed(Throwable exc, Void att) {
17. }
18. });
20. }
22. public void handle(AsynchronousSocketChannel ch) {
23. ByteBuffer byteBuffer = ByteBuffer.allocate(32);
24. try {
25. ch.read(byteBuffer).get();
26. } catch (InterruptedException e) {
27. // TODO Auto-generated catch block
28. e.printStackTrace();
29. } catch (ExecutionException e) {
30. // TODO Auto-generated catch block
31. e.printStackTrace();
32. }
33. byteBuffer.flip();
34. System.out.println(byteBuffer.get());
35. // Do something
36. }
38. }

public class SimpleServer {

    public SimpleServer(int port) throws IOException {
        final AsynchronousServerSocketChannel listener = AsynchronousServerSocketChannel.open().bind(new InetSocketAddress(port));

        listener.accept(null, new CompletionHandler<AsynchronousSocketChannel, Void>() {
            public void completed(AsynchronousSocketChannel ch, Void att) {
                // 接受下一个连接
                listener.accept(null, this);

                // 处理当前连接
                handle(ch);
            }

            public void failed(Throwable exc, Void att) {

            }
        });

    }

    public void handle(AsynchronousSocketChannel ch) {
        ByteBuffer byteBuffer = ByteBuffer.allocate(32);
        try {
            ch.read(byteBuffer).get();
        } catch (InterruptedException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        } catch (ExecutionException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
        byteBuffer.flip();
        System.out.println(byteBuffer.get());
        // Do something
    }
    
}

跟着是客户端实现：

客户端代码

SimpleClient:

Java代码

1. public class SimpleClient {
3. private AsynchronousSocketChannel client;
5. public SimpleClient(String host, int port) throws IOException, InterruptedException, ExecutionException {
6. this.client = AsynchronousSocketChannel.open();
7. Future<?> future = client.connect(new InetSocketAddress(host, port));
8. future.get();
9. }
11. public void write(byte b) {
12. ByteBuffer byteBuffer = ByteBuffer.allocate(32);
13. byteBuffer.put(b);
14. byteBuffer.flip();
15. client.write(byteBuffer);
16. }
18. }

public class SimpleClient {
    
    private AsynchronousSocketChannel client;
    
    public SimpleClient(String host, int port) throws IOException, InterruptedException, ExecutionException {
        this.client = AsynchronousSocketChannel.open();
        Future<?> future = client.connect(new InetSocketAddress(host, port));
        future.get();
    }
    
    public void write(byte b) {
        ByteBuffer byteBuffer = ByteBuffer.allocate(32);
        byteBuffer.put(b);
        byteBuffer.flip();
        client.write(byteBuffer);
    }

}

写一个简单的测试用例来跑服务端和客户端，先运行testServer()，在运行testClient();

测试用例

AIOTest

Java代码

1. public class AIOTest {
3. @Test
4. public void testServer() throws IOException, InterruptedException {
5. SimpleServer server = new SimpleServer(7788);
7. Thread.sleep(10000);
8. }
10. @Test
11. public void testClient() throws IOException, InterruptedException, ExecutionException {
12. SimpleClient client = new SimpleClient("localhost", 7788);
13. client.write((byte) 11);
14. }
16. }

public class AIOTest {
    
    @Test
    public void testServer() throws IOException, InterruptedException {
        SimpleServer server = new SimpleServer(7788);
        
        Thread.sleep(10000);
    }
    
    @Test
    public void testClient() throws IOException, InterruptedException, ExecutionException {
        SimpleClient client = new SimpleClient("localhost", 7788);
        client.write((byte) 11);
    }

}

因为是异步的，所以在运行server的时候没有发生同步阻塞，在这里我加了一个线程sleep()，如果没有的话，程序会直接跑完回收掉。