if isinstance(message[1], list): # 消息类型为列表 data = json.dumps(message[1]) for i in range(len(users)): # 向所有用户发送在线列表 users[i][0].send(data.encode()) if isinstance(message[1], dict): # 消息类型为字典 # friend_list_all = {} # 存放所有用户的好友 for i in range(len(users)): # 遍历所有用户 username = users[i][1] # 获取用户名字 friend_list = Userfriend(username) # 获取用户好友列表 data = json.dumps(friend_list) users[i][0].send(data.encode()) 在此基础上,为用户在线列表和用户好友列表增加标识以区分两种类型的消息
时间: 2024-01-13 10:03:23 浏览: 23
可以在发送消息时增加标识,以区分消息类型。
具体实现方法如下:
1. 在发送在线列表消息时,增加标识 `"online_list"`:
```python
if isinstance(message[1], list): # 消息类型为列表
data = json.dumps({"type": "online_list", "data": message[1]})
for i in range(len(users)): # 向所有用户发送在线列表
users[i][0].send(data.encode())
```
2. 在发送好友列表消息时,增加标识 `"friend_list"`:
```python
if isinstance(message[1], dict): # 消息类型为字典
for i in range(len(users)): # 遍历所有用户
username = users[i][1] # 获取用户名字
friend_list = Userfriend(username) # 获取用户好友列表
data = json.dumps({"type": "friend_list", "data": friend_list})
users[i][0].send(data.encode())
```
3. 在客户端接收消息时,根据标识区分消息类型:
```python
data = sock.recv(1024).decode()
message = json.loads(data)
if message["type"] == "online_list":
# 在线列表消息
online_list = message["data"]
# 更新在线列表
update_online_list(online_list)
elif message["type"] == "friend_list":
# 好友列表消息
friend_list = message["data"]
# 更新好友列表
update_friend_list(friend_list)
```
完整的代码如下:
```python
if isinstance(message[1], list): # 消息类型为列表
data = json.dumps({"type": "online_list", "data": message[1]})
for i in range(len(users)): # 向所有用户发送在线列表
users[i][0].send(data.encode())
if isinstance(message[1], dict): # 消息类型为字典
for i in range(len(users)): # 遍历所有用户
username = users[i][1] # 获取用户名字
friend_list = Userfriend(username) # 获取用户好友列表
data = json.dumps({"type": "friend_list", "data": friend_list})
users[i][0].send(data.encode())
```
```python
data = sock.recv(1024).decode()
message = json.loads(data)
if message["type"] == "online_list":
# 在线列表消息
online_list = message["data"]
# 更新在线列表
update_online_list(online_list)
elif message["type"] == "friend_list":
# 好友列表消息
friend_list = message["data"]
# 更新好友列表
update_friend_list(friend_list)
```