-
Notifications
You must be signed in to change notification settings - Fork 1
/
conn.go
113 lines (95 loc) · 2.6 KB
/
conn.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
110
111
112
113
package pitcher
import (
"bytes"
"encoding/json"
"fmt"
"io"
"net/http"
"net/url"
)
type connection interface {
Close()
hasToken() bool
setToken(string)
Get(string) ([]byte, *ErrorResponse)
Post(string, interface{}) ([]byte, *ErrorResponse)
// WriteMessage([]byte) error
// ReadMessage() ([]byte, error)
}
type hConn struct {
client *http.Client
baseURL string
sessionToken string
}
func createNewConnection(url string) connection {
return &hConn{client: &http.Client{}, baseURL: url}
}
func (c *hConn) Close() {
}
func (c *hConn) hasToken() bool {
fmt.Println("- ? Has Token : ", c.sessionToken)
return c.sessionToken != ""
}
func (c *hConn) setToken(token string) {
c.sessionToken = token
}
func (c *hConn) request(method string, path string, content io.Reader) ([]byte, *ErrorResponse) {
url, _ := c.buildURL(path)
request, err := http.NewRequest(method, url.String(), content)
if err != nil {
fmt.Println("===> ERROR CODE : ", err.Error())
return nil, nil
}
c.setDefaultHeaders(request)
response, err := c.client.Do(request)
if err != nil {
fmt.Println("===> ERROR CODE : ", err.Error())
return nil, nil
}
if response.StatusCode >= 400 && response.StatusCode < 600 {
return nil, c.processErrorMessage(response.Body)
}
return c.streamToByte(response.Body), nil
}
func (c *hConn) Get(path string) ([]byte, *ErrorResponse) {
return c.request("GET", path, nil)
}
func (c *hConn) Post(path string, obj interface{}) ([]byte, *ErrorResponse) {
payloadJSON, _ := json.Marshal(obj)
return c.request("POST", path, bytes.NewBuffer(payloadJSON))
}
func (c *hConn) buildURL(pathOrURL string) (*url.URL, error) {
u, err := url.ParseRequestURI(pathOrURL)
if err != nil {
u, err = url.Parse(c.baseURL)
if err != nil {
return nil, err
}
return u.Parse(pathOrURL)
}
return u, nil
}
func (c *hConn) setDefaultHeaders(request *http.Request) {
request.Header.Set("Accept", "application/json")
request.Header.Set("Content-Type", "application/json")
request.Header.Set("Auth-Server-Token", c.sessionToken)
// if c.hasToken() {
// cookie := &http.Cookie{Name: "auth_token", Value: c.sessionToken}
// request.AddCookie(cookie)
// }
}
func (c *hConn) processErrorMessage(body io.ReadCloser) *ErrorResponse {
res := ErrorResponse{}
// readio := bytes.NewReader(body)
err := json.NewDecoder(body).Decode(&res)
if err != nil {
fmt.Printf("===!!!!==== ERROR Read Error : %s \n", err.Error())
}
fmt.Println("===> MESSAGE : ", res.Description)
return &res
}
func (c *hConn) streamToByte(stream io.Reader) []byte {
buf := new(bytes.Buffer)
buf.ReadFrom(stream)
return buf.Bytes()
}