/
endpoints.go
117 lines (98 loc) · 2.92 KB
/
endpoints.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
114
115
116
117
package ticketsvc
import (
"context"
"net/url"
"strings"
"github.com/go-kit/kit/endpoint"
httptransport "github.com/go-kit/kit/transport/http"
)
type EndpointSet struct {
GetEndpoint endpoint.Endpoint
SetEndpoint endpoint.Endpoint
IncrementEndpoint endpoint.Endpoint
}
func MakeServerEndpoints(svc Service) EndpointSet {
return EndpointSet{
GetEndpoint: MakeGetEndpoint(svc),
SetEndpoint: MakeSetEndpoint(svc),
IncrementEndpoint: MakeIncrementEndpoint(svc),
}
}
func MakeClientEndpoints(instance string) (EndpointSet, error) {
if !strings.HasPrefix(instance, "http") {
instance = "http://" + instance
}
tgt, err := url.Parse(instance)
if err != nil {
return EndpointSet{}, err
}
tgt.Path = ""
options := []httptransport.ClientOption{}
return EndpointSet{
GetEndpoint: httptransport.NewClient("GET", tgt, encodeGetRequest, decodeResponse, options...).Endpoint(),
SetEndpoint: httptransport.NewClient("PUT", tgt, encodeSetRequest, decodeResponse, options...).Endpoint(),
IncrementEndpoint: httptransport.NewClient("POST", tgt, encodeIncrementRequest, decodeResponse, options...).Endpoint(),
}, nil
}
func (e *EndpointSet) Get(ctx context.Context, ids ...int) ([]Tickets, error) {
request := getRequest{Ids: ids}
r, err := e.GetEndpoint(ctx, request)
if err != nil {
return nil, err
}
resp := r.(response)
return resp.Tickets, nil
}
func (e *EndpointSet) Set(ctx context.Context, tickets ...Tickets) ([]Tickets, error) {
request := setRequest{Tickets: tickets}
r, err := e.SetEndpoint(ctx, request)
if err != nil {
return nil, err
}
resp := r.(response)
return resp.Tickets, nil
}
func (e *EndpointSet) Increment(ctx context.Context, ids ...int) ([]Tickets, error) {
request := incrementRequest{Ids: ids}
r, err := e.IncrementEndpoint(ctx, request)
if err != nil {
return nil, err
}
resp := r.(response)
return resp.Tickets, nil
}
func MakeGetEndpoint(svc Service) endpoint.Endpoint {
return func(ctx context.Context, request interface{}) (interface{}, error) {
req := request.(getRequest)
tickets, err := svc.Get(ctx, req.Ids...)
return response{tickets, err}, nil
}
}
func MakeSetEndpoint(svc Service) endpoint.Endpoint {
return func(ctx context.Context, request interface{}) (interface{}, error) {
req := request.(setRequest)
tickets, err := svc.Set(ctx, req.Tickets...)
return response{tickets, err}, nil
}
}
func MakeIncrementEndpoint(svc Service) endpoint.Endpoint {
return func(ctx context.Context, request interface{}) (interface{}, error) {
req := request.(incrementRequest)
tickets, err := svc.Increment(ctx, req.Ids...)
return response{tickets, err}, nil
}
}
type getRequest struct {
Ids []int
}
type incrementRequest struct {
Ids []int
}
type setRequest struct {
Tickets []Tickets
}
type response struct {
Tickets []Tickets `json:"tickets,omitempty"`
Err error `json:"err,omitempty"`
}
func (r response) error() error { return r.Err }