fix fail connection issue

matishsiao · Sep 10, 2015 · e97be24 · e97be24
1 parent c28c2bf
commit e97be24
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/GoHAProxy b/GoHAProxy
diff --git a/forwardServer.go b/forwardServer.go
@@ -7,6 +7,7 @@ import (
 	"net"
 	"strings"
 	"time"
+	"log"
 )
 
 
@@ -24,16 +25,19 @@ type Client struct {
 }
 
 func (fs *ForwardServer) CheckHealth(connType string, uri string) (bool, int) {
-
+	if uri == "" {
+		log.Println("Check health failed:uri is empty.")
+		return false,0
+	}
 	conn, err := net.Dial(connType, uri)
 	errCode := 0
 
 	if err != nil {
 		errCode = 1
 		return false, errCode
 	}
-
-	defer conn.Close()
+	
+
 
 	conn.SetReadDeadline(time.Now().Add(5 * time.Second))
 
@@ -60,6 +64,7 @@ func (fs *ForwardServer) CheckHealth(connType string, uri string) (bool, int) {
 		}
 		break
 	}
+	conn.Close()
 	return true, errCode
 }