-
Notifications
You must be signed in to change notification settings - Fork 6
/
transport.go
85 lines (71 loc) · 1.62 KB
/
transport.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
package signalr
import (
"net/http"
"net/url"
"time"
"github.com/rs/zerolog"
"github.com/rs/zerolog/log"
"github.com/pkg/errors"
)
// Version - protocol version
type Version string
// Versions
const (
Version0 Version = "0"
Version1 Version = "1"
)
// Transport -
type Transport struct {
url string
client *http.Client
log zerolog.Logger
}
// NewTransport -
func NewTransport(baseURL string) *Transport {
t := http.DefaultTransport.(*http.Transport).Clone()
t.MaxIdleConns = 100
t.MaxConnsPerHost = 100
t.MaxIdleConnsPerHost = 100
return &Transport{
url: baseURL,
client: &http.Client{
Timeout: time.Minute,
Transport: t,
},
log: log.Logger,
}
}
// Negotiate - is used to establish a connection between the client and the server.
func (t *Transport) Negotiate(version Version) (response NegotiateResponse, err error) {
u, err := url.Parse(t.url)
if err != nil {
return
}
u.Path += "/negotiate"
q := u.Query()
q.Set("negotiateVersion", string(version))
u.RawQuery = q.Encode()
t.log.Debug().Str("url", u.String()).Msg("send negotiate request...")
req, err := http.NewRequest(http.MethodPost, u.String(), nil)
if err != nil {
return
}
resp, err := t.client.Do(req)
if err != nil {
return
}
defer resp.Body.Close()
switch resp.StatusCode {
case http.StatusOK:
err = json.NewDecoder(resp.Body).Decode(&response)
return
case http.StatusInternalServerError:
var e Error
if err = json.NewDecoder(resp.Body).Decode(&e); err != nil {
return
}
return response, errors.Wrap(ErrNegotiate, e.Error)
default:
return response, errors.Wrapf(ErrInvalidStatusCode, resp.Status)
}
}