-
Notifications
You must be signed in to change notification settings - Fork 0
/
request_option.go
69 lines (62 loc) · 1.58 KB
/
request_option.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
package httpclient
import (
"net/http"
"net/url"
)
// RequestOption allows additional modifications to request object before
// http.Do is called
type RequestOption func(*http.Request) error
// AddHeaders allows for additional headers to be added when making a request
func AddHeaders(headers http.Header) RequestOption {
return func(req *http.Request) error {
for k, v := range headers {
for _, dv := range v {
req.Header.Add(k, dv)
}
}
return nil
}
}
// DelHeaders allows for certain headers to be removed when making requests
func DelHeaders(headers http.Header) RequestOption {
return func(req *http.Request) error {
for k := range headers {
req.Header.Del(k)
}
return nil
}
}
// SetHeaders allows for certain headers to be replaced when making a request
func SetHeaders(headers http.Header) RequestOption {
return func(req *http.Request) error {
for k, vs := range headers {
req.Header.Del(k)
for i := range vs {
req.Header.Add(k, vs[i])
}
}
return nil
}
}
// AddQueryParams is an additive option that will not replace existing query
// parameters
func AddQueryParams(values url.Values) RequestOption {
return func(req *http.Request) error {
params := req.URL.Query()
for k, vals := range values {
for _, v := range vals {
params.Add(k, v)
}
}
req.URL.RawQuery = params.Encode()
return nil
}
}
// WithQueryParams will override any existing query parameters bound to the
// request
func WithQueryParams(values url.Values) RequestOption {
return func(req *http.Request) error {
req.URL.RawQuery = values.Encode()
return nil
}
}