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的处理。。。         * */    }}