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


    }
}






上一篇:聊聊并发(十)生产者消费者模式


下一篇:OceanBase时序数据库CeresDB正式商用 为用户提供安全可靠的数据存储管理服务