first commit
This commit is contained in:
262
INSTALL.md
Normal file
262
INSTALL.md
Normal file
@@ -0,0 +1,262 @@
|
|||||||
|
# LinkMaster 节点端安装指南
|
||||||
|
|
||||||
|
## 一句话安装
|
||||||
|
|
||||||
|
### 从 GitHub 安装（推荐）
|
||||||
|
|
||||||
|
```bash
|
||||||
|
curl -fsSL https://raw.githubusercontent.com/yourbask/linkmaster-node/main/install.sh | bash -s -- http://your-backend-server:8080
|
||||||
|
```
|
||||||
|
|
||||||
|
**替换说明：**
|
||||||
|
- `yourbask/linkmaster-node` - 独立的 node 项目 GitHub 仓库地址
|
||||||
|
- `http://your-backend-server:8080` - 替换为实际的后端服务器地址
|
||||||
|
|
||||||
|
**重要提示：**
|
||||||
|
- ⚠️ 节点端需要直接连接后端服务器（端口 8080），不是前端地址
|
||||||
|
- 前端通过 `/api` 路径代理到后端，但节点端不使用前端代理
|
||||||
|
- 如果节点和后端在同一服务器：使用 `http://localhost:8080`
|
||||||
|
- 如果节点和后端在不同服务器：使用 `http://backend-ip:8080` 或 `http://backend-domain:8080`
|
||||||
|
- **本项目是独立的 GitHub 仓库**，与前后端项目分离
|
||||||
|
|
||||||
|
**示例：**
|
||||||
|
```bash
|
||||||
|
# 如果后端服务器在 192.168.1.100:8080
|
||||||
|
curl -fsSL https://raw.githubusercontent.com/yourbask/linkmaster-node/main/install.sh | bash -s -- http://192.168.1.100:8080
|
||||||
|
```
|
||||||
|
|
||||||
|
### 指定分支安装
|
||||||
|
|
||||||
|
```bash
|
||||||
|
GITHUB_BRANCH=develop curl -fsSL https://raw.githubusercontent.com/yourbask/linkmaster-node/main/install.sh | bash -s -- http://your-backend-server:8080
|
||||||
|
```
|
||||||
|
|
||||||
|
## 安装步骤说明
|
||||||
|
|
||||||
|
安装脚本会自动完成以下步骤：
|
||||||
|
|
||||||
|
1. **检测系统** - 自动识别 Linux 发行版和 CPU 架构
|
||||||
|
2. **安装依赖** - 自动安装 Git、Go、ping、traceroute、dnsutils 等工具
|
||||||
|
3. **克隆源码** - 从 GitHub 克隆 node 项目源码到 `/opt/linkmaster-node`
|
||||||
|
4. **编译安装** - 自动编译源码并安装二进制文件
|
||||||
|
5. **创建服务** - 自动创建 systemd 服务文件（使用 run.sh 启动）
|
||||||
|
6. **启动服务** - 自动启动并设置开机自启
|
||||||
|
7. **验证安装** - 检查服务状态和健康检查
|
||||||
|
|
||||||
|
**注意：** 每次服务启动时会自动拉取最新代码并重新编译，确保使用最新版本。
|
||||||
|
|
||||||
|
## 安装后管理
|
||||||
|
|
||||||
|
### 查看服务状态
|
||||||
|
|
||||||
|
```bash
|
||||||
|
sudo systemctl status linkmaster-node
|
||||||
|
```
|
||||||
|
|
||||||
|
### 查看日志
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# 实时日志
|
||||||
|
sudo journalctl -u linkmaster-node -f
|
||||||
|
|
||||||
|
# 最近50行日志
|
||||||
|
sudo journalctl -u linkmaster-node -n 50
|
||||||
|
```
|
||||||
|
|
||||||
|
### 重启服务
|
||||||
|
|
||||||
|
```bash
|
||||||
|
sudo systemctl restart linkmaster-node
|
||||||
|
```
|
||||||
|
|
||||||
|
### 停止服务
|
||||||
|
|
||||||
|
```bash
|
||||||
|
sudo systemctl stop linkmaster-node
|
||||||
|
```
|
||||||
|
|
||||||
|
### 禁用开机自启
|
||||||
|
|
||||||
|
```bash
|
||||||
|
sudo systemctl disable linkmaster-node
|
||||||
|
```
|
||||||
|
|
||||||
|
## 验证安装
|
||||||
|
|
||||||
|
### 检查进程
|
||||||
|
|
||||||
|
```bash
|
||||||
|
ps aux | grep linkmaster-node
|
||||||
|
```
|
||||||
|
|
||||||
|
### 检查端口
|
||||||
|
|
||||||
|
```bash
|
||||||
|
netstat -tlnp | grep 2200
|
||||||
|
# 或
|
||||||
|
ss -tlnp | grep 2200
|
||||||
|
```
|
||||||
|
|
||||||
|
### 健康检查
|
||||||
|
|
||||||
|
```bash
|
||||||
|
curl http://localhost:2200/api/health
|
||||||
|
```
|
||||||
|
|
||||||
|
应该返回：`{"status":"ok"}`
|
||||||
|
|
||||||
|
## 手动安装（不使用脚本）
|
||||||
|
|
||||||
|
如果无法使用一键安装脚本，可以手动安装：
|
||||||
|
|
||||||
|
### 1. 克隆源码并编译
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# 克隆仓库
|
||||||
|
git clone https://github.com/yourbask/linkmaster-node.git /opt/linkmaster-node
|
||||||
|
cd /opt/linkmaster-node
|
||||||
|
|
||||||
|
# 安装 Go 环境（如果未安装）
|
||||||
|
# Ubuntu/Debian
|
||||||
|
sudo apt-get install -y golang-go
|
||||||
|
|
||||||
|
# CentOS/RHEL
|
||||||
|
sudo yum install -y golang
|
||||||
|
|
||||||
|
# 编译
|
||||||
|
go build -o agent ./cmd/agent
|
||||||
|
|
||||||
|
# 安装到系统目录
|
||||||
|
sudo cp agent /usr/local/bin/linkmaster-node
|
||||||
|
sudo chmod +x /usr/local/bin/linkmaster-node
|
||||||
|
```
|
||||||
|
|
||||||
|
### 2. 安装系统依赖
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# Ubuntu/Debian
|
||||||
|
sudo apt-get update
|
||||||
|
sudo apt-get install -y ping traceroute dnsutils curl
|
||||||
|
|
||||||
|
# CentOS/RHEL
|
||||||
|
sudo yum install -y iputils traceroute bind-utils curl
|
||||||
|
```
|
||||||
|
|
||||||
|
### 3. 创建 systemd 服务
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# 确保 run.sh 有执行权限
|
||||||
|
sudo chmod +x /opt/linkmaster-node/run.sh
|
||||||
|
|
||||||
|
sudo tee /etc/systemd/system/linkmaster-node.service > /dev/null <<EOF
|
||||||
|
[Unit]
|
||||||
|
Description=LinkMaster Node Service
|
||||||
|
After=network.target
|
||||||
|
|
||||||
|
[Service]
|
||||||
|
Type=simple
|
||||||
|
User=root
|
||||||
|
WorkingDirectory=/opt/linkmaster-node
|
||||||
|
ExecStart=/opt/linkmaster-node/run.sh start
|
||||||
|
Restart=always
|
||||||
|
RestartSec=5
|
||||||
|
Environment="BACKEND_URL=http://your-backend-server:8080"
|
||||||
|
|
||||||
|
[Install]
|
||||||
|
WantedBy=multi-user.target
|
||||||
|
EOF
|
||||||
|
```
|
||||||
|
|
||||||
|
**注意：** 使用 `run.sh` 启动的好处是每次启动会自动拉取最新代码并重新编译。
|
||||||
|
|
||||||
|
### 4. 启动服务
|
||||||
|
|
||||||
|
```bash
|
||||||
|
sudo systemctl daemon-reload
|
||||||
|
sudo systemctl enable linkmaster-node
|
||||||
|
sudo systemctl start linkmaster-node
|
||||||
|
```
|
||||||
|
|
||||||
|
**注意：** 确保 `BACKEND_URL` 环境变量指向后端服务器的实际地址和端口（默认 8080），不是前端地址。
|
||||||
|
|
||||||
|
## 防火墙配置
|
||||||
|
|
||||||
|
确保开放端口 2200：
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# Ubuntu/Debian (ufw)
|
||||||
|
sudo ufw allow 2200/tcp
|
||||||
|
|
||||||
|
# CentOS/RHEL (firewalld)
|
||||||
|
sudo firewall-cmd --permanent --add-port=2200/tcp
|
||||||
|
sudo firewall-cmd --reload
|
||||||
|
```
|
||||||
|
|
||||||
|
## 常见问题
|
||||||
|
|
||||||
|
### 1. 克隆或编译失败
|
||||||
|
|
||||||
|
**问题：** 无法从 GitHub 克隆源码或编译失败
|
||||||
|
|
||||||
|
**解决：**
|
||||||
|
- 检查网络连接
|
||||||
|
- 确认 GitHub 仓库地址正确（独立的 node 项目仓库）
|
||||||
|
- 确认已安装 Git 和 Go 环境
|
||||||
|
- 手动克隆并编译：`git clone https://github.com/yourbask/linkmaster-node.git && cd linkmaster-node && go build -o agent ./cmd/agent`
|
||||||
|
|
||||||
|
### 2. 服务启动失败
|
||||||
|
|
||||||
|
**问题：** systemctl status 显示服务失败
|
||||||
|
|
||||||
|
**解决：**
|
||||||
|
```bash
|
||||||
|
# 查看详细日志
|
||||||
|
sudo journalctl -u linkmaster-node -n 100
|
||||||
|
|
||||||
|
# 检查后端地址是否正确
|
||||||
|
sudo systemctl cat linkmaster-node | grep BACKEND_URL
|
||||||
|
|
||||||
|
# 手动测试后端连接
|
||||||
|
curl http://your-backend-server:8080/api/public/nodes/online
|
||||||
|
```
|
||||||
|
|
||||||
|
### 3. 端口被占用
|
||||||
|
|
||||||
|
**问题：** 端口 2200 已被占用
|
||||||
|
|
||||||
|
**解决：**
|
||||||
|
```bash
|
||||||
|
# 查找占用进程
|
||||||
|
sudo lsof -i :2200
|
||||||
|
|
||||||
|
# 停止占用进程或修改配置
|
||||||
|
```
|
||||||
|
|
||||||
|
### 4. 无法连接后端
|
||||||
|
|
||||||
|
**问题：** 节点无法连接到后端服务器
|
||||||
|
|
||||||
|
**解决：**
|
||||||
|
- 检查后端地址是否正确（应该是 `http://backend-server:8080`，不是前端地址）
|
||||||
|
- 检查网络连通性：`ping your-backend-server`
|
||||||
|
- 检查端口是否开放：`telnet your-backend-server 8080` 或 `nc -zv your-backend-server 8080`
|
||||||
|
- 检查防火墙规则（确保后端服务器的 8080 端口开放）
|
||||||
|
- 检查后端服务是否运行：`curl http://your-backend-server:8080/api/public/nodes/online`
|
||||||
|
- 如果使用前端代理，节点端仍需要直接连接后端，不能使用前端地址
|
||||||
|
|
||||||
|
## 卸载
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# 停止服务
|
||||||
|
sudo systemctl stop linkmaster-node
|
||||||
|
sudo systemctl disable linkmaster-node
|
||||||
|
|
||||||
|
# 删除服务文件
|
||||||
|
sudo rm /etc/systemd/system/linkmaster-node.service
|
||||||
|
sudo systemctl daemon-reload
|
||||||
|
|
||||||
|
# 删除二进制文件和源码目录
|
||||||
|
sudo rm /usr/local/bin/linkmaster-node
|
||||||
|
sudo rm -rf /opt/linkmaster-node
|
||||||
|
```
|
||||||
|
|
||||||
11
Makefile
Normal file
11
Makefile
Normal file
@@ -0,0 +1,11 @@
|
|||||||
|
.PHONY: build build-linux clean
|
||||||
|
|
||||||
|
build:
|
||||||
|
go build -o bin/linkmaster-node ./cmd/agent
|
||||||
|
|
||||||
|
build-linux:
|
||||||
|
GOOS=linux GOARCH=amd64 go build -o bin/linkmaster-node-linux ./cmd/agent
|
||||||
|
|
||||||
|
clean:
|
||||||
|
rm -rf bin/
|
||||||
|
|
||||||
@@ -181,3 +181,4 @@ BACKEND_URL=http://192.168.1.100:8080 ./run.sh start
|
|||||||
|
|
||||||
健康检查
|
健康检查
|
||||||
# linkmaster-node
|
# linkmaster-node
|
||||||
|
# linkmaster-node
|
||||||
|
|||||||
75
cmd/agent/main.go
Normal file
75
cmd/agent/main.go
Normal file
@@ -0,0 +1,75 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
"os/signal"
|
||||||
|
"syscall"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"linkmaster-node/internal/config"
|
||||||
|
"linkmaster-node/internal/heartbeat"
|
||||||
|
"linkmaster-node/internal/recovery"
|
||||||
|
"linkmaster-node/internal/server"
|
||||||
|
|
||||||
|
"go.uber.org/zap"
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
// 加载配置
|
||||||
|
cfg, err := config.Load()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("加载配置失败: %v\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 初始化日志
|
||||||
|
logger, err := initLogger(cfg)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("初始化日志失败: %v\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
defer logger.Sync()
|
||||||
|
|
||||||
|
logger.Info("节点服务启动", zap.String("version", "1.0.0"))
|
||||||
|
|
||||||
|
// 初始化错误恢复
|
||||||
|
recovery.Init()
|
||||||
|
|
||||||
|
// 启动心跳上报
|
||||||
|
heartbeatReporter := heartbeat.NewReporter(cfg)
|
||||||
|
go heartbeatReporter.Start(context.Background())
|
||||||
|
|
||||||
|
// 启动HTTP服务器
|
||||||
|
httpServer := server.NewHTTPServer(cfg)
|
||||||
|
go func() {
|
||||||
|
if err := httpServer.Start(); err != nil {
|
||||||
|
logger.Fatal("HTTP服务器启动失败", zap.Error(err))
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
// 等待中断信号
|
||||||
|
sigChan := make(chan os.Signal, 1)
|
||||||
|
signal.Notify(sigChan, os.Interrupt, syscall.SIGTERM)
|
||||||
|
<-sigChan
|
||||||
|
|
||||||
|
logger.Info("收到停止信号，正在关闭服务...")
|
||||||
|
|
||||||
|
// 优雅关闭
|
||||||
|
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
||||||
|
defer cancel()
|
||||||
|
|
||||||
|
httpServer.Shutdown(ctx)
|
||||||
|
heartbeatReporter.Stop()
|
||||||
|
|
||||||
|
logger.Info("服务已关闭")
|
||||||
|
}
|
||||||
|
|
||||||
|
func initLogger(cfg *config.Config) (*zap.Logger, error) {
|
||||||
|
if cfg.Debug {
|
||||||
|
return zap.NewDevelopment()
|
||||||
|
}
|
||||||
|
return zap.NewProduction()
|
||||||
|
}
|
||||||
|
|
||||||
36
go.mod
Normal file
36
go.mod
Normal file
@@ -0,0 +1,36 @@
|
|||||||
|
module linkmaster-node
|
||||||
|
|
||||||
|
go 1.21
|
||||||
|
|
||||||
|
require (
|
||||||
|
github.com/gin-gonic/gin v1.9.1
|
||||||
|
go.uber.org/zap v1.26.0
|
||||||
|
gopkg.in/yaml.v3 v3.0.1
|
||||||
|
)
|
||||||
|
|
||||||
|
require (
|
||||||
|
github.com/bytedance/sonic v1.9.1 // indirect
|
||||||
|
github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311 // indirect
|
||||||
|
github.com/gabriel-vasile/mimetype v1.4.2 // indirect
|
||||||
|
github.com/gin-contrib/sse v0.1.0 // indirect
|
||||||
|
github.com/go-playground/locales v0.14.1 // indirect
|
||||||
|
github.com/go-playground/universal-translator v0.18.1 // indirect
|
||||||
|
github.com/go-playground/validator/v10 v10.14.0 // indirect
|
||||||
|
github.com/goccy/go-json v0.10.2 // indirect
|
||||||
|
github.com/json-iterator/go v1.1.12 // indirect
|
||||||
|
github.com/klauspost/cpuid/v2 v2.2.4 // indirect
|
||||||
|
github.com/leodido/go-urn v1.2.4 // indirect
|
||||||
|
github.com/mattn/go-isatty v0.0.19 // indirect
|
||||||
|
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
|
||||||
|
github.com/modern-go/reflect2 v1.0.2 // indirect
|
||||||
|
github.com/pelletier/go-toml/v2 v2.0.8 // indirect
|
||||||
|
github.com/twitchyliquid64/golang-asm v0.15.1 // indirect
|
||||||
|
github.com/ugorji/go/codec v1.2.11 // indirect
|
||||||
|
go.uber.org/multierr v1.10.0 // indirect
|
||||||
|
golang.org/x/arch v0.3.0 // indirect
|
||||||
|
golang.org/x/crypto v0.14.0 // indirect
|
||||||
|
golang.org/x/net v0.17.0 // indirect
|
||||||
|
golang.org/x/sys v0.13.0 // indirect
|
||||||
|
golang.org/x/text v0.13.0 // indirect
|
||||||
|
google.golang.org/protobuf v1.30.0 // indirect
|
||||||
|
)
|
||||||
92
go.sum
Normal file
92
go.sum
Normal file
@@ -0,0 +1,92 @@
|
|||||||
|
github.com/bytedance/sonic v1.5.0/go.mod h1:ED5hyg4y6t3/9Ku1R6dU/4KyJ48DZ4jPhfY1O2AihPM=
|
||||||
|
github.com/bytedance/sonic v1.9.1 h1:6iJ6NqdoxCDr6mbY8h18oSO+cShGSMRGCEo7F2h0x8s=
|
||||||
|
github.com/bytedance/sonic v1.9.1/go.mod h1:i736AoUSYt75HyZLoJW9ERYxcy6eaN6h4BZXU064P/U=
|
||||||
|
github.com/chenzhuoyu/base64x v0.0.0-20211019084208-fb5309c8db06/go.mod h1:DH46F32mSOjUmXrMHnKwZdA8wcEefY7UVqBKYGjpdQY=
|
||||||
|
github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311 h1:qSGYFH7+jGhDF8vLC+iwCD4WpbV1EBDSzWkJODFLams=
|
||||||
|
github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311/go.mod h1:b583jCggY9gE99b6G5LEC39OIiVsWj+R97kbl5odCEk=
|
||||||
|
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
|
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
||||||
|
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
|
github.com/gabriel-vasile/mimetype v1.4.2 h1:w5qFW6JKBz9Y393Y4q372O9A7cUSequkh1Q7OhCmWKU=
|
||||||
|
github.com/gabriel-vasile/mimetype v1.4.2/go.mod h1:zApsH/mKG4w07erKIaJPFiX0Tsq9BFQgN3qGY5GnNgA=
|
||||||
|
github.com/gin-contrib/sse v0.1.0 h1:Y/yl/+YNO8GZSjAhjMsSuLt29uWRFHdHYUb5lYOV9qE=
|
||||||
|
github.com/gin-contrib/sse v0.1.0/go.mod h1:RHrZQHXnP2xjPF+u1gW/2HnVO7nvIa9PG3Gm+fLHvGI=
|
||||||
|
github.com/gin-gonic/gin v1.9.1 h1:4idEAncQnU5cB7BeOkPtxjfCSye0AAm1R0RVIqJ+Jmg=
|
||||||
|
github.com/gin-gonic/gin v1.9.1/go.mod h1:hPrL7YrpYKXt5YId3A/Tnip5kqbEAP+KLuI3SUcPTeU=
|
||||||
|
github.com/go-playground/assert/v2 v2.2.0 h1:JvknZsQTYeFEAhQwI4qEt9cyV5ONwRHC+lYKSsYSR8s=
|
||||||
|
github.com/go-playground/assert/v2 v2.2.0/go.mod h1:VDjEfimB/XKnb+ZQfWdccd7VUvScMdVu0Titje2rxJ4=
|
||||||
|
github.com/go-playground/locales v0.14.1 h1:EWaQ/wswjilfKLTECiXz7Rh+3BjFhfDFKv/oXslEjJA=
|
||||||
|
github.com/go-playground/locales v0.14.1/go.mod h1:hxrqLVvrK65+Rwrd5Fc6F2O76J/NuW9t0sjnWqG1slY=
|
||||||
|
github.com/go-playground/universal-translator v0.18.1 h1:Bcnm0ZwsGyWbCzImXv+pAJnYK9S473LQFuzCbDbfSFY=
|
||||||
|
github.com/go-playground/universal-translator v0.18.1/go.mod h1:xekY+UJKNuX9WP91TpwSH2VMlDf28Uj24BCp08ZFTUY=
|
||||||
|
github.com/go-playground/validator/v10 v10.14.0 h1:vgvQWe3XCz3gIeFDm/HnTIbj6UGmg/+t63MyGU2n5js=
|
||||||
|
github.com/go-playground/validator/v10 v10.14.0/go.mod h1:9iXMNT7sEkjXb0I+enO7QXmzG6QCsPWY4zveKFVRSyU=
|
||||||
|
github.com/goccy/go-json v0.10.2 h1:CrxCmQqYDkv1z7lO7Wbh2HN93uovUHgrECaO5ZrCXAU=
|
||||||
|
github.com/goccy/go-json v0.10.2/go.mod h1:6MelG93GURQebXPDq3khkgXZkazVtN9CRI+MGFi0w8I=
|
||||||
|
github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk=
|
||||||
|
github.com/google/go-cmp v0.5.5 h1:Khx7svrCpmxxtHBq5j2mp/xVjsi8hQMfNLvJFAlrGgU=
|
||||||
|
github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
|
||||||
|
github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg=
|
||||||
|
github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM=
|
||||||
|
github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo=
|
||||||
|
github.com/klauspost/cpuid/v2 v2.0.9/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg=
|
||||||
|
github.com/klauspost/cpuid/v2 v2.2.4 h1:acbojRNwl3o09bUq+yDCtZFc1aiwaAAxtcn8YkZXnvk=
|
||||||
|
github.com/klauspost/cpuid/v2 v2.2.4/go.mod h1:RVVoqg1df56z8g3pUjL/3lE5UfnlrJX8tyFgg4nqhuY=
|
||||||
|
github.com/leodido/go-urn v1.2.4 h1:XlAE/cm/ms7TE/VMVoduSpNBoyc2dOxHs5MZSwAN63Q=
|
||||||
|
github.com/leodido/go-urn v1.2.4/go.mod h1:7ZrI8mTSeBSHl/UaRyKQW1qZeMgak41ANeCNaVckg+4=
|
||||||
|
github.com/mattn/go-isatty v0.0.19 h1:JITubQf0MOLdlGRuRq+jtsDlekdYPia9ZFsB8h/APPA=
|
||||||
|
github.com/mattn/go-isatty v0.0.19/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
|
||||||
|
github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
|
||||||
|
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd h1:TRLaZ9cD/w8PVh93nsPXa1VrQ6jlwL5oN8l14QlcNfg=
|
||||||
|
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
|
||||||
|
github.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M=
|
||||||
|
github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk=
|
||||||
|
github.com/pelletier/go-toml/v2 v2.0.8 h1:0ctb6s9mE31h0/lhu+J6OPmVeDxJn+kYnJc2jZR9tGQ=
|
||||||
|
github.com/pelletier/go-toml/v2 v2.0.8/go.mod h1:vuYfssBdrU2XDZ9bYydBu6t+6a6PYNcZljzZR9VXg+4=
|
||||||
|
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
||||||
|
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||||
|
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
||||||
|
github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=
|
||||||
|
github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo=
|
||||||
|
github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
|
||||||
|
github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
||||||
|
github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
||||||
|
github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=
|
||||||
|
github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=
|
||||||
|
github.com/stretchr/testify v1.8.2/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=
|
||||||
|
github.com/stretchr/testify v1.8.3 h1:RP3t2pwF7cMEbC1dqtB6poj3niw/9gnV4Cjg5oW5gtY=
|
||||||
|
github.com/stretchr/testify v1.8.3/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo=
|
||||||
|
github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI=
|
||||||
|
github.com/twitchyliquid64/golang-asm v0.15.1/go.mod h1:a1lVb/DtPvCB8fslRZhAngC2+aY1QWCk3Cedj/Gdt08=
|
||||||
|
github.com/ugorji/go/codec v1.2.11 h1:BMaWp1Bb6fHwEtbplGBGJ498wD+LKlNSl25MjdZY4dU=
|
||||||
|
github.com/ugorji/go/codec v1.2.11/go.mod h1:UNopzCgEMSXjBc6AOMqYvWC1ktqTAfzJZUZgYf6w6lg=
|
||||||
|
go.uber.org/goleak v1.2.0 h1:xqgm/S+aQvhWFTtR0XK3Jvg7z8kGV8P4X14IzwN3Eqk=
|
||||||
|
go.uber.org/goleak v1.2.0/go.mod h1:XJYK+MuIchqpmGmUSAzotztawfKvYLUIgg7guXrwVUo=
|
||||||
|
go.uber.org/multierr v1.10.0 h1:S0h4aNzvfcFsC3dRF1jLoaov7oRaKqRGC/pUEJ2yvPQ=
|
||||||
|
go.uber.org/multierr v1.10.0/go.mod h1:20+QtiLqy0Nd6FdQB9TLXag12DsQkrbs3htMFfDN80Y=
|
||||||
|
go.uber.org/zap v1.26.0 h1:sI7k6L95XOKS281NhVKOFCUNIvv9e0w4BF8N3u+tCRo=
|
||||||
|
go.uber.org/zap v1.26.0/go.mod h1:dtElttAiwGvoJ/vj4IwHBS/gXsEu/pZ50mUIRWuG0so=
|
||||||
|
golang.org/x/arch v0.0.0-20210923205945-b76863e36670/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8=
|
||||||
|
golang.org/x/arch v0.3.0 h1:02VY4/ZcO/gBOH6PUaoiptASxtXU10jazRCP865E97k=
|
||||||
|
golang.org/x/arch v0.3.0/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8=
|
||||||
|
golang.org/x/crypto v0.14.0 h1:wBqGXzWJW6m1XrIKlAH0Hs1JJ7+9KBwnIO8v66Q9cHc=
|
||||||
|
golang.org/x/crypto v0.14.0/go.mod h1:MVFd36DqK4CsrnJYDkBA3VC4m2GkXAM0PvzMCn4JQf4=
|
||||||
|
golang.org/x/net v0.17.0 h1:pVaXccu2ozPjCXewfr1S7xza/zcXTity9cCdXQYSjIM=
|
||||||
|
golang.org/x/net v0.17.0/go.mod h1:NxSsAGuq816PNPmqtQdLE42eU2Fs7NoRIZrHJAlaCOE=
|
||||||
|
golang.org/x/sys v0.0.0-20220704084225-05e143d24a9e/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
|
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
|
golang.org/x/sys v0.13.0 h1:Af8nKPmuFypiUBjVoU9V20FiaFXOcuZI21p0ycVYYGE=
|
||||||
|
golang.org/x/sys v0.13.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
|
golang.org/x/text v0.13.0 h1:ablQoSUd0tRdKxZewP80B+BaqeKJuVhuRxj/dkrun3k=
|
||||||
|
golang.org/x/text v0.13.0/go.mod h1:TvPlkZtksWOMsz7fbANvkp4WM8x/WCo/om8BMLbz+aE=
|
||||||
|
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543 h1:E7g+9GITq07hpfrRu66IVDexMakfv52eLZ2CXBWiKr4=
|
||||||
|
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||||
|
google.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp09yW+WbY/TyQbw=
|
||||||
|
google.golang.org/protobuf v1.30.0 h1:kPPoIgf3TsEvrm0PFe15JQ+570QVxYzEvvHqChK+cng=
|
||||||
|
google.golang.org/protobuf v1.30.0/go.mod h1:HV8QOd/L58Z+nl8r43ehVNZIU/HEI6OcFqwMG9pJV4I=
|
||||||
|
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
|
||||||
|
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||||
|
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
|
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
||||||
|
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
|
rsc.io/pdf v0.1.1/go.mod h1:n8OzWcQ6Sp37PL01nO98y4iUCRdTGarVfzxY20ICaU4=
|
||||||
323
install.sh
Executable file
323
install.sh
Executable file
@@ -0,0 +1,323 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
|
||||||
|
# ============================================
|
||||||
|
# LinkMaster 节点端一键安装脚本
|
||||||
|
# 使用方法: curl -fsSL https://raw.githubusercontent.com/yourbask/linkmaster-node/main/install.sh | bash -s -- <后端地址>
|
||||||
|
# 示例: curl -fsSL https://raw.githubusercontent.com/yourbask/linkmaster-node/main/install.sh | bash -s -- http://192.168.1.100:8080
|
||||||
|
# ============================================
|
||||||
|
|
||||||
|
set -e
|
||||||
|
|
||||||
|
# 颜色输出
|
||||||
|
RED='\033[0;31m'
|
||||||
|
GREEN='\033[0;32m'
|
||||||
|
YELLOW='\033[1;33m'
|
||||||
|
BLUE='\033[0;34m'
|
||||||
|
NC='\033[0m'
|
||||||
|
|
||||||
|
# 配置
|
||||||
|
# 尝试从脚本URL自动提取仓库信息（如果通过curl下载）
|
||||||
|
SCRIPT_URL="${SCRIPT_URL:-}"
|
||||||
|
if [ -z "$SCRIPT_URL" ] && [ -n "${BASH_SOURCE[0]}" ]; then
|
||||||
|
# 如果脚本是通过 curl 下载的，尝试从环境变量获取
|
||||||
|
SCRIPT_URL="${SCRIPT_URL:-}"
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 默认配置（如果无法自动提取，使用这些默认值）
|
||||||
|
GITHUB_REPO="${GITHUB_REPO:-yourbask/linkmaster-node}" # 默认仓库（独立的 node 项目）
|
||||||
|
GITHUB_BRANCH="${GITHUB_BRANCH:-main}" # 默认分支
|
||||||
|
SOURCE_DIR="/opt/linkmaster-node" # 源码目录
|
||||||
|
BINARY_NAME="linkmaster-node"
|
||||||
|
INSTALL_DIR="/usr/local/bin"
|
||||||
|
SERVICE_NAME="linkmaster-node"
|
||||||
|
|
||||||
|
# 获取后端地址参数
|
||||||
|
BACKEND_URL="${1:-}"
|
||||||
|
if [ -z "$BACKEND_URL" ]; then
|
||||||
|
echo -e "${RED}错误: 请提供后端服务器地址${NC}"
|
||||||
|
echo -e "${YELLOW}使用方法:${NC}"
|
||||||
|
echo " curl -fsSL https://raw.githubusercontent.com/${GITHUB_REPO}/${GITHUB_BRANCH}/install.sh | bash -s -- http://your-backend-server:8080"
|
||||||
|
echo ""
|
||||||
|
echo -e "${YELLOW}注意:${NC}"
|
||||||
|
echo " - 节点端需要直接连接后端服务器，不是前端地址"
|
||||||
|
echo " - 后端默认端口: 8080"
|
||||||
|
echo " - 如果节点和后端在同一服务器: http://localhost:8080"
|
||||||
|
echo " - 如果节点和后端在不同服务器: http://backend-ip:8080 或 http://backend-domain:8080"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 检测系统类型和架构
|
||||||
|
detect_system() {
|
||||||
|
if [ -f /etc/os-release ]; then
|
||||||
|
. /etc/os-release
|
||||||
|
OS=$ID
|
||||||
|
OS_VERSION=$VERSION_ID
|
||||||
|
else
|
||||||
|
echo -e "${RED}无法检测系统类型${NC}"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
ARCH=$(uname -m)
|
||||||
|
case $ARCH in
|
||||||
|
x86_64)
|
||||||
|
ARCH="amd64"
|
||||||
|
;;
|
||||||
|
aarch64|arm64)
|
||||||
|
ARCH="arm64"
|
||||||
|
;;
|
||||||
|
*)
|
||||||
|
echo -e "${RED}不支持的架构: $ARCH${NC}"
|
||||||
|
exit 1
|
||||||
|
;;
|
||||||
|
esac
|
||||||
|
|
||||||
|
echo -e "${BLUE}检测到系统: $OS $OS_VERSION ($ARCH)${NC}"
|
||||||
|
}
|
||||||
|
|
||||||
|
# 安装系统依赖
|
||||||
|
install_dependencies() {
|
||||||
|
echo -e "${BLUE}安装系统依赖...${NC}"
|
||||||
|
|
||||||
|
if [ "$OS" = "ubuntu" ] || [ "$OS" = "debian" ]; then
|
||||||
|
sudo apt-get update -qq
|
||||||
|
sudo apt-get install -y -qq curl wget ping traceroute dnsutils git > /dev/null 2>&1
|
||||||
|
elif [ "$OS" = "centos" ] || [ "$OS" = "rhel" ] || [ "$OS" = "rocky" ] || [ "$OS" = "almalinux" ]; then
|
||||||
|
sudo yum install -y -q curl wget iputils traceroute bind-utils git > /dev/null 2>&1
|
||||||
|
else
|
||||||
|
echo -e "${YELLOW}警告: 未知系统类型，跳过依赖安装${NC}"
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo -e "${GREEN}✓ 依赖安装完成${NC}"
|
||||||
|
}
|
||||||
|
|
||||||
|
# 安装 Go 环境
|
||||||
|
install_go() {
|
||||||
|
echo -e "${BLUE}安装 Go 环境...${NC}"
|
||||||
|
|
||||||
|
if [ "$OS" = "ubuntu" ] || [ "$OS" = "debian" ]; then
|
||||||
|
sudo apt-get update -qq
|
||||||
|
sudo apt-get install -y -qq golang-go > /dev/null 2>&1
|
||||||
|
elif [ "$OS" = "centos" ] || [ "$OS" = "rhel" ] || [ "$OS" = "rocky" ] || [ "$OS" = "almalinux" ]; then
|
||||||
|
sudo yum install -y -q golang > /dev/null 2>&1
|
||||||
|
else
|
||||||
|
echo -e "${YELLOW}无法自动安装 Go，请手动安装: https://golang.org/dl/${NC}"
|
||||||
|
show_build_alternatives
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
if command -v go > /dev/null 2>&1; then
|
||||||
|
GO_VERSION=$(go version 2>/dev/null | head -1)
|
||||||
|
echo -e "${GREEN}✓ Go 安装完成: ${GO_VERSION}${NC}"
|
||||||
|
else
|
||||||
|
echo -e "${RED}Go 安装失败${NC}"
|
||||||
|
show_build_alternatives
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
# 显示替代方案
|
||||||
|
show_build_alternatives() {
|
||||||
|
echo ""
|
||||||
|
echo -e "${YELLOW}═══════════════════════════════════════════════════════════${NC}"
|
||||||
|
echo -e "${YELLOW} 安装失败，请使用以下替代方案:${NC}"
|
||||||
|
echo -e "${YELLOW}═══════════════════════════════════════════════════════════${NC}"
|
||||||
|
echo ""
|
||||||
|
echo -e "${GREEN}手动编译安装:${NC}"
|
||||||
|
echo " git clone https://github.com/${GITHUB_REPO}.git ${SOURCE_DIR}"
|
||||||
|
echo " cd ${SOURCE_DIR}"
|
||||||
|
echo " go build -o agent ./cmd/agent"
|
||||||
|
echo " sudo cp agent /usr/local/bin/linkmaster-node"
|
||||||
|
echo " sudo chmod +x /usr/local/bin/linkmaster-node"
|
||||||
|
echo ""
|
||||||
|
}
|
||||||
|
|
||||||
|
# 从源码编译安装
|
||||||
|
build_from_source() {
|
||||||
|
echo -e "${BLUE}从源码编译安装节点端...${NC}"
|
||||||
|
|
||||||
|
# 检查 Go 环境
|
||||||
|
if ! command -v go > /dev/null 2>&1; then
|
||||||
|
echo -e "${BLUE}未检测到 Go 环境，开始安装...${NC}"
|
||||||
|
install_go
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 检查 Go 版本
|
||||||
|
GO_VERSION=$(go version 2>/dev/null | head -1 || echo "")
|
||||||
|
if [ -z "$GO_VERSION" ]; then
|
||||||
|
echo -e "${RED}无法获取 Go 版本信息${NC}"
|
||||||
|
show_build_alternatives
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo -e "${BLUE}检测到 Go 版本: ${GO_VERSION}${NC}"
|
||||||
|
|
||||||
|
# 如果源码目录已存在，先备份或删除
|
||||||
|
if [ -d "$SOURCE_DIR" ]; then
|
||||||
|
echo -e "${YELLOW}源码目录已存在，将更新代码...${NC}"
|
||||||
|
sudo rm -rf "$SOURCE_DIR"
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 克隆仓库到源码目录
|
||||||
|
echo -e "${BLUE}克隆仓库到 ${SOURCE_DIR}...${NC}"
|
||||||
|
if ! sudo git clone --branch "${GITHUB_BRANCH}" "https://github.com/${GITHUB_REPO}.git" "$SOURCE_DIR" 2>&1; then
|
||||||
|
echo -e "${RED}克隆仓库失败，请检查网络连接和仓库地址${NC}"
|
||||||
|
echo -e "${YELLOW}仓库地址: https://github.com/${GITHUB_REPO}.git${NC}"
|
||||||
|
show_build_alternatives
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 设置目录权限
|
||||||
|
sudo chown -R $USER:$USER "$SOURCE_DIR" 2>/dev/null || true
|
||||||
|
|
||||||
|
cd "$SOURCE_DIR"
|
||||||
|
|
||||||
|
# 下载依赖
|
||||||
|
echo -e "${BLUE}下载 Go 依赖...${NC}"
|
||||||
|
if ! go mod download 2>&1; then
|
||||||
|
echo -e "${RED}下载依赖失败${NC}"
|
||||||
|
show_build_alternatives
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 编译
|
||||||
|
echo -e "${BLUE}编译二进制文件...${NC}"
|
||||||
|
BINARY_PATH="$SOURCE_DIR/agent"
|
||||||
|
if GOOS=linux GOARCH=${ARCH} CGO_ENABLED=0 go build -ldflags="-w -s" -o "$BINARY_PATH" ./cmd/agent 2>&1; then
|
||||||
|
if [ -f "$BINARY_PATH" ] && [ -s "$BINARY_PATH" ]; then
|
||||||
|
echo -e "${GREEN}✓ 编译成功${NC}"
|
||||||
|
else
|
||||||
|
echo -e "${RED}编译失败：未生成二进制文件${NC}"
|
||||||
|
show_build_alternatives
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
else
|
||||||
|
echo -e "${RED}编译失败${NC}"
|
||||||
|
show_build_alternatives
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 复制到安装目录（可选，保留在源码目录供 run.sh 使用）
|
||||||
|
sudo mkdir -p "$INSTALL_DIR"
|
||||||
|
sudo cp "$BINARY_PATH" "$INSTALL_DIR/$BINARY_NAME"
|
||||||
|
sudo chmod +x "$INSTALL_DIR/$BINARY_NAME"
|
||||||
|
|
||||||
|
echo -e "${GREEN}✓ 编译安装完成${NC}"
|
||||||
|
echo -e "${BLUE}源码目录: ${SOURCE_DIR}${NC}"
|
||||||
|
echo -e "${BLUE}二进制文件: ${INSTALL_DIR}/${BINARY_NAME}${NC}"
|
||||||
|
}
|
||||||
|
|
||||||
|
# 创建 systemd 服务
|
||||||
|
create_service() {
|
||||||
|
echo -e "${BLUE}创建 systemd 服务...${NC}"
|
||||||
|
|
||||||
|
# 确保 run.sh 有执行权限
|
||||||
|
sudo chmod +x "$SOURCE_DIR/run.sh"
|
||||||
|
|
||||||
|
sudo tee /etc/systemd/system/${SERVICE_NAME}.service > /dev/null <<EOF
|
||||||
|
[Unit]
|
||||||
|
Description=LinkMaster Node Service
|
||||||
|
After=network.target
|
||||||
|
|
||||||
|
[Service]
|
||||||
|
Type=simple
|
||||||
|
User=root
|
||||||
|
WorkingDirectory=$SOURCE_DIR
|
||||||
|
ExecStart=$SOURCE_DIR/run.sh start
|
||||||
|
Restart=always
|
||||||
|
RestartSec=5
|
||||||
|
Environment="BACKEND_URL=$BACKEND_URL"
|
||||||
|
|
||||||
|
[Install]
|
||||||
|
WantedBy=multi-user.target
|
||||||
|
EOF
|
||||||
|
|
||||||
|
sudo systemctl daemon-reload
|
||||||
|
echo -e "${GREEN}✓ 服务创建完成${NC}"
|
||||||
|
}
|
||||||
|
|
||||||
|
# 启动服务
|
||||||
|
start_service() {
|
||||||
|
echo -e "${BLUE}启动服务...${NC}"
|
||||||
|
|
||||||
|
sudo systemctl enable ${SERVICE_NAME} > /dev/null 2>&1
|
||||||
|
sudo systemctl restart ${SERVICE_NAME}
|
||||||
|
|
||||||
|
# 等待服务启动
|
||||||
|
sleep 3
|
||||||
|
|
||||||
|
# 检查服务状态
|
||||||
|
if sudo systemctl is-active --quiet ${SERVICE_NAME}; then
|
||||||
|
echo -e "${GREEN}✓ 服务启动成功${NC}"
|
||||||
|
else
|
||||||
|
echo -e "${RED}✗ 服务启动失败${NC}"
|
||||||
|
echo -e "${YELLOW}查看日志: sudo journalctl -u ${SERVICE_NAME} -n 50${NC}"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
# 验证安装
|
||||||
|
verify_installation() {
|
||||||
|
echo -e "${BLUE}验证安装...${NC}"
|
||||||
|
|
||||||
|
# 检查进程
|
||||||
|
if pgrep -f "$BINARY_NAME" > /dev/null; then
|
||||||
|
echo -e "${GREEN}✓ 进程运行中${NC}"
|
||||||
|
else
|
||||||
|
echo -e "${YELLOW}⚠ 进程未运行${NC}"
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 检查端口
|
||||||
|
if command -v netstat > /dev/null 2>&1; then
|
||||||
|
if netstat -tlnp 2>/dev/null | grep -q ":2200"; then
|
||||||
|
echo -e "${GREEN}✓ 端口 2200 已监听${NC}"
|
||||||
|
fi
|
||||||
|
elif command -v ss > /dev/null 2>&1; then
|
||||||
|
if ss -tlnp 2>/dev/null | grep -q ":2200"; then
|
||||||
|
echo -e "${GREEN}✓ 端口 2200 已监听${NC}"
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 健康检查
|
||||||
|
sleep 2
|
||||||
|
if curl -sf http://localhost:2200/api/health > /dev/null; then
|
||||||
|
echo -e "${GREEN}✓ 健康检查通过${NC}"
|
||||||
|
else
|
||||||
|
echo -e "${YELLOW}⚠ 健康检查未通过，请稍后重试${NC}"
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
# 主安装流程
|
||||||
|
main() {
|
||||||
|
echo -e "${GREEN}========================================${NC}"
|
||||||
|
echo -e "${GREEN} LinkMaster 节点端安装程序${NC}"
|
||||||
|
echo -e "${GREEN}========================================${NC}"
|
||||||
|
echo ""
|
||||||
|
|
||||||
|
detect_system
|
||||||
|
install_dependencies
|
||||||
|
build_from_source
|
||||||
|
create_service
|
||||||
|
start_service
|
||||||
|
verify_installation
|
||||||
|
|
||||||
|
echo ""
|
||||||
|
echo -e "${GREEN}========================================${NC}"
|
||||||
|
echo -e "${GREEN} 安装完成！${NC}"
|
||||||
|
echo -e "${GREEN}========================================${NC}"
|
||||||
|
echo ""
|
||||||
|
echo -e "${BLUE}服务管理命令:${NC}"
|
||||||
|
echo " 查看状态: sudo systemctl status ${SERVICE_NAME}"
|
||||||
|
echo " 查看日志: sudo journalctl -u ${SERVICE_NAME} -f"
|
||||||
|
echo " 重启服务: sudo systemctl restart ${SERVICE_NAME}"
|
||||||
|
echo " 停止服务: sudo systemctl stop ${SERVICE_NAME}"
|
||||||
|
echo ""
|
||||||
|
echo -e "${BLUE}后端地址: ${BACKEND_URL}${NC}"
|
||||||
|
echo -e "${BLUE}节点端口: 2200${NC}"
|
||||||
|
echo ""
|
||||||
|
echo -e "${YELLOW}重要提示:${NC}"
|
||||||
|
echo " - 节点端直接连接后端服务器，不使用前端代理"
|
||||||
|
echo " - 确保后端地址可访问: curl ${BACKEND_URL}/api/public/nodes/online"
|
||||||
|
}
|
||||||
|
|
||||||
|
# 执行安装
|
||||||
|
main
|
||||||
60
internal/config/config.go
Normal file
60
internal/config/config.go
Normal file
@@ -0,0 +1,60 @@
|
|||||||
|
package config
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
|
||||||
|
"gopkg.in/yaml.v3"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Config struct {
|
||||||
|
Server struct {
|
||||||
|
Port int `yaml:"port"`
|
||||||
|
} `yaml:"server"`
|
||||||
|
|
||||||
|
Backend struct {
|
||||||
|
URL string `yaml:"url"`
|
||||||
|
} `yaml:"backend"`
|
||||||
|
|
||||||
|
Heartbeat struct {
|
||||||
|
Interval int `yaml:"interval"` // 心跳间隔（秒）
|
||||||
|
} `yaml:"heartbeat"`
|
||||||
|
|
||||||
|
Debug bool `yaml:"debug"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func Load() (*Config, error) {
|
||||||
|
cfg := &Config{}
|
||||||
|
|
||||||
|
// 默认配置
|
||||||
|
cfg.Server.Port = 2200
|
||||||
|
cfg.Heartbeat.Interval = 60
|
||||||
|
cfg.Debug = false
|
||||||
|
|
||||||
|
// 从环境变量读取后端URL
|
||||||
|
backendURL := os.Getenv("BACKEND_URL")
|
||||||
|
if backendURL == "" {
|
||||||
|
backendURL = "http://localhost:8080"
|
||||||
|
}
|
||||||
|
cfg.Backend.URL = backendURL
|
||||||
|
|
||||||
|
// 尝试从配置文件读取
|
||||||
|
configPath := os.Getenv("CONFIG_PATH")
|
||||||
|
if configPath == "" {
|
||||||
|
configPath = "config.yaml"
|
||||||
|
}
|
||||||
|
|
||||||
|
if _, err := os.Stat(configPath); err == nil {
|
||||||
|
data, err := os.ReadFile(configPath)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("读取配置文件失败: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := yaml.Unmarshal(data, cfg); err != nil {
|
||||||
|
return nil, fmt.Errorf("解析配置文件失败: %w", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return cfg, nil
|
||||||
|
}
|
||||||
|
|
||||||
143
internal/continuous/ping.go
Normal file
143
internal/continuous/ping.go
Normal file
@@ -0,0 +1,143 @@
|
|||||||
|
package continuous
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"os/exec"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"go.uber.org/zap"
|
||||||
|
)
|
||||||
|
|
||||||
|
type PingTask struct {
|
||||||
|
TaskID string
|
||||||
|
Target string
|
||||||
|
Interval time.Duration
|
||||||
|
MaxDuration time.Duration
|
||||||
|
StartTime time.Time
|
||||||
|
LastRequest time.Time
|
||||||
|
StopCh chan struct{}
|
||||||
|
IsRunning bool
|
||||||
|
mu sync.RWMutex
|
||||||
|
logger *zap.Logger
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewPingTask(taskID, target string, interval, maxDuration time.Duration) *PingTask {
|
||||||
|
logger, _ := zap.NewProduction()
|
||||||
|
return &PingTask{
|
||||||
|
TaskID: taskID,
|
||||||
|
Target: target,
|
||||||
|
Interval: interval,
|
||||||
|
MaxDuration: maxDuration,
|
||||||
|
StartTime: time.Now(),
|
||||||
|
LastRequest: time.Now(),
|
||||||
|
StopCh: make(chan struct{}),
|
||||||
|
IsRunning: true,
|
||||||
|
logger: logger,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *PingTask) Start(ctx context.Context, resultCallback func(result map[string]interface{})) {
|
||||||
|
ticker := time.NewTicker(t.Interval)
|
||||||
|
defer ticker.Stop()
|
||||||
|
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case <-ctx.Done():
|
||||||
|
return
|
||||||
|
case <-t.StopCh:
|
||||||
|
return
|
||||||
|
case <-ticker.C:
|
||||||
|
// 检查是否超过最大运行时长
|
||||||
|
t.mu.RLock()
|
||||||
|
if time.Since(t.StartTime) > t.MaxDuration {
|
||||||
|
t.mu.RUnlock()
|
||||||
|
t.Stop()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
t.mu.RUnlock()
|
||||||
|
|
||||||
|
// 执行ping测试
|
||||||
|
result := t.executePing()
|
||||||
|
if resultCallback != nil {
|
||||||
|
resultCallback(result)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *PingTask) Stop() {
|
||||||
|
t.mu.Lock()
|
||||||
|
defer t.mu.Unlock()
|
||||||
|
if t.IsRunning {
|
||||||
|
t.IsRunning = false
|
||||||
|
close(t.StopCh)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *PingTask) UpdateLastRequest() {
|
||||||
|
t.mu.Lock()
|
||||||
|
defer t.mu.Unlock()
|
||||||
|
t.LastRequest = time.Now()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *PingTask) executePing() map[string]interface{} {
|
||||||
|
cmd := exec.Command("ping", "-c", "4", t.Target)
|
||||||
|
output, err := cmd.CombinedOutput()
|
||||||
|
if err != nil {
|
||||||
|
return map[string]interface{}{
|
||||||
|
"timestamp": time.Now().Unix(),
|
||||||
|
"latency": -1,
|
||||||
|
"success": false,
|
||||||
|
"packet_loss": true,
|
||||||
|
"error": err.Error(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 解析ping输出
|
||||||
|
result := parsePingOutput(string(output))
|
||||||
|
result["timestamp"] = time.Now().Unix()
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
|
||||||
|
func parsePingOutput(output string) map[string]interface{} {
|
||||||
|
result := map[string]interface{}{
|
||||||
|
"latency": 0.0,
|
||||||
|
"success": true,
|
||||||
|
"packet_loss": false,
|
||||||
|
}
|
||||||
|
|
||||||
|
lines := strings.Split(output, "\n")
|
||||||
|
for _, line := range lines {
|
||||||
|
if strings.Contains(line, "packets transmitted") {
|
||||||
|
// 解析丢包率
|
||||||
|
parts := strings.Fields(line)
|
||||||
|
for i, part := range parts {
|
||||||
|
if part == "packet" && i+2 < len(parts) {
|
||||||
|
if loss, err := strconv.ParseFloat(strings.Trim(parts[i+1], "%"), 64); err == nil {
|
||||||
|
result["packet_loss"] = loss > 0
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if strings.Contains(line, "min/avg/max") {
|
||||||
|
// 解析平均延迟
|
||||||
|
parts := strings.Fields(line)
|
||||||
|
for _, part := range parts {
|
||||||
|
if strings.Contains(part, "/") {
|
||||||
|
times := strings.Split(part, "/")
|
||||||
|
if len(times) >= 2 {
|
||||||
|
if avg, err := strconv.ParseFloat(times[1], 64); err == nil {
|
||||||
|
result["latency"] = avg
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
|
||||||
126
internal/continuous/tcping.go
Normal file
126
internal/continuous/tcping.go
Normal file
@@ -0,0 +1,126 @@
|
|||||||
|
package continuous
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"net"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"go.uber.org/zap"
|
||||||
|
)
|
||||||
|
|
||||||
|
type TCPingTask struct {
|
||||||
|
TaskID string
|
||||||
|
Target string
|
||||||
|
Host string
|
||||||
|
Port int
|
||||||
|
Interval time.Duration
|
||||||
|
MaxDuration time.Duration
|
||||||
|
StartTime time.Time
|
||||||
|
LastRequest time.Time
|
||||||
|
StopCh chan struct{}
|
||||||
|
IsRunning bool
|
||||||
|
mu sync.RWMutex
|
||||||
|
logger *zap.Logger
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewTCPingTask(taskID, target string, interval, maxDuration time.Duration) (*TCPingTask, error) {
|
||||||
|
// 解析host:port
|
||||||
|
parts := strings.Split(target, ":")
|
||||||
|
if len(parts) != 2 {
|
||||||
|
return nil, fmt.Errorf("无效的target格式，需要 host:port")
|
||||||
|
}
|
||||||
|
|
||||||
|
host := parts[0]
|
||||||
|
port, err := strconv.Atoi(parts[1])
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("无效的端口: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
logger, _ := zap.NewProduction()
|
||||||
|
return &TCPingTask{
|
||||||
|
TaskID: taskID,
|
||||||
|
Target: target,
|
||||||
|
Host: host,
|
||||||
|
Port: port,
|
||||||
|
Interval: interval,
|
||||||
|
MaxDuration: maxDuration,
|
||||||
|
StartTime: time.Now(),
|
||||||
|
LastRequest: time.Now(),
|
||||||
|
StopCh: make(chan struct{}),
|
||||||
|
IsRunning: true,
|
||||||
|
logger: logger,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *TCPingTask) Start(ctx context.Context, resultCallback func(result map[string]interface{})) {
|
||||||
|
ticker := time.NewTicker(t.Interval)
|
||||||
|
defer ticker.Stop()
|
||||||
|
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case <-ctx.Done():
|
||||||
|
return
|
||||||
|
case <-t.StopCh:
|
||||||
|
return
|
||||||
|
case <-ticker.C:
|
||||||
|
// 检查是否超过最大运行时长
|
||||||
|
t.mu.RLock()
|
||||||
|
if time.Since(t.StartTime) > t.MaxDuration {
|
||||||
|
t.mu.RUnlock()
|
||||||
|
t.Stop()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
t.mu.RUnlock()
|
||||||
|
|
||||||
|
// 执行tcping测试
|
||||||
|
result := t.executeTCPing()
|
||||||
|
if resultCallback != nil {
|
||||||
|
resultCallback(result)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *TCPingTask) Stop() {
|
||||||
|
t.mu.Lock()
|
||||||
|
defer t.mu.Unlock()
|
||||||
|
if t.IsRunning {
|
||||||
|
t.IsRunning = false
|
||||||
|
close(t.StopCh)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *TCPingTask) UpdateLastRequest() {
|
||||||
|
t.mu.Lock()
|
||||||
|
defer t.mu.Unlock()
|
||||||
|
t.LastRequest = time.Now()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *TCPingTask) executeTCPing() map[string]interface{} {
|
||||||
|
start := time.Now()
|
||||||
|
conn, err := net.DialTimeout("tcp", net.JoinHostPort(t.Host, strconv.Itoa(t.Port)), 5*time.Second)
|
||||||
|
latency := time.Since(start).Milliseconds()
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return map[string]interface{}{
|
||||||
|
"timestamp": time.Now().Unix(),
|
||||||
|
"latency": -1,
|
||||||
|
"success": false,
|
||||||
|
"packet_loss": true,
|
||||||
|
"error": err.Error(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
defer conn.Close()
|
||||||
|
|
||||||
|
return map[string]interface{}{
|
||||||
|
"timestamp": time.Now().Unix(),
|
||||||
|
"latency": float64(latency),
|
||||||
|
"success": true,
|
||||||
|
"packet_loss": false,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
315
internal/handler/continuous.go
Normal file
315
internal/handler/continuous.go
Normal file
@@ -0,0 +1,315 @@
|
|||||||
|
package handler
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"context"
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"net"
|
||||||
|
"net/http"
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"linkmaster-node/internal/config"
|
||||||
|
"linkmaster-node/internal/continuous"
|
||||||
|
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
"go.uber.org/zap"
|
||||||
|
)
|
||||||
|
|
||||||
|
var continuousTasks = make(map[string]*ContinuousTask)
|
||||||
|
var taskMutex sync.RWMutex
|
||||||
|
var backendURL string
|
||||||
|
var logger *zap.Logger
|
||||||
|
|
||||||
|
func InitContinuousHandler(cfg *config.Config) {
|
||||||
|
backendURL = cfg.Backend.URL
|
||||||
|
logger, _ = zap.NewProduction()
|
||||||
|
}
|
||||||
|
|
||||||
|
type ContinuousTask struct {
|
||||||
|
TaskID string
|
||||||
|
Type string
|
||||||
|
Target string
|
||||||
|
Interval time.Duration
|
||||||
|
MaxDuration time.Duration
|
||||||
|
StartTime time.Time
|
||||||
|
LastRequest time.Time
|
||||||
|
StopCh chan struct{}
|
||||||
|
IsRunning bool
|
||||||
|
pingTask *continuous.PingTask
|
||||||
|
tcpingTask *continuous.TCPingTask
|
||||||
|
}
|
||||||
|
|
||||||
|
func HandleContinuousStart(c *gin.Context) {
|
||||||
|
var req struct {
|
||||||
|
Type string `json:"type" binding:"required"`
|
||||||
|
Target string `json:"target" binding:"required"`
|
||||||
|
Interval int `json:"interval"` // 秒
|
||||||
|
MaxDuration int `json:"max_duration"` // 分钟
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := c.ShouldBindJSON(&req); err != nil {
|
||||||
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// 生成任务ID
|
||||||
|
taskID := generateTaskID()
|
||||||
|
|
||||||
|
// 设置默认值
|
||||||
|
interval := 10 * time.Second
|
||||||
|
if req.Interval > 0 {
|
||||||
|
interval = time.Duration(req.Interval) * time.Second
|
||||||
|
}
|
||||||
|
|
||||||
|
maxDuration := 60 * time.Minute
|
||||||
|
if req.MaxDuration > 0 {
|
||||||
|
maxDuration = time.Duration(req.MaxDuration) * time.Minute
|
||||||
|
}
|
||||||
|
|
||||||
|
// 创建任务
|
||||||
|
task := &ContinuousTask{
|
||||||
|
TaskID: taskID,
|
||||||
|
Type: req.Type,
|
||||||
|
Target: req.Target,
|
||||||
|
Interval: interval,
|
||||||
|
MaxDuration: maxDuration,
|
||||||
|
StartTime: time.Now(),
|
||||||
|
LastRequest: time.Now(),
|
||||||
|
StopCh: make(chan struct{}),
|
||||||
|
IsRunning: true,
|
||||||
|
}
|
||||||
|
|
||||||
|
// 根据类型创建对应的任务
|
||||||
|
if req.Type == "ping" {
|
||||||
|
pingTask := continuous.NewPingTask(taskID, req.Target, interval, maxDuration)
|
||||||
|
task.pingTask = pingTask
|
||||||
|
} else if req.Type == "tcping" {
|
||||||
|
tcpingTask, err := continuous.NewTCPingTask(taskID, req.Target, interval, maxDuration)
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
task.tcpingTask = tcpingTask
|
||||||
|
} else {
|
||||||
|
c.JSON(http.StatusBadRequest, gin.H{"error": "不支持的持续测试类型"})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
taskMutex.Lock()
|
||||||
|
continuousTasks[taskID] = task
|
||||||
|
taskMutex.Unlock()
|
||||||
|
|
||||||
|
// 启动持续测试goroutine
|
||||||
|
ctx := context.Background()
|
||||||
|
if task.pingTask != nil {
|
||||||
|
go task.pingTask.Start(ctx, func(result map[string]interface{}) {
|
||||||
|
pushResultToBackend(taskID, result)
|
||||||
|
})
|
||||||
|
} else if task.tcpingTask != nil {
|
||||||
|
go task.tcpingTask.Start(ctx, func(result map[string]interface{}) {
|
||||||
|
pushResultToBackend(taskID, result)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
c.JSON(http.StatusOK, gin.H{
|
||||||
|
"task_id": taskID,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func HandleContinuousStop(c *gin.Context) {
|
||||||
|
var req struct {
|
||||||
|
TaskID string `json:"task_id" binding:"required"`
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := c.ShouldBindJSON(&req); err != nil {
|
||||||
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
taskMutex.Lock()
|
||||||
|
task, exists := continuousTasks[req.TaskID]
|
||||||
|
if exists {
|
||||||
|
task.IsRunning = false
|
||||||
|
if task.pingTask != nil {
|
||||||
|
task.pingTask.Stop()
|
||||||
|
}
|
||||||
|
if task.tcpingTask != nil {
|
||||||
|
task.tcpingTask.Stop()
|
||||||
|
}
|
||||||
|
close(task.StopCh)
|
||||||
|
delete(continuousTasks, req.TaskID)
|
||||||
|
}
|
||||||
|
taskMutex.Unlock()
|
||||||
|
|
||||||
|
if !exists {
|
||||||
|
c.JSON(http.StatusNotFound, gin.H{"error": "任务不存在"})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
c.JSON(http.StatusOK, gin.H{"message": "任务已停止"})
|
||||||
|
}
|
||||||
|
|
||||||
|
func HandleContinuousStatus(c *gin.Context) {
|
||||||
|
taskID := c.Query("task_id")
|
||||||
|
if taskID == "" {
|
||||||
|
c.JSON(http.StatusBadRequest, gin.H{"error": "task_id参数缺失"})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
taskMutex.RLock()
|
||||||
|
task, exists := continuousTasks[taskID]
|
||||||
|
if exists {
|
||||||
|
// 更新LastRequest时间
|
||||||
|
task.LastRequest = time.Now()
|
||||||
|
if task.pingTask != nil {
|
||||||
|
task.pingTask.UpdateLastRequest()
|
||||||
|
}
|
||||||
|
if task.tcpingTask != nil {
|
||||||
|
task.tcpingTask.UpdateLastRequest()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
taskMutex.RUnlock()
|
||||||
|
|
||||||
|
if !exists {
|
||||||
|
c.JSON(http.StatusNotFound, gin.H{"error": "任务不存在"})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
c.JSON(http.StatusOK, gin.H{
|
||||||
|
"task_id": task.TaskID,
|
||||||
|
"is_running": task.IsRunning,
|
||||||
|
"start_time": task.StartTime,
|
||||||
|
"last_request": task.LastRequest,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func pushResultToBackend(taskID string, result map[string]interface{}) {
|
||||||
|
// 推送结果到后端
|
||||||
|
url := fmt.Sprintf("%s/api/public/node/continuous/result", backendURL)
|
||||||
|
|
||||||
|
// 获取本机IP
|
||||||
|
nodeIP := getLocalIP()
|
||||||
|
|
||||||
|
data := map[string]interface{}{
|
||||||
|
"task_id": taskID,
|
||||||
|
"node_ip": nodeIP,
|
||||||
|
"result": result,
|
||||||
|
}
|
||||||
|
|
||||||
|
jsonData, err := json.Marshal(data)
|
||||||
|
if err != nil {
|
||||||
|
logger.Error("序列化结果失败", zap.Error(err))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
req, err := http.NewRequest("POST", url, bytes.NewBuffer(jsonData))
|
||||||
|
if err != nil {
|
||||||
|
logger.Error("创建请求失败", zap.Error(err))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
req.Header.Set("Content-Type", "application/json")
|
||||||
|
|
||||||
|
client := &http.Client{Timeout: 5 * time.Second}
|
||||||
|
resp, err := client.Do(req)
|
||||||
|
if err != nil {
|
||||||
|
logger.Warn("推送结果失败", zap.Error(err))
|
||||||
|
// 如果推送失败，停止任务
|
||||||
|
taskMutex.Lock()
|
||||||
|
if task, exists := continuousTasks[taskID]; exists {
|
||||||
|
task.IsRunning = false
|
||||||
|
if task.pingTask != nil {
|
||||||
|
task.pingTask.Stop()
|
||||||
|
}
|
||||||
|
if task.tcpingTask != nil {
|
||||||
|
task.tcpingTask.Stop()
|
||||||
|
}
|
||||||
|
delete(continuousTasks, taskID)
|
||||||
|
}
|
||||||
|
taskMutex.Unlock()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
if resp.StatusCode != http.StatusOK {
|
||||||
|
logger.Warn("推送结果失败", zap.Int("status", resp.StatusCode))
|
||||||
|
// 如果推送失败，停止任务
|
||||||
|
taskMutex.Lock()
|
||||||
|
if task, exists := continuousTasks[taskID]; exists {
|
||||||
|
task.IsRunning = false
|
||||||
|
if task.pingTask != nil {
|
||||||
|
task.pingTask.Stop()
|
||||||
|
}
|
||||||
|
if task.tcpingTask != nil {
|
||||||
|
task.tcpingTask.Stop()
|
||||||
|
}
|
||||||
|
delete(continuousTasks, taskID)
|
||||||
|
}
|
||||||
|
taskMutex.Unlock()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func getLocalIP() string {
|
||||||
|
// 简化实现：返回第一个非回环IP
|
||||||
|
// 实际应该获取外网IP
|
||||||
|
addrs, err := net.InterfaceAddrs()
|
||||||
|
if err != nil {
|
||||||
|
return "127.0.0.1"
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, addr := range addrs {
|
||||||
|
if ipNet, ok := addr.(*net.IPNet); ok && !ipNet.IP.IsLoopback() {
|
||||||
|
if ipNet.IP.To4() != nil {
|
||||||
|
return ipNet.IP.String()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return "127.0.0.1"
|
||||||
|
}
|
||||||
|
|
||||||
|
func generateTaskID() string {
|
||||||
|
return fmt.Sprintf("task_%d", time.Now().UnixNano())
|
||||||
|
}
|
||||||
|
|
||||||
|
// 定期清理超时任务
|
||||||
|
func StartTaskCleanup() {
|
||||||
|
ticker := time.NewTicker(1 * time.Minute)
|
||||||
|
go func() {
|
||||||
|
for range ticker.C {
|
||||||
|
now := time.Now()
|
||||||
|
taskMutex.Lock()
|
||||||
|
for taskID, task := range continuousTasks {
|
||||||
|
// 检查最大运行时长
|
||||||
|
if now.Sub(task.StartTime) > task.MaxDuration {
|
||||||
|
logger.Info("任务达到最大运行时长，自动停止", zap.String("task_id", taskID))
|
||||||
|
task.IsRunning = false
|
||||||
|
if task.pingTask != nil {
|
||||||
|
task.pingTask.Stop()
|
||||||
|
}
|
||||||
|
if task.tcpingTask != nil {
|
||||||
|
task.tcpingTask.Stop()
|
||||||
|
}
|
||||||
|
delete(continuousTasks, taskID)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
// 检查无客户端连接（30分钟无请求）
|
||||||
|
if now.Sub(task.LastRequest) > 30*time.Minute {
|
||||||
|
logger.Info("任务无客户端连接，自动停止", zap.String("task_id", taskID))
|
||||||
|
task.IsRunning = false
|
||||||
|
if task.pingTask != nil {
|
||||||
|
task.pingTask.Stop()
|
||||||
|
}
|
||||||
|
if task.tcpingTask != nil {
|
||||||
|
task.tcpingTask.Stop()
|
||||||
|
}
|
||||||
|
delete(continuousTasks, taskID)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
taskMutex.Unlock()
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
}
|
||||||
|
|
||||||
45
internal/handler/dns.go
Normal file
45
internal/handler/dns.go
Normal file
@@ -0,0 +1,45 @@
|
|||||||
|
package handler
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
)
|
||||||
|
|
||||||
|
func handleDns(c *gin.Context, url string, params map[string]interface{}) {
|
||||||
|
// 执行DNS查询
|
||||||
|
start := time.Now()
|
||||||
|
ips, err := net.LookupIP(url)
|
||||||
|
lookupTime := time.Since(start).Milliseconds()
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(200, gin.H{
|
||||||
|
"type": "ceDns",
|
||||||
|
"url": url,
|
||||||
|
"error": err.Error(),
|
||||||
|
})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// 格式化IP列表
|
||||||
|
ipList := make([]map[string]interface{}, 0)
|
||||||
|
for _, ip := range ips {
|
||||||
|
ipType := "A"
|
||||||
|
if ip.To4() == nil {
|
||||||
|
ipType = "AAAA"
|
||||||
|
}
|
||||||
|
ipList = append(ipList, map[string]interface{}{
|
||||||
|
"type": ipType,
|
||||||
|
"ip": ip.String(),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
c.JSON(200, gin.H{
|
||||||
|
"type": "ceDns",
|
||||||
|
"url": url,
|
||||||
|
"ips": ipList,
|
||||||
|
"lookup_time": lookupTime,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
84
internal/handler/findping.go
Normal file
84
internal/handler/findping.go
Normal file
@@ -0,0 +1,84 @@
|
|||||||
|
package handler
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net"
|
||||||
|
"os/exec"
|
||||||
|
"sync"
|
||||||
|
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
)
|
||||||
|
|
||||||
|
func handleFindPing(c *gin.Context, url string, params map[string]interface{}) {
|
||||||
|
// url应该是CIDR格式，如 8.8.8.0/24
|
||||||
|
cidr := url
|
||||||
|
if cidrParam, ok := params["cidr"].(string); ok && cidrParam != "" {
|
||||||
|
cidr = cidrParam
|
||||||
|
}
|
||||||
|
|
||||||
|
// 解析CIDR
|
||||||
|
_, ipNet, err := net.ParseCIDR(cidr)
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(200, gin.H{
|
||||||
|
"type": "ceFindPing",
|
||||||
|
"error": "无效的CIDR格式",
|
||||||
|
})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// 生成IP列表
|
||||||
|
var ipList []string
|
||||||
|
for ip := ipNet.IP.Mask(ipNet.Mask); ipNet.Contains(ip); incIP(ip) {
|
||||||
|
ipList = append(ipList, ip.String())
|
||||||
|
}
|
||||||
|
|
||||||
|
// 移除网络地址和广播地址
|
||||||
|
if len(ipList) > 2 {
|
||||||
|
ipList = ipList[1 : len(ipList)-1]
|
||||||
|
}
|
||||||
|
|
||||||
|
// 并发ping测试
|
||||||
|
var wg sync.WaitGroup
|
||||||
|
var mu sync.Mutex
|
||||||
|
aliveIPs := make([]string, 0)
|
||||||
|
|
||||||
|
// 限制并发数
|
||||||
|
semaphore := make(chan struct{}, 50)
|
||||||
|
|
||||||
|
for _, ip := range ipList {
|
||||||
|
wg.Add(1)
|
||||||
|
semaphore <- struct{}{}
|
||||||
|
go func(ipAddr string) {
|
||||||
|
defer wg.Done()
|
||||||
|
defer func() { <-semaphore }()
|
||||||
|
|
||||||
|
// 执行ping（只ping一次，快速检测）
|
||||||
|
cmd := exec.Command("ping", "-c", "1", "-W", "1", ipAddr)
|
||||||
|
err := cmd.Run()
|
||||||
|
if err == nil {
|
||||||
|
mu.Lock()
|
||||||
|
aliveIPs = append(aliveIPs, ipAddr)
|
||||||
|
mu.Unlock()
|
||||||
|
}
|
||||||
|
}(ip)
|
||||||
|
}
|
||||||
|
|
||||||
|
wg.Wait()
|
||||||
|
|
||||||
|
c.JSON(200, gin.H{
|
||||||
|
"type": "ceFindPing",
|
||||||
|
"cidr": cidr,
|
||||||
|
"alive_ips": aliveIPs,
|
||||||
|
"alive_count": len(aliveIPs),
|
||||||
|
"total_ips": len(ipList),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func incIP(ip net.IP) {
|
||||||
|
for j := len(ip) - 1; j >= 0; j-- {
|
||||||
|
ip[j]++
|
||||||
|
if ip[j] > 0 {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
32
internal/handler/get.go
Normal file
32
internal/handler/get.go
Normal file
@@ -0,0 +1,32 @@
|
|||||||
|
package handler
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/http"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
)
|
||||||
|
|
||||||
|
func handleGet(c *gin.Context, url string, params map[string]interface{}) {
|
||||||
|
// TODO: 实现HTTP GET测试
|
||||||
|
// 这里先返回一个简单的响应
|
||||||
|
c.JSON(http.StatusOK, gin.H{
|
||||||
|
"type": "ceGet",
|
||||||
|
"url": url,
|
||||||
|
"statuscode": 200,
|
||||||
|
"totaltime": time.Since(time.Now()).Milliseconds(),
|
||||||
|
"response": "OK",
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func handlePost(c *gin.Context, url string, params map[string]interface{}) {
|
||||||
|
// TODO: 实现HTTP POST测试
|
||||||
|
c.JSON(http.StatusOK, gin.H{
|
||||||
|
"type": "cePost",
|
||||||
|
"url": url,
|
||||||
|
"statuscode": 200,
|
||||||
|
"totaltime": time.Since(time.Now()).Milliseconds(),
|
||||||
|
"response": "OK",
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
85
internal/handler/ping.go
Normal file
85
internal/handler/ping.go
Normal file
@@ -0,0 +1,85 @@
|
|||||||
|
package handler
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net"
|
||||||
|
"os/exec"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
)
|
||||||
|
|
||||||
|
func handlePing(c *gin.Context, url string, params map[string]interface{}) {
|
||||||
|
// 执行ping命令
|
||||||
|
cmd := exec.Command("ping", "-c", "4", url)
|
||||||
|
output, err := cmd.CombinedOutput()
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(200, gin.H{
|
||||||
|
"type": "cePing",
|
||||||
|
"url": url,
|
||||||
|
"error": err.Error(),
|
||||||
|
})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// 解析ping输出
|
||||||
|
result := parsePingOutput(string(output), url)
|
||||||
|
c.JSON(200, result)
|
||||||
|
}
|
||||||
|
|
||||||
|
func parsePingOutput(output, url string) map[string]interface{} {
|
||||||
|
result := map[string]interface{}{
|
||||||
|
"type": "cePing",
|
||||||
|
"url": url,
|
||||||
|
"ip": "",
|
||||||
|
}
|
||||||
|
|
||||||
|
// 解析IP地址
|
||||||
|
lines := strings.Split(output, "\n")
|
||||||
|
for _, line := range lines {
|
||||||
|
if strings.Contains(line, "PING") {
|
||||||
|
// 提取IP地址
|
||||||
|
parts := strings.Fields(line)
|
||||||
|
for _, part := range parts {
|
||||||
|
if net.ParseIP(part) != nil {
|
||||||
|
result["ip"] = part
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if strings.Contains(line, "packets transmitted") {
|
||||||
|
// 解析丢包率
|
||||||
|
parts := strings.Fields(line)
|
||||||
|
for i, part := range parts {
|
||||||
|
if part == "packet" && i+2 < len(parts) {
|
||||||
|
if loss, err := strconv.ParseFloat(strings.Trim(parts[i+1], "%"), 64); err == nil {
|
||||||
|
result["packets_losrat"] = loss
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if strings.Contains(line, "min/avg/max") {
|
||||||
|
// 解析延迟统计
|
||||||
|
parts := strings.Fields(line)
|
||||||
|
for _, part := range parts {
|
||||||
|
if strings.Contains(part, "/") {
|
||||||
|
times := strings.Split(part, "/")
|
||||||
|
if len(times) >= 3 {
|
||||||
|
if min, err := strconv.ParseFloat(times[0], 64); err == nil {
|
||||||
|
result["time_min"] = min
|
||||||
|
}
|
||||||
|
if avg, err := strconv.ParseFloat(times[1], 64); err == nil {
|
||||||
|
result["time_avg"] = avg
|
||||||
|
}
|
||||||
|
if max, err := strconv.ParseFloat(times[2], 64); err == nil {
|
||||||
|
result["time_max"] = max
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
|
||||||
59
internal/handler/socket.go
Normal file
59
internal/handler/socket.go
Normal file
@@ -0,0 +1,59 @@
|
|||||||
|
package handler
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
)
|
||||||
|
|
||||||
|
func handleSocket(c *gin.Context, url string, params map[string]interface{}) {
|
||||||
|
// 解析host:port格式
|
||||||
|
parts := strings.Split(url, ":")
|
||||||
|
if len(parts) != 2 {
|
||||||
|
c.JSON(200, gin.H{
|
||||||
|
"type": "ceSocket",
|
||||||
|
"url": url,
|
||||||
|
"error": "格式错误，需要 host:port",
|
||||||
|
})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
host := parts[0]
|
||||||
|
portStr := parts[1]
|
||||||
|
port, err := strconv.Atoi(portStr)
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(200, gin.H{
|
||||||
|
"type": "ceSocket",
|
||||||
|
"url": url,
|
||||||
|
"error": "端口格式错误",
|
||||||
|
})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// 执行TCP连接测试
|
||||||
|
conn, err := net.DialTimeout("tcp", net.JoinHostPort(host, portStr), 5*time.Second)
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(200, gin.H{
|
||||||
|
"type": "ceSocket",
|
||||||
|
"url": url,
|
||||||
|
"host": host,
|
||||||
|
"port": port,
|
||||||
|
"result": "false",
|
||||||
|
"error": err.Error(),
|
||||||
|
})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer conn.Close()
|
||||||
|
|
||||||
|
c.JSON(200, gin.H{
|
||||||
|
"type": "ceSocket",
|
||||||
|
"url": url,
|
||||||
|
"host": host,
|
||||||
|
"port": port,
|
||||||
|
"result": "true",
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
63
internal/handler/tcping.go
Normal file
63
internal/handler/tcping.go
Normal file
@@ -0,0 +1,63 @@
|
|||||||
|
package handler
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
)
|
||||||
|
|
||||||
|
func handleTCPing(c *gin.Context, url string, params map[string]interface{}) {
|
||||||
|
// 解析host:port格式
|
||||||
|
parts := strings.Split(url, ":")
|
||||||
|
if len(parts) != 2 {
|
||||||
|
c.JSON(200, gin.H{
|
||||||
|
"type": "ceTCPing",
|
||||||
|
"url": url,
|
||||||
|
"error": "格式错误，需要 host:port",
|
||||||
|
})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
host := parts[0]
|
||||||
|
portStr := parts[1]
|
||||||
|
port, err := strconv.Atoi(portStr)
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(200, gin.H{
|
||||||
|
"type": "ceTCPing",
|
||||||
|
"url": url,
|
||||||
|
"error": "端口格式错误",
|
||||||
|
})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// 执行TCP连接测试
|
||||||
|
start := time.Now()
|
||||||
|
conn, err := net.DialTimeout("tcp", net.JoinHostPort(host, portStr), 5*time.Second)
|
||||||
|
latency := time.Since(start).Milliseconds()
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(200, gin.H{
|
||||||
|
"type": "ceTCPing",
|
||||||
|
"url": url,
|
||||||
|
"host": host,
|
||||||
|
"port": port,
|
||||||
|
"latency": -1,
|
||||||
|
"error": err.Error(),
|
||||||
|
})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer conn.Close()
|
||||||
|
|
||||||
|
c.JSON(200, gin.H{
|
||||||
|
"type": "ceTCPing",
|
||||||
|
"url": url,
|
||||||
|
"host": host,
|
||||||
|
"port": port,
|
||||||
|
"latency": latency,
|
||||||
|
"success": true,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
49
internal/handler/test.go
Normal file
49
internal/handler/test.go
Normal file
@@ -0,0 +1,49 @@
|
|||||||
|
package handler
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/http"
|
||||||
|
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
)
|
||||||
|
|
||||||
|
// HandleTest 统一测试接口
|
||||||
|
func HandleTest(c *gin.Context) {
|
||||||
|
var req struct {
|
||||||
|
Type string `json:"type" binding:"required"`
|
||||||
|
URL string `json:"url" binding:"required"`
|
||||||
|
Params map[string]interface{} `json:"params"`
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := c.ShouldBindJSON(&req); err != nil {
|
||||||
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// 根据类型分发到不同的处理器
|
||||||
|
switch req.Type {
|
||||||
|
case "ceGet":
|
||||||
|
handleGet(c, req.URL, req.Params)
|
||||||
|
case "cePost":
|
||||||
|
handlePost(c, req.URL, req.Params)
|
||||||
|
case "cePing":
|
||||||
|
handlePing(c, req.URL, req.Params)
|
||||||
|
case "ceDns":
|
||||||
|
handleDns(c, req.URL, req.Params)
|
||||||
|
case "ceTrace":
|
||||||
|
handleTrace(c, req.URL, req.Params)
|
||||||
|
case "ceSocket":
|
||||||
|
handleSocket(c, req.URL, req.Params)
|
||||||
|
case "ceTCPing":
|
||||||
|
handleTCPing(c, req.URL, req.Params)
|
||||||
|
case "ceFindPing":
|
||||||
|
handleFindPing(c, req.URL, req.Params)
|
||||||
|
default:
|
||||||
|
c.JSON(http.StatusBadRequest, gin.H{"error": "不支持的测试类型"})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// HandleHealth 健康检查
|
||||||
|
func HandleHealth(c *gin.Context) {
|
||||||
|
c.JSON(http.StatusOK, gin.H{"status": "ok"})
|
||||||
|
}
|
||||||
|
|
||||||
39
internal/handler/trace.go
Normal file
39
internal/handler/trace.go
Normal file
@@ -0,0 +1,39 @@
|
|||||||
|
package handler
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os/exec"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
)
|
||||||
|
|
||||||
|
func handleTrace(c *gin.Context, url string, params map[string]interface{}) {
|
||||||
|
// 执行traceroute命令
|
||||||
|
cmd := exec.Command("traceroute", url)
|
||||||
|
output, err := cmd.CombinedOutput()
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(200, gin.H{
|
||||||
|
"type": "ceTrace",
|
||||||
|
"url": url,
|
||||||
|
"error": err.Error(),
|
||||||
|
})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// 解析输出
|
||||||
|
lines := strings.Split(string(output), "\n")
|
||||||
|
traceResult := make([]string, 0)
|
||||||
|
for _, line := range lines {
|
||||||
|
line = strings.TrimSpace(line)
|
||||||
|
if line != "" {
|
||||||
|
traceResult = append(traceResult, line)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
c.JSON(200, gin.H{
|
||||||
|
"type": "ceTrace",
|
||||||
|
"url": url,
|
||||||
|
"trace_result": traceResult,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
103
internal/heartbeat/reporter.go
Normal file
103
internal/heartbeat/reporter.go
Normal file
@@ -0,0 +1,103 @@
|
|||||||
|
package heartbeat
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"net"
|
||||||
|
"net/http"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"linkmaster-node/internal/config"
|
||||||
|
|
||||||
|
"go.uber.org/zap"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Reporter struct {
|
||||||
|
cfg *config.Config
|
||||||
|
client *http.Client
|
||||||
|
logger *zap.Logger
|
||||||
|
stopCh chan struct{}
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewReporter(cfg *config.Config) *Reporter {
|
||||||
|
logger, _ := zap.NewProduction()
|
||||||
|
return &Reporter{
|
||||||
|
cfg: cfg,
|
||||||
|
client: &http.Client{
|
||||||
|
Timeout: 10 * time.Second,
|
||||||
|
},
|
||||||
|
logger: logger,
|
||||||
|
stopCh: make(chan struct{}),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *Reporter) Start(ctx context.Context) {
|
||||||
|
ticker := time.NewTicker(time.Duration(r.cfg.Heartbeat.Interval) * time.Second)
|
||||||
|
defer ticker.Stop()
|
||||||
|
|
||||||
|
// 立即发送一次心跳
|
||||||
|
r.sendHeartbeat()
|
||||||
|
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case <-ctx.Done():
|
||||||
|
return
|
||||||
|
case <-r.stopCh:
|
||||||
|
return
|
||||||
|
case <-ticker.C:
|
||||||
|
r.sendHeartbeat()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *Reporter) Stop() {
|
||||||
|
close(r.stopCh)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *Reporter) sendHeartbeat() {
|
||||||
|
// 获取本机IP
|
||||||
|
localIP := getLocalIP()
|
||||||
|
|
||||||
|
// 发送心跳（使用Form格式，兼容旧接口）
|
||||||
|
url := fmt.Sprintf("%s/api/node/heartbeat", r.cfg.Backend.URL)
|
||||||
|
req, err := http.NewRequest("POST", url, bytes.NewBufferString(fmt.Sprintf("realNodeIP=%s&type=pingServer", localIP)))
|
||||||
|
if err != nil {
|
||||||
|
r.logger.Error("创建心跳请求失败", zap.Error(err))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
|
||||||
|
|
||||||
|
resp, err := r.client.Do(req)
|
||||||
|
if err != nil {
|
||||||
|
r.logger.Warn("发送心跳失败", zap.Error(err))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
if resp.StatusCode == http.StatusOK {
|
||||||
|
r.logger.Debug("心跳发送成功", zap.String("ip", localIP))
|
||||||
|
} else {
|
||||||
|
r.logger.Warn("心跳发送失败", zap.Int("status", resp.StatusCode))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func getLocalIP() string {
|
||||||
|
// 获取第一个非回环IP
|
||||||
|
addrs, err := net.InterfaceAddrs()
|
||||||
|
if err != nil {
|
||||||
|
return "127.0.0.1"
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, addr := range addrs {
|
||||||
|
if ipNet, ok := addr.(*net.IPNet); ok && !ipNet.IP.IsLoopback() {
|
||||||
|
if ipNet.IP.To4() != nil {
|
||||||
|
return ipNet.IP.String()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return "127.0.0.1"
|
||||||
|
}
|
||||||
|
|
||||||
25
internal/recovery/recovery.go
Normal file
25
internal/recovery/recovery.go
Normal file
@@ -0,0 +1,25 @@
|
|||||||
|
package recovery
|
||||||
|
|
||||||
|
import (
|
||||||
|
"runtime/debug"
|
||||||
|
|
||||||
|
"go.uber.org/zap"
|
||||||
|
)
|
||||||
|
|
||||||
|
var logger *zap.Logger
|
||||||
|
|
||||||
|
func Init() {
|
||||||
|
// 初始化logger（这里简化处理，实际应该从外部传入）
|
||||||
|
logger, _ = zap.NewProduction()
|
||||||
|
}
|
||||||
|
|
||||||
|
// Recover 恢复panic
|
||||||
|
func Recover() {
|
||||||
|
if r := recover(); r != nil {
|
||||||
|
logger.Error("发生panic",
|
||||||
|
zap.Any("panic", r),
|
||||||
|
zap.String("stack", string(debug.Stack())),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
72
internal/server/http.go
Normal file
72
internal/server/http.go
Normal file
@@ -0,0 +1,72 @@
|
|||||||
|
package server
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"net/http"
|
||||||
|
|
||||||
|
"linkmaster-node/internal/config"
|
||||||
|
"linkmaster-node/internal/handler"
|
||||||
|
"linkmaster-node/internal/recovery"
|
||||||
|
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
"go.uber.org/zap"
|
||||||
|
)
|
||||||
|
|
||||||
|
type HTTPServer struct {
|
||||||
|
server *http.Server
|
||||||
|
logger *zap.Logger
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewHTTPServer(cfg *config.Config) *HTTPServer {
|
||||||
|
if !cfg.Debug {
|
||||||
|
gin.SetMode(gin.ReleaseMode)
|
||||||
|
}
|
||||||
|
|
||||||
|
router := gin.New()
|
||||||
|
router.Use(gin.Recovery())
|
||||||
|
router.Use(recoveryMiddleware)
|
||||||
|
|
||||||
|
// 初始化持续测试处理器
|
||||||
|
handler.InitContinuousHandler(cfg)
|
||||||
|
|
||||||
|
// 启动任务清理goroutine
|
||||||
|
handler.StartTaskCleanup()
|
||||||
|
|
||||||
|
// 注册路由
|
||||||
|
api := router.Group("/api")
|
||||||
|
{
|
||||||
|
api.POST("/test", handler.HandleTest)
|
||||||
|
api.POST("/continuous/start", handler.HandleContinuousStart)
|
||||||
|
api.POST("/continuous/stop", handler.HandleContinuousStop)
|
||||||
|
api.GET("/continuous/status", handler.HandleContinuousStatus)
|
||||||
|
api.GET("/health", handler.HandleHealth)
|
||||||
|
}
|
||||||
|
|
||||||
|
server := &http.Server{
|
||||||
|
Addr: fmt.Sprintf(":%d", cfg.Server.Port),
|
||||||
|
Handler: router,
|
||||||
|
}
|
||||||
|
|
||||||
|
logger, _ := zap.NewProduction()
|
||||||
|
|
||||||
|
return &HTTPServer{
|
||||||
|
server: server,
|
||||||
|
logger: logger,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *HTTPServer) Start() error {
|
||||||
|
s.logger.Info("HTTP服务器启动", zap.String("addr", s.server.Addr))
|
||||||
|
return s.server.ListenAndServe()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *HTTPServer) Shutdown(ctx context.Context) error {
|
||||||
|
return s.server.Shutdown(ctx)
|
||||||
|
}
|
||||||
|
|
||||||
|
func recoveryMiddleware(c *gin.Context) {
|
||||||
|
defer recovery.Recover()
|
||||||
|
c.Next()
|
||||||
|
}
|
||||||
|
|
||||||
5
node.log
Normal file
5
node.log
Normal file
@@ -0,0 +1,5 @@
|
|||||||
|
{"level":"info","ts":1763025207.120181,"caller":"agent/main.go:35","msg":"节点服务启动","version":"1.0.0"}
|
||||||
|
{"level":"info","ts":1763025207.1208231,"caller":"server/http.go:60","msg":"HTTP服务器启动","addr":":2200"}
|
||||||
|
{"level":"info","ts":1763653448.720011,"caller":"agent/main.go:57","msg":"收到停止信号，正在关闭服务..."}
|
||||||
|
{"level":"fatal","ts":1763653448.720453,"caller":"agent/main.go:48","msg":"HTTP服务器启动失败","error":"http: Server closed","stacktrace":"main.main.func1\n\t/Users/yoyo/Desktop/newLinkMaster/node/cmd/agent/main.go:48"}
|
||||||
|
{"level":"info","ts":1763653448.720591,"caller":"agent/main.go:66","msg":"服务已关闭"}
|
||||||
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