add udp and manager
This commit is contained in:
zj
234
templates/control.sh.tpl
Normal file
234
templates/control.sh.tpl
Normal file
@@ -0,0 +1,234 @@
|
||||
#!/bin/bash
|
||||
# Auto-generated control script
|
||||
# Usage: ./control.sh {start|stop|restart|status}
|
||||
|
||||
BASE_DIR="{base_dir}"
|
||||
PYTHON_PATH="{python_path}"
|
||||
|
||||
# 设备列表: ID 名称 协议
|
||||
DEVICES=(
|
||||
{device_defs}
|
||||
)
|
||||
|
||||
# 颜色定义
|
||||
RED='\033[0;31m'
|
||||
GREEN='\033[0;32m'
|
||||
YELLOW='\033[1;33m'
|
||||
NC='\033[0m' # No Color
|
||||
|
||||
show_help() {
|
||||
echo "Usage: $0 {start|stop|restart|status}"
|
||||
echo ""
|
||||
echo "Commands:"
|
||||
echo " start 启动所有服务"
|
||||
echo " stop 停止所有服务"
|
||||
echo " restart 重启所有服务"
|
||||
echo " status 查看服务状态"
|
||||
}
|
||||
|
||||
check_root() {
|
||||
if [[ $EUID -ne 0 ]]; then
|
||||
echo -e "${YELLOW}Warning: 建议使用 root 权限运行${NC}"
|
||||
fi
|
||||
}
|
||||
|
||||
start_service() {
|
||||
local device_id=$1
|
||||
local device_name=$2
|
||||
local protocol=$3
|
||||
local script_path="$BASE_DIR/bin/$device_id.py"
|
||||
|
||||
# 检查进程是否已存在（使用更精确的匹配）
|
||||
local pid=$(pgrep -f "${PYTHON_PATH}.*${script_path}" | head -1)
|
||||
if [[ -n "$pid" ]]; then
|
||||
echo -e "[${YELLOW}SKIP${NC}] $device_name ($device_id) - ${protocol} 已在运行 (PID: $pid)"
|
||||
return 0
|
||||
fi
|
||||
|
||||
echo -n "[START] $device_name ($device_id) - ${protocol} ... "
|
||||
nohup $PYTHON_PATH "$script_path" > /dev/null 2>&1 &
|
||||
sleep 1
|
||||
|
||||
# 检查是否启动成功
|
||||
pid=$(pgrep -f "${PYTHON_PATH}.*${script_path}" | head -1)
|
||||
if [[ -n "$pid" ]]; then
|
||||
echo -e "${GREEN}OK${NC} (PID: $pid)"
|
||||
return 0
|
||||
else
|
||||
echo -e "${RED}FAILED${NC}"
|
||||
return 1
|
||||
fi
|
||||
}
|
||||
|
||||
stop_service() {
|
||||
local device_id=$1
|
||||
local device_name=$2
|
||||
local protocol=$3
|
||||
local script_path="$BASE_DIR/bin/$device_id.py"
|
||||
|
||||
# 使用更精确的匹配模式
|
||||
local pids=$(pgrep -f "${PYTHON_PATH}.*${script_path}")
|
||||
if [[ -z "$pids" ]]; then
|
||||
echo -e "[${YELLOW}SKIP${NC}] $device_name ($device_id) - 未运行"
|
||||
return 0
|
||||
fi
|
||||
|
||||
# 获取第一个匹配的 PID
|
||||
local pid=$(echo "$pids" | head -1)
|
||||
echo -n "[STOP] $device_name ($device_id) - ${protocol} (PID: $pid) ... "
|
||||
|
||||
# 结束所有匹配的进程
|
||||
echo "$pids" | while read -r p; do
|
||||
kill "$p" 2>/dev/null
|
||||
done
|
||||
|
||||
# 等待进程结束
|
||||
local i
|
||||
for i in {1..10}; do
|
||||
sleep 0.5
|
||||
pids=$(pgrep -f "${PYTHON_PATH}.*${script_path}" || true)
|
||||
if [[ -z "$pids" ]]; then
|
||||
break
|
||||
fi
|
||||
done
|
||||
|
||||
# 检查是否还有残留进程
|
||||
pids=$(pgrep -f "${PYTHON_PATH}.*${script_path}" || true)
|
||||
if [[ -n "$pids" ]]; then
|
||||
# 强制结束残留进程
|
||||
echo "$pids" | while read -r p; do
|
||||
kill -9 "$p" 2>/dev/null
|
||||
done
|
||||
sleep 0.5
|
||||
fi
|
||||
|
||||
# 最终检查
|
||||
pids=$(pgrep -f "${PYTHON_PATH}.*${script_path}" || true)
|
||||
if [[ -z "$pids" ]]; then
|
||||
echo -e "${GREEN}OK${NC}"
|
||||
return 0
|
||||
else
|
||||
echo -e "${RED}FAILED${NC}"
|
||||
return 1
|
||||
fi
|
||||
}
|
||||
|
||||
status_service() {
|
||||
local device_id=$1
|
||||
local device_name=$2
|
||||
local protocol=$3
|
||||
local script_path="$BASE_DIR/bin/$device_id.py"
|
||||
|
||||
local pid=$(pgrep -f "${PYTHON_PATH}.*${script_path}" | head -1)
|
||||
if [[ -n "$pid" ]]; then
|
||||
echo -e "${GREEN}[RUNNING]${NC} $device_name ($device_id) - ${protocol} (PID: $pid)"
|
||||
return 0
|
||||
else
|
||||
echo -e "${RED}[STOPPED]${NC} $device_name ($device_id) - ${protocol}"
|
||||
return 1
|
||||
fi
|
||||
}
|
||||
|
||||
cmd_start() {
|
||||
echo "======================================="
|
||||
echo "启动 UPC Resent 服务"
|
||||
echo "======================================="
|
||||
check_root
|
||||
echo ""
|
||||
|
||||
local count=0
|
||||
local device_id device_name protocol
|
||||
for ((i=0; i<${#DEVICES[@]}; i+=3)); do
|
||||
device_id="${DEVICES[i]}"
|
||||
device_name="${DEVICES[i+1]}"
|
||||
protocol="${DEVICES[i+2]}"
|
||||
|
||||
if start_service "$device_id" "$device_name" "$protocol"; then
|
||||
((count++))
|
||||
fi
|
||||
done
|
||||
|
||||
echo ""
|
||||
echo "======================================="
|
||||
echo "启动完成: $count/$((${#DEVICES[@]}/3)) 个服务"
|
||||
echo "======================================="
|
||||
}
|
||||
|
||||
cmd_stop() {
|
||||
echo "======================================="
|
||||
echo "停止 UPC Resent 服务"
|
||||
echo "======================================="
|
||||
echo ""
|
||||
|
||||
local count=0
|
||||
local device_id device_name protocol
|
||||
for ((i=0; i<${#DEVICES[@]}; i+=3)); do
|
||||
device_id="${DEVICES[i]}"
|
||||
device_name="${DEVICES[i+1]}"
|
||||
protocol="${DEVICES[i+2]}"
|
||||
|
||||
if stop_service "$device_id" "$device_name" "$protocol"; then
|
||||
((count++))
|
||||
fi
|
||||
done
|
||||
|
||||
echo ""
|
||||
echo "======================================="
|
||||
echo "停止完成: $count/$((${#DEVICES[@]}/3)) 个服务"
|
||||
echo "======================================="
|
||||
}
|
||||
|
||||
cmd_restart() {
|
||||
cmd_stop
|
||||
echo ""
|
||||
sleep 1
|
||||
cmd_start
|
||||
}
|
||||
|
||||
cmd_status() {
|
||||
echo "======================================="
|
||||
echo "UPC Resent 服务状态"
|
||||
echo "======================================="
|
||||
echo ""
|
||||
|
||||
local running=0
|
||||
local stopped=0
|
||||
local device_id device_name protocol
|
||||
|
||||
for ((i=0; i<${#DEVICES[@]}; i+=3)); do
|
||||
device_id="${DEVICES[i]}"
|
||||
device_name="${DEVICES[i+1]}"
|
||||
protocol="${DEVICES[i+2]}"
|
||||
|
||||
if status_service "$device_id" "$device_name" "$protocol"; then
|
||||
((running++))
|
||||
else
|
||||
((stopped++))
|
||||
fi
|
||||
done
|
||||
|
||||
echo ""
|
||||
echo "======================================="
|
||||
echo "运行中: $running, 已停止: $stopped, 总计: $((${#DEVICES[@]}/3))"
|
||||
echo "======================================="
|
||||
}
|
||||
|
||||
# 主逻辑
|
||||
case "${1:-}" in
|
||||
start)
|
||||
cmd_start
|
||||
;;
|
||||
stop)
|
||||
cmd_stop
|
||||
;;
|
||||
restart)
|
||||
cmd_restart
|
||||
;;
|
||||
status)
|
||||
cmd_status
|
||||
;;
|
||||
*)
|
||||
show_help
|
||||
exit 1
|
||||
;;
|
||||
esac
|
||||
@@ -59,9 +59,13 @@ def check_process_running():
|
||||
"""通过进程名检查服务是否正在运行"""
|
||||
try:
|
||||
# 检查是否有对应设备的Python进程在运行
|
||||
cmd = f"ps aux | grep '{{DEVICE_ID}}.py' | grep -v grep"
|
||||
result = subprocess.run(cmd, shell=True, capture_output=True, text=True)
|
||||
if result.returncode == 0 and result.stdout.strip():
|
||||
# 注意：需要排除 crond 脚本自身的进程 (upcrond-{{DEVICE_ID}}.py)
|
||||
cmd = f"ps aux | grep '{{DEVICE_ID}}.py' | grep -v grep | grep -v upcrond"
|
||||
# 兼容 Python 3.6: 使用 PIPE 代替 capture_output
|
||||
result = subprocess.run(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
||||
# 兼容 Python 3.6: 手动解码输出
|
||||
stdout = result.stdout.decode('utf-8') if result.stdout else ''
|
||||
if result.returncode == 0 and stdout.strip():
|
||||
print(f"[{{DEVICE_ID}}] 进程检查: 服务正在运行")
|
||||
logging.info(f"进程检查: 服务正在运行")
|
||||
return True
|
||||
|
||||
Reference in New Issue
Block a user