初始化项目

resopnse.go

@@ -0,0 +1,124 @@
+package curlx
+
+import (
+	"net"
+	"net/http"
+	"strings"
+
+	"github.com/tidwall/gjson"
+)
+
+// Response response object
+type Response struct {
+	resp *http.Response
+	req  *http.Request
+	body []byte
+	err  error
+}
+
+// ResponseBody response body
+type ResponseBody []byte
+
+// String fmt outout
+func (r ResponseBody) String() string {
+	return string(r)
+}
+
+// Read get slice of response body
+func (r ResponseBody) Read(length int) []byte {
+	if length > len(r) {
+		length = len(r)
+	}
+
+	return r[:length]
+}
+
+// GetContents format response body as string
+func (r ResponseBody) GetContents() string {
+	return string(r)
+}
+
+// GetRequest get request object
+func (r *Response) GetRequest() *http.Request {
+	return r.req
+}
+
+// GetBody parse response body
+func (r *Response) GetBody() (ResponseBody, error) {
+	return ResponseBody(r.body), r.err
+}
+
+// GetParsedBody parse response body with gjson
+func (r *Response) GetParsedBody() (*gjson.Result, error) {
+	pb := gjson.ParseBytes(r.body)
+
+	return &pb, nil
+}
+
+// GetStatusCode get response status code
+func (r *Response) GetStatusCode() int {
+	return r.resp.StatusCode
+}
+
+// GetReasonPhrase get response reason phrase
+func (r *Response) GetReasonPhrase() string {
+	status := r.resp.Status
+	arr := strings.Split(status, " ")
+
+	return arr[1]
+}
+
+// IsTimeout get if request is timeout
+func (r *Response) IsTimeout() bool {
+	if r.err == nil {
+		return false
+	}
+	netErr, ok := r.err.(net.Error)
+	if !ok {
+		return false
+	}
+	if netErr.Timeout() {
+		return true
+	}
+
+	return false
+}
+
+// GetHeaders get response headers
+func (r *Response) GetHeaders() map[string][]string {
+	return r.resp.Header
+}
+
+// GetHeader get response header
+func (r *Response) GetHeader(name string) []string {
+	headers := r.GetHeaders()
+	for k, v := range headers {
+		if strings.ToLower(name) == strings.ToLower(k) {
+			return v
+		}
+	}
+
+	return nil
+}
+
+// GetHeaderLine get a single response header
+func (r *Response) GetHeaderLine(name string) string {
+	header := r.GetHeader(name)
+	if len(header) > 0 {
+		return header[0]
+	}
+
+	return ""
+}
+
+// HasHeader get if header exsits in response headers
+func (r *Response) HasHeader(name string) bool {
+	headers := r.GetHeaders()
+	for k := range headers {
+		if strings.ToLower(name) == strings.ToLower(k) {
+			return true
+		}
+	}
+
+	return false
+}