优化文件结构

sysmodule/httpclientmodule/httpclientpoolmodule.go

@@ -0,0 +1,111 @@
+package httpclientmodule
+
+import (
+	"bytes"
+	"crypto/tls"
+	"fmt"
+	"io/ioutil"
+	"net"
+	"net/http"
+	"net/url"
+	"time"
+
+	"github.com/duanhf2012/origin/service"
+)
+
+type HttpClientModule struct {
+	service.Module
+	client *http.Client
+}
+
+type HttpRespone struct {
+	Err        error
+	Header     http.Header
+	StatusCode int
+	Status     string
+	Body       []byte
+}
+
+type SyncHttpRespone struct {
+	resp chan HttpRespone
+}
+
+func (slf *SyncHttpRespone) Get(timeoutMs int) HttpRespone {
+	timerC := time.NewTicker(time.Millisecond * time.Duration(timeoutMs)).C
+	select {
+	case <-timerC:
+		break
+	case rsp := <-slf.resp:
+		return rsp
+	}
+	return HttpRespone{
+		Err: fmt.Errorf("Getting the return result timeout [%d]ms", timeoutMs),
+	}
+}
+
+func (slf *HttpClientModule) 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{
+				Timeout:   5 * time.Second,
+				KeepAlive: 30 * time.Second,
+			}).DialContext,
+			MaxIdleConns:        maxpool,
+			MaxIdleConnsPerHost: maxpool,
+			IdleConnTimeout:     60 * time.Second,
+			Proxy:               proxyfun,
+			TLSClientConfig:     &tls.Config{InsecureSkipVerify: true},
+		},
+	}
+}
+
+func (slf *HttpClientModule) 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, header)
+		ret.resp <- rsp
+	}()
+	return ret
+}
+
+func (slf *HttpClientModule) Request(method string, url string, body []byte, header http.Header) HttpRespone {
+	if slf.client == nil {
+		panic("Call the init function first")
+	}
+	ret := HttpRespone{}
+	req, err := http.NewRequest(method, url, bytes.NewReader(body))
+	if err != nil {
+		ret.Err = err
+		return ret
+	}
+	if header != nil {
+		req.Header = header
+	}
+	rsp, err := slf.client.Do(req)
+	if err != nil {
+		ret.Err = err
+		return ret
+	}
+	defer rsp.Body.Close()
+
+	ret.Body, err = ioutil.ReadAll(rsp.Body)
+	if err != nil {
+		ret.Err = err
+		return ret
+	}
+	ret.StatusCode = rsp.StatusCode
+	ret.Status = rsp.Status
+	ret.Header = rsp.Header
+
+	return ret
+}