first commit

response.go

@@ -0,0 +1,116 @@
+/**
+ * @Author: fuxiao
+ * @Email: 576101059@qq.com
+ * @Date: 2021/8/15 4:56 下午
+ * @Desc: TODO
+ */
+
+package http
+
+import (
+	"io/ioutil"
+	"net/http"
+	
+	"github.com/dobyte/http/internal"
+)
+
+type Response struct {
+	*http.Response
+	Request *http.Request
+	body    []byte
+	cookies map[string]string
+}
+
+// Bytes retrieves and returns the response content as []byte.
+func (r *Response) Bytes() []byte {
+	if r == nil || r.Response == nil {
+		return []byte{}
+	}
+	
+	if r.body == nil {
+		var err error
+		if r.body, err = ioutil.ReadAll(r.Response.Body); err != nil {
+			return nil
+		}
+	}
+	
+	return r.body
+}
+
+// String retrieves and returns the response content as string.
+func (r *Response) String() string {
+	return internal.UnsafeBytesToString(r.Bytes())
+}
+
+// Scan convert the response into a complex data structure.
+func (r *Response) Scan(any interface{}) error {
+	return internal.Scan(r.Bytes(), any)
+}
+
+// Close closes the response when it will never be used.
+func (r *Response) Close() error {
+	if r == nil || r.Response == nil || r.Response.Close {
+		return nil
+	}
+	r.Response.Close = true
+	return r.Response.Body.Close()
+}
+
+// HasHeader Determine if a header exists in the cache.
+func (r *Response) HasHeader(key string) bool {
+	for k, _ := range r.Header {
+		if k == key {
+			return true
+		}
+	}
+	
+	return false
+}
+
+// GetHeader Retrieve header's value from the response.
+func (r *Response) GetHeader(key string) string {
+	return r.Header.Get(key)
+}
+
+// GetHeader Retrieve all header's value from the response.
+func (r *Response) GetHeaders() map[string]interface{} {
+	headers := make(map[string]interface{})
+	for k, v := range r.Header {
+		if len(v) > 1 {
+			headers[k] = v
+		} else {
+			headers[k] = v[0]
+		}
+	}
+	
+	return headers
+}
+
+// HasCookie Determine if a cookie exists in the cache.
+func (r *Response) HasCookie(key string) bool {
+	if r.cookies == nil {
+		r.cookies = r.GetCookies()
+	}
+	_, ok := r.cookies[key]
+	
+	return ok
+}
+
+// GetCookie Retrieve cookie's value from the response.
+func (r *Response) GetCookie(key string) string {
+	if r.cookies == nil {
+		r.cookies = r.GetCookies()
+	}
+	return r.cookies[key]
+}
+
+// GetCookies Retrieve all cookie's value from the response.
+func (r *Response) GetCookies() map[string]string {
+	cookies := make(map[string]string)
+	if r != nil && r.Response != nil {
+		for _, cookie := range r.Cookies() {
+			cookies[cookie.Name] = cookie.Value
+		}
+	}
+	return cookies
+}