新增 消息定时推送功能

db/task_db.py

@@ -496,3 +496,55 @@ class TaskDBOperator(BaseDBOperator):
         except Exception as e:
             logger.error(f"获取任务日志失败: {e}")
             return {'logs': [], 'total': 0, 'page': page, 'limit': limit}
+
+    def get_tasks_count(self) -> int:
+        """获取任务总数
+        
+        Returns:
+            任务总数
+        """
+        try:
+            sql = "SELECT COUNT(*) as total FROM t_push_tasks"
+            result = self.execute_query(sql, fetch_one=True)
+            return result['total'] if result else 0
+        except Exception as e:
+            logger.error(f"获取任务总数失败: {e}")
+            return 0
+
+    def get_tasks_count_by_status(self, status: str) -> int:
+        """获取指定状态的任务数量
+        
+        Args:
+            status: 任务状态
+            
+        Returns:
+            任务数量
+        """
+        try:
+            sql = "SELECT COUNT(*) as total FROM t_push_tasks WHERE status = %s"
+            result = self.execute_query(sql, (status,), fetch_one=True)
+            return result['total'] if result else 0
+        except Exception as e:
+            logger.error(f"获取{status}状态任务数量失败: {e}")
+            return 0
+
+    def get_tasks_count_by_date(self, date: str) -> int:
+        """获取指定日期的任务数量
+        
+        Args:
+            date: 日期，格式：YYYY-MM-DD
+            
+        Returns:
+            任务数量
+        """
+        try:
+            sql = """
+                SELECT COUNT(*) as total 
+                FROM t_push_tasks 
+                WHERE DATE(created_at) = %s
+            """
+            result = self.execute_query(sql, (date,), fetch_one=True)
+            return result['total'] if result else 0
+        except Exception as e:
+            logger.error(f"获取{date}任务数量失败: {e}")
+            return 0