// choose http GET/method to download
func connectByHttp(p *context.Response, req *context.Request) (*http.Response, error) {
client := &http.Client{
CheckRedirect: req.GetRedirectFunc(),
}
httpreq, err := http.NewRequest(req.GetMethod(), req.GetUrl(), strings.NewReader(req.GetPostdata()))
if header := req.GetHeader(); header != nil {
httpreq.Header = req.GetHeader()
}
if cookies := req.GetCookies(); cookies != nil {
for i := range cookies {
httpreq.AddCookie(cookies[i])
}
}
var resp *http.Response
if resp, err = client.Do(httpreq); err != nil {
if e, ok := err.(*url.Error); ok && e.Err != nil && e.Err.Error() == "normal" {
// normal
} else {
reporter.Log.Println(err.Error())
p.SetStatus(true, err.Error())
//fmt.Printf("client do error %v \r\n", err)
return nil, err
}
}
return resp, nil
}
func (self *scheduler) Push(req *context.Request) {
if self.status == STOP {
return
}
is := self.Compare(req.GetUrl() + req.GetMethod())
// 有重复则返回
if is {
return
}
self.SrcManage.Push(req)
}
func (self *Surfer) Download(cReq *context.Request) *context.Response {
cResp := context.NewResponse(nil)
resp, err := self.download.Download(cReq.GetMethod(), cReq.GetUrl(), cReq.GetReferer(), cReq.GetPostData(), cReq.GetHeader(), cReq.GetCookies())
cResp.SetRequest(cReq)
if err != nil {
cResp.SetStatus(true, err.Error())
return cResp
}
// get converter to utf-8
body := self.changeCharsetEncodingAuto(resp.Body, resp.Header.Get("Content-Type"))
//fmt.Printf("utf-8 body %v \r\n", bodyStr)
defer resp.Body.Close()
cResp.SetText(body)
cResp.SetStatus(false, "")
return cResp
}