github/nps
1
0
Fork 0
mirror of https://github.com/ehang-io/nps.git synced 2025-07-03 04:53:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #207 from evangwt/master

Browse Source 
fix: close health check tcp connection
This commit is contained in:
he liu 2019-09-19 09:49:39 +08:00 committed by GitHub
commit 445ed1dd60
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
client/health.go
View File

@ -71,7 +71,11 @@ func check(t *file.Health) {
var rs *http.Response var rs *http.Response
for _, v := range arr { for _, v := range arr {
if t.HealthCheckType == "tcp" { if t.HealthCheckType == "tcp" {
_, err = net.DialTimeout("tcp", v, time.Duration(t.HealthCheckTimeout)*time.Second) var c net.Conn
c, err = net.DialTimeout("tcp", v, time.Duration(t.HealthCheckTimeout)*time.Second)
if err == nil {
c.Close()
}
} else { } else {
client := &http.Client{} client := &http.Client{}
client.Timeout = time.Duration(t.HealthCheckTimeout) * time.Second client.Timeout = time.Duration(t.HealthCheckTimeout) * time.Second