http/response.go
2022-10-31 17:22:49 +08:00

120 lines
2.4 KiB
Go

/**
* @Author: Echo
* @Email: 1711788888@qq.com
* @Date: 2021/8/15 4:56 下午
* @Desc: TODO
*/
package http
import (
"git.echol.cn/loser/http/internal"
"io/ioutil"
"net/http"
"sync"
)
type Response struct {
*http.Response
Request *http.Request
body []byte
cookies map[string]string
mu sync.Mutex
}
// ReadBytes retrieves and returns the response content as []byte.
func (r *Response) ReadBytes() []byte {
if r == nil || r.Response == nil {
return []byte{}
}
if r.body == nil {
var err error
r.mu.Lock()
defer r.mu.Unlock()
if r.body, err = ioutil.ReadAll(r.Response.Body); err != nil {
return nil
}
}
return r.body
}
// ReadString retrieves and returns the response content as string.
func (r *Response) ReadString() string {
return internal.UnsafeBytesToString(r.ReadBytes())
}
// Scan convert the response into a complex data structure.
func (r *Response) Scan(any interface{}) error {
return internal.Scan(r.ReadBytes(), 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
}