admin/bl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

1

Browse Source
This commit is contained in:
1
2026-01-23 21:53:54 +00:00
parent 5682ff2d42
commit eda9d955fe
1 changed files with 58 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files

121
modules/base/middleware/server.go
View File

@@ -198,12 +198,10 @@ else
fi
#!/bin/bash
# ===== 优雅终止logic会话（先等内部程序退出 → 再退screen）=====
# ===== 优雅终止logic会话（解决screen -ls卡住问题）=====
echo "===== 优雅终止logic会话 ====="
SCREEN_PID=""
# 替换为你实际的screen名称（示例：logic）
SCREEN_NAME="%s{screen_name}"
# 日志文件路径（可根据需要调整）
SCREEN_NAME="logic"
LOG_FILE="./screen_logic_exit.log"
# 调试开关（如需详细日志，取消set -x注释）
@@ -211,20 +209,17 @@ set -o pipefail
export PS4='[DEBUG] ${BASH_SOURCE}:${LINENO} - ${FUNCNAME[0]:+${FUNCNAME[0]}(): }'
# set -x
# ========== 核心函数：基于kill -0的等待进程退出函数 ==========
# 等待进程结束（带60秒超时，用kill -0检测存活，输出进度点）
# ========== 核心函数（保留你的原版） ==========
wait_for_process_exit() {
local pidKilled=$1
local begin=$(date +%s) # 记录开始时间（秒）
local begin=$(date +%s)
local end
local timeout=60 # 最大等待时间（秒）
local timeout=60
# 循环检测进程是否存活
while kill -0 $pidKilled > /dev/null 2>&1; do
echo -n "." # 输出进度点，直观显示等待过程
echo -n "."
sleep 1;
# 检查是否超时
end=$(date +%s)
if [ $((end - begin)) -gt $timeout ]; then
echo -e "\n⚠️ 等待进程$pidKilled退出超时（已等${timeout}秒）"
@@ -232,50 +227,57 @@ wait_for_process_exit() {
fi
done
# 最终状态提示
if ! kill -0 $pidKilled > /dev/null 2>&1; then
echo -e "\n✅ 进程$pidKilled已退出"
fi
}
# 定义：检查PID是否存活的函数（复用kill -0逻辑）
pid_is_alive() {
local pid=$1
if [ -n "$pid" ] && kill -0 "$pid" > /dev/null 2>&1; then
return 0 # PID存活
return 0
else
return 1 # PID不存在
return 1
fi
}
# 定义：获取screen会话下的所有子进程PID
get_inner_procs() {
local screen_pid=$1
# 5秒超时，避免pstree卡住
local procs=$(timeout 5 pstree -p "$screen_pid" 2>/dev/null | grep -oE '\([0-9]+\)' | tr -d '()' | grep -v "$screen_pid" | sort -u)
# 兜底：pstree失败时用pgrep按screen名称查找
if [ -z "$procs" ]; then
procs=$(pgrep -f "SCREEN -S $SCREEN_NAME" 2>/dev/null | grep -v "$screen_pid")
fi
echo "$procs"
}
# 第一步：提取screen主PID并校验
SCREEN_PID=$(screen -ls "$SCREEN_NAME" | grep -oE '[0-9]+\.'"$SCREEN_NAME" | head -1 | cut -d. -f1)
if [ -z "$SCREEN_PID" ]; then
echo "ℹ️ 未找到$SCREEN_NAME会话对应的PID，跳过终止流程"
# 去掉exit 1，直接跳过后续逻辑，不影响其他脚本执行
else
# ========== 仅当找到PID时，才执行以下终止流程 ==========
echo "✅ 找到$SCREEN_NAME，主PID：$SCREEN_PID"
screen_send_cmd() {
local cmd="$1"
local screen_full_id="$2"
# ^M需手动生成：Ctrl+v+回车
screen -S "$screen_full_id" -p 0 -X stuff "${cmd}^M"
sleep 1
}
# 第二步：导出【退出前】的实时日志（终端+文件双输出）
# ========== 核心修复：给screen -ls加超时，避免卡住 ==========
echo "===== 检测screen会话（5秒超时） ====="
# 关键修改：给整个提取命令加5秒超时，超时则直接设为空
SCREEN_FULL_ID=$(timeout 5 screen -ls 2>/dev/null | grep -E "[0-9]+\.$SCREEN_NAME" | grep -v "Dead\|Invalid" | head -1 | awk '{print $1}')
# 无论是否超时/卡住，只要SCREEN_FULL_ID为空，就直接走后续
if [ -z "$SCREEN_FULL_ID" ]; then
echo "ℹ️ 未找到$SCREEN_NAME会话（或screen -ls执行超时），直接执行后续脚本"
else
# 找到会话：执行终止逻辑
SCREEN_PID=$(echo "$SCREEN_FULL_ID" | cut -d. -f1)
echo "✅ 找到$SCREEN_NAME，主PID：$SCREEN_PID | 完整ID：$SCREEN_FULL_ID"
# 导出退出前日志
echo -e "\n===== 【退出前】$SCREEN_NAME 内程序实时log ====="
screen -S "$SCREEN_NAME" -p 0 -X hardcopy -h "$LOG_FILE" 2>/dev/null
screen -S "$SCREEN_FULL_ID" -p 0 -X hardcopy -h "$LOG_FILE" 2>/dev/null
cat "$LOG_FILE"
# 第三步：给所有子进程发SIGTERM信号，并等待进程退出
echo -e "\n===== 开始给所有子进程发优雅退出信号(SIGTERM) ====="
# 给子进程发SIGTERM并等待
echo -e "\n===== 给子进程发优雅退出信号(SIGTERM) ====="
INNER_PROCS=$(get_inner_procs "$SCREEN_PID")
if [ -z "$INNER_PROCS" ]; then
echo "ℹ 未检测到$SCREEN_NAME下的子进程"
@@ -284,9 +286,9 @@ else
for pid in $INNER_PROCS; do
if pid_is_alive "$pid"; then
echo -n "📌 终止进程$pid并等待退出："
kill -15 "$pid" # 发送SIGTERM优雅退出信号
kill -15 "$pid"
if [ $? -eq 0 ]; then
wait_for_process_exit "$pid" # 调用等待函数
wait_for_process_exit "$pid"
else
echo "❌ 进程$pid：发送SIGTERM失败"
fi
@@ -296,54 +298,47 @@ else
done
fi
# 验证子进程是否全部退出
# 验证子进程退出状态
echo -e "\n===== 验证子进程退出状态 ====="
REMAIN_PROCS=$(get_inner_procs "$SCREEN_PID")
INNER_PROC_EXIST=true
if [ -z "$REMAIN_PROCS" ]; then
INNER_PROC_EXIST=false
echo "✅ $SCREEN_NAME内部所有程序已退出"
else
echo "⚠️ 仍有残留进程：$REMAIN_PROCS"
fi
# 退出screen会话
if [ "$INNER_PROC_EXIST" = false ]; then
echo "内部程序已退出，开始退出$SCREEN_NAME会话..."
# 尝试正常退出screen会话
if screen -S "$SCREEN_NAME" -X quit 2>/dev/null; then
echo "✅ $SCREEN_NAME会话已通过screen -X quit退出"
else
echo "ℹ️ screen -X quit执行失败，检查PID是否存活..."
# 仅当PID存活时，执行kill
if pid_is_alive "$SCREEN_PID"; then
echo "📌 PID $SCREEN_PID 仍存活，尝试kill终止..."
timeout 5 kill -15 "$SCREEN_PID" 2>/dev/null
if [ $? -eq 0 ]; then
echo "✅ 已给screen主进程$SCREEN_PID发送SIGTERM信号"
else
echo "⚠️ kill $SCREEN_PID 失败"
fi
else
echo "ℹ️ PID $SCREEN_PID 已不存在，跳过kill操作"
fi
fi
sleep 2
else
echo "⚠️ 内部程序未完全退出，跳过退出screen会话"
fi
# 投递exit命令退出screen
echo -e "\n===== 优雅退出screen会话 ====="
echo "向$SCREEN_NAME投递exit命令..."
screen_send_cmd "exit" "$SCREEN_FULL_ID"
# 等待并验证最终状态
echo -n "等待screen会话自动退出"
begin=$(date +%s)
while timeout 1 screen -ls "$SCREEN_NAME" 2>/dev/null | grep -q -E "[0-9]+\.$SCREEN_NAME"; do
echo -n "."
sleep 1
if [ $((date +%s - begin)) -gt 30 ]; then
echo -e "\n⚠️ 等待screen退出超时（30秒）"
break
fi
done
# 最终验证
echo -e "\n===== 最终验证 ====="
if screen -ls "$SCREEN_NAME" 2>/dev/null | grep -q -E "[0-9]+\.$SCREEN_NAME"; then
if timeout 1 screen -ls "$SCREEN_NAME" 2>/dev/null | grep -q -E "[0-9]+\.$SCREEN_NAME"; then
echo "❌ $SCREEN_NAME会话最终仍未退出"
else
echo "✅ $SCREEN_NAME会话已完全退出"
fi
fi
# 后续脚本可以从这里继续执行，不受上述逻辑影响
# ========== 后续脚本：必执行，永不卡住 ==========
echo -e "\n===== 终止logic会话流程结束，继续执行后续脚本 ====="
# 示例后续逻辑
# echo "执行后续任务：备份日志、启动新进程等..."
# ===== 准备下载目录 =====
echo "创建工作目录：%s{work_dir}"