/
client_response.go
78 lines (68 loc) · 1.86 KB
/
client_response.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
// Copyright GoFrame Author(https://goframe.org). All Rights Reserved.
//
// This Source Code Form is subject to the terms of the MIT License.
// If a copy of the MIT was not distributed with this file,
// You can obtain one at https://github.com/gogf/gf.
package client
import (
"io/ioutil"
"net/http"
"github.com/gogf/gf/util/gconv"
)
// Response is the struct for client request response.
type Response struct {
*http.Response
request *http.Request
requestBody []byte
cookies map[string]string
}
// initCookie initializes the cookie map attribute of Response.
func (r *Response) initCookie() {
if r.cookies == nil {
r.cookies = make(map[string]string)
// Response might be nil.
if r != nil && r.Response != nil {
for _, v := range r.Cookies() {
r.cookies[v.Name] = v.Value
}
}
}
}
// GetCookie retrieves and returns the cookie value of specified <key>.
func (r *Response) GetCookie(key string) string {
r.initCookie()
return r.cookies[key]
}
// GetCookieMap retrieves and returns a copy of current cookie values map.
func (r *Response) GetCookieMap() map[string]string {
r.initCookie()
m := make(map[string]string, len(r.cookies))
for k, v := range r.cookies {
m[k] = v
}
return m
}
// ReadAll retrieves and returns the response content as []byte.
func (r *Response) ReadAll() []byte {
// Response might be nil.
if r == nil || r.Response == nil {
return []byte{}
}
body, err := ioutil.ReadAll(r.Response.Body)
if err != nil {
return nil
}
return body
}
// ReadAllString retrieves and returns the response content as string.
func (r *Response) ReadAllString() string {
return gconv.UnsafeBytesToStr(r.ReadAll())
}
// 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()
}