为聊天模型定义 Room 实体之间的一对多关系

定义 Room 实体之间的一对多关系

python
class Room:
    def __init__(self, room_id, users=[]):
        self.room_id = room_id
        self.users = users
    def add_user(self, user):
        self.users.append(user)
    def remove_user(self, user):
        self.users.remove(user)

python
class User:
    def __init__(self, user_id, username):
        self.user_id = user_id
        self.username = username
# 创建两个用户
user1 = User(1, "Alice")
user2 = User(2, "Bob")
# 创建两个 Room
room1 = Room(101, [user1])
room2 = Room(102, [user2])
# 将 user2 添加到 room1 中，建立一对多关系
room1.add_user(user2)

python
from queue import Queue
from threading import Thread
class Room:
    def __init__(self, room_id, users=[]):
        self.room_id = room_id
        self.users = users
        self.message_queue = Queue()
    def add_user(self, user):
        self.users.append(user)
    def remove_user(self, user):
        self.users.remove(user)
    def send_message(self, message):
        self.message_queue.put(message)
    def process_messages(self):
        while True:
            message = self.message_queue.get()
            # 处理消息的逻辑
            print(f"Room {self.room_id} received message: {message}")
# 创建一个 Room
room = Room(201)
# 创建一个处理消息的线程
message_thread = Thread(target=room.process_messages)
message_thread.start()
# 向 Room 发送消息
room.send_message("Hello, everyone!")
room.send_message("How are you doing?")
# 等待消息处理线程结束
message_thread.join()