fix：增加定时推动

server/run.sh

@@ -11,24 +11,28 @@ set -euo pipefail
 #   ./run.sh --env prod      # 使用 .env.prod（若存在且可被 source）
 #   ./run.sh --port 9000     # 改端口
 #   ./run.sh --no-reload     # 关闭热更新
+#   ./run.sh --with-worker --with-beat   # 同时启动 Celery Worker + Beat（用于定时推送）
 #   ./run.sh --install-only  # 只安装依赖，不启动
 
 usage() {
   cat <<'EOF'
 用法：
-  ./run.sh [--env dev|prod] [--host 0.0.0.0] [--port 8000] [--no-reload] [--skip-install] [--install-only]
+  ./run.sh [--env dev|prod] [--host 0.0.0.0] [--port 8000] [--no-reload] [--with-worker] [--with-beat] [--skip-install] [--install-only]
 
 参数：
   --env dev|prod     优先尝试加载 .env.dev 或 .env.prod（如果存在）。
   --host <host>      uvicorn host（默认 0.0.0.0）
   --port <port>      uvicorn port（默认 8000）
   --no-reload        关闭 uvicorn --reload
+  --with-worker      同时启动 Celery Worker（处理异步/ETA 任务）
+  --with-beat        同时启动 Celery Beat（定时调度，例如每日生成推送排程）
   --skip-install     跳过依赖安装（默认会安装/更新 requirements.txt）
   --install-only     只安装依赖，不启动服务
   -h, --help         显示帮助
 
 说明：
   - 若你的 .env.* 不是 shell 可 source 的格式（例如包含空格/特殊字符未加引号），建议改成 KEY=value 形式。
+  - 仅启动 API 并不会生成 `push_send_log`；要测试“定时推送”，需要 Beat 调度 `tasks.push.generate_daily_schedule`，并由 Worker 执行后续 ETA 任务。
   - 启动后访问：
       /healthz  健康检查
       /docs     OpenAPI 文档
@@ -45,6 +49,8 @@ PORT="8000"
 RELOAD="1"
 SKIP_INSTALL="0"
 INSTALL_ONLY="0"
+WITH_WORKER="0"
+WITH_BEAT="0"
 
 while [[ $# -gt 0 ]]; do
   case "$1" in
@@ -64,6 +70,14 @@ while [[ $# -gt 0 ]]; do
       RELOAD="0"
       shift 1
       ;;
+    --with-worker)
+      WITH_WORKER="1"
+      shift 1
+      ;;
+    --with-beat)
+      WITH_BEAT="1"
+      shift 1
+      ;;
     --skip-install)
       SKIP_INSTALL="1"
       shift 1
@@ -99,6 +113,15 @@ if [[ -f "$ENV_FILE" ]]; then
   set +a
 fi
 
+# 让 API/Celery 统一使用同一个 APP_ENV（影响 Redis key 前缀、定时任务配置等）
+export APP_ENV="${APP_ENV:-$ENV_NAME}"
+
+# Beat 只负责“投递任务到队列”，真正执行仍需要 Worker；这里自动补齐，避免误用。
+if [[ "$WITH_BEAT" == "1" && "$WITH_WORKER" == "0" ]]; then
+  echo "提示：已启用 --with-beat，自动同时启用 --with-worker（否则队列无人执行）。"
+  WITH_WORKER="1"
+fi
+
 # 选择 python 命令（优先 python3）
 PY_BIN=""
 if command -v python3 >/dev/null 2>&1; then
@@ -140,6 +163,45 @@ if [[ "$RELOAD" == "1" ]]; then
   UVICORN_ARGS+=(--reload)
 fi
 
-echo "启动服务：uvicorn ${UVICORN_ARGS[*]}"
-exec uvicorn "${UVICORN_ARGS[@]}"
+if [[ "$WITH_WORKER" == "0" && "$WITH_BEAT" == "0" ]]; then
+  echo "启动服务：uvicorn ${UVICORN_ARGS[*]}"
+  exec uvicorn "${UVICORN_ARGS[@]}"
+fi
+
+PIDS=()
+
+cleanup() {
+  # 避免重复清理导致脚本退出码被覆盖
+  set +e
+  if [[ ${#PIDS[@]} -gt 0 ]]; then
+    echo ""
+    echo "正在停止后台进程..."
+    # 先尝试温和退出
+    for pid in "${PIDS[@]}"; do
+      kill -TERM "$pid" >/dev/null 2>&1 || true
+    done
+    # 等待一点时间，再强制杀掉仍存活的（防止残留）
+    sleep 1
+    for pid in "${PIDS[@]}"; do
+      kill -KILL "$pid" >/dev/null 2>&1 || true
+    done
+  fi
+}
+
+trap cleanup EXIT INT TERM
+
+if [[ "$WITH_WORKER" == "1" ]]; then
+  echo "启动 Celery Worker：celery -A app.worker:celery_app worker -l info"
+  celery -A app.worker:celery_app worker -l info &
+  PIDS+=("$!")
+fi
+
+if [[ "$WITH_BEAT" == "1" ]]; then
+  echo "启动 Celery Beat：celery -A app.worker:celery_app beat -l info"
+  celery -A app.worker:celery_app beat -l info &
+  PIDS+=("$!")
+fi
+
+echo "启动服务：uvicorn ${UVICORN_ARGS[*]}"
+uvicorn "${UVICORN_ARGS[@]}"