feat: initial XCClaw基础架构
- 基于 FastAPI 的 Web API 服务 - OpenCode API 客户端封装 - 会话管理器(同步/异步任务执行) - APScheduler 定时任务调度 - 完整的 REST API 端点
This commit is contained in:
79
app/services/scheduler.py
Normal file
79
app/services/scheduler.py
Normal file
@@ -0,0 +1,79 @@
|
||||
import uuid
|
||||
from pydantic import BaseModel
|
||||
from apscheduler.schedulers.asyncio import AsyncIOScheduler
|
||||
from apscheduler.triggers.cron import CronTrigger
|
||||
from app.services.session_manager import session_manager
|
||||
from app.core.logging import logger
|
||||
|
||||
|
||||
class ScheduleTask(BaseModel):
|
||||
id: str
|
||||
name: str
|
||||
cron: str
|
||||
prompt: str
|
||||
enabled: bool = True
|
||||
|
||||
|
||||
class SchedulerService:
|
||||
def __init__(self):
|
||||
self.scheduler = AsyncIOScheduler()
|
||||
self.scheduled_tasks: dict[str, ScheduleTask] = {}
|
||||
|
||||
def start(self):
|
||||
self.scheduler.start()
|
||||
logger.info("Scheduler started")
|
||||
|
||||
def shutdown(self):
|
||||
self.scheduler.shutdown()
|
||||
logger.info("Scheduler shutdown")
|
||||
|
||||
async def add_schedule(self, task: ScheduleTask) -> ScheduleTask:
|
||||
self.scheduled_tasks[task.id] = task
|
||||
|
||||
if task.enabled:
|
||||
trigger = CronTrigger.from_crontab(task.cron)
|
||||
self.scheduler.add_job(
|
||||
self._run_scheduled_task,
|
||||
trigger,
|
||||
args=[task.id],
|
||||
id=task.id,
|
||||
)
|
||||
logger.info(f"Added schedule: {task.name} ({task.cron})")
|
||||
|
||||
return task
|
||||
|
||||
async def remove_schedule(self, task_id: str) -> bool:
|
||||
task = self.scheduled_tasks.pop(task_id, None)
|
||||
if not task:
|
||||
return False
|
||||
|
||||
self.scheduler.remove_job(task_id)
|
||||
logger.info(f"Removed schedule: {task.name}")
|
||||
return True
|
||||
|
||||
async def list_schedules(self) -> list[ScheduleTask]:
|
||||
return list(self.scheduled_tasks.values())
|
||||
|
||||
async def _run_scheduled_task(self, task_id: str):
|
||||
task = self.scheduled_tasks.get(task_id)
|
||||
if not task:
|
||||
return
|
||||
|
||||
logger.info(f"Running scheduled task: {task.name}")
|
||||
|
||||
from app.models.session import CreateSessionRequest, SessionType
|
||||
|
||||
request = CreateSessionRequest(
|
||||
type=SessionType.SCHEDULED,
|
||||
prompt=task.prompt,
|
||||
)
|
||||
task_obj = await session_manager.create_task(request)
|
||||
|
||||
try:
|
||||
await session_manager.execute_task_async(task_obj.id)
|
||||
logger.info(f"Scheduled task {task.name} started")
|
||||
except Exception as e:
|
||||
logger.error(f"Scheduled task {task.name} failed to start: {e}")
|
||||
|
||||
|
||||
scheduler_service = SchedulerService()
|
||||
Reference in New Issue
Block a user