用 Java 实现的简易实时聊天系统

本文介绍如何使用 Java 编写一个简易的实时聊天系统。该系统使用 Socket 编程实现，允许多个客户端与服务器之间进行实时的消息传递。服务器将处理客户端的连接、消息转发以及管理所有活跃的连接。客户端通过控制台输入消息并实时接收来自其他用户的消息。
项目结构
/chat-app
│
├── /src
│   ├── Server.java        # 服务器端代码，负责接收客户端连接并广播消息
│   ├── Client.java        # 客户端代码，用户输入并接收消息
│
└── README.md             # 使用说明

1. 创建实时聊天服务器端
Server.java — 服务器端代码
服务器负责监听客户端的连接，接收客户端发送的消息并将其转发给所有其他连接的客户端。
import java.io.*;
import java.net.*;
import java.util.*;

public class Server {

    private static final int PORT = 12345; // 服务器监听端口
    private static Set<PrintWriter> clientWriters = new HashSet<>();

    public static void main(String[] args) {
        System.out.println("聊天服务器启动，等待连接...");

        try (ServerSocket serverSocket = new ServerSocket(PORT)) {
            while (true) {
                new ClientHandler(serverSocket.accept()).start();  // 每有一个新的客户端连接，启动一个线程
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    // 处理每个客户端的连接
    private static class ClientHandler extends Thread {
        private Socket socket;
        private PrintWriter out;
        private BufferedReader in;

        public ClientHandler(Socket socket) {
            this.socket = socket;
        }

        @Override
        public void run() {
            try {
                // 设置输入输出流
                in = new BufferedReader(new InputStreamReader(socket.getInputStream()));
                out = new PrintWriter(socket.getOutputStream(), true);

                synchronized (clientWriters) {
                    clientWriters.add(out);  // 将新的客户端输出流添加到集合中
                }

                // 处理客户端发送的消息
                String message;
                while ((message = in.readLine()) != null) {
                    System.out.println("收到消息: " + message);
                    broadcast(message);  // 将消息广播给所有客户端
                }
            } catch (IOException e) {
                e.printStackTrace();
            } finally {
                try {
                    socket.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }

                synchronized (clientWriters) {
                    clientWriters.remove(out);  // 客户端断开连接时，从集合中移除该客户端
                }
            }
        }

        // 广播消息给所有客户端
        private void broadcast(String message) {
            synchronized (clientWriters) {
                for (PrintWriter writer : clientWriters) {
                    writer.println(message);
                }
            }
        }
    }
}

2. 创建实时聊天客户端
Client.java — 客户端代码
客户端通过控制台输入消息并通过 Socket 连接服务器。同时，客户端不断监听来自服务器的消息并显示。
import java.io.*;
import java.net.*;

public class Client {

    private static final String SERVER_ADDRESS = "localhost"; // 服务器地址
    private static final int SERVER_PORT = 12345;            // 服务器端口

    public static void main(String[] args) {
        try (Socket socket = new Socket(SERVER_ADDRESS, SERVER_PORT);
             BufferedReader userInput = new BufferedReader(new InputStreamReader(System.in));
             PrintWriter out = new PrintWriter(socket.getOutputStream(), true);
             BufferedReader in = new BufferedReader(new InputStreamReader(socket.getInputStream()))) {

            // 启动接收消息的线程
            new Thread(new IncomingMessageHandler(in)).start();

            // 用户输入消息并发送给服务器
            String message;
            while (true) {
                System.out.print("请输入消息：");
                message = userInput.readLine();
                if (message != null) {
                    out.println(message); // 发送消息给服务器
                }
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    // 处理从服务器接收到的消息
    private static class IncomingMessageHandler implements Runnable {
        private BufferedReader in;

        public IncomingMessageHandler(BufferedReader in) {
            this.in = in;
        }

        @Override
        public void run() {
            String message;
            try {
                while ((message = in.readLine()) != null) {
                    System.out.println("收到消息: " + message); // 打印服务器广播的消息
                }
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
}

3. 编译与运行
3.1 编译 Java 文件
打开终端或命令行窗口，进入项目目录下的 src 文件夹，执行以下命令来编译 Java 文件：
javac Server.java Client.java

3.2 运行服务器
在终端中启动服务器：
java Server

此时，服务器将在 localhost 的 12345 端口上等待客户端连接。
3.3 运行客户端
打开多个终端窗口或命令行窗口，启动多个客户端，连接到聊天服务器。每个客户端都需要执行以下命令：
java Client

客户端启动后，可以在控制台输入消息，消息将被发送到服务器并转发给所有其他连接的客户端。
4. 使用说明

1.启动聊天服务器：

2.运行 java Server 启动聊天服务器，服务器将等待客户端连接。

3.连接到服务器：

4.运行 java Client 启动客户端。客户端将连接到 localhost 地址的 12345 端口。

5.发送和接收消息：

6.在任意客户端的控制台中输入消息并回车，消息将通过 WebSocket 发送到服务器，并且所有已连接的客户端都会接收到该消息。

7.退出客户端：

8.客户端可以通过 Ctrl + C 或直接关闭终端窗口来断开与服务器的连接。

5. 代码解析

9.服务器端 Server.java：
10.使用 ServerSocket 类创建一个服务器，该服务器监听特定的端口（在此例中为 12345）。
11.每当一个客户端连接时，服务器就会为其创建一个新的 ClientHandler 线程，负责接收客户端消息并将其广播给所有其他连接的客户端。
12.服务器通过一个 HashSet<PrintWriter> 集合来保存所有客户端的输出流，并使用 broadcast() 方法将消息发送给所有客户端。
13.客户端 Client.java：
14.客户端首先连接到服务器的指定 IP 地址和端口。
15.客户端通过 PrintWriter 发送消息，并通过 BufferedReader 接收服务器返回的消息。
16.客户端的消息接收通过一个独立的线程 IncomingMessageHandler 来处理，这样可以确保客户端可以同时发送和接收消息。

6. 注意事项

17.并发问题：本实现使用 synchronized 关键字来确保线程安全，特别是在操作 clientWriters 集合时。
18.性能优化：该示例适用于少量客户端的聊天应用。如果要处理大量的并发连接，可能需要引入线程池或使用更复杂的异步 IO 机制（例如 NIO）。
19.网络安全：此系统使用的是普通的 TCP/IP 套接字连接，没有加密或身份验证。在实际的生产环境中，建议使用加密（如 TLS）来保护通信内容。

7. 扩展功能建议

20.用户身份管理：为每个客户端分配一个唯一的用户名，显示发送者的名称。
21.聊天记录保存：可以将聊天记录保存在服务器端，方便回顾历史记录。
22.图形界面（GUI）：将该控制台程序改为图形界面的聊天工具，增加用户体验。
23.群组聊天：支持创建多个聊天室，用户可以选择加入不同的聊天室进行聊天。