### Python Socket 聊天室实现房间管理 #### 创建服务器端脚本 为了创建一个带有房间管理功能的聊天室，服务器端需要能够处理多个客户端连接并维护不同的聊天房间。每个房间可以视为一组特定主题下的参与者集合。 ```python import socket from threading import Thread import json class ChatServer: def __init__(self, host='localhost', port=9001): server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM) server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) server_socket.bind((host, port)) server_socket.listen(5) self.server_socket = server_socket self.clients = {} self.rooms = {} def broadcast(self, room_name, message, sender=None): if room_name not in self.rooms: return for client in self.rooms[room_name]: if client != sender: try: client.sendall(message.encode()) except Exception as e: print(f"Error broadcasting to {client}: {e}") def handle_client(self, conn, addr): while True: try: data = conn.recv(1024).decode() msg = json.loads(data) action = msg.get('action') content = msg.get('content') if action == 'join': room_name = content['room'] user_name = content['name'] if room_name not in self.rooms: self.rooms[room_name] = set() self.rooms[room_name].add(conn) self.clients[conn] = {'name': user_name, 'room': room_name} join_message = f"{user_name} has joined the room." self.broadcast(room_name, join_message, None) elif action == 'message' and conn in self.clients: current_room = self.clients[conn]['room'] send_message = f"[{self.clients[conn]['name']}]: {content}" self.broadcast(current_room, send_message, conn) except ConnectionResetError: break # Handle disconnecting clients if conn in self.clients: leave_message = f"{self.clients[conn]['name']} has left the chat." self.broadcast(self.clients[conn]['room'], leave_message, None) del self.clients[conn] self.rooms[self.clients[conn]['room']].remove(conn) conn.close() def start_server(self): print("Chat Server started.") while True: conn, addr = self.server_socket.accept() thread = Thread(target=self.handle_client, args=(conn, addr)) thread.start() if __name__ == "__main__": chat_server = ChatServer() chat_server.start_server() ``` 这段代码展示了如何设置一个多线程TCP服务器来接收来自不同用户的连接请求，并允许他们加入指定名称的聊天室[^1]。 #### 客户端部分 对于客户端而言，则要负责向服务器发送消息以及显示接收到的信息。这里同样利用了多线程机制以便于同时监听键盘输入和网络数据流的变化。 ```python import socket import threading import json def receive_messages(client_socket): """Function to continuously listen for incoming messages from the server.""" while True: try: message = client_socket.recv(1024).decode() print("\n" + message) except OSError: break def main(): nickname = input("Choose your username: ") client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM) client_socket.connect(('localhost', 9001)) room_to_join = input("Enter room name you want to join/create: ") join_msg = { "action": "join", "content": {"room": room_to_join, "name": nickname} } client_socket.send(json.dumps(join_msg).encode()) receiver_thread = threading.Thread(target=receive_messages, args=(client_socket,)) receiver_thread.daemon = True receiver_thread.start() while True: message_content = input("") if message_content.lower().strip() == '/exit': break message_data = { "action": "message", "content": message_content, } client_socket.send(json.dumps(message_data).encode()) client_socket.shutdown(socket.SHUT_RDWR) client_socket.close() if __name__ == '__main__': main() ``` 上述程序实现了基本的功能需求——即用户可以选择进入某个具体的讨论组（或称为“房间”），并且可以在其中与其他成员交流互动[^3]。