新增客服重定向接口

添加专用脚本用于启动rag2_0.api.intent_recognition_api服务
脚本功能包括检测并结束现有screen会话，清理占用端口，最后启动新服务

rag2_0/api/intent_recognition_api.py

@@ -16,6 +16,7 @@ load_dotenv()
 
 import sys
 from rag2_0.intent_recognition import AsyncIntentRecognizer
+from rag2_0.api.kefu_redirect_url import router as kefu_router
 
 # 确保日志目录存在
 os.makedirs('data/logs', exist_ok=True)
@@ -84,6 +85,9 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
+# 注册外部路由
+app.include_router(kefu_router)
+
 # 全局变量存储AsyncIntentRecognizer实例
 _instance = None
 

rag2_0/api/kefu_redirect_url.py

@@ -0,0 +1,92 @@
+from fastapi import APIRouter
+from fastapi.responses import RedirectResponse
+import os
+import sqlite3
+import threading
+import time
+from queue import Queue, Full
+
+
+router = APIRouter()
+
+# 以当前文件为基准的相对路径：../../data/db
+PROJECT_ROOT = os.getcwd()
+DB_DIR = os.path.join(PROJECT_ROOT, "data", "db")
+DB_FILE = os.path.join(DB_DIR, "redirects.sqlite3")
+TABLE_SQL = (
+    "CREATE TABLE IF NOT EXISTS redirects ("
+    " msg_id TEXT PRIMARY KEY,"
+    " url TEXT NOT NULL"
+    ")"
+)
+
+
+def _ensure_db():
+    """确保数据库与表存在。"""
+    os.makedirs(DB_DIR, exist_ok=True)
+    with sqlite3.connect(DB_FILE) as conn:
+        cur = conn.cursor()
+        cur.execute(TABLE_SQL)
+        conn.commit()
+
+
+def save_redirect(msg_id: str, url: str) -> None:
+    """将 msg_id 与 url 写入 SQLite，若已存在则忽略。
+
+    使用 INSERT OR IGNORE 结合 PRIMARY KEY(msg_id) 来避免重复写入。
+    """
+    _ensure_db()
+    with sqlite3.connect(DB_FILE) as conn:
+        cur = conn.cursor()
+        cur.execute(
+            "INSERT OR IGNORE INTO redirects (msg_id, url) VALUES (?, ?)",
+            (msg_id, url),
+        )
+        conn.commit()
+
+
+# ========= 异步写库队列与后台线程 =========
+_write_queue: "Queue[tuple[str, str]]" = Queue(maxsize=10000)
+
+
+def _write_worker():
+    _ensure_db()
+    while True:
+        try:
+            msg_id, url = _write_queue.get()
+            try:
+                with sqlite3.connect(DB_FILE) as conn:
+                    cur = conn.cursor()
+                    cur.execute(
+                        "INSERT OR IGNORE INTO redirects (msg_id, url) VALUES (?, ?)",
+                        (msg_id, url),
+                    )
+                    conn.commit()
+            except Exception:
+                # 失败忽略，避免阻断工作线程
+                pass
+            finally:
+                _write_queue.task_done()
+        except Exception:
+            # 防御性 sleep，避免异常导致CPU空转
+            time.sleep(0.1)
+
+
+_worker_thread = threading.Thread(target=_write_worker, daemon=True)
+_worker_thread.start()
+
+
+@router.get("/kefu_login", summary="客服登录页重定向")
+async def kefu_redirect(msg_id:str):
+    """重定向到客服登录页。"""
+    target_url = "https://www.booway.com.cn/kefu/toLoginPage"
+    # 写入 SQLite：若 msg_id 已存在将不会重复写入
+    try:
+        if msg_id:
+            # 走异步队列
+            _write_queue.put_nowait((msg_id, target_url))
+    except Exception:
+        # 出于稳健性考虑，即使写库失败也不影响重定向
+        pass
+    return RedirectResponse(target_url, status_code=302)
+

start_intent_api.sh

