-
Notifications
You must be signed in to change notification settings - Fork 77
/
api_request.go
77 lines (66 loc) · 1.82 KB
/
api_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
package common
import (
"encoding/json"
"fmt"
"io"
"io/ioutil"
"net"
"net/http"
"time"
)
type deadlinedConn struct {
Timeout time.Duration
net.Conn
}
func (c *deadlinedConn) Read(b []byte) (n int, err error) {
c.Conn.SetReadDeadline(time.Now().Add(c.Timeout))
return c.Conn.Read(b)
}
func (c *deadlinedConn) Write(b []byte) (n int, err error) {
c.Conn.SetWriteDeadline(time.Now().Add(c.Timeout))
return c.Conn.Write(b)
}
func NewDeadlineTransport(timeout time.Duration) *http.Transport {
transport := &http.Transport{
Dial: func(netw, addr string) (net.Conn, error) {
c, err := net.DialTimeout(netw, addr, timeout)
if err != nil {
return nil, err
}
return &deadlinedConn{timeout, c}, nil
},
}
return transport
}
// stores the result in the value pointed to by ret(must be a pointer)
func APIRequest(method string, endpoint string, body io.Reader, timeout time.Duration, ret interface{}) (int, error) {
httpclient := &http.Client{Transport: NewDeadlineTransport(timeout)}
req, err := http.NewRequest(method, endpoint, body)
if err != nil {
return 0, err
}
req.Header.Add("Accept", "application/zanredisdb; version=1.0")
resp, err := httpclient.Do(req)
if err != nil {
return 0, fmt.Errorf("req %v error %v",
endpoint, err.Error())
}
respBody, err := ioutil.ReadAll(resp.Body)
resp.Body.Close()
if err != nil {
return resp.StatusCode, fmt.Errorf("req %v read body error %v",
endpoint, err.Error())
}
// return status code and error info if not status ok
// no error means if is status ok
if resp.StatusCode != http.StatusOK {
return resp.StatusCode, fmt.Errorf("req %v got error response %s %q", endpoint, resp.Status, respBody)
}
if ret == nil {
return resp.StatusCode, nil
}
if len(respBody) == 0 {
respBody = []byte("{}")
}
return resp.StatusCode, json.Unmarshal(respBody, ret)
}