shuguang/nezha
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: close connection when reconnect

Browse Source
This commit is contained in:
奶爸 2020-02-09 22:04:59 +08:00
parent 70f0e92343
commit b497b29402
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
cmd/agent/main.go
View File

@ -59,10 +59,6 @@ func run(cmd *cobra.Command, args []string) {
ClientID: clientID, ClientID: clientID,
ClientSecret: clientSecret, ClientSecret: clientSecret,
} }
retry := func() {
time.Sleep(delayWhenError)
log.Println("Try to reconnect ...")
}
// 上报服务器信息 // 上报服务器信息
go reportState() go reportState()
@ -71,6 +67,15 @@ func run(cmd *cobra.Command, args []string) {
var conn *grpc.ClientConn var conn *grpc.ClientConn
var hc pb.NezhaService_HeartbeatClient var hc pb.NezhaService_HeartbeatClient
retry := func() {
log.Println("Error to close connection ...")
if conn != nil {
conn.Close()
}
time.Sleep(delayWhenError)
log.Println("Try to reconnect ...")
}
for { for {
conn, err = grpc.Dial(server, grpc.WithInsecure(), grpc.WithPerRPCCredentials(&auth)) conn, err = grpc.Dial(server, grpc.WithInsecure(), grpc.WithPerRPCCredentials(&auth))
if err != nil { if err != nil {