Java网络编程三 非阻塞通信TCP

 

1. nio服务端

import java.io.BufferedReader;
import java.io.ByteArrayInputStream;
import java.io.DataInputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.InetSocketAddress;
import java.net.Socket;
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.Set;

public class NioTcpServer {

    public static void main(String[] args) {
        try {
            Selector selector = Selector.open();
            ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
            // 将其设为non-blocking状态，这样才能进行异步IO操作
            serverSocketChannel.configureBlocking(false);   
            serverSocketChannel.socket().setReuseAddress(true);
            serverSocketChannel.socket().bind(new InetSocketAddress(1000));
           
            // 将ServerSocketChannel注册到Selector上，返回对应的SelectionKey
            serverSocketChannel.register(selector, SelectionKey.OP_ACCEPT); //ServerSocketChannelImpl
           
            final int PACKAGE_SIZE = 10;
            ByteBuffer readBuff= ByteBuffer.allocate(1024);
            while(true){
                int n = selector.select();
                if(n == 0){
                    continue;
                }
                Set<SelectionKey> readyKeys = selector.selectedKeys();
                for(SelectionKey key : readyKeys){
                    readyKeys.remove(key);

                    if(key.isAcceptable()){
                        ServerSocketChannel ssc = (ServerSocketChannel)key.channel();
                        SocketChannel sc = ssc.accept();
                        sc.configureBlocking(false);
                        sc.register(selector, SelectionKey.OP_READ);
                        key.interestOps(SelectionKey.OP_ACCEPT);
                    }
                    if(key.isReadable()){
                        SocketChannel clientChannel = (SocketChannel)key.channel();
                        Socket client = clientChannel.socket();
                        clientChannel.read(readBuff);
                        key.interestOps(SelectionKey.OP_READ);
                       
                        System.out.println("client ----> IP: " + client.getInetAddress().getHostAddress() + ", port: " + client.getPort());
                        if(readBuff.position() >= PACKAGE_SIZE){
                            readBuff.flip();
                            DataInputStream dis = new DataInputStream(new ByteArrayInputStream(readBuff.array()));
                            System.out.println(dis.readInt());
                            BufferedReader d = new BufferedReader(new InputStreamReader(dis));
                            System.out.println(d.readLine());
                           
                            key.cancel();
                            key.channel().close();
                           
                            readBuff.clear();
                        }else{
                            clientChannel.register(selector, SelectionKey.OP_READ);
                        }
                       
                    }//if
                   
                }
            }
           
            //关闭一个已经注册的SelectableChannel需要两个步骤
            //1. 取消注册的key，这个可以通过SelectionKey.cancel方法，也可以通过SelectableChannel.close方法，或者中断阻塞在该channel上的IO操作的线程来做到
            //2. 后续的Selector.selectXXX方法的调用才真正地关闭本地Socket
            //如果在取消SelectionKey后没有调用到selector的select方法, 那么本地socket将进入CLOSE-WAIT状态（等待本地Socket关闭）.
//            serverSocketChannel.close();
//            selector.selectNow();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

}

 

 

2. 客户端可以使用《Java网络编程一 TCP通信》中的客户端程序来连接此NIO服务端测试。