/
interface.go
62 lines (52 loc) · 1.6 KB
/
interface.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
package nap
import (
"context"
"net/http"
"net/url"
"github.com/gojek/heimdall/v7/httpclient"
"github.com/gojek/heimdall/v7/plugins"
"go.uber.org/zap"
)
//go:generate mockgen --source=client.go -destination=./mocks/client.go -package=mocks
type Client interface {
Get(ctx context.Context, path string, request interface{}, header http.Header, out interface{}) (int, error)
Post(ctx context.Context, path string, request BodyProvider, header http.Header, out interface{}) (int, error)
}
type client struct {
// Logger
logger *zap.SugaredLogger
// HTTP Client, wrap by httpClient of heimdall
httpClient *httpclient.Client
// Host of service
baseHost string
// Response decoder/unmarshal
// Set default by JsonDecoder. Improve custom own options
responseDecoder ResponseDecoder
}
func New(baseHost string, proxyURL string, logger *zap.SugaredLogger, opts ...httpclient.Option) Client {
c := &client{
logger: logger,
baseHost: baseHost,
responseDecoder: jsonDecoder{},
httpClient: getHTTPClient(proxyURL, opts...),
}
return c
}
func getHTTPClient(proxyURL string, opts ...httpclient.Option) *httpclient.Client {
if proxyURL != "" {
opts = append(opts, httpclient.WithHTTPClient(&http.Client{
Transport: getTransport(proxyURL),
}))
}
httpClient := httpclient.NewClient(opts...)
requestLogger := plugins.NewRequestLogger(nil, nil)
httpClient.AddPlugin(requestLogger)
return httpClient
}
func getTransport(proxyURLCfg string) http.RoundTripper {
proxyURL, _ := url.Parse(proxyURLCfg)
transport := &http.Transport{
Proxy: http.ProxyURL(proxyURL),
}
return transport
}