NIO编程入门示例

r囧r小猫 2022-09-02 08:10 146阅读 0赞

首先在服务端创建一个ServerSocketChannel，监听指定端口，然后创建一个选择器，ServerSocketChannel在选择器上注册客户端连接事件；

当客户端发送连接请求时，事件触发，基于ServerSocketChannel创建SocketChannel，在选择器上注册相关事件（比如数据读取事件）；

通过选择器的select方法对选择器上注册的所有通道进行事件监听，当有事件触发时会把发生事件的通道关联的SelectionKey保存到内部集合中，并返回发生事件的通道数量；

根据SelectionKey反向获取通道及事件类型，进行业务处理。

服务端：

package com.tech.netty.nio;
    
    import java.net.InetSocketAddress;
    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.Iterator;
    import java.util.Set;
    
    /**
     * @author lw
     * @since 2021/8/2
     **/
    public class NIOServer {
        public static void main(String[] args) throws Exception {
            //创建ServerSocketChannel
            ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
            //创建选择器
            Selector selector = Selector.open();
            //绑定端口 在服务端进行监听
            serverSocketChannel.bind(new InetSocketAddress(6666));
            //设置为非阻塞
            serverSocketChannel.configureBlocking(false);
            //在选择器上注册监听事件 事件类型为客户端连接
            serverSocketChannel.register(selector, SelectionKey.OP_ACCEPT);
    
            //循环等待客户端连接及读取客户端发送的数据
            while(true){
                //监听事件1秒钟，如果没有事件发生则重新监听
                if(selector.select(1000)==0){
                    System.out.println("服务器等待了1秒，无连接");
                    continue;
                }
                //表示已获取到监听的事件，获取selectionKey
                Set<SelectionKey> selectionKeys = selector.selectedKeys();
                //进行遍历
                Iterator<SelectionKey> iterator = selectionKeys.iterator();
                while (iterator.hasNext()){
                    SelectionKey selectionKey = iterator.next();
                    //说明SelectionKey对应的通道上发生了客户端连接事件
                    if(selectionKey.isAcceptable()){
                        //给客户端创建一个SocketChannel进行通信
                        SocketChannel socketChannel = serverSocketChannel.accept();
                        System.out.println("有客户端连接 socketChannel hashCode="+socketChannel.hashCode());
                        //SocketChannel设置非阻塞
                        socketChannel.configureBlocking(false);
                        //在选择器上注册读取事件，并关联一个缓冲区用于数据读写
                        socketChannel.register(selector,SelectionKey.OP_READ, ByteBuffer.allocate(1024));
                    }else if(selectionKey.isReadable()){ //说明SelectionKey对应的通道上发生了读取事件
                        //通过SelectionKey反向获取通道
                        SocketChannel channel = (SocketChannel) selectionKey.channel();
                        //获取通道关联的缓冲区
                        ByteBuffer byteBuffer = (ByteBuffer) selectionKey.attachment();
    
                        channel.read(byteBuffer);
                        System.out.println("from 客户端："+new String(byteBuffer.array()));
                    }
                    //手动移除SelectionKey 防止重复操作
                    iterator.remove();
                }
            }
        }
    }

客户端：

package com.tech.netty.nio;
    
    import java.net.InetSocketAddress;
    import java.nio.ByteBuffer;
    import java.nio.channels.SocketChannel;
    
    /**
     * @author lw
     * @since 2021/8/2
     **/
    public class NIOClient {
        public static void main(String[] args) throws Exception {
            //得到一个网络通道
            SocketChannel socketChannel = SocketChannel.open();
            //设置非阻塞
            socketChannel.configureBlocking(false);
            InetSocketAddress inetSocketAddress = new InetSocketAddress("127.0.0.1", 6666);
           //连接服务器
            if(!socketChannel.connect(inetSocketAddress)){
                while (!socketChannel.finishConnect()){
                    System.out.println("连接需要时间，客户端不会阻塞，可以做一些其他操作...");
                }
            }
    
            //如果连接成功，就发送数据
            String str="hello 中国";
            ByteBuffer byteBuffer = ByteBuffer.wrap(str.getBytes());
            //将buffer数据写入通道
            socketChannel.write(byteBuffer);
    
            System.in.read();
        }
    }