init

day01/src/com/inmind/s_test_11/chatroom1/ChatServer.java

@@ -0,0 +1,111 @@
+package com.inmind.s_test_11.chatroom1;
+
+import java.io.*;
+import java.net.*;
+import java.util.*;
+import java.io.*;
+import java.net.*;
+import java.util.*;
+
+public class ChatServer {
+    // 服务器监听的端口号
+    private static final int PORT = 12345;
+    // 存储所有客户端输出流的集合，用于广播消息
+    private static Set<PrintWriter> clientWriters = new HashSet<>();
+
+    public static void main(String[] args) {
+        System.out.println("多人聊天服务器启动，监听端口: " + PORT);
+
+        try (ServerSocket serverSocket = new ServerSocket(PORT)) {
+            // 循环接受客户端连接
+            while (true) {
+                // 阻塞等待客户端连接
+                Socket clientSocket = serverSocket.accept();
+                System.out.println("新客户端连接: " + clientSocket);
+
+                // 创建PrintWriter用于向客户端发送消息
+                PrintWriter out = new PrintWriter(clientSocket.getOutputStream(), true);
+
+                // 将新客户端的输出流添加到集合中，使用同步块确保线程安全
+                synchronized (clientWriters) {
+                    clientWriters.add(out);
+                }
+
+                // 为每个客户端创建并启动一个新的处理线程
+                new Thread(new ClientHandler(clientSocket, out)).start();
+            }
+        } catch (IOException e) {
+            System.err.println("服务器错误: " + e.getMessage());
+        }
+    }
+
+    // 客户端处理线程类，负责接收客户端消息并广播
+    private static class ClientHandler implements Runnable {
+        private Socket clientSocket;    // 客户端Socket
+        private PrintWriter out;        // 向客户端输出的流
+        private String username;        // 客户端用户名
+
+        public ClientHandler(Socket socket, PrintWriter out) {
+            this.clientSocket = socket;
+            this.out = out;
+        }
+
+        public void run() {
+            try {
+                // 读取客户端发送的数据
+                BufferedReader in = new BufferedReader(
+                        new InputStreamReader(clientSocket.getInputStream()));
+
+                // 读取客户端发送的用户名
+                username = in.readLine();
+                System.out.println("用户 " + username + " 加入聊天");
+
+                // 广播新用户加入的消息
+                broadcast(username + " 加入了聊天！");
+
+                // 循环读取客户端发送的消息
+                String input;
+                while ((input = in.readLine()) != null) {
+                    // 如果用户输入exit，退出循环
+                    if (input.equalsIgnoreCase("exit")) {
+                        break;
+                    }
+
+                    // 广播用户发送的消息
+                    broadcast(username + ": " + input);
+                }
+
+                // 用户退出时的处理
+                System.out.println("用户 " + username + " 离开聊天");
+                broadcast(username + " 离开了聊天！");
+
+            } catch (IOException e) {
+                System.err.println("与客户端通信错误: " + e.getMessage());
+            } finally {
+                // 清理资源
+                if (out != null) {
+                    // 从客户端集合中移除当前客户端的输出流
+                    synchronized (clientWriters) {
+                        clientWriters.remove(out);
+                    }
+                }
+                try {
+                    // 关闭客户端连接
+                    clientSocket.close();
+                } catch (IOException e) {
+                    System.err.println("关闭客户端连接错误: " + e.getMessage());
+                }
+            }
+        }
+
+        // 广播消息给所有连接的客户端
+        private void broadcast(String message) {
+            // 同步遍历客户端输出流集合，确保线程安全
+            synchronized (clientWriters) {
+                for (PrintWriter writer : clientWriters) {
+                    writer.println(message);
+                }
+            }
+        }
+    }
+}