NIO系列：4.NIO快速入门

怼烎@ 2023-07-14 05:57 6阅读 0赞

前面的文章介绍了NIO的三大核心组件，现在编写一个 NIO 入门案例，实现服务器端和客户端之间的数据简单通讯（非阻塞）：  
服务端代码：

import java.net.InetSocketAddress;
    import java.nio.ByteBuffer;
    import java.nio.channels.*;
    import java.util.Iterator;
    import java.util.Set;
     
    public class NIOServer { 
        public static void main(String[] args) throws Exception{ 
     
            //创建ServerSocketChannel -> ServerSocket
            ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
     
            //得到一个Selecor对象
            Selector selector = Selector.open();
     
            //绑定一个端口6666, 在服务器端监听
            serverSocketChannel.socket().bind(new InetSocketAddress(6666));
            //设置为非阻塞
            serverSocketChannel.configureBlocking(false);
     
            //把 serverSocketChannel 注册到 selector 关心 事件为 OP_ACCEPT
            serverSocketChannel.register(selector, SelectionKey.OP_ACCEPT);
     
            System.out.println("注册后的selectionkey 数量=" + selector.keys().size()); // 1
     
     
     
            //循环等待客户端连接
            while (true) { 
     
                //这里我们等待1秒，如果没有事件发生, 返回
                if(selector.select(1000) == 0) {  //没有事件发生
                    System.out.println("服务器等待了1秒，无连接");
                    continue;
                }
     
                //如果返回的>0, 就获取到相关的 selectionKey集合
                //1.如果返回的>0， 表示已经获取到关注的事件
                //2.selector.selectedKeys() 返回关注事件的集合
                //通过 selectionKeys 反向获取通道
                Set<SelectionKey> selectionKeys = selector.selectedKeys();
                System.out.println("selectionKeys 数量 = " + selectionKeys.size());
     
                //遍历Set<SelectionKey>, 使用迭代器遍历
                Iterator<SelectionKey> keyIterator = selectionKeys.iterator();
     
                while (keyIterator.hasNext()) { 
                    //获取到SelectionKey
                    SelectionKey key = keyIterator.next();
                    //根据key 对应的通道发生的事件做相应处理
                    if(key.isAcceptable()) {  //如果是 OP_ACCEPT, 有新的客户端连接
                        //该该客户端生成一个 SocketChannel
                        SocketChannel socketChannel = serverSocketChannel.accept();
                        System.out.println("客户端连接成功 生成了一个 socketChannel " + socketChannel.hashCode());
                        //将 SocketChannel 设置为非阻塞
                        socketChannel.configureBlocking(false);
                        //将socketChannel 注册到selector, 关注事件为 OP_READ， 同时给socketChannel
                        //关联一个Buffer
                        socketChannel.register(selector, SelectionKey.OP_READ, ByteBuffer.allocate(1024));
                        System.out.println("客户端连接后 ，注册的selectionkey 数量=" + selector.keys().size()); //2,3,4..
                    }
                    if(key.isReadable()) {   //发生 OP_READ
     
                        //通过key 反向获取到对应channel
                        SocketChannel channel = (SocketChannel)key.channel();
     
                        //获取到该channel关联的buffer
                        ByteBuffer buffer = (ByteBuffer)key.attachment();
                        channel.read(buffer);
                        System.out.println("form 客户端 " + new String(buffer.array()));
                    }
     
                    //手动从集合中移动当前的selectionKey, 防止重复操作
                    keyIterator.remove();
                }
            }
        }
    }

客户端代码：

import java.net.InetSocketAddress;
    import java.nio.ByteBuffer;
    import java.nio.channels.SocketChannel;
     
    public class NIOClient { 
        public static void main(String[] args) throws Exception{ 
     
            //得到一个网络通道
            SocketChannel socketChannel = SocketChannel.open();
            //设置非阻塞
            socketChannel.configureBlocking(false);
            //提供服务器端的ip 和 端口
            InetSocketAddress inetSocketAddress = new InetSocketAddress("127.0.0.1", 6666);
            //连接服务器
            if (!socketChannel.connect(inetSocketAddress)) { 
                while (!socketChannel.finishConnect()) { 
                    System.out.println("因为连接需要时间，客户端不会阻塞，可以做其它工作..");
                }
            }
     
            //...如果连接成功，就发送数据
            String str = "hello, nio~";
            //Wraps a byte array into a buffer
            ByteBuffer buffer = ByteBuffer.wrap(str.getBytes());
            //发送数据，将 buffer 数据写入 channel
            socketChannel.write(buffer);
            System.in.read();
        }
    }

注意：必须设置通道为非阻塞，才能向Selector注册，否则报 java.nio.channels.IllegalBlockingModeException 错