-
-
Notifications
You must be signed in to change notification settings - Fork 1.5k
/
gclient_response.go
86 lines (75 loc) · 2.38 KB
/
gclient_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
79
80
81
82
83
84
85
86
// 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 gclient
import (
"bytes"
"io/ioutil"
"net/http"
"github.com/gogf/gf/v2/internal/intlog"
)
// Response is the struct for client request response.
type Response struct {
*http.Response // Response is the underlying http.Response object of certain request.
request *http.Request // Request is the underlying http.Request object of certain request.
requestBody []byte // The body bytes of certain request, only available in Dump feature.
cookies map[string]string // Response cookies, which are only parsed once.
}
// 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 {
intlog.Errorf(r.request.Context(), `%+v`, err)
return nil
}
return body
}
// ReadAllString retrieves and returns the response content as string.
func (r *Response) ReadAllString() string {
return string(r.ReadAll())
}
// SetBodyContent overwrites response content with custom one.
func (r *Response) SetBodyContent(content []byte) {
buffer := bytes.NewBuffer(content)
r.Body = ioutil.NopCloser(buffer)
r.ContentLength = int64(buffer.Len())
}
// Close closes the response when it will never be used.
func (r *Response) Close() error {
if r == nil || r.Response == nil {
return nil
}
return r.Response.Body.Close()
}