init

at-common/http.go

@@ -0,0 +1,110 @@
+package at_common
+
+import (
+	"crypto/tls"
+	"io/ioutil"
+	"net/http"
+	"net/url"
+	"strings"
+	"time"
+)
+
+func HttpGetNotInternal(URL string) (body []byte, code int, err error) {
+	return _HttpGet(URL, false)
+}
+func HttpPostNotInternal(URL string, data map[string]string, header map[string]string) (body []byte, code int, err error) {
+	return _HttpPost(URL, data, header, false)
+}
+func HttpPost(URL string, data map[string]string, header map[string]string) (body []byte, code int, err error) {
+	return _HttpPost(URL, data, header, true)
+}
+func HttpGet(URL string) (body []byte, code int, err error) {
+	return _HttpGet(URL, true)
+}
+func _HttpGet(URL string, isInternal bool) (body []byte, code int, err error) {
+	client, tr, err := getRequestClient(URL, isInternal)
+	if err != nil {
+		return
+	}
+	resp, err := client.Get(URL)
+	if err != nil {
+		return
+	}
+	defer resp.Body.Close()
+	defer tr.CloseIdleConnections()
+	code = resp.StatusCode
+	body, err = ioutil.ReadAll(resp.Body)
+	return
+}
+func _HttpPost(URL string, data map[string]string, header map[string]string, isInternal bool) (body []byte, code int, err error) {
+	postParamsString := ""
+	if data != nil {
+		postParams := []string{}
+		for k, v := range data {
+			postParams = append(postParams, url.QueryEscape(k)+"="+url.QueryEscape(v))
+		}
+		postParamsString = strings.Join(postParams, "&")
+	}
+	return HttpPostRaw(URL, postParamsString, header, isInternal)
+}
+
+func HttpPostRaw(URL, postParamsString string, header map[string]string, isInternal bool) (body []byte, code int, err error) {
+	var resp *http.Response
+	var client *http.Client
+	var tr *http.Transport
+	defer func() {
+		if resp != nil && resp.Body != nil {
+			resp.Body.Close()
+		}
+		if tr != nil {
+			tr.CloseIdleConnections()
+		}
+	}()
+	req, err := http.NewRequest("POST", URL, strings.NewReader(postParamsString))
+	if err != nil {
+		return
+	}
+	client, tr, err = getRequestClient(URL, isInternal)
+	if err != nil {
+		return
+	}
+	if header != nil {
+		for k, v := range header {
+			req.Header.Set(k, v)
+		}
+	}
+	req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
+	resp, err = client.Do(req)
+	if err != nil {
+		return
+	}
+	body, err = ioutil.ReadAll(resp.Body)
+	code = resp.StatusCode
+	if err != nil {
+		return
+	}
+	return
+}
+
+func getRequestClient(url string, isInternal bool) (client *http.Client, tr *http.Transport, err error) {
+	var conf *tls.Config
+	conf, err = getRequestTlsConfig(isInternal)
+	if err != nil {
+		return
+	}
+	if strings.Contains(url, "https://") {
+		tr = &http.Transport{TLSClientConfig: conf}
+		client = &http.Client{Timeout: time.Second * 5, Transport: tr}
+	} else {
+		tr = &http.Transport{}
+		client = &http.Client{Timeout: time.Second * 5, Transport: tr}
+	}
+	return
+}
+
+func UrlArgs(url, args string) string {
+	if strings.Contains(url, "?") {
+		return url + "&" + args
+	}
+	return url + "?" + args
+}