3 | 3 |
"golang.org/x/net/context"
|
4 | 4 |
"google.golang.org/grpc"
|
5 | 5 |
|
|
6 |
"github.com/go-kit/kit/endpoint"
|
6 | 7 |
"github.com/go-kit/kit/examples/addsvc/pb"
|
7 | 8 |
"github.com/go-kit/kit/examples/addsvc/server"
|
8 | 9 |
"github.com/go-kit/kit/log"
|
|
10 |
grpctransport "github.com/go-kit/kit/transport/grpc"
|
9 | 11 |
)
|
10 | 12 |
|
11 | 13 |
// New returns an AddService that's backed by the provided ClientConn.
|
12 | 14 |
func New(ctx context.Context, cc *grpc.ClientConn, logger log.Logger) server.AddService {
|
13 | |
return client{ctx, pb.NewAddClient(cc), logger}
|
|
15 |
return client{
|
|
16 |
Context: ctx,
|
|
17 |
Logger: logger,
|
|
18 |
sum: grpctransport.NewClient(cc, "Add", "sum", encodeSumRequest, decodeSumResponse, pb.SumReply{}).Endpoint(),
|
|
19 |
concat: grpctransport.NewClient(cc, "Add", "concat", encodeConcatRequest, decodeConcatResponse, pb.ConcatReply{}).Endpoint(),
|
|
20 |
}
|
14 | 21 |
}
|
15 | 22 |
|
16 | 23 |
type client struct {
|
17 | 24 |
context.Context
|
18 | |
pb.AddClient
|
19 | 25 |
log.Logger
|
|
26 |
sum endpoint.Endpoint
|
|
27 |
concat endpoint.Endpoint
|
20 | 28 |
}
|
21 | 29 |
|
22 | 30 |
// TODO(pb): If your service interface methods don't return an error, we have
|
|
35 | 43 |
// important.
|
36 | 44 |
|
37 | 45 |
func (c client) Sum(a, b int) int {
|
38 | |
request := &pb.SumRequest{
|
39 | |
A: int64(a),
|
40 | |
B: int64(b),
|
|
46 |
request := &server.SumRequest{
|
|
47 |
A: a,
|
|
48 |
B: b,
|
41 | 49 |
}
|
42 | |
reply, err := c.AddClient.Sum(c.Context, request)
|
|
50 |
reply, err := c.sum(c.Context, request)
|
43 | 51 |
if err != nil {
|
44 | 52 |
c.Logger.Log("err", err) // Without an error return parameter, we can't do anything else...
|
45 | 53 |
return 0
|
46 | 54 |
}
|
47 | |
return int(reply.V)
|
|
55 |
|
|
56 |
r := reply.(server.SumResponse)
|
|
57 |
return r.V
|
48 | 58 |
}
|
49 | 59 |
|
50 | 60 |
func (c client) Concat(a, b string) string {
|
51 | |
request := &pb.ConcatRequest{
|
|
61 |
request := &server.ConcatRequest{
|
52 | 62 |
A: a,
|
53 | 63 |
B: b,
|
54 | 64 |
}
|
55 | |
reply, err := c.AddClient.Concat(c.Context, request)
|
|
65 |
reply, err := c.concat(c.Context, request)
|
56 | 66 |
if err != nil {
|
57 | |
c.Logger.Log("err", err)
|
|
67 |
c.Logger.Log("err", err) // Without an error return parameter, we can't do anything else...
|
58 | 68 |
return ""
|
59 | 69 |
}
|
60 | |
return reply.V
|
|
70 |
|
|
71 |
r := reply.(server.ConcatResponse)
|
|
72 |
return r.V
|
61 | 73 |
}
|