-
Notifications
You must be signed in to change notification settings - Fork 27
/
common.go
82 lines (76 loc) · 1.79 KB
/
common.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
package impl
import (
"context"
"io"
"net"
"net/http"
"github.com/deadblue/elevengo/lowlevel/client"
)
func isTimeoutError(err error) bool {
if err == nil {
return false
}
ne, ok := err.(net.Error)
return ok && ne.Timeout()
}
// |send| sends an HTTP request, returns HTTP response or an error.
func (c *ClientImpl) send(req *http.Request) (resp *http.Response, err error) {
req.Header.Set("Accept", "*/*")
// Always override user-agent
ua := c.ua
if ua == "" {
ua = defaultUserAgent
}
req.Header.Set(headerUserAgent, ua)
if c.mc {
// Add cookie
for _, cookie := range c.cj.Cookies(req.URL) {
req.AddCookie(cookie)
}
}
// Send request with retry
for {
if resp, err = c.hc.Do(req); !isTimeoutError(err) {
break
}
}
if err == nil && c.mc {
// Save cookie
c.cj.SetCookies(req.URL, resp.Cookies())
}
return
}
// |post| performs an HTTP POST request to specific URL with given payload.
func (c *ClientImpl) post(url string, payload client.Payload, context context.Context) (body io.ReadCloser, err error) {
req, err := http.NewRequestWithContext(context, http.MethodPost, url, payload)
if err != nil {
return
}
req.Header.Set(headerContentType, payload.ContentType())
if size := payload.ContentLength(); size > 0 {
req.ContentLength = size
}
var resp *http.Response
if resp, err = c.send(req); err == nil {
body = resp.Body
}
return
}
func (c *ClientImpl) Get(
url string, headers map[string]string, context context.Context,
) (body io.ReadCloser, err error) {
req, err := http.NewRequestWithContext(context, http.MethodGet, url, nil)
if err != nil {
return
}
if len(headers) > 0 {
for name, value := range headers {
req.Header.Add(name, value)
}
}
var resp *http.Response = nil
if resp, err = c.send(req); err == nil {
body = resp.Body
}
return
}