add

2026-10-15 20:36:59 +08:00
commit 6bf949949c
7923 changed files with 1369582 additions and 0 deletions
@@ -0,0 +1,166 @@
+"""结算服务：周期任务（推荐激活 / 视频创作收益 / 订单超时关单）
+
+由 main.py lifespan 内的 asyncio 周期协程驱动（开发环境单进程够用）；
+各结算函数均为幂等扫描，测试可直接调用。
+"""
+import asyncio
+import logging
+from datetime import datetime, timedelta
+from decimal import Decimal
+
+from sqlalchemy import func, select
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from app.config import biz_config
+from app.database import SessionFactory
+from app.models.billing import RechargeOrder, WatchRecord
+from app.models.incentive import Referral, VideoEarning
+from app.models.user import User
+from app.services import incentive_service, membership_service, wallet_service
+
+logger = logging.getLogger(__name__)
+
+# 单轮扫描批量上限
+BATCH_SIZE = 500
+
+# 周期间隔（秒）
+INTERVAL_REFERRAL = 600   # 推荐激活结算 10min
+INTERVAL_EARNINGS = 300   # 创作收益结算 5min
+INTERVAL_ORDER_CLOSE = 60  # 订单超时关单 1min
+
+
+def _referral_cfg() -> dict:
+    return biz_config.get("referral") or {}
+
+
+def _asset_cfg() -> dict:
+    return biz_config.get("asset") or {}
+
+
+# ---------- 推荐激活结算 ----------
+
+async def settle_referral_activations(db: AsyncSession) -> dict:
+    """待激活推荐关系结算（《01-开发文档》2.8：72h 内观看≥N 且次日登录 → 有效；超期 → 无效）。
+
+    激活时点校验推荐人会员身份，会员才发放注册奖励积分（point biz_type=8）。
+    """
+    cfg = _referral_cfg()
+    hours = int(cfg.get("activate_hours", 72))
+    watch_count = int(cfg.get("activate_watch_count", 3))
+    reward = int(cfg.get("register_reward_points", 10))
+    now = datetime.now()
+
+    rows = (await db.execute(
+        select(Referral).where(Referral.status == 0).order_by(Referral.id).limit(BATCH_SIZE)
+    )).scalars().all()
+    activated = expired = 0
+    for r in rows:
+        if r.created_at + timedelta(hours=hours) < now:
+            r.status = 2  # 超期未达标
+            expired += 1
+            continue
+        watches = (await db.execute(
+            select(func.count(WatchRecord.id)).where(
+                WatchRecord.subject_type == 1, WatchRecord.subject_id == str(r.invitee_id)
+            )
+        )).scalar() or 0
+        if watches < watch_count:
+            continue
+        invitee = await db.get(User, r.invitee_id)
+        next_day_login = bool(
+            invitee and invitee.last_login_at and invitee.last_login_at.date() > r.created_at.date()
+        )
+        if not next_day_login:
+            continue
+        r.status = 1
+        r.activated_at = now
+        activated += 1
+        if reward > 0 and await membership_service.is_member(db, r.inviter_id):
+            await wallet_service.change_points(
+                db, r.inviter_id, reward, wallet_service.BIZ_REFERRAL,
+                biz_id=str(r.id), remark="推荐新用户激活奖励",
+            )
+    await db.commit()
+    return {"scanned": len(rows), "activated": activated, "expired": expired}
+
+
+# ---------- 视频创作收益结算 ----------
+
+async def settle_video_earnings(db: AsyncSession) -> dict:
+    """创作收益按增量结算：完整观看每满 1000 发 1 资产、点赞每满 100 发 1 资产（余数保留）。
+
+    结算时点校验作者会员身份：非会员只保留计数，不推进结算刻度。
+    """
+    cfg = _asset_cfg()
+    views_per = int(cfg.get("complete_views_per_asset", 1000))
+    likes_per = int(cfg.get("likes_per_asset", 100))
+
+    rows = (await db.execute(
+        select(VideoEarning).order_by(VideoEarning.video_id).limit(BATCH_SIZE)
+    )).scalars().all()
+    settled_videos = 0
+    granted = Decimal(0)
+    for e in rows:
+        if not await membership_service.is_member(db, e.author_id):
+            continue
+        view_assets = (e.complete_views - e.views_settled) // views_per if views_per > 0 else 0
+        like_assets = (e.likes_accrued - e.likes_settled) // likes_per if likes_per > 0 else 0
+        if view_assets <= 0 and like_assets <= 0:
+            continue
+        if view_assets > 0:
+            await incentive_service.change_assets(
+                db, e.author_id, Decimal(view_assets), incentive_service.ASSET_BIZ_COMPLETE,
+                biz_id=str(e.video_id), remark="完整观看收益",
+            )
+            e.views_settled += view_assets * views_per
+        if like_assets > 0:
+            await incentive_service.change_assets(
+                db, e.author_id, Decimal(like_assets), incentive_service.ASSET_BIZ_LIKE,
+                biz_id=str(e.video_id), remark="点赞收益",
+            )
+            e.likes_settled += like_assets * likes_per
+        settled_videos += 1
+        granted += view_assets + like_assets
+    await db.commit()
+    return {"scanned": len(rows), "settled_videos": settled_videos, "granted_assets": float(granted)}
+
+
+# ---------- 订单超时关单 ----------
+
+async def close_expired_orders(db: AsyncSession) -> dict:
+    """待支付订单超过 expire_at → 已关闭"""
+    rows = (await db.execute(
+        select(RechargeOrder)
+        .where(RechargeOrder.status == 0, RechargeOrder.expire_at < datetime.now())
+        .order_by(RechargeOrder.id)
+        .limit(BATCH_SIZE)
+    )).scalars().all()
+    for o in rows:
+        o.status = 2
+    await db.commit()
+    return {"closed": len(rows)}
+
+
+# ---------- 周期协程（lifespan 挂载） ----------
+
+async def _run_loop(name: str, fn, interval: int) -> None:
+    while True:
+        try:
+            async with SessionFactory() as db:
+                result = await fn(db)
+            if result:
+                logger.info("周期任务[%s]完成: %s", name, result)
+        except asyncio.CancelledError:
+            raise
+        except Exception:
+            logger.exception("周期任务[%s]执行失败", name)
+        await asyncio.sleep(interval)
+
+
+def start_periodic_tasks() -> list[asyncio.Task]:
+    """启动全部周期任务，返回 Task 列表（lifespan 退出时取消）"""
+    return [
+        asyncio.create_task(_run_loop("推荐激活结算", settle_referral_activations, INTERVAL_REFERRAL)),
+        asyncio.create_task(_run_loop("创作收益结算", settle_video_earnings, INTERVAL_EARNINGS)),
+        asyncio.create_task(_run_loop("订单超时关单", close_expired_orders, INTERVAL_ORDER_CLOSE)),
+    ]