84 lines
3.4 KiB
Python
84 lines
3.4 KiB
Python
"""定时调度（工作台侧，dev 默认关）：每日组合再平衡（遍历 signed 客户）。
|
||
|
||
架构 §7 已选型 APScheduler；仅当 config.advisor.scheduler_enabled=true 时启动。
|
||
dev 常以 uvicorn --reload 运行（重载会重复调度），故默认关闭，生产单进程开启。
|
||
|
||
边界：调度器无需前端/无需请求上下文，需自造投顾 JWT（service.auth.create_token）与
|
||
trace_id（new_request_id）调用 Agent；大额申赎扫描/回访到期/风评到期等本地定时任务
|
||
P1 后续补（依赖 fin_transaction 模型等）。
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
import logging
|
||
|
||
from common_const import CRON_PORTFOLIO_REBALANCE, CUSTOMER_REL_STATUS_SIGNED
|
||
from config.settings import settings
|
||
from repositories.customer_relation import CustomerRelationRepo
|
||
|
||
logger = logging.getLogger("service.advisor.scheduler")
|
||
|
||
|
||
class AdvisorScheduler:
|
||
def __init__(self):
|
||
self._scheduler = None
|
||
|
||
def start(self) -> None:
|
||
if not settings.advisor.scheduler_enabled:
|
||
return
|
||
# 懒加载：未安装 apscheduler 时不阻塞应用启动（默认关闭本调度器）
|
||
from apscheduler.schedulers.asyncio import AsyncIOScheduler
|
||
from apscheduler.triggers.cron import CronTrigger
|
||
|
||
self._scheduler = AsyncIOScheduler()
|
||
self._scheduler.add_job(
|
||
self.daily_rebalance,
|
||
CronTrigger.from_crontab(CRON_PORTFOLIO_REBALANCE),
|
||
id="advisor_daily_rebalance",
|
||
name="每日组合再平衡（遍历 signed 客户）",
|
||
)
|
||
self._scheduler.start()
|
||
logger.info("advisor scheduler started (cron=%s)", CRON_PORTFOLIO_REBALANCE)
|
||
|
||
def shutdown(self) -> None:
|
||
if self._scheduler is not None:
|
||
self._scheduler.shutdown(wait=False)
|
||
self._scheduler = None
|
||
|
||
async def daily_rebalance(self) -> None:
|
||
"""每日遍历已签约客户，按投顾逐个调 Agent rebalance/run（异步受理）。
|
||
|
||
结果靠事件 event:rebalance_draft_created 回执 → 工作台消费生成待办。
|
||
"""
|
||
from config.database.mysql import get_session_factory
|
||
from service.advisor.agent_client import get_agent_client
|
||
from service.auth import create_token
|
||
from utils.request_id import new_request_id
|
||
|
||
client = get_agent_client()
|
||
if not client.configured:
|
||
logger.warning("投顾Agent 未配置，跳过每日 rebalance")
|
||
return
|
||
|
||
async with get_session_factory()() as session:
|
||
relations = await CustomerRelationRepo(session).list_by_status(
|
||
CUSTOMER_REL_STATUS_SIGNED
|
||
)
|
||
|
||
by_advisor: dict[int, list[int]] = {}
|
||
for rel in relations:
|
||
by_advisor.setdefault(rel.advisor_id, []).append(rel.customer_id)
|
||
|
||
for advisor_id, customer_ids in by_advisor.items():
|
||
# 后台任务无请求上下文，自造投顾 JWT + trace_id
|
||
token = create_token(int(advisor_id))
|
||
auth = f"Bearer {token}"
|
||
for customer_id in customer_ids:
|
||
try:
|
||
await client.rebalance_run(
|
||
customer_id, auth_header=auth, trace_id=new_request_id()
|
||
)
|
||
except Exception:
|
||
logger.exception(
|
||
"rebalance run failed advisor=%s customer=%s", advisor_id, customer_id
|
||
)
|