first commit
This commit is contained in:
339
run.sh
Executable file
339
run.sh
Executable file
@@ -0,0 +1,339 @@
|
||||
#!/bin/bash
|
||||
|
||||
# ============================================
|
||||
# LinkMaster 节点端运行脚本
|
||||
# 用途：启动、停止、重启节点端服务
|
||||
# ============================================
|
||||
|
||||
set -e
|
||||
|
||||
# 颜色输出
|
||||
RED='\033[0;31m'
|
||||
GREEN='\033[0;32m'
|
||||
YELLOW='\033[1;33m'
|
||||
BLUE='\033[0;34m'
|
||||
NC='\033[0m' # No Color
|
||||
|
||||
# 脚本目录
|
||||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||
cd "$SCRIPT_DIR"
|
||||
|
||||
# 配置
|
||||
BINARY_NAME="agent"
|
||||
LOG_FILE="node.log"
|
||||
PID_FILE="node.pid"
|
||||
BACKEND_URL="${BACKEND_URL:-http://localhost:8080}"
|
||||
|
||||
# 获取PID
|
||||
get_pid() {
|
||||
if [ -f "$PID_FILE" ]; then
|
||||
PID=$(cat "$PID_FILE")
|
||||
if ps -p "$PID" > /dev/null 2>&1; then
|
||||
echo "$PID"
|
||||
else
|
||||
rm -f "$PID_FILE"
|
||||
echo ""
|
||||
fi
|
||||
else
|
||||
echo ""
|
||||
fi
|
||||
}
|
||||
|
||||
# 拉取最新源码并编译
|
||||
update_and_build() {
|
||||
echo -e "${BLUE}拉取最新源码...${NC}"
|
||||
|
||||
# 检查是否在 Git 仓库中
|
||||
if [ ! -d ".git" ]; then
|
||||
echo -e "${YELLOW}警告: 当前目录不是 Git 仓库，跳过代码更新${NC}"
|
||||
return 0
|
||||
fi
|
||||
|
||||
# 拉取最新代码
|
||||
if git pull 2>&1; then
|
||||
echo -e "${GREEN}✓ 代码更新完成${NC}"
|
||||
else
|
||||
PULL_EXIT_CODE=$?
|
||||
echo -e "${YELLOW}警告: Git 拉取失败（退出码: $PULL_EXIT_CODE），将使用当前代码继续${NC}"
|
||||
echo -e "${YELLOW}可能原因: 网络问题、权限问题或本地有未提交的更改${NC}"
|
||||
fi
|
||||
|
||||
# 检查 Go 环境
|
||||
if ! command -v go > /dev/null 2>&1; then
|
||||
echo -e "${RED}错误: 未找到 Go 环境，无法编译${NC}"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# 更新依赖
|
||||
echo -e "${BLUE}更新 Go 依赖...${NC}"
|
||||
if ! go mod download 2>&1; then
|
||||
echo -e "${YELLOW}警告: 依赖更新失败，尝试继续编译${NC}"
|
||||
else
|
||||
echo -e "${GREEN}✓ 依赖更新完成${NC}"
|
||||
fi
|
||||
|
||||
# 编译
|
||||
echo -e "${BLUE}编译二进制文件...${NC}"
|
||||
ARCH=$(uname -m)
|
||||
case $ARCH in
|
||||
x86_64)
|
||||
ARCH="amd64"
|
||||
;;
|
||||
aarch64|arm64)
|
||||
ARCH="arm64"
|
||||
;;
|
||||
*)
|
||||
ARCH="amd64"
|
||||
;;
|
||||
esac
|
||||
|
||||
if GOOS=linux GOARCH=${ARCH} CGO_ENABLED=0 go build -ldflags="-w -s" -o "$BINARY_NAME" ./cmd/agent 2>&1; then
|
||||
if [ -f "$BINARY_NAME" ] && [ -s "$BINARY_NAME" ]; then
|
||||
chmod +x "$BINARY_NAME"
|
||||
echo -e "${GREEN}✓ 编译成功${NC}"
|
||||
else
|
||||
echo -e "${RED}错误: 编译失败，未生成二进制文件${NC}"
|
||||
exit 1
|
||||
fi
|
||||
else
|
||||
echo -e "${RED}错误: 编译失败${NC}"
|
||||
exit 1
|
||||
fi
|
||||
}
|
||||
|
||||
# 检查二进制文件
|
||||
check_binary() {
|
||||
if [ ! -f "$BINARY_NAME" ]; then
|
||||
echo -e "${RED}错误: 找不到二进制文件 $BINARY_NAME${NC}"
|
||||
echo -e "${YELLOW}尝试编译...${NC}"
|
||||
update_and_build
|
||||
return
|
||||
fi
|
||||
|
||||
if [ ! -x "$BINARY_NAME" ]; then
|
||||
chmod +x "$BINARY_NAME"
|
||||
fi
|
||||
}
|
||||
|
||||
# 检查端口占用
|
||||
check_port() {
|
||||
if command -v lsof > /dev/null 2>&1; then
|
||||
PORT_PID=$(lsof -ti :2200 2>/dev/null || echo "")
|
||||
if [ -n "$PORT_PID" ]; then
|
||||
# 检查是否是我们的进程
|
||||
if [ -f "$PID_FILE" ] && [ "$PORT_PID" = "$(cat "$PID_FILE")" ]; then
|
||||
return 0
|
||||
fi
|
||||
echo -e "${YELLOW}警告: 端口2200已被占用 (PID: $PORT_PID)${NC}"
|
||||
echo -e "${YELLOW}是否要停止该进程? (y/n)${NC}"
|
||||
read -r answer
|
||||
if [ "$answer" = "y" ] || [ "$answer" = "Y" ]; then
|
||||
kill "$PORT_PID" 2>/dev/null || true
|
||||
sleep 1
|
||||
else
|
||||
echo -e "${RED}取消启动${NC}"
|
||||
exit 1
|
||||
fi
|
||||
fi
|
||||
fi
|
||||
}
|
||||
|
||||
# 启动服务
|
||||
start() {
|
||||
PID=$(get_pid)
|
||||
if [ -n "$PID" ]; then
|
||||
echo -e "${YELLOW}节点端已在运行 (PID: $PID)${NC}"
|
||||
return 0
|
||||
fi
|
||||
|
||||
# 拉取最新源码并编译
|
||||
update_and_build
|
||||
|
||||
check_port
|
||||
|
||||
echo -e "${BLUE}启动节点端服务...${NC}"
|
||||
echo -e "${BLUE}后端地址: $BACKEND_URL${NC}"
|
||||
|
||||
# 设置环境变量
|
||||
export BACKEND_URL="$BACKEND_URL"
|
||||
|
||||
# 后台运行
|
||||
nohup ./"$BINARY_NAME" > "$LOG_FILE" 2>&1 &
|
||||
NEW_PID=$!
|
||||
|
||||
# 保存PID
|
||||
echo "$NEW_PID" > "$PID_FILE"
|
||||
|
||||
# 等待启动
|
||||
sleep 2
|
||||
|
||||
# 检查是否启动成功
|
||||
if ps -p "$NEW_PID" > /dev/null 2>&1; then
|
||||
# 再次检查健康状态
|
||||
sleep 1
|
||||
if curl -s http://localhost:2200/api/health > /dev/null 2>&1; then
|
||||
echo -e "${GREEN}✓ 节点端已启动 (PID: $NEW_PID)${NC}"
|
||||
echo -e "${BLUE}日志文件: $LOG_FILE${NC}"
|
||||
echo -e "${BLUE}查看日志: tail -f $LOG_FILE${NC}"
|
||||
else
|
||||
echo -e "${GREEN}✓ 节点端进程已启动 (PID: $NEW_PID)${NC}"
|
||||
echo -e "${YELLOW}⚠ 健康检查未通过，请稍后查看日志${NC}"
|
||||
fi
|
||||
else
|
||||
echo -e "${RED}✗ 节点端启动失败${NC}"
|
||||
echo -e "${YELLOW}请查看日志: cat $LOG_FILE${NC}"
|
||||
rm -f "$PID_FILE"
|
||||
exit 1
|
||||
fi
|
||||
}
|
||||
|
||||
# 停止服务
|
||||
stop() {
|
||||
PID=$(get_pid)
|
||||
|
||||
# 如果没有PID文件，尝试通过端口查找
|
||||
if [ -z "$PID" ]; then
|
||||
if command -v lsof > /dev/null 2>&1; then
|
||||
PORT_PID=$(lsof -ti :2200 2>/dev/null || echo "")
|
||||
if [ -n "$PORT_PID" ]; then
|
||||
PID="$PORT_PID"
|
||||
echo -e "${YELLOW}通过端口找到进程 (PID: $PID)${NC}"
|
||||
fi
|
||||
fi
|
||||
fi
|
||||
|
||||
if [ -z "$PID" ]; then
|
||||
echo -e "${YELLOW}节点端未运行${NC}"
|
||||
rm -f "$PID_FILE"
|
||||
return 0
|
||||
fi
|
||||
|
||||
echo -e "${BLUE}停止节点端服务 (PID: $PID)...${NC}"
|
||||
|
||||
# 发送TERM信号
|
||||
kill "$PID" 2>/dev/null || true
|
||||
|
||||
# 等待进程结束
|
||||
for i in {1..10}; do
|
||||
if ! ps -p "$PID" > /dev/null 2>&1; then
|
||||
break
|
||||
fi
|
||||
sleep 1
|
||||
done
|
||||
|
||||
# 如果还在运行，强制杀死
|
||||
if ps -p "$PID" > /dev/null 2>&1; then
|
||||
echo -e "${YELLOW}强制停止节点端...${NC}"
|
||||
kill -9 "$PID" 2>/dev/null || true
|
||||
sleep 1
|
||||
fi
|
||||
|
||||
rm -f "$PID_FILE"
|
||||
echo -e "${GREEN}✓ 节点端已停止${NC}"
|
||||
}
|
||||
|
||||
# 重启服务
|
||||
restart() {
|
||||
echo -e "${BLUE}重启节点端服务...${NC}"
|
||||
stop
|
||||
sleep 1
|
||||
start
|
||||
}
|
||||
|
||||
# 查看状态
|
||||
status() {
|
||||
PID=$(get_pid)
|
||||
if [ -n "$PID" ]; then
|
||||
echo -e "${GREEN}节点端运行中 (PID: $PID)${NC}"
|
||||
|
||||
# 检查健康状态
|
||||
if command -v curl > /dev/null 2>&1; then
|
||||
HEALTH=$(curl -s http://localhost:2200/api/health 2>/dev/null || echo "failed")
|
||||
if [ "$HEALTH" = '{"status":"ok"}' ]; then
|
||||
echo -e "${GREEN}✓ 健康检查: 正常${NC}"
|
||||
else
|
||||
echo -e "${YELLOW}⚠ 健康检查: 异常${NC}"
|
||||
fi
|
||||
fi
|
||||
|
||||
# 显示进程信息
|
||||
ps -p "$PID" -o pid,ppid,cmd,%mem,%cpu,etime 2>/dev/null || true
|
||||
else
|
||||
echo -e "${RED}节点端未运行${NC}"
|
||||
fi
|
||||
}
|
||||
|
||||
# 查看日志
|
||||
logs() {
|
||||
if [ -f "$LOG_FILE" ]; then
|
||||
tail -f "$LOG_FILE"
|
||||
else
|
||||
echo -e "${YELLOW}日志文件不存在: $LOG_FILE${NC}"
|
||||
fi
|
||||
}
|
||||
|
||||
# 查看完整日志
|
||||
logs_all() {
|
||||
if [ -f "$LOG_FILE" ]; then
|
||||
cat "$LOG_FILE"
|
||||
else
|
||||
echo -e "${YELLOW}日志文件不存在: $LOG_FILE${NC}"
|
||||
fi
|
||||
}
|
||||
|
||||
# 显示帮助
|
||||
help() {
|
||||
echo "LinkMaster 节点端运行脚本"
|
||||
echo ""
|
||||
echo "使用方法:"
|
||||
echo " $0 {start|stop|restart|status|logs|logs-all|help}"
|
||||
echo ""
|
||||
echo "命令说明:"
|
||||
echo " start - 启动节点端服务（会自动拉取最新代码并编译）"
|
||||
echo " stop - 停止节点端服务"
|
||||
echo " restart - 重启节点端服务"
|
||||
echo " status - 查看运行状态"
|
||||
echo " logs - 实时查看日志"
|
||||
echo " logs-all - 查看完整日志"
|
||||
echo " help - 显示帮助信息"
|
||||
echo ""
|
||||
echo "环境变量:"
|
||||
echo " BACKEND_URL - 后端服务地址 (默认: http://localhost:8080)"
|
||||
echo ""
|
||||
echo "示例:"
|
||||
echo " BACKEND_URL=http://192.168.1.100:8080 $0 start"
|
||||
echo " $0 status"
|
||||
echo " $0 logs"
|
||||
}
|
||||
|
||||
# 主逻辑
|
||||
case "${1:-help}" in
|
||||
start)
|
||||
start
|
||||
;;
|
||||
stop)
|
||||
stop
|
||||
;;
|
||||
restart)
|
||||
restart
|
||||
;;
|
||||
status)
|
||||
status
|
||||
;;
|
||||
logs)
|
||||
logs
|
||||
;;
|
||||
logs-all)
|
||||
logs_all
|
||||
;;
|
||||
help|--help|-h)
|
||||
help
|
||||
;;
|
||||
*)
|
||||
echo -e "${RED}未知命令: $1${NC}"
|
||||
echo ""
|
||||
help
|
||||
exit 1
|
||||
;;
|
||||
esac
|
||||
|
||||
Reference in New Issue
Block a user