jettask 0.2.15__py3-none-any.whl → 0.2.16__py3-none-any.whl

jettask/schemas/queue.py

@@ -0,0 +1,83 @@
+"""
+队列相关的数据模型
+"""
+from typing import Optional, List, Dict, Any
+from datetime import datetime
+from pydantic import BaseModel, Field
+
+
+class TimeRangeQuery(BaseModel):
+    """时间范围查询模型"""
+    queues: Optional[List[str]] = Field(None, description="队列名称列表，为空时统计所有队列")
+    time_range: Optional[str] = Field(None, description="时间范围（如 '1h', '24h', '7d'），与 start_time/end_time 二选一")
+    start_time: Optional[str] = Field(None, description="开始时间，与 time_range 二选一")
+    end_time: Optional[str] = Field(None, description="结束时间，与 time_range 二选一")
+    interval: Optional[str] = Field(default="1h", description="时间间隔")
+
+
+class QueueStatsResponse(BaseModel):
+    """队列统计响应模型"""
+    queue_name: str = Field(..., description="队列名称")
+    pending_count: int = Field(default=0, description="待处理任务数")
+    processing_count: int = Field(default=0, description="处理中任务数")
+    completed_count: int = Field(default=0, description="已完成任务数")
+    failed_count: int = Field(default=0, description="失败任务数")
+    success_rate: float = Field(default=0.0, description="成功率")
+    avg_processing_time: Optional[float] = Field(None, description="平均处理时间")
+
+
+class QueueTimelineResponse(BaseModel):
+    """队列时间线响应模型"""
+    queue_name: str = Field(..., description="队列名称")
+    timeline: List[Dict[str, Any]] = Field(default=[], description="时间线数据")
+
+
+class TrimQueueRequest(BaseModel):
+    """修剪队列请求模型"""
+    queue_name: str = Field(..., description="队列名称")
+    max_length: int = Field(..., ge=1, description="最大长度")
+    trim_strategy: str = Field(default="oldest", description="修剪策略", pattern="^(oldest|newest)$")
+
+
+class QueueInfo(BaseModel):
+    """队列信息模型"""
+    name: str = Field(..., description="队列名称")
+    namespace: str = Field(..., description="命名空间")
+    pending_count: int = Field(default=0, description="待处理任务数")
+    processing_count: int = Field(default=0, description="处理中任务数")
+    completed_count: int = Field(default=0, description="已完成任务数")
+    failed_count: int = Field(default=0, description="失败任务数")
+    created_at: Optional[datetime] = Field(None, description="创建时间")
+    updated_at: Optional[datetime] = Field(None, description="更新时间")
+    
+    class Config:
+        json_encoders = {
+            datetime: lambda v: v.isoformat() if v else None
+        }
+
+
+class QueueStats(BaseModel):
+    """队列统计模型"""
+    queue_name: str = Field(..., description="队列名称")
+    total_tasks: int = Field(default=0, description="总任务数")
+    pending_tasks: int = Field(default=0, description="待处理任务数")
+    processing_tasks: int = Field(default=0, description="处理中任务数")
+    completed_tasks: int = Field(default=0, description="已完成任务数")
+    failed_tasks: int = Field(default=0, description="失败任务数")
+    success_rate: float = Field(default=0.0, description="成功率")
+    avg_processing_time: Optional[float] = Field(None, description="平均处理时间")
+    throughput: Optional[float] = Field(None, description="吞吐量（任务/秒）")
+    last_activity: Optional[datetime] = Field(None, description="最后活动时间")
+    
+    class Config:
+        json_encoders = {
+            datetime: lambda v: v.isoformat() if v else None
+        }
+
+
+class QueueActionRequest(BaseModel):
+    """队列操作请求模型"""
+    queue_names: List[str] = Field(..., min_items=1, description="队列名称列表")
+    action: str = Field(..., description="操作类型", pattern="^(pause|resume|clear|delete)$")
+    reason: Optional[str] = Field(None, description="操作原因")
+    force: bool = Field(default=False, description="是否强制执行")

jettask/schemas/scheduled_task.py

