Ma/origin
1
0
Fork 0
mirror of https://github.com/duanhf2012/origin.git synced 2026-02-11 21:54:50 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

http请求添加header和代理

Browse Source
This commit is contained in:
learnmore2019
2019-02-15 10:49:38 +08:00
parent 6945fc8532
commit bc82d200b0
2 changed files with 19 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
sysmodule/HttpClientPoolModule.go
View File

@@ -6,6 +6,7 @@ import (
"io/ioutil"
"net"
"net/http"
"net/url"
"time"
"github.com/duanhf2012/origin/service"
@@ -41,7 +42,15 @@ func (slf *SyncHttpRespone) Get(timeoutMs int) HttpRespone {
}
}
func (slf *HttpClientPoolModule) Init(maxpool int) {
func (slf *HttpClientPoolModule) Init(maxpool int, proxyUrl string) {
type ProxyFun func(_ *http.Request) (*url.URL, error)
var proxyfun ProxyFun
if proxyUrl != "" {
proxyfun = func(_ *http.Request) (*url.URL, error) {
return url.Parse(proxyUrl)
}
}
slf.client = &http.Client{
Transport: &http.Transport{
DialContext: (&net.Dialer{
@@ -51,22 +60,23 @@ func (slf *HttpClientPoolModule) Init(maxpool int) {
MaxIdleConns: maxpool,
MaxIdleConnsPerHost: maxpool,
IdleConnTimeout: 60 * time.Second,
Proxy: proxyfun,
},
}
}
func (slf *HttpClientPoolModule) SyncRequest(method string, url string, body []byte) SyncHttpRespone {
func (slf *HttpClientPoolModule) SyncRequest(method string, url string, body []byte, header http.Header) SyncHttpRespone {
ret := SyncHttpRespone{
resp: make(chan HttpRespone, 1),
}
go func() {
rsp := slf.Request(method, url, body)
rsp := slf.Request(method, url, body, header)
ret.resp <- rsp
}()
return ret
}
func (slf *HttpClientPoolModule) Request(method string, url string, body []byte) HttpRespone {
func (slf *HttpClientPoolModule) Request(method string, url string, body []byte, header http.Header) HttpRespone {
if slf.client == nil {
panic("Call the init function first")
}
@@ -76,6 +86,9 @@ func (slf *HttpClientPoolModule) Request(method string, url string, body []byte)
ret.Err = err
return ret
}
if header != nil {
req.Header = header
}
rsp, err := slf.client.Do(req)
if err != nil {
ret.Err = err