sacred/http/client.go

package http

import (
	"compress/gzip"
	"crypto/tls"
	"crypto/x509"
	"io"
	"net"
	"net/http"
	"time"
)

// readBody read response
func readBody(res *http.Response) (msg Message, err error) {
	var (
		body   []byte
		reader io.Reader
	)

	encoding := res.Header.Get("Content-Encoding")
	switch encoding {
	case "gzip":
		reader, err = gzip.NewReader(res.Body)
		if err == nil {
			body, err = io.ReadAll(reader)
		}
	default:
		body, err = io.ReadAll(res.Body)
	}
	if err != nil {
		return
	}

	msg.StatusCode = res.StatusCode
	msg.Header = res.Header
	msg.Body = body
	return
}

// newRequest new request
func newRequest(method, uri string, body io.Reader, opt RequestOption) (res *http.Response, err error) {
	t := &http.Transport{
		Dial: func(netw, addr string) (net.Conn, error) {
			conn, err := net.DialTimeout(netw, addr, opt.RequestTimeOut)
			if err != nil {
				return nil, err
			}
			conn.SetDeadline(time.Now().Add(opt.RequestTimeOut))
			return conn, nil
		},
		ResponseHeaderTimeout: time.Second * opt.ResponseHeaderTimeout,
	}
	if len(opt.CertFile) > 0 {
		cert, e := tls.LoadX509KeyPair(opt.CertFile, opt.KeyFile)
		if e != nil {
			t.TLSClientConfig = &tls.Config{InsecureSkipVerify: true}
		} else {
			pool := x509.NewCertPool()
			t.TLSClientConfig = &tls.Config{InsecureSkipVerify: true, Certificates: []tls.Certificate{cert}, RootCAs: pool}
		}
	}
	c := &http.Client{Transport: t}
	var req *http.Request
	req, err = http.NewRequest(method, uri, body)
	if err != nil {
		return
	}

	for k, v := range opt.Headers {
		req.Header.Add(k, v)
	}

	res, err = c.Do(req)
	return
}

// Get HTTP request GET
func Get(uri string, opt RequestOption) (msg Message, err error) {
	var res *http.Response
	if res, err = newRequest("GET", uri, nil, opt); err != nil {
		return
	}

	defer res.Body.Close()
	msg, err = readBody(res)
	return
}

// GetJSON HTTP request GET, response JSON
func GetJSON(v interface{}, uri string, opt RequestOption) (msg Message, err error) {
	var res *http.Response
	if res, err = newRequest("GET", uri, nil, opt); err != nil {
		return
	}

	defer res.Body.Close()
	msg, err = readBody(res)
	if err != nil {
		return
	}
	err = msg.JSON(v)
	return
}

// GetXML HTTP request GET, response XML
func GetXML(v interface{}, uri string, opt RequestOption) (msg Message, err error) {
	var res *http.Response
	if res, err = newRequest("GET", uri, nil, opt); err != nil {
		return
	}

	defer res.Body.Close()
	msg, err = readBody(res)
	if err != nil {
		return
	}
	err = msg.XML(v)
	return
}

// Post HTTP request POST
func Post(uri string, body io.Reader, opt RequestOption) (msg Message, err error) {
	var res *http.Response
	if res, err = newRequest("POST", uri, body, opt); err != nil {
		return
	}
	defer res.Body.Close()
	msg, err = readBody(res)
	return
}