fixbug:调整排行榜写库动作

This commit is contained in:
liuwei
2025-03-18 10:32:36 +08:00
parent 727d2d3938
commit c26e6a2d8d
2 changed files with 62 additions and 27 deletions

View File

@@ -51,7 +51,6 @@ class MessageStorageDB(BaseDBOperator):
"""
return self.execute_query(sql, (date,)) or []
# 在 MessageStorageDB 类中添加以下方法
def get_speech_ranking(self, date: str, group_id: str, limit: int = 20) -> List[Dict]:
"""获取指定日期和群组的发言排名"""
@@ -66,4 +65,24 @@ class MessageStorageDB(BaseDBOperator):
"""
params = (date, group_id, limit)
results = self.execute_query(sql, params)
return results or []
return results or []
def insert_speech_count(self, group_id: str, wx_id: str, date: str, count: int) -> bool:
"""插入发言统计数据
Args:
group_id: 群组ID
wx_id: 微信ID
date: 日期,格式为 YYYY-MM-DD
count: 发言计数
Returns:
是否成功插入
"""
sql = """
INSERT INTO speech_counts (group_id, wx_id, date, count)
VALUES (%s, %s, %s, %s)
ON DUPLICATE KEY UPDATE count = VALUES(count)
"""
params = (group_id, wx_id, date, count)
return self.execute_update(sql, params)