/
request.go
173 lines (152 loc) · 5.08 KB
/
request.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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
package builder
import (
"bytes"
"fmt"
"golang.org/x/net/context"
"io"
"net/http"
"net/url"
"strings"
"sync"
)
type Request struct {
URL *url.URL
ctx context.Context
Method string // HTTP 请求的 Method 部分
Body any
bodyBuf *bytes.Buffer
bodyBytes []byte
client *Client // 指向 Client 的指针
Header sync.Map
QueryParam sync.Map
Cookies []*http.Cookie
NewRequest *http.Request
}
func (request *Request) SetBody(v interface{}) *Request {
request.Body = v
return request
}
// SetHeader 方法用于设置 HTTP 请求的 Header 部分。它接收两个 string 类型的参数,
func (request *Request) SetHeader(key, value string) *Request {
request.Header.Store(key, value)
return request
}
func (request *Request) SetHeaders(headers map[string]string) *Request {
for key, value := range headers {
request.SetHeader(key, value)
}
return request
}
// SetCookies 方法用于设置 HTTP 请求的 Cookies 部分。它接收一个 []*http.Cookie 类型的参数,
func (request *Request) SetCookies(cookie []*http.Cookie) *Request {
for _, c := range cookie {
request.SetCookie(c)
}
return request
}
// SetCookie 方法用于设置 HTTP 请求的 Cookie 部分。它接收一个 *http.Cookie 类型的参数,
func (request *Request) SetCookie(cookie *http.Cookie) *Request {
request.Cookies = append(request.Cookies, cookie)
return request
}
// SetQueryParams 方法用于设置 HTTP 请求的 Query 部分。它接收一个 map[string]interface{} 类型的参数,
func (request *Request) SetQueryParams(query map[string]any) *Request {
for key, value := range query {
request.SetQueryParam(key, value)
}
return request
}
// SetQueryParam 方法用于设置 HTTP 请求的 Query 部分。它接收两个 string 类型的参数,
func (request *Request) SetQueryParam(key string, value any) *Request {
request.QueryParam.Store(key, value)
return request
}
// SetQueryString 方法用于设置 HTTP 请求的 Query 部分。它接收一个 string 类型的参数,
func (request *Request) SetQueryString(query string) *Request {
if params, err := url.ParseQuery(strings.TrimSpace(query)); err == nil {
for key, value := range params {
request.SetQueryParam(key, value[0])
}
} else {
request.client.LogError(err, query, "request.go", "SetQueryString")
}
return request
}
func (request *Request) SetHeaderContentType(contentType string) *Request {
request.SetHeader("Content-Type", contentType)
return request
}
// GetQueryParamsEncode 方法用于获取 HTTP 请求的 Query 部分的 URL 编码字符串。
func (request *Request) GetQueryParamsEncode() string {
var parts []string
request.QueryParam.Range(func(key any, value any) bool {
k, _ := key.(string)
v, _ := value.(string)
parts = append(parts, fmt.Sprintf("%s=%s", url.QueryEscape(k), url.QueryEscape(v)))
return true
})
return strings.Join(parts, "&")
}
// GetQueryParamsNopCloser 方法用于获取 HTTP 请求的 Query 部分的 ReadCloser。
func (request *Request) GetQueryParamsNopCloser() io.ReadCloser {
// 将字符串转换为 io.ReadCloser, 并返回
return io.NopCloser(strings.NewReader(request.GetQueryParamsEncode()))
}
// GetQueryParams 方法用于获取 HTTP 请求的 Query 部分的 url.Values。
//func (request *Request) GetQueryParams() url.Values {
// return request.QueryParam
//}
// GetHost 方法用于获取 HTTP 请求的 Host 部分的字符串。
func (request *Request) GetHost() string {
return request.client.baseUrl
}
// GetPath 方法用于获取 HTTP 请求的 Path 部分的字符串。
func (request *Request) GetPath() string {
return request.URL.Path
}
// GetUrl 方法用于获取 HTTP 请求的 URL 部分的字符串。
func (request *Request) GetUrl() string {
return request.URL.String()
}
// GetMethod 方法用于获取 HTTP 请求的 Method 部分的字符串。
func (request *Request) GetMethod() string {
return request.Method
}
// GetRequestHeader 方法用于获取 HTTP 请求的 Header 部分的 http.Header。
func (request *Request) GetRequestHeader() http.Header {
header := make(http.Header)
request.Header.Range(func(key, value interface{}) bool {
keyStr, _ := key.(string)
valueStr, _ := value.(string)
if keyStr != "" && valueStr != "" {
header.Add(keyStr, valueStr)
}
return true
})
return header
}
func (request *Request) GetHeaderContentType() string {
return request.GetRequestHeader().Get("Content-Type")
}
func (request *Request) jsonToMap(jsonStr string) map[string]any {
var result map[string]any
err := request.client.JSONUnmarshal([]byte(jsonStr), &result)
if err != nil {
request.client.LogError(err, jsonStr, "request.go", "jsonToMap")
}
return result
}
func (request *Request) mapToJson(params any) string {
jsonStr, err := request.client.JSONMarshal(params)
if err != nil {
request.client.LogError(err, params, "request.go", "mapToJson")
}
return string(jsonStr)
}
func (request *Request) structToJson(params any) string {
jsonStr, err := request.client.JSONMarshal(params)
if err != nil {
request.client.LogError(err, params, "request.go", "structToJson")
}
return string(jsonStr)
}