@@ -0,0 +1,128 @@
+"""
+定时任务相关的数据模型
+"""
+from typing import Optional, Dict, Any
+from datetime import datetime
+from pydantic import BaseModel, Field
+
+
+class ScheduleConfig(BaseModel):
+    """调度配置模型"""
+    schedule_type: str = Field(..., description="调度类型", pattern="^(cron|interval)$")
+    cron_expression: Optional[str] = Field(None, description="Cron表达式（当schedule_type为cron时）")
+    interval_seconds: Optional[int] = Field(None, ge=1, description="间隔秒数（当schedule_type为interval时）")
+    timezone: str = Field(default="UTC", description="时区")
+    max_instances: int = Field(default=1, ge=1, description="最大实例数")
+    
+    def validate_schedule_config(self):
+        """验证调度配置"""
+        if self.schedule_type == "cron" and not self.cron_expression:
+            raise ValueError("Cron类型调度必须提供cron_expression")
+        if self.schedule_type == "interval" and not self.interval_seconds:
+            raise ValueError("Interval类型调度必须提供interval_seconds")
+
+
+class ScheduledTaskRequest(BaseModel):
+    """定时任务请求模型"""
+    task_name: str = Field(..., description="任务名称")
+    task_function: str = Field(..., description="任务函数")
+    schedule_config: ScheduleConfig = Field(..., description="调度配置")
+    task_args: Optional[list] = Field(default=[], description="任务参数")
+    task_kwargs: Optional[Dict[str, Any]] = Field(default={}, description="任务关键字参数")
+    queue_name: Optional[str] = Field(None, description="队列名称")
+    priority: int = Field(default=0, description="优先级")
+    enabled: bool = Field(default=True, description="是否启用")
+    description: Optional[str] = Field(None, description="任务描述")
+
+
+class ScheduledTaskCreate(BaseModel):
+    """创建定时任务请求模型"""
+    name: str = Field(..., description="任务名称")
+    task_function: str = Field(..., description="任务函数")
+    schedule_type: str = Field(..., description="调度类型", pattern="^(cron|interval)$")
+    cron_expression: Optional[str] = Field(None, description="Cron表达式")
+    interval_seconds: Optional[int] = Field(None, ge=1, description="间隔秒数")
+    task_args: Optional[list] = Field(default=[], description="任务参数")
+    task_kwargs: Optional[Dict[str, Any]] = Field(default={}, description="任务关键字参数")
+    enabled: bool = Field(default=True, description="是否启用")
+    description: Optional[str] = Field(None, description="任务描述")
+
+
+class ScheduledTaskUpdate(BaseModel):
+    """更新定时任务请求模型"""
+    name: Optional[str] = Field(None, description="任务名称")
+    task_function: Optional[str] = Field(None, description="任务函数")
+    schedule_type: Optional[str] = Field(None, description="调度类型")
+    cron_expression: Optional[str] = Field(None, description="Cron表达式")
+    interval_seconds: Optional[int] = Field(None, ge=1, description="间隔秒数")
+    task_args: Optional[list] = Field(None, description="任务参数")
+    task_kwargs: Optional[Dict[str, Any]] = Field(None, description="任务关键字参数")
+    enabled: Optional[bool] = Field(None, description="是否启用")
+    description: Optional[str] = Field(None, description="任务描述")
+
+
+class ScheduledTaskCreateRequest(BaseModel):
+    """定时任务创建请求模型（详细版）"""
+    task_name: str = Field(..., min_length=1, max_length=255, description="任务名称")
+    task_function: str = Field(..., min_length=1, description="任务函数路径")
+    schedule_type: str = Field(..., description="调度类型", pattern="^(cron|interval|once)$")
+    cron_expression: Optional[str] = Field(None, description="Cron表达式")
+    interval_seconds: Optional[int] = Field(None, ge=1, le=86400*365, description="间隔秒数")
+    start_time: Optional[datetime] = Field(None, description="开始时间")
+    end_time: Optional[datetime] = Field(None, description="结束时间")
+    task_args: list = Field(default=[], description="任务参数")
+    task_kwargs: Dict[str, Any] = Field(default={}, description="任务关键字参数")
+    queue_name: Optional[str] = Field(None, description="目标队列")
+    priority: int = Field(default=0, ge=0, le=100, description="优先级")
+    max_retries: int = Field(default=3, ge=0, le=10, description="最大重试次数")
+    timeout: Optional[int] = Field(None, ge=1, description="超时时间（秒）")
+    enabled: bool = Field(default=True, description="是否启用")
+    description: Optional[str] = Field(None, max_length=1000, description="任务描述")
+
+
+class ScheduledTaskResponse(BaseModel):
+    """定时任务响应模型"""
+    id: str = Field(..., description="任务ID")
+    task_name: str = Field(..., description="任务名称")
+    task_function: str = Field(..., description="任务函数")
+    schedule_type: str = Field(..., description="调度类型")
+    next_run_time: Optional[datetime] = Field(None, description="下次运行时间")
+    last_run_time: Optional[datetime] = Field(None, description="上次运行时间")
+    status: str = Field(..., description="任务状态")
+    enabled: bool = Field(..., description="是否启用")
+    
+    class Config:
+        json_encoders = {
+            datetime: lambda v: v.isoformat() if v else None
+        }
+
+
+class ScheduledTaskInfo(BaseModel):
+    """定时任务详细信息模型"""
+    id: str = Field(..., description="任务ID")
+    task_name: str = Field(..., description="任务名称")
+    task_function: str = Field(..., description="任务函数")
+    schedule_type: str = Field(..., description="调度类型")
+    cron_expression: Optional[str] = Field(None, description="Cron表达式")
+    interval_seconds: Optional[int] = Field(None, description="间隔秒数")
+    task_args: list = Field(default=[], description="任务参数")
+    task_kwargs: Dict[str, Any] = Field(default={}, description="任务关键字参数")
+    queue_name: Optional[str] = Field(None, description="队列名称")
+    priority: int = Field(default=0, description="优先级")
+    max_retries: int = Field(default=3, description="最大重试次数")
+    timeout: Optional[int] = Field(None, description="超时时间")
+    enabled: bool = Field(default=True, description="是否启用")
+    status: str = Field(..., description="任务状态")
+    created_at: datetime = Field(..., description="创建时间")
+    updated_at: Optional[datetime] = Field(None, description="更新时间")
+    next_run_time: Optional[datetime] = Field(None, description="下次运行时间")
+    last_run_time: Optional[datetime] = Field(None, description="上次运行时间")
+    run_count: int = Field(default=0, description="运行次数")
+    success_count: int = Field(default=0, description="成功次数")
+    failure_count: int = Field(default=0, description="失败次数")
+    description: Optional[str] = Field(None, description="任务描述")
+    
+    class Config:
+        json_encoders = {
+            datetime: lambda v: v.isoformat() if v else None
+        }

jettask/schemas/task.py

@@ -0,0 +1,70 @@
+"""
+任务相关的数据模型
+"""
+from typing import Optional, List, Dict, Any
+from datetime import datetime
+from pydantic import BaseModel, Field
+
+
+class TasksRequest(BaseModel):
+    """任务查询请求模型"""
+    queue_name: str = Field(..., description="队列名称")
+    page: int = Field(default=1, ge=1, description="页码")
+    page_size: int = Field(default=20, ge=1, le=100, description="每页大小")
+    filters: Optional[List[Dict]] = Field(default=[], description="筛选条件")
+    time_range: Optional[str] = Field(None, description="时间范围，如 '15m', '1h', '7d'")
+    start_time: Optional[str] = Field(None, description="开始时间")
+    end_time: Optional[str] = Field(None, description="结束时间")
+
+
+class TaskInfo(BaseModel):
+    """任务信息模型"""
+    id: str = Field(..., description="任务ID")
+    queue_name: str = Field(..., description="队列名称")
+    status: str = Field(..., description="任务状态")
+    created_at: datetime = Field(..., description="创建时间")
+    started_at: Optional[datetime] = Field(None, description="开始时间")
+    completed_at: Optional[datetime] = Field(None, description="完成时间")
+    worker_id: Optional[str] = Field(None, description="工作节点ID")
+    task_name: Optional[str] = Field(None, description="任务名称")
+    task_data: Optional[Dict[str, Any]] = Field(None, description="任务数据")
+    result: Optional[Any] = Field(None, description="任务结果")
+    error: Optional[str] = Field(None, description="错误信息")
+    retry_count: int = Field(default=0, description="重试次数")
+    execution_time: Optional[float] = Field(None, description="执行时间（秒）")
+    priority: Optional[int] = Field(None, description="优先级")
+    consumer_group: Optional[str] = Field(None, description="消费者组")
+    
+    class Config:
+        json_encoders = {
+            datetime: lambda v: v.isoformat() if v else None
+        }
+
+
+class TaskDetailResponse(BaseModel):
+    """任务详细信息响应模型"""
+    success: bool = Field(default=True, description="请求是否成功")
+    data: Optional[TaskInfo] = Field(None, description="任务详细信息")
+    error: Optional[str] = Field(None, description="错误信息")
+    
+    class Config:
+        json_encoders = {
+            datetime: lambda v: v.isoformat() if v else None
+        }
+
+
+class TaskActionRequest(BaseModel):
+    """任务操作请求模型"""
+    task_ids: List[str] = Field(..., min_items=1, description="任务ID列表")
+    action: str = Field(..., description="操作类型", pattern="^(retry|cancel|delete)$")
+    reason: Optional[str] = Field(None, description="操作原因")
+
+
+class TaskListResponse(BaseModel):
+    """任务列表响应模型"""
+    success: bool = Field(default=True, description="请求是否成功")
+    data: List[TaskInfo] = Field(default=[], description="任务列表")
+    total: int = Field(default=0, description="总记录数")
+    page: int = Field(default=1, description="当前页码")
+    page_size: int = Field(default=20, description="每页大小")
+    error: Optional[str] = Field(None, description="错误信息")

jettask/services/__init__.py

@@ -0,0 +1,24 @@
+"""
+业务逻辑服务层
+包含所有业务逻辑处理
+"""
+
+from .overview_service import OverviewService
+from .queue_service import QueueService
+from .scheduled_task_service import ScheduledTaskService
+from .alert_service import AlertService
+from .analytics_service import AnalyticsService
+from .settings_service import SettingsService
+from .task_service import TaskService
+from .redis_monitor_service import RedisMonitorService
+
+__all__ = [
+    'OverviewService',
+    'QueueService', 
+    'ScheduledTaskService',
+    'AlertService',
+    'AnalyticsService',
+    'SettingsService',
+    'TaskService',
+    'RedisMonitorService'
+]