ghttp add SetDisableDNS

util/http/client.go

@@ -106,6 +106,7 @@ type HTTPClient struct {
 	proxyUsed         *url.URL
 	beforeDo          []BeforeDoClientFunc
 	afterDo           []AfterDoClientFunc
+	disableDNS        bool
 }
 
 // NewHTTPClient new a HTTPClient, all request shared one http.Client object, keep cookies, keepalive etc.
@@ -117,6 +118,10 @@ func NewHTTPClient() *HTTPClient {
 	}
 }
 
+func (s *HTTPClient) SetDisableDNS(disableDNS bool) {
+	s.disableDNS = disableDNS
+}
+
 // SetBeforeDo sets callback call before request sent.
 func (s *HTTPClient) SetBeforeDo(beforeDo BeforeDoClientFunc) *HTTPClient {
 	return s.setBeforeDo(beforeDo, true)
@@ -523,16 +528,18 @@ func (s *HTTPClient) newTransport(timeout time.Duration) (tr *http.Transport, er
 				}
 				conn, err = j.Dial(addr)
 			} else {
-				host, port, _ := net.SplitHostPort(addr)
-				iparr, e := resolver.LookupIPAddr(ctx, host)
-				if e != nil {
-					return nil, e
-				}
-				if len(iparr) == 0 {
-					return nil, fmt.Errorf("can not resolve domain %s", host)
+				if !s.disableDNS {
+					host, port, _ := net.SplitHostPort(addr)
+					iparr, e := resolver.LookupIPAddr(ctx, host)
+					if e != nil {
+						return nil, e
+					}
+					if len(iparr) == 0 {
+						return nil, fmt.Errorf("can not resolve domain %s", host)
+					}
+					ip := iparr[0]
+					addr = net.JoinHostPort(ip.String(), port)
 				}
-				ip := iparr[0]
-				addr = net.JoinHostPort(ip.String(), port)
 				if s.dialer != nil {
 					conn, err = s.dialer(network, addr, timeout)
 				} else {

util/http/client_test.go

@@ -235,6 +235,7 @@ func TestHTTPClient_SetDNS(t *testing.T) {
 	assert2.Equal(t, "value1", jar.Cookies(u1)[0].Value)
 	assert := assert2.New(t)
 	client := NewHTTPClient()
+	client.SetDisableDNS(false)
 	client.SetDNS("114.114.114.114:53")
 	body1, _, _, _ := client.Get("http://www.baidu.com/", time.Second*5, nil, map[string]string{"token": "200"})
 	client.SetDNS("8.8.8.8:53")