使用NIO实现非阻塞式网络编程

最新推荐文章于 2024-07-31 20:45:17 发布

风～叶～树

最新推荐文章于 2024-07-31 20:45:17 发布

阅读量236

点赞数

文章标签：网络 nio socket 网络通信

本文链接：https://blog.csdn.net/qq_41004971/article/details/108044823

版权

服务端

/**
 * @description:
 * @author: yrm
 * @create: 2020-08-16 20:07
 **/
public class NIOServer {
    public static void main(String[] args) throws IOException {
        ServerSocketChannel listener = null;
        System.out.println("服务器已启动...");
        try {
            //创建服务器通道
            listener = ServerSocketChannel.open();
            //绑定端口号
            listener.bind(new InetSocketAddress("10.9.63.249",5555));
            //设置非阻塞
            listener.configureBlocking(false);
            //创建selector选择器
            Selector selector = Selector.open();
            //将通道注册到选择器上，并设置事件
            listener.register(selector, SelectionKey.OP_ACCEPT);
            //监听有没有事件发生(轮询法，没有事件发生阻塞)
            while (selector.select() > 0) {
                Set<SelectionKey> selectionKeys = selector.selectedKeys();
                Iterator<SelectionKey> it = selectionKeys.iterator();
                while (it.hasNext()) {
                    SelectionKey selectionKey = it.next();
                    //判断selectKey
                    if (selectionKey.isAcceptable()) {
                        //表示有客户端请求，接受请求
                        SocketChannel socketChannel = listener.accept();
                        System.out.println(socketChannel.getRemoteAddress()+"进入了聊天室");
                        //设置非阻塞
                        socketChannel.configureBlocking(false);
                        //将监听到的通道注册到选择器上
                        socketChannel.register(selector, SelectionKey.OP_READ);
                    }else if (selectionKey.isReadable()) {
                        SocketChannel sChannel = (SocketChannel) selectionKey.channel();
                        //创建缓冲区
                        ByteBuffer buffer = ByteBuffer.allocate(1024);
                        int len = 0;
                        try {
                            while ((len = sChannel.read(buffer)) > 0) {
                                //每进行读取则flip一次
                                buffer.flip();
                                System.out.println(sChannel.getRemoteAddress() +"说了:"+new String(buffer.array(), 0, buffer.limit()));
                                buffer.clear();
                            }
                            if (len == -1) {
                                sChannel.close();
                            }
                        } catch (IOException e) {
                            System.out.println(sChannel.getRemoteAddress()+"异常退出了！");
                            sChannel.close();
                        }
                    }
                }
                it.remove();
            }
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            listener.close();
        }

    }
}

客户端

/**
 * @description:
 * @author: yrm
 * @create: 2020-08-16 13:55
 * 使用NIO实现非阻塞式网络编程
 **/
public class NIOClient {
    public static void main(String[] args) throws Exception {
        //创建socket通道
        SocketChannel socketChannel = SocketChannel.open(new InetSocketAddress("10.9.63.249", 5555));
        //给通道设置非阻塞
        socketChannel.configureBlocking(false);
        Scanner scanner = new Scanner(System.in);
        while (true) {
            //创建缓冲区
            ByteBuffer buffer = ByteBuffer.allocate(1024);
            String data = scanner.nextLine();
            //将数据放入到缓冲区中
            buffer.put(data.getBytes());
            buffer.flip();
            //将缓冲区中的数据写入到通道中
            socketChannel.write(buffer);
            buffer.clear();
            if (data.equals("886")) {
                break;
            }
        }
        socketChannel.close();
    }
}