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)
cResp.SetResponse(resp)
if err != nil {
cResp.SetStatus(true, err.Error())
return cResp
}
cResp.SetStatus(false, "")
return cResp
}
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
}
// 添加请求到队列
func (self *scheduler) Push(req *context.Request) {
pushMutex.Lock()
defer func() {
pushMutex.Unlock()
}()
if self.status == status.STOP {
return
}
is := self.Compare(req.GetUrl() + req.GetMethod())
// 有重复则返回
if is {
return
}
// 留作未来分发请求用
// if pholcus.Self.GetRunMode() == config.SERVER || req.CanOutsource() {
// return
// }
self.SrcManage.Push(req)
}