137 lines
3.2 KiB
Bash
137 lines
3.2 KiB
Bash
#!/bin/bash
|
|
|
|
: << 'EOF'
|
|
执行远程脚本，以实现任务的状态监控。
|
|
监控机上部署发送通知（企微）的脚本，把结果发送出来。
|
|
EOF
|
|
|
|
# 颜色定义
|
|
RED='\033[0;31m'
|
|
GREEN='\033[0;32m'
|
|
YELLOW='\033[0;33m'
|
|
NC='\033[0m' # 无颜色
|
|
|
|
# 定义命令映射和函数
|
|
declare -A COMMAND_MAP
|
|
declare -a COMMAND_ORDER # 记录命令注册顺序，用于帮助信息
|
|
|
|
# 注册命令函数并添加到映射
|
|
register_command() {
|
|
local cmd_name=$1
|
|
local cmd_desc=$2
|
|
COMMAND_MAP[$cmd_name]=$cmd_desc
|
|
COMMAND_ORDER+=($cmd_name)
|
|
}
|
|
|
|
# 执行远程命令的通用函数
|
|
execute_remote() {
|
|
local server=$1
|
|
local user=$2
|
|
local ssh_opts=$3
|
|
local cmd=$4
|
|
local desc=$5
|
|
|
|
echo -e "${YELLOW}执行命令: ${desc}${NC}"
|
|
echo "----------------------------------------"
|
|
echo -e "${YELLOW}服务器: ${user}@${server}${NC}"
|
|
|
|
# 执行远程命令
|
|
RESULT=$(ssh $ssh_opts $user@$server "$cmd")
|
|
EXIT_CODE=$?
|
|
|
|
# 输出结果
|
|
if [ $EXIT_CODE -eq 0 ]; then
|
|
echo -e "${GREEN}命令执行成功${NC}"
|
|
echo "$RESULT"
|
|
|
|
# 执行本地脚本并将远程命令结果作为输入
|
|
python3 ./send_to_wecom.py "$RESULT"
|
|
return 0
|
|
else
|
|
echo -e "${RED}命令执行失败，退出代码: $EXIT_CODE${NC}"
|
|
return $EXIT_CODE
|
|
fi
|
|
}
|
|
|
|
|
|
# 定义具体命令函数
|
|
cmd_javbus() {
|
|
local SERVER="170.106.191.35"
|
|
local USER="root"
|
|
local SSH_OPTS="-o StrictHostKeyChecking=no -o ConnectTimeout=10"
|
|
|
|
execute_remote "$SERVER" "$USER" "$SSH_OPTS" \
|
|
"cd /root/projects/resources && python3 -m src.monitor.check_javbus" \
|
|
"检查服务状态"
|
|
}
|
|
|
|
# 定义具体命令函数
|
|
cmd_aabook() {
|
|
local SERVER="170.106.191.35"
|
|
local USER="root"
|
|
local SSH_OPTS="-o StrictHostKeyChecking=no -o ConnectTimeout=10"
|
|
|
|
execute_remote "$SERVER" "$USER" "$SSH_OPTS" \
|
|
"cd ~/resources/aabook/src/ && python3 ./check_status.py" \
|
|
"检查服务状态"
|
|
}
|
|
|
|
# 注册命令
|
|
register_command "aabook" "查询 aabook 任务进度"
|
|
register_command "javbus" "查询 javbus 任务进度"
|
|
|
|
# 显示帮助信息
|
|
show_help() {
|
|
echo "用法: $0 [命令1] [命令2] [...]"
|
|
echo "可用命令:"
|
|
|
|
for cmd in "${COMMAND_ORDER[@]}"; do
|
|
printf " %-10s %s\n" "$cmd" "${COMMAND_MAP[$cmd]}"
|
|
done
|
|
|
|
echo
|
|
echo "示例:"
|
|
echo " $0 check fetch"
|
|
echo " $0 backup"
|
|
}
|
|
|
|
# 主函数
|
|
main() {
|
|
# 检查是否提供了命令参数
|
|
if [ $# -eq 0 ]; then
|
|
show_help
|
|
exit 1
|
|
fi
|
|
|
|
local success=true
|
|
|
|
# 处理每个命令参数
|
|
for cmd in "$@"; do
|
|
# 检查命令是否存在
|
|
if [[ -z "${COMMAND_MAP[$cmd]}" ]]; then
|
|
echo -e "${RED}错误: 未知命令 '$cmd'${NC}"
|
|
show_help
|
|
exit 1
|
|
fi
|
|
|
|
# 执行对应的命令函数
|
|
cmd_${cmd}
|
|
if [ $? -ne 0 ]; then
|
|
success=false
|
|
fi
|
|
|
|
echo
|
|
done
|
|
|
|
# 输出总体执行结果
|
|
if [ "$success" = true ]; then
|
|
echo -e "${GREEN}所有命令执行成功${NC}"
|
|
exit 0
|
|
else
|
|
echo -e "${RED}部分命令执行失败${NC}"
|
|
exit 1
|
|
fi
|
|
}
|
|
|
|
# 执行主函数
|
|
main "$@" |