46 lines
1.1 KiB
Bash
Executable File
46 lines
1.1 KiB
Bash
Executable File
#!/usr/bin/env bash
|
||
# 停止 GPU worker（由 run_worker.sh 后台启动的进程）。
|
||
# 优先用 worker.pid；找不到则按端口兜底查找。
|
||
set -euo pipefail
|
||
cd "$(dirname "$0")"
|
||
|
||
PID_FILE="worker.pid"
|
||
PORT="${PORT:-8187}"
|
||
|
||
stop_pid() {
|
||
local pid="$1"
|
||
if kill -0 "$pid" 2>/dev/null; then
|
||
echo ">> 停止 worker (PID $pid)"
|
||
kill "$pid"
|
||
# 等待优雅退出，最多 10 秒
|
||
for _ in $(seq 1 10); do
|
||
kill -0 "$pid" 2>/dev/null || return 0
|
||
sleep 1
|
||
done
|
||
echo ">> 未在 10 秒内退出，强制 kill -9"
|
||
kill -9 "$pid" 2>/dev/null || true
|
||
fi
|
||
}
|
||
|
||
# 1) 用 PID 文件
|
||
if [ -f "$PID_FILE" ]; then
|
||
PID="$(cat "$PID_FILE")"
|
||
if [ -n "$PID" ] && kill -0 "$PID" 2>/dev/null; then
|
||
stop_pid "$PID"
|
||
rm -f "$PID_FILE"
|
||
echo ">> 已停止"
|
||
exit 0
|
||
fi
|
||
rm -f "$PID_FILE"
|
||
fi
|
||
|
||
# 2) 兜底：按监听端口查找
|
||
PIDS="$(ss -tlnp 2>/dev/null | grep ":${PORT} " | grep -oE 'pid=[0-9]+' | grep -oE '[0-9]+' | sort -u || true)"
|
||
if [ -n "$PIDS" ]; then
|
||
for p in $PIDS; do stop_pid "$p"; done
|
||
echo ">> 已停止（按端口 ${PORT} 查找）"
|
||
exit 0
|
||
fi
|
||
|
||
echo ">> 没有发现正在运行的 worker（端口 ${PORT}）"
|