/
option.go
109 lines (91 loc) · 2.6 KB
/
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
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
package httputil
import (
"net/http"
"strings"
"time"
)
// ClientOptions http client options
type ClientOptions struct {
userAgent string
timeout time.Duration
proxy string
dialTimeout time.Duration
keepAliveTimeout time.Duration
maxIdleConns int
idleConnTimeout time.Duration
tlsHandshakeTimeout time.Duration
expectContinueTimeout time.Duration
}
type ClientOption func(*ClientOptions)
func WithUserAgent(userAgent string) func(*ClientOptions) {
return func(options *ClientOptions) {
options.userAgent = userAgent
}
}
func WithTimeout(timeout time.Duration) func(*ClientOptions) {
return func(options *ClientOptions) {
options.timeout = timeout
}
}
func WithProxy(proxy string) func(*ClientOptions) {
return func(options *ClientOptions) {
options.proxy = strings.TrimSpace(proxy)
}
}
func WithDialTimeout(dialTimeout time.Duration) func(*ClientOptions) {
return func(options *ClientOptions) {
options.dialTimeout = dialTimeout
}
}
func WithKeepAliveTimeout(keepAliveTimeout time.Duration) func(*ClientOptions) {
return func(options *ClientOptions) {
options.keepAliveTimeout = keepAliveTimeout
}
}
func WithMaxIdleConns(maxIdleConns int) func(*ClientOptions) {
return func(options *ClientOptions) {
options.maxIdleConns = maxIdleConns
}
}
func WithIdleConnTimeout(idleConnTimeout time.Duration) func(*ClientOptions) {
return func(options *ClientOptions) {
options.idleConnTimeout = idleConnTimeout
}
}
func WithTLSHandshakeTimeout(tlsHandshakeTimeout time.Duration) func(*ClientOptions) {
return func(options *ClientOptions) {
options.tlsHandshakeTimeout = tlsHandshakeTimeout
}
}
func WithExpectContinueTimeout(expectContinueTimeout time.Duration) func(*ClientOptions) {
return func(options *ClientOptions) {
options.expectContinueTimeout = expectContinueTimeout
}
}
type RequestOptions struct {
headers http.Header
referer string
contentType string
timeout time.Duration
}
type RequestOption func(*RequestOptions)
func WithHeaders(headers http.Header) func(*RequestOptions) {
return func(options *RequestOptions) {
options.headers = headers
}
}
func WithReferer(referer string) func(*RequestOptions) {
return func(options *RequestOptions) {
options.referer = strings.TrimSpace(referer)
}
}
func WithContentType(contentType string) func(*RequestOptions) {
return func(options *RequestOptions) {
options.contentType = strings.TrimSpace(contentType)
}
}
func WithRequestTimeout(timeout time.Duration) func(*RequestOptions) {
return func(options *RequestOptions) {
options.timeout = timeout
}
}