@@ -10,14 +10,16 @@ import (
1010 "time"
1111
1212 "github.com/juju/errors"
13- "github.com/zssky/log"
13+
14+ "github.com/dearcode/crab/log"
1415)
1516
1617//HTTPClient 带超时重试控制的http客户端.
1718type HTTPClient struct {
1819 retryTimes int
1920 timeout time.Duration
2021 client http.Client
22+ logger * log.Logger
2123}
2224
2325type StatusError struct {
@@ -37,17 +39,17 @@ func (c *HTTPClient) dial(network, addr string) (net.Conn, error) {
3739 if err == nil {
3840 break
3941 }
40- log .Errorf ("DialTimeout %s:%s error:%v retry:%v" , network , addr , err , i + 1 )
42+ c . logger .Errorf ("DialTimeout %s:%s error:%v retry:%v" , network , addr , err , i + 1 )
4143 }
4244
4345 if err != nil {
44- log .Errorf ("DialTimeout %s:%s error:%v" , network , addr , err )
46+ c . logger .Errorf ("DialTimeout %s:%s error:%v" , network , addr , err )
4547 return nil , errors .Trace (err )
4648 }
4749
4850 deadline := time .Now ().Add (c .timeout )
4951 if err = conn .SetDeadline (deadline ); err != nil {
50- log .Errorf ("SetDeadline %s:%s" , network , addr )
52+ c . logger .Errorf ("SetDeadline %s:%s" , network , addr )
5153 conn .Close ()
5254 return nil , errors .Trace (err )
5355 }
@@ -78,8 +80,9 @@ func (c *HTTPClient) Timeout(t int) *HTTPClient {
7880 return c
7981}
8082
81- func (c * HTTPClient ) SetLogLevel (l string ) * HTTPClient {
82- log .SetLevelByString (l )
83+ //SetLogger 开启日志.
84+ func (c * HTTPClient ) SetLogger (l * log.Logger ) * HTTPClient {
85+ c .logger = l
8386 return c
8487}
8588
@@ -128,7 +131,7 @@ func (c HTTPClient) GetJSON(url string, headers map[string]string, resp interfac
128131 if err != nil {
129132 return errors .Trace (err )
130133 }
131- log .Debugf ("url:%v, resp:%s" , url , buf )
134+ c . logger .Debugf ("url:%v, resp:%s" , url , buf )
132135 return errors .Trace (json .Unmarshal (buf , resp ))
133136}
134137
@@ -153,7 +156,7 @@ func (c HTTPClient) PostJSON(url string, headers map[string]string, data interfa
153156 return errors .Trace (err )
154157 }
155158
156- log .Debugf ("url:%v, resp:%s" , url , buf )
159+ c . logger .Debugf ("url:%v, resp:%s" , url , buf )
157160 return json .Unmarshal (buf , resp )
158161}
159162
@@ -177,7 +180,7 @@ func (c HTTPClient) PutJSON(url string, headers map[string]string, data interfac
177180 if buf , err = c .do ("PUT" , url , headers , buf ); err != nil {
178181 return errors .Trace (err )
179182 }
180- log .Debugf ("url:%v, resp:%s" , url , buf )
183+ c . logger .Debugf ("url:%v, resp:%s" , url , buf )
181184 return json .Unmarshal (buf , resp )
182185}
183186
@@ -192,6 +195,6 @@ func (c HTTPClient) DeleteJSON(url string, headers map[string]string, resp inter
192195 if err != nil {
193196 return errors .Trace (err )
194197 }
195- log .Debugf ("url:%v, resp:%s" , url , buf )
198+ c . logger .Debugf ("url:%v, resp:%s" , url , buf )
196199 return json .Unmarshal (buf , resp )
197200}
0 commit comments