@@ -0,0 +1,111 @@
+#!/usr/bin/env bash
+
+# 专用脚本：启动 rag2_0.api.intent_recognition_api 服务
+# 功能：启动前检测screen是否存在，存在则结束，最后启动服务
+
+set -euo pipefail
+
+SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+SESSION_NAME="intent_recognition_api"
+SERVICE_PORT="8001"
+START_COMMAND="cd \"$SCRIPT_DIR\" && uv run uvicorn rag2_0.api.intent_recognition_api:app --host 0.0.0.0 --port 8001 --workers 4"
+
+echo "[脚本] 启动 intent_recognition_api 服务..."
+
+# 检查screen会话是否存在
+exists_session() {
+  # 使用严格匹配，避免误判
+  if screen -ls 2>/dev/null | grep -q "\\.${SESSION_NAME}\\s"; then
+    return 0
+  fi
+  return 1
+}
+
+# 按端口获取监听该端口的任意一个PID，优先用 ss，其次 lsof
+pids_on_port() {
+  # 从 ss 提取 pid 列表
+  local ss_pids
+  ss_pids=$(ss -lptn 2>/dev/null \
+    | grep -E ":${SERVICE_PORT}\\b" \
+    | awk '{print $NF}' \
+    | sed -n 's/.*pid=\([0-9]\+\),.*/\1/p' \
+    | sort -u)
+  if [[ -n "$ss_pids" ]]; then
+    echo "$ss_pids"
+    return 0
+  fi
+  # 从 lsof 提取 pid 列表
+  if command -v lsof >/dev/null 2>&1; then
+    local lsof_pids
+    lsof_pids=$(lsof -nP -i :"$SERVICE_PORT" -sTCP:LISTEN -t 2>/dev/null | sort -u)
+    if [[ -n "$lsof_pids" ]]; then
+      echo "$lsof_pids"
+      return 0
+    fi
+  fi
+  return 1
+}
+
+# 根据端口优雅终止(TERM)并在必要时强制(KILL)清理进程
+kill_by_port() {
+  local pids
+  pids=$(pids_on_port || true)
+  if [[ -z "$pids" ]]; then
+    return 0
+  fi
+  echo "[清理] 端口 $SERVICE_PORT 仍被占用，发送 SIGTERM 到: $pids"
+  kill -TERM $pids 2>/dev/null || true
+  sleep 2
+  # 再次检查
+  local left
+  left=$(pids_on_port || true)
+  if [[ -n "$left" ]]; then
+    echo "[强制] 端口 $SERVICE_PORT 仍占用，发送 SIGKILL 到: $left"
+    kill -KILL $left 2>/dev/null || true
+  fi
+}
+
+# 停止已存在的服务
+stop_existing_service() {
+  # 1) 先尝试关闭 screen 会话
+  if exists_session "$SESSION_NAME"; then
+    echo "[停止] 发现已存在的 screen 会话 '$SESSION_NAME'，正在结束..."
+    screen -S "$SESSION_NAME" -X quit || true
+    echo "[停止] screen 会话 '$SESSION_NAME' 已结束"
+  else
+    echo "[提示] 未发现 screen 会话: $SESSION_NAME"
+  fi
+
+  # 2) 等待释放端口
+  sleep 2
+
+  # 3) 如果仍占用，按端口清理
+  if ss -lptn 2>/dev/null | grep -E -q ":${SERVICE_PORT}\\b" || (command -v lsof >/dev/null 2>&1 && lsof -i :"$SERVICE_PORT" -sTCP:LISTEN >/dev/null 2>&1); then
+    echo "[清理] 端口 $SERVICE_PORT 仍被占用，正在清理..."
+    kill_by_port
+    echo "[清理] 端口 $SERVICE_PORT 清理完成"
+  fi
+}
+
+# 启动服务
+start_new_service() {
+  echo "[启动] 准备启动 intent_recognition_api 服务..."
+  echo "[启动] 启动命令: $START_COMMAND"
+  screen -dmS "$SESSION_NAME" bash -c "$START_COMMAND"
+  echo "[启动] intent_recognition_api 服务已启动，screen 会话名: '$SESSION_NAME'"
+  echo "[启动] 服务运行在端口: $SERVICE_PORT"
+  echo "[提示] 可使用 'screen -r $SESSION_NAME' 查看服务输出"
+}
+
+# 主流程
+main() {
+  # 1. 停止已存在的服务
+  stop_existing_service
+  
+  # 2. 启动新服务
+  start_new_service
+  
+  echo "[完成] intent_recognition_api 服务启动脚本执行完成"
+}
+
+main