forked from devimteam/microgen
/
client.microgen.go
50 lines (45 loc) · 1.46 KB
/
client.microgen.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
// Code generated by microgen 0.9.0. DO NOT EDIT.
package transport
import (
"context"
"errors"
opentracing "github.com/go-kit/kit/tracing/opentracing"
opentracinggo "github.com/opentracing/opentracing-go"
codes "google.golang.org/grpc/codes"
status "google.golang.org/grpc/status"
)
// TraceClientEndpoints is used for tracing endpoints on client side.
func TraceClientEndpoints(endpoints EndpointsSet, tracer opentracinggo.Tracer) EndpointsSet {
return EndpointsSet{
ConcatEndpoint: opentracing.TraceClient(tracer, "Concat")(endpoints.ConcatEndpoint),
SumEndpoint: opentracing.TraceClient(tracer, "Sum")(endpoints.SumEndpoint),
}
}
func (set EndpointsSet) Sum(arg0 context.Context, arg1 int, arg2 int) (res0 int, res1 error) {
request := SumRequest{
A: arg1,
B: arg2,
}
response, res1 := set.SumEndpoint(arg0, &request)
if res1 != nil {
if e, ok := status.FromError(res1); ok || e.Code() == codes.Internal || e.Code() == codes.Unknown {
res1 = errors.New(e.Message())
}
return
}
return response.(*SumResponse).Result, res1
}
func (set EndpointsSet) Concat(arg0 context.Context, arg1 string, arg2 string) (res0 string, res1 error) {
request := ConcatRequest{
A: arg1,
B: arg2,
}
response, res1 := set.ConcatEndpoint(arg0, &request)
if res1 != nil {
if e, ok := status.FromError(res1); ok || e.Code() == codes.Internal || e.Code() == codes.Unknown {
res1 = errors.New(e.Message())
}
return
}
return response.(*ConcatResponse).Result, res1
}