diff --git a/README.md b/README.md
index 09c17bd..01dc78d 100644
--- a/README.md
+++ b/README.md
@@ -322,10 +322,10 @@ restart() {
- Agent 连接 Dashboard 域名开启 Cloudflare CDN
-根据 Cloudflare gRPC 的要求：gRPC 服务必须侦听 443 端口 且必须支持 TLS 和 HTTP/2。我们可以使用 nginx 反向代理 gRPC 并配置 SSL/TLS 证书。
+ 反向代理 gRPC 端口（支持 Cloudflare CDN）
+使用 Nginx 或者 Caddy 反向代理 gRPC
-- nginx 配置，比如 Agent 连接 Dashboard 的域名为 ip-to-dashboard.nai.ba，为 nginx 添加如下配置，然后重新启动 nginx 或者重新加载配置文件。
+- Nginx 配置
```nginx
server {
@@ -339,18 +339,44 @@ server {
underscores_in_headers on;
location / {
+ grpc_read_timeout 300s;
+ grpc_send_timeout 300s;
grpc_pass grpc://localhost:5555;
}
}
```
-- Agent 端配置，编辑 `/etc/systemd/system/nezha-agent.service`，在 `ExecStart=` 这一行的末尾加上 `--tls`，然后重启 nezha-agent.service。例如：
+- Caddy 配置
-```bash
-ExecStart=/opt/nezha/agent/nezha-agent -s ip-to-dashboard.nai.ba:443 -p xxxxxx --tls
+```Caddyfile
+ip-to-dashboard.nai.ba:443 {
+ reverse_proxy {
+ to localhost:5555
+ transport http {
+ versions h2c 2
+ }
+ }
+}
```
-- 在 Cloudflare 中将对应的域名解析设置橙色云开启CDN，并在网络选项中启用gRPC。
+
+Dashboard 面板端配置
+
+- 首先登录面板进入管理后台 打开设置页面，在 `未接入CDN的面板服务器域名/IP` 中填入上一步在 Nginx 或 Caddy 中配置的域名 比如 `ip-to-dashboard.nai.ba` ，并保存。
+- 然后在面板服务器中，打开 /opt/nezha/dashboard/data/config.yaml 文件，将 `proxygrpcport` 修改为 Nginx 或 Caddy 监听的端口，比如上一步设置的 `443` ；因为我们在 Nginx 或 Caddy 中开启了 SSL/TLS，所以需要将 `tls` 设置为 `true` ；修改完成后重启面板。
+
+
+Agent 端配置
+
+- 登录面板管理后台，复制一键安装命令，在对应的服务器上面执行一键安装命令重新安装 agent 端即可。
+
+
+开启 Cloudflare CDN（可选）
+
+根据 Cloudflare gRPC 的要求：gRPC 服务必须侦听 443 端口 且必须支持 TLS 和 HTTP/2。
+所以如果需要开启CDN，必须在配置 Nginx 或者 Caddy 反向代理 gRPC 时使用 443 端口，并配置证书（Caddy 会自动申请并配置证书）。
+
+- 登录 Cloudflare，选择使用的域名。打开 `网络` 选项将 `gRPC` 开关打开，打开 `DNS` 选项，找到 Nginx 或 Caddy 反代 gRPC 配置的域名的解析记录，打开橙色云启用CDN。
diff --git a/cmd/agent/main.go b/cmd/agent/main.go
index 06a6979..7ddd9f7 100644
--- a/cmd/agent/main.go
+++ b/cmd/agent/main.go
@@ -218,6 +218,8 @@ func doTask(task *pb.Task) {
handleCommandTask(task, &result)
case model.TaskTypeUpgrade:
handleUpgradeTask(task, &result)
+ case model.TaskTypeKeepalive:
+ return
default:
println("不支持的任务：", task)
}
diff --git a/cmd/dashboard/main.go b/cmd/dashboard/main.go
index bbb83b9..9645944 100644
--- a/cmd/dashboard/main.go
+++ b/cmd/dashboard/main.go
@@ -193,6 +193,7 @@ func main() {
go rpc.ServeRPC(dao.Conf.GRPCPort)
serviceSentinelDispatchBus := make(chan model.Monitor)
go rpc.DispatchTask(serviceSentinelDispatchBus)
+ go rpc.DispatchKeepalive()
go dao.AlertSentinelStart()
dao.NewServiceSentinel(serviceSentinelDispatchBus)
srv := controller.ServeWeb(dao.Conf.HTTPPort)
diff --git a/cmd/dashboard/rpc/rpc.go b/cmd/dashboard/rpc/rpc.go
index 2e52a5d..9dcb9f0 100644
--- a/cmd/dashboard/rpc/rpc.go
+++ b/cmd/dashboard/rpc/rpc.go
@@ -57,3 +57,17 @@ func DispatchTask(serviceSentinelDispatchBus <-chan model.Monitor) {
dao.SortedServerLock.RUnlock()
}
}
+
+func DispatchKeepalive() {
+ dao.Cron.AddFunc("@every 60s", func() {
+ dao.SortedServerLock.RLock()
+ defer dao.SortedServerLock.RUnlock()
+ for i := 0; i < len(dao.SortedServerList); i++ {
+ if dao.SortedServerList[i] == nil || dao.SortedServerList[i].TaskStream == nil || dao.SortedServerList[i].TaskStream.Context().Err() != nil {
+ continue
+ }
+
+ dao.SortedServerList[i].TaskStream.Send(&pb.Task{Type: model.TaskTypeKeepalive})
+ }
+ })
+}
diff --git a/model/config.go b/model/config.go
index 7224b89..3634636 100644
--- a/model/config.go
+++ b/model/config.go
@@ -39,6 +39,8 @@ type Config struct {
GRPCPort uint
GRPCHost string
EnableIPChangeNotification bool
+ ProxyGRPCPort uint
+ TLS bool
// IP变更提醒
Cover uint8 // 覆盖范围
diff --git a/model/monitor.go b/model/monitor.go
index daf26c3..39b2a7c 100644
--- a/model/monitor.go
+++ b/model/monitor.go
@@ -17,6 +17,7 @@ const (
TaskTypeCommand
TaskTypeTerminal
TaskTypeUpgrade
+ TaskTypeKeepalive
)
type TerminalTask struct {
diff --git a/resource/template/component/server.html b/resource/template/component/server.html
index db5ef5e..ab88344 100644
--- a/resource/template/component/server.html
+++ b/resource/template/component/server.html
@@ -31,7 +31,8 @@
{{if .Conf.GRPCHost}}
curl -L https://raw.githubusercontent.com/naiba/nezha/master/script/install.sh -o nezha.sh && chmod
+x nezha.sh && sudo ./nezha.sh install_agent {{.Conf.GRPCHost}} {{.Conf.GRPCPort}} {{if .Conf.TLS}}--tls{{end}}
{{else}}
请先在设置页面配置 未接入CDN的面板服务器域名/IP
{{end}}
diff --git a/resource/template/dashboard/server.html b/resource/template/dashboard/server.html
index 3b27046..3a00318 100644
--- a/resource/template/dashboard/server.html
+++ b/resource/template/dashboard/server.html
@@ -40,7 +40,7 @@
{{$server.Secret}} |
diff --git a/script/install.sh b/script/install.sh
index c07afa0..121c584 100755
--- a/script/install.sh
+++ b/script/install.sh
@@ -203,8 +203,8 @@ install_agent() {
mv nezha-agent $NZ_AGENT_PATH &&
rm -rf nezha-agent_linux_${os_arch}.tar.gz README.md
- if [[ $# == 3 ]]; then
- modify_agent_config $1 $2 $3
+ if [ $# -ge 3 ]; then
+ modify_agent_config "$@"
else
modify_agent_config 0
fi
@@ -223,7 +223,7 @@ modify_agent_config() {
return 0
fi
- if [[ $# != 3 ]]; then
+ if [ $# -lt 3 ]; then
echo "请先在管理面板上添加Agent，记录下密钥" &&
read -ep "请输入一个解析到面板所在IP的域名（不可套CDN）: " nz_grpc_host &&
read -ep "请输入面板RPC端口: (5555)" nz_grpc_port &&
@@ -242,12 +242,16 @@ modify_agent_config() {
nz_client_secret=$3
fi
-
-
sed -i "s/nz_grpc_host/${nz_grpc_host}/" ${NZ_AGENT_SERVICE}
sed -i "s/nz_grpc_port/${nz_grpc_port}/" ${NZ_AGENT_SERVICE}
sed -i "s/nz_client_secret/${nz_client_secret}/" ${NZ_AGENT_SERVICE}
+ shift 3
+ if [ $# -gt 0 ]; then
+ args=" $*"
+ sed -i "/ExecStart/ s/$/${args}/" ${NZ_AGENT_SERVICE}
+ fi
+
echo -e "Agent配置 ${green}修改成功，请稍等重启生效${plain}"
systemctl daemon-reload
@@ -558,8 +562,9 @@ if [[ $# > 0 ]]; then
uninstall_dashboard 0
;;
"install_agent")
- if [[ $# == 4 ]]; then
- install_agent $2 $3 $4
+ shift
+ if [ $# -ge 3 ]; then
+ install_agent "$@"
else
install_agent 0
fi
|