定时任务订阅模式

offline-backend-20260512/backend/app/gateway/routers/scheduled_tasks.py

@@ -43,6 +43,10 @@ class ScheduledTaskDeliveryProfileUpdateRequest(BaseModel):
     element_user_id: str | None = None
 
 
+class ScheduledTaskSubscribeRequest(BaseModel):
+    notify_element: bool = False
+
+
 async def _user_id(request: Request) -> str:
     user_id = await get_current_user(request)
     if not user_id:
@@ -325,3 +329,63 @@ async def trigger_scheduled_task(task_id: str, request: Request) -> dict:
     if task is None:
         raise HTTPException(status_code=404, detail=f"Task {task_id} not found")
     return {"success": True, "task": task}
+
+
+# --- Publish / subscribe ---
+
+
+@router.get("/published")
+async def list_published_scheduled_tasks(request: Request, limit: int = 50, offset: int = 0) -> list[dict]:
+    service = get_scheduled_task_service()
+    await _user_id(request)
+    return await service.list_published(limit=max(1, min(limit, 200)), offset=max(0, offset))
+
+
+@router.post("/{task_id}/publish")
+async def publish_scheduled_task(task_id: str, request: Request) -> dict:
+    service = get_scheduled_task_service()
+    user_id = await _user_id(request)
+    task = await service.publish_task(user_id, task_id)
+    if task is None:
+        raise HTTPException(status_code=404, detail=f"Task {task_id} not found")
+    return task
+
+
+@router.post("/{task_id}/unpublish")
+async def unpublish_scheduled_task(task_id: str, request: Request) -> dict:
+    service = get_scheduled_task_service()
+    user_id = await _user_id(request)
+    task = await service.unpublish_task(user_id, task_id)
+    if task is None:
+        raise HTTPException(status_code=404, detail=f"Task {task_id} not found")
+    return task
+
+
+@router.post("/{task_id}/subscribe")
+async def subscribe_scheduled_task(task_id: str, body: ScheduledTaskSubscribeRequest, request: Request) -> dict:
+    service = get_scheduled_task_service()
+    user_id = await _user_id(request)
+    sub = await service.subscribe_task(user_id, task_id, notify_element=body.notify_element)
+    if sub is None:
+        raise HTTPException(status_code=404, detail="Task is not available for subscription")
+    return sub
+
+
+@router.patch("/{task_id}/subscribe")
+async def update_scheduled_task_subscription(task_id: str, body: ScheduledTaskSubscribeRequest, request: Request) -> dict:
+    service = get_scheduled_task_service()
+    user_id = await _user_id(request)
+    sub = await service.subscribe_task(user_id, task_id, notify_element=body.notify_element)
+    if sub is None:
+        raise HTTPException(status_code=404, detail="Subscription not found")
+    return sub
+
+
+@router.delete("/{task_id}/subscribe")
+async def delete_scheduled_task_subscription(task_id: str, request: Request) -> dict:
+    service = get_scheduled_task_service()
+    user_id = await _user_id(request)
+    deleted = await service.unsubscribe_task(user_id, task_id)
+    if not deleted:
+        raise HTTPException(status_code=404, detail="Subscription not found")
+    return {"success": True}