构建多客户端聊天室，开启Netty实战之旅

后端

2024-02-03 09:19:25

Netty入门：构建一个多客户端聊天室

简介

踏入Netty的世界，开启一段网络编程的探索之旅。本指南将带领你一步步构建一个多客户端聊天室，从零开始掌握Netty的基础知识。

准备工作

踏上旅程之前，你需要准备好以下装备：

JDK 1.8或更高版本
Netty 4.1.73或更高版本
一个趁手的文本编辑器或IDE

创建项目

作为任何编程项目的基础，首先创建一个Java项目。利用你熟稔的IDE，如IntelliJ IDEA或Eclipse，打造你的代码乐园。

添加Netty依赖

为了将Netty的强大功能纳入你的项目，你需要添加其依赖项。在你的项目构建文件中（例如pom.xml或build.gradle），输入以下依赖项代码：

<dependency>
  <groupId>io.netty</groupId>
  <artifactId>netty-all</artifactId>
  <version>4.1.73.Final</version>
</dependency>

编写服务器端代码

服务器端代码是聊天室的基石，它负责倾听客户端的呼唤，处理消息的传递。

public class ChatServer {

    public static void main(String[] args) throws Exception {
        // 创建一个ServerSocketChannel，守候客户端的到来
        ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();

        // 为ServerSocketChannel绑定一个端口号，开启聆听模式
        serverSocketChannel.bind(new InetSocketAddress(9090));

        // 创建一个Selector，时刻关注服务器端的动静
        Selector selector = Selector.open();

        // 将ServerSocketChannel注册到Selector上，等待客户端的敲门声
        serverSocketChannel.register(selector, SelectionKey.OP_ACCEPT);

        // 进入事件监听循环，等待客户端的连接和消息
        while (true) {
            // 调用select方法，耐心等待事件的发生
            selector.select();

            // 获取所有发生的事件
            Set<SelectionKey> selectionKeys = selector.selectedKeys();

            // 遍历所有事件
            for (SelectionKey selectionKey : selectionKeys) {
                // 如果是ACCEPT事件，说明有新的客户端连接
                if (selectionKey.isAcceptable()) {
                    // 接受客户端的连接，并获取客户端的SocketChannel
                    SocketChannel socketChannel = serverSocketChannel.accept();

                    // 将SocketChannel注册到Selector上，并指定感兴趣的事件为READ
                    socketChannel.register(selector, SelectionKey.OP_READ);

                    // 向客户端发送欢迎消息，表示热情款待
                    socketChannel.write(ByteBuffer.wrap("欢迎来到聊天室！".getBytes()));
                } else if (selectionKey.isReadable()) {
                    // 如果是READ事件，说明有客户端发送消息
                    // 获取客户端发送的消息
                    ByteBuffer byteBuffer = ByteBuffer.allocate(1024);
                    int readBytes = socketChannel.read(byteBuffer);

                    // 如果读取到的字节数为0，则说明客户端已断开连接
                    if (readBytes == 0) {
                        socketChannel.close();
                    } else {
                        // 将消息转为字符串，便于理解
                        String message = new String(byteBuffer.array(), 0, readBytes);

                        // 将消息广播给所有在线客户端，分享快乐
                        for (SelectionKey key : selector.keys()) {
                            if (key.channel() != socketChannel && key.channel() instanceof SocketChannel) {
                                SocketChannel client = (SocketChannel) key.channel();
                                client.write(ByteBuffer.wrap(message.getBytes()));
                            }
                        }
                    }
                }
            }

            // 清空事件集合，迎接新的事件
            selectionKeys.clear();
        }
    }
}

编写客户端代码

客户端代码是与服务器端交互的桥梁，它负责连接服务器，发送和接收消息。

public class ChatClient {

    public static void main(String[] args) throws Exception {
        // 创建一个SocketChannel，准备与服务器建立联系
        SocketChannel socketChannel = SocketChannel.open();

        // 将SocketChannel连接到服务器，敲开聊天室的大门
        socketChannel.connect(new InetSocketAddress("localhost", 9090));

        // 创建一个BufferedReader，从控制台读取消息
        BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(System.in));

        // 进入消息发送循环，与其他客户端尽情交流
        while (true) {
            // 从控制台读取消息
            String message = bufferedReader.readLine();

            // 将消息发送给服务器，传递你的心声
            socketChannel.write(ByteBuffer.wrap(message.getBytes()));

            // 如果输入了"exit"，则挥别聊天室，结束旅程
            if ("exit".equals(message)) {
                break;
            }
        }

        // 关闭连接，优雅地退出聊天室
        socketChannel.close();
    }
}