158 lines
6.0 KiB
Bash
Executable File
158 lines
6.0 KiB
Bash
Executable File
#!/usr/bin/env bash
|
||
# 须用 bash 执行；若误用 sh/dash 会自动改用 bash 再跑一遍
|
||
[ -n "${BASH_VERSION:-}" ] || exec /usr/bin/env bash "$0" ${1+"$@"}
|
||
# 重启 langchain-chat（startup.py --all-api，与 start_all.sh 一致）
|
||
# 日志：gangyan/logs/langchain-chat.log
|
||
set -u
|
||
source "$(cd "$(dirname "$0")" && pwd)/common-restart.sh"
|
||
LOG_FILE="$LOG_DIR/langchain-chat.log"
|
||
TRIM_LOG="$LOG_DIR/log-trim.log"
|
||
|
||
CONDA_SH="/opt/software/miniconda3/etc/profile.d/conda.sh"
|
||
LC_ROOT="$GANGYAN_ROOT/langchain-chat"
|
||
|
||
# 与 configs/server_config.py 一致，用于旧进程未退出时按端口释放
|
||
LANGCHAIN_LISTEN_PORTS=(7861 20000 20101 20002)
|
||
|
||
_kill_by_listen_ports() {
|
||
local port pid
|
||
for port in "${LANGCHAIN_LISTEN_PORTS[@]}"; do
|
||
for pid in $(lsof -ti :"$port" -sTCP:LISTEN 2>/dev/null || true); do
|
||
kill -TERM "$pid" 2>/dev/null && log_tee "按端口 ${port} TERM PID=${pid}"
|
||
done
|
||
done
|
||
sleep 2
|
||
for port in "${LANGCHAIN_LISTEN_PORTS[@]}"; do
|
||
for pid in $(lsof -ti :"$port" -sTCP:LISTEN 2>/dev/null || true); do
|
||
kill -KILL "$pid" 2>/dev/null && log_tee "按端口 ${port} KILL PID=${pid}"
|
||
done
|
||
done
|
||
}
|
||
|
||
# 用 ss 判断端口是否在监听。仅用 lsof 会漏掉 root 监听的端口（非 root 看不到 PID），
|
||
# 会误判为已释放并再次启动，导致 [Errno 98] address already in use。
|
||
_langchain_ports_still_listen() {
|
||
local port
|
||
for port in "${LANGCHAIN_LISTEN_PORTS[@]}"; do
|
||
if ss -tln 2>/dev/null | grep -qE ":${port}[[:space:]]"; then
|
||
return 0
|
||
fi
|
||
done
|
||
return 1
|
||
}
|
||
|
||
_sudo_fuser_kill_ports() {
|
||
command -v sudo >/dev/null 2>&1 || return 1
|
||
if ! sudo -n true 2>/dev/null; then
|
||
log_tee "提示: 相关端口仍被占用（可能是 root 启动的进程）。当前用户无法用 lsof 看到其 PID。"
|
||
log_tee "请在本机执行其一: sudo bash $0 或 sudo fuser -k 7861/tcp 20000/tcp 20101/tcp 20002/tcp"
|
||
return 1
|
||
fi
|
||
local port
|
||
for port in "${LANGCHAIN_LISTEN_PORTS[@]}"; do
|
||
if ss -tln 2>/dev/null | grep -qE ":${port}[[:space:]]"; then
|
||
sudo -n fuser -k "${port}/tcp" 2>/dev/null && log_tee "sudo fuser -k ${port}/tcp"
|
||
fi
|
||
done
|
||
sleep 2
|
||
return 0
|
||
}
|
||
|
||
_stop_langchain() {
|
||
log_tee "======== 停止 langchain-chat ========"
|
||
|
||
# 多种命令行形式（-a / --all-api、conda 路径、直接 python）
|
||
for pat in \
|
||
"/opt/software/miniconda3/envs/langchain-chat/bin/python"'.''*'"startup.py" \
|
||
"conda run -n langchain-chat"'.''*'"startup.py" \
|
||
"python"'.''*'"startup.py --all-api" \
|
||
"python"'.''*'"startup.py -a"
|
||
do
|
||
if pkill -f "$pat" 2>/dev/null; then
|
||
log_tee "pkill 已匹配并发送信号: $pat"
|
||
fi
|
||
done
|
||
|
||
# cwd 在 langchain-chat 且命令行含 startup.py（兜底）
|
||
while read -r pid; do
|
||
[ -z "${pid:-}" ] && continue
|
||
[ ! -d "/proc/$pid" ] && continue
|
||
local cwd cmdl
|
||
cwd=$(readlink -f "/proc/$pid/cwd" 2>/dev/null || true)
|
||
cmdl=$(tr '\0' ' ' <"/proc/$pid/cmdline" 2>/dev/null || true)
|
||
if [[ "$cmdl" == *startup.py* ]] && [[ "$cwd" == "$LC_ROOT" ]]; then
|
||
kill -TERM "$pid" 2>/dev/null && log_tee "按 cwd+argv TERM PID=$pid"
|
||
fi
|
||
done < <(pgrep -f 'startup\.py' 2>/dev/null || true)
|
||
|
||
sleep 2
|
||
|
||
# pkill 未匹配到旧进程时，端口仍可能被旧实例占用（含 root 启动、非 root 无法用 lsof 杀）
|
||
if _langchain_ports_still_listen; then
|
||
log_tee "langchain 相关端口仍监听 (7861/20000/20101/20002)，按 lsof 可见 PID 清理…"
|
||
_kill_by_listen_ports
|
||
else
|
||
log_tee "相关端口已释放，跳过端口强杀"
|
||
fi
|
||
|
||
if _langchain_ports_still_listen; then
|
||
log_tee "端口仍被占用，尝试免密 sudo fuser 清理…"
|
||
_sudo_fuser_kill_ports || true
|
||
fi
|
||
|
||
local _w=0
|
||
while _langchain_ports_still_listen && [ "$_w" -lt 30 ]; do
|
||
sleep 1
|
||
_w=$((_w + 1))
|
||
done
|
||
|
||
if _langchain_ports_still_listen; then
|
||
log_tee "错误: 7861/20000/20101/20002 仍有端口在监听，无法安全启动第二个实例。"
|
||
log_tee "请先释放端口后再执行本脚本（见上方 sudo 提示）。"
|
||
exit 1
|
||
fi
|
||
|
||
sleep 1
|
||
}
|
||
|
||
_stop_langchain
|
||
|
||
# 停止旧的日志裁剪守护（避免重复多开）
|
||
pkill -f "log-trim-daemon\.sh --file ${LOG_FILE}" 2>/dev/null || true
|
||
|
||
log_tee "======== 启动 langchain-chat (--all-api) ========"
|
||
if [ ! -f "$CONDA_SH" ]; then
|
||
log_tee "错误: 未找到 $CONDA_SH"
|
||
exit 1
|
||
fi
|
||
# shellcheck source=/dev/null
|
||
source "$CONDA_SH"
|
||
conda activate langchain-chat
|
||
cd "$LC_ROOT"
|
||
export PYTHONPATH="$LC_ROOT"
|
||
# PDF 预览与 kb_config.PDF_CONVERT_KB_ROOT 一致（地址归一化在 Python 内完成）
|
||
export PDF_CONVERT_KB_ROOT="${PDF_CONVERT_KB_ROOT:-$LC_ROOT/knowledge_base}"
|
||
# 明确使用可请求的地址（:- 无法覆盖已设置的 0.0.0.0，此处强制纠正）
|
||
if [[ "${PDF_CONVERT_API_URL:-}" == *0.0.0.0* ]] || [[ -z "${PDF_CONVERT_API_URL:-}" ]]; then
|
||
export PDF_CONVERT_API_URL="http://127.0.0.1:6006/convert/"
|
||
fi
|
||
# 避免旧 .pyc 仍含硬编码的 0.0.0.0:6006
|
||
rm -f "$LC_ROOT/configs/__pycache__/kb_config."*.pyc 2>/dev/null || true
|
||
rm -f "$LC_ROOT/server/knowledge_base/__pycache__/file_converter."*.pyc 2>/dev/null || true
|
||
rm -f "$LC_ROOT/server/knowledge_base/__pycache__/cleanpdf."*.pyc 2>/dev/null || true
|
||
rm -f "$LC_ROOT/server/knowledge_base/__pycache__/pdf_convert_url."*.pyc 2>/dev/null || true
|
||
|
||
# 控制日志文件总大小（保留最后 5MB，避免 langchain-chat.log 无限增长）
|
||
nohup bash "$GANGYAN_ROOT/scripts/log-trim-daemon.sh" --file "$LOG_FILE" --max-mb 5 --interval-sec 3 >> "$TRIM_LOG" 2>&1 &
|
||
log_tee "已启动日志裁剪守护，日志: $TRIM_LOG"
|
||
|
||
nohup python startup.py --all-api >> "$LOG_FILE" 2>&1 &
|
||
log_tee "已后台启动，PID=$! ，API 约 7861 ，日志: $LOG_FILE"
|
||
|
||
# 本地 PDF 预览依赖的转换微服务（:6006）；未监听时再启动
|
||
if ! ss -tln 2>/dev/null | grep -qE ':6006[[:space:]]'; then
|
||
bash "$GANGYAN_ROOT/scripts/pdf-convert-service.sh" >> "$LOG_DIR/pdf-convert-service.log" 2>&1 || log_tee "提示: pdf-convert-service 启动失败，见 $LOG_DIR/pdf-convert-service.log"
|
||
else
|
||
log_tee "pdf-convert-service 端口 6006 已在监听，跳过"
|
||
fi
|