address comments

Signed-off-by: Ryan Leung <[email protected]>

.gitignore

@@ -10,3 +10,4 @@ tags
 vendor
 default*
 *.bak
+.vscode/

server/api/redirector.go

@@ -19,6 +19,7 @@ import (
 	"net/http"
 	"net/url"
 	"strings"
+	"sync"
 
 	"github.com/pingcap/log"
 	"github.com/pingcap/pd/server"
@@ -35,6 +36,8 @@ const (
 	errRedirectToNotLeader = "redirect to not leader"
 )
 
+var initHTTPClientOnce sync.Once
+
 type redirector struct {
 	s *server.Server
 }
@@ -69,27 +72,28 @@ func (h *redirector) ServeHTTP(w http.ResponseWriter, r *http.Request, next http
 		http.Error(w, err.Error(), http.StatusInternalServerError)
 		return
 	}
-	tlsConfig, err := h.s.GetSecurityConfig().ToTLSConfig()
+	initHTTPClientOnce.Do(func() {
+		var tlsConfig *tls.Config
+		tlsConfig, err = h.s.GetSecurityConfig().ToTLSConfig()
+		dialClient = &http.Client{
+			Transport: &http.Transport{
+				TLSClientConfig: tlsConfig,
+			},
+		}
+	})
 	if err != nil {
 		http.Error(w, err.Error(), http.StatusInternalServerError)
 		return
 	}
-	newCustomReverseProxies(urls, tlsConfig).ServeHTTP(w, r)
+	newCustomReverseProxies(urls).ServeHTTP(w, r)
 }
 
 type customReverseProxies struct {
 	urls   []url.URL
 	client *http.Client
 }
 
-func newCustomReverseProxies(urls []url.URL, cfg *tls.Config) *customReverseProxies {
-	dialClient = &http.Client{
-		Transport: &http.Transport{
-			TLSClientConfig:   cfg,
-			DisableKeepAlives: true,
-		},
-	}
-
+func newCustomReverseProxies(urls []url.URL) *customReverseProxies {
 	p := &customReverseProxies{
 		client: dialClient,
 	}

server/api/util.go

@@ -28,11 +28,7 @@ import (
 )
 
 // dialClient used to dail http request.
-var dialClient = &http.Client{
-	Transport: &http.Transport{
-		DisableKeepAlives: true,
-	},
-}
+var dialClient = &http.Client{}
 
 // Respond to the client about the given error, integrating with errcode.ErrorCode.
 //

tests/pdctl/cluster/cluster_test.go

@@ -71,6 +71,7 @@ func (s *clusterTestSuite) TestClusterAndPing(c *C) {
 	temp.Close()
 	os.Stdout = old
 	out, _ := ioutil.ReadFile(fname)
+	os.Remove(fname)
 	c.Assert(strings.Contains(string(out), "no such file or directory"), IsTrue)
 
 	// cluster status

tests/pdctl/region/region_test.go

@@ -68,6 +68,7 @@ func (s *regionTestSuite) TestRegionKeyFormat(c *C) {
 	temp.Close()
 	os.Stdout = old
 	out, _ := ioutil.ReadFile(fname)
+	os.Remove(fname)
 	c.Assert(strings.Contains(string(out), "unknown flag"), IsFalse)
 }