fix(application): 修复普通用户接收消息数量统计错误

- 移除了对 user_id == 1 的特殊判断,统一了消息数量统计逻辑
- 优化了代码结构,提高了代码的可读性和维护性
This commit is contained in:
李小涛
2025-06-26 17:51:36 +08:00
parent 344f754fc7
commit ad95bea301

View File

@@ -17,10 +17,7 @@ def event_stream(user_id):
last_db_change_time = cache.get('last_db_change_time', 0) last_db_change_time = cache.get('last_db_change_time', 0)
# 只有当数据库发生变化时才检查总数 # 只有当数据库发生变化时才检查总数
if last_db_change_time and last_db_change_time > last_sent_time: if last_db_change_time and last_db_change_time > last_sent_time:
if user_id == 1: count = MessageCenterTargetUser.objects.filter(users=user_id, is_read=False).count()
count = MessageCenterTargetUser.objects.filter(is_read=False).count()
else:
count = MessageCenterTargetUser.objects.filter(users=user_id, is_read=False).count()
yield f"data: {count}\n\n" yield f"data: {count}\n\n"
last_sent_time = time.time() last_sent_time = time.time()