Java——基于java自身包实现消息系统间的通信（TCP/IP+NIO）

/**
 * Created by LiuHuiChao on 2016/11/15.
 * description:based on TCP/IP+NIO to deliver the message
 */
public class TCP_IP_NIO {

    @Test
    public void clientStart() throws IOException {
        SocketChannel channel=SocketChannel.open();
        channel.configureBlocking(false);//设置为非阻塞方式
        SocketAddress remote=new InetSocketAddress("127.0.0.1",8888);
        channel.connect(remote);
        Selector selector= Selector.open();
        channel.register(selector, SelectionKey.OP_CONNECT);
        /**阻塞至有感兴趣的IO事件发生，或到达超时时间，如果希望一直等至有感兴趣的IO事件发生，可调用无参数select方法，
         * 如果希望不阻塞直接返回目前是否有感兴趣的事件发生，可以调用selectNow方法
         * */
        int nkeys=selector.select();//如果nkeys大于0，说明有感兴趣的IO事件发生
        SelectionKey selectionKey=null;
        if(nkeys>0){
            Set<SelectionKey> keys=selector.selectedKeys();
            for(SelectionKey key : keys){
                //对于发生连接的事件
                if(key.isConnectable()){
                    SocketChannel sc= (SocketChannel) key.channel();
                    sc.configureBlocking(false);
                    /**注册感兴趣的IO读事件，通常不直接注册写事件，在发送缓冲区未满的情况下，一直是可写的，
                     * 因此，如注册了写事件，而又不用写数据，很容易造成CUP消耗100%的情况；
                     * */
                    selectionKey=sc.register(selector,SelectionKey.OP_READ);
                    sc.finishConnect();
                }else if(key.isReadable()){/**有流可读*/
                    ByteBuffer byteBuffer=ByteBuffer.allocate(1024);
                    SocketChannel sc= (SocketChannel) key.channel();
                    int readBytes=0;
                    try{
                        int ret=0;
                        try{
                            /**读取目前可读的流，sc.read返回的为成功复制到bytebuffer中的字节数；
                             * 此步骤为阻塞操作，值可能为0；当已经是流的结尾时，返回-1
                             * */
                            while((ret=sc.read(byteBuffer))>0){
                                readBytes+=ret;
                            }
                        }finally{
                            byteBuffer.flip();
                        }
                    }finally{
                        if(byteBuffer!=null){
                            byteBuffer.clear();
                        }
                    }
                }else if(key.isWritable()){/**可写入流*/
                    //取消对OP_WRITE事件的注册
                    key.interestOps(key.interestOps()&(~selectionKey.OP_WRITE));
                    SocketChannel sc= (SocketChannel) key.channel();
                    /**此步骤为阻塞操作，直到写入操作系统发送缓冲区或网路IO出现异常，返回的为成功写入的字节数，当操作系统的发送缓冲区已满，此处返回0*/
                    ByteBuffer byteBuffer=ByteBuffer.allocate(1024);
                    sc.read(byteBuffer);
                    int writtenedSize=sc.write(byteBuffer);
                    //如未写入，则继续注册感兴趣的OP_WRITE事件
                    if(writtenedSize==0){
                        key.interestOps(key.interestOps() | selectionKey.OP_WRITE);
                    }
                }
            }
            selector.selectedKeys().clear();
        }

    }

    @Test
    public void serverStart() throws IOException {
        ServerSocketChannel ssc=ServerSocketChannel.open();
        ServerSocket serverSocket=ssc.socket();
        //绑定要监听的端口
        serverSocket.bind(new InetSocketAddress(8888));
        ssc.configureBlocking(false);
        Selector selector= Selector.open();
        //注册感兴趣的事件连接
        ssc.register(selector,SelectionKey.OP_ACCEPT);
        /**
         * 之后采取和客户端相同的方式对selector.select进行轮询。。。但是要增加一个key.isAcceptable的处理。。。
         * */

    }
}