-
Notifications
You must be signed in to change notification settings - Fork 3
/
options.go
93 lines (79 loc) · 1.54 KB
/
options.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
package httpClientKit
import (
"crypto/tls"
"net/http"
"time"
)
const (
// DefaultTimeout 发送请求的默认超时时间.
DefaultTimeout = time.Second * 10
)
type (
options struct {
// timeout 请求的超时时间
timeout time.Duration
// safe 默认false(跳过ssl证书验证,即使url 以https开头 && 证书非法,请求也能成功)
safe bool
// queryParams 适用于: POST、GET
queryParams map[string][]string
// postParams 适用于: POST
postParams map[string][]string
}
Option func(opts *options)
)
func (opts *options) newHttpClient() *http.Client {
return &http.Client{
Timeout: opts.timeout,
Transport: &http.Transport{
TLSClientConfig: &tls.Config{
InsecureSkipVerify: !opts.safe,
},
},
}
}
func loadOptions(s ...Option) *options {
opts := &options{}
for _, option := range s {
option(opts)
}
if opts.timeout <= 0 {
opts.timeout = DefaultTimeout
}
return opts
}
// WithTimeout
/*
适用于: GET、POST
*/
func WithTimeout(timeout time.Duration) Option {
return func(opts *options) {
opts.timeout = timeout
}
}
// WithSafe
/*
适用于: GET、POST
*/
func WithSafe(safe bool) Option {
return func(opts *options) {
opts.safe = safe
}
}
// WithQueryParams
/*
适用于: GET、POST
*/
func WithQueryParams(queryParams map[string][]string) Option {
return func(opts *options) {
opts.queryParams = queryParams
}
}
// WithPostParams
/*
适用于: POST
*/
func WithPostParams(postParams map[string][]string) Option {
return func(opts *options) {
opts.postParams = postParams
}
}