Java Socket编程：实现支持消息广播的多线程聊天服务器

嘿，我来帮你搞定这个消息广播的问题！核心问题在于你的服务器目前没有维护所有在线客户端的连接列表，每个客户端线程只自顾自处理自己的读写，所以没法把消息转发给其他人。下面是具体的修改方案，一步步来：

1. 先搞一个线程安全的客户端连接集合 #

我们需要一个全局的、线程安全的集合来保存所有活跃客户端的BufferedWriter（直接存Writer比存Socket更方便，因为每次发送消息都要用到它）。推荐用CopyOnWriteArrayList，它天生就是线程安全的，完美适配多线程场景下的读写操作。

在你的服务器类里添加这个静态集合：

// 全局保存所有在线客户端的Writer，线程安全
private static final List<BufferedWriter> clientWriters = new CopyOnWriteArrayList<>();

2. 修改客户端连接的处理逻辑 #

当有新客户端连进来时，我们要先创建它的Writer并加入集合；当客户端断开时，记得从集合里把它移除，避免无效连接占坑。

更新你的服务器主循环代码：

try (ServerSocket server = new ServerSocket(port)){
    System.out.println("Waiting for client connection-");
    while (true){
        Socket client = server.accept();
        System.out.println("Client applies for connection: " + client.getInetAddress());
        
        // 提前创建客户端的Writer并加入全局集合
        BufferedWriter writer = new BufferedWriter(new OutputStreamWriter(client.getOutputStream()));
        clientWriters.add(writer);
        
        Thread t = new Thread(() -> {
            try {
                serveClient(client, writer);
            } catch (IOException e) {
                System.out.println("Client error: " + e.getMessage());
            } finally {
                // 客户端断开时，清理资源并移除集合中的Writer
                clientWriters.remove(writer);
                try {
                    client.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        });
        t.start();
    }
}

3. 改造serveClient方法实现广播 #

现在，当服务器收到某个客户端的消息时，只需要遍历全局的客户端Writer集合，把消息发送给每一个人就行啦：

public void serveClient(Socket client, BufferedWriter clientWriter) throws IOException {
    BufferedReader reader = new BufferedReader(new InputStreamReader(client.getInputStream()));
    String clientAddr = client.getInetAddress().toString(); // 获取客户端标识，可选
    
    try {
        String line;
        // 循环读取客户端消息，直到连接断开
        while ((line = reader.readLine()) != null){
            System.out.println("I received from " + clientAddr + ": " + line);
            // 构造广播消息，这里可以带上发送者的标识，方便客户端区分
            String broadcastMessage = "[" + clientAddr + "] I received: " + line;
            
            // 遍历所有客户端，发送广播消息
            for (BufferedWriter writer : clientWriters) {
                try {
                    writer.write(broadcastMessage);
                    writer.newLine();
                    writer.flush();
                } catch (IOException e) {
                    // 如果发送失败，说明这个客户端可能已经断开，直接从集合移除
                    clientWriters.remove(writer);
                }
            }
        }
    } catch (IOException e) {
        System.out.println("Client disconnected: " + clientAddr);
    } finally {
        // 确保关闭所有资源
        reader.close();
        clientWriter.close();
    }
}

4. 客户端代码的小说明 #

你的客户端代码基本不用改！因为现在服务器广播的消息会被客户端的reader.readLine()接收到，然后显示出来。如果想让体验更好，可以在客户端收到消息时，不要和自己输入的提示混在一起，比如可以把输入提示放在单独的线程里，不过这属于优化项，核心功能已经实现了。

几个关键注意点 #

• 线程安全：用CopyOnWriteArrayList是因为它的迭代器不会抛出并发修改异常，遍历的时候很安全，适合这种多线程添加/移除元素的场景。
• 异常处理：发送消息给某个客户端失败时，一定要把它从集合里移除，不然下次还会尝试发送，浪费资源。
• 资源清理：客户端断开时，必须关闭对应的Socket、Reader和Writer，避免资源泄漏。

这样改完之后，不管哪个客户端发消息，服务器都会把消息同步发给所有在线的客户端啦！

内容的提问来源于stack exchange，提问作者Yige Song