Skip to content

Commit 4b5ce77

Browse files
committed
cleaned up code formatting and some doc comments per PR review
1 parent 2a23fa6 commit 4b5ce77

File tree

4 files changed

+26
-32
lines changed

4 files changed

+26
-32
lines changed

endpoint/endpoint.go

Lines changed: 7 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -27,16 +27,14 @@ func Chain(outer Middleware, others ...Middleware) Middleware {
2727
}
2828
}
2929

30-
// Failer is an interface that should be implemented by response types that
31-
// hold error properties as to separate business errors from transport errors.
32-
// If the response type can hold business errors it is highly advised to
33-
// implement Failer.
34-
// Response encoders can check if responses are Failer, and if so if they've
35-
// failed encode them using a separate write path based on the error.
36-
// Endpoint middlewares can test if a response type failed and also act or
37-
// report upon it.
30+
// Failer may be implemented by Go kit response types that contain business
31+
// logic error details. If Failed returns a non-nil error, the Go kit transport
32+
// layer may interpret this as a business logic error, and may encode it
33+
// differently than a regular, successful response.
3834
//
39-
// The addsvc example shows Failer's intended usage.
35+
// It's not necessary for your response types to implement Failer, but it may
36+
// help for more sophisticated use cases. The addsvc example shows how Failer
37+
// should be used by a complete application.
4038
type Failer interface {
4139
Failed() error
4240
}

tracing/opencensus/endpoint_test.go

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -23,10 +23,10 @@ const (
2323
)
2424

2525
var (
26-
err1 = errors.New("someError")
27-
err2 = errors.New("otherError")
28-
err3 = errors.New("someBusinessError")
29-
err4 = errors.New("otherBusinessError")
26+
err1 = errors.New("some error")
27+
err2 = errors.New("other error")
28+
err3 = errors.New("some business error")
29+
err4 = errors.New("other business error")
3030
)
3131

3232
// compile time assertion

tracing/opencensus/grpc.go

Lines changed: 14 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -55,8 +55,7 @@ func GRPCClientTrace(options ...TracerOption) kitgrpc.ClientOption {
5555
clientFinalizer := kitgrpc.ClientFinalizer(
5656
func(ctx context.Context, err error) {
5757
if span := trace.FromContext(ctx); span != nil {
58-
s, ok := status.FromError(err)
59-
if ok {
58+
if s, ok := status.FromError(err); ok {
6059
span.SetStatus(trace.Status{Code: int32(s.Code()), Message: s.Message()})
6160
} else {
6261
span.SetStatus(trace.Status{Code: int32(codes.Unknown), Message: err.Error()})
@@ -70,7 +69,6 @@ func GRPCClientTrace(options ...TracerOption) kitgrpc.ClientOption {
7069
clientBefore(c)
7170
clientFinalizer(c)
7271
}
73-
7472
}
7573

7674
// GRPCServerTrace enables OpenCensus tracing of a Go kit gRPC transport server.
@@ -83,33 +81,33 @@ func GRPCServerTrace(options ...TracerOption) kitgrpc.ServerOption {
8381

8482
serverBefore := kitgrpc.ServerBefore(
8583
func(ctx context.Context, md metadata.MD) context.Context {
86-
var (
87-
spanContext trace.SpanContext
88-
ok bool
89-
name string
90-
)
84+
var name string
9185

9286
if cfg.name != "" {
9387
name = cfg.name
9488
} else {
95-
name, ok = ctx.Value(kitgrpc.ContextKeyRequestMethod).(string)
96-
if !ok || name == "" {
89+
name, _ = ctx.Value(kitgrpc.ContextKeyRequestMethod).(string)
90+
if name == "" {
9791
// we can't find the gRPC method. probably the
9892
// unaryInterceptor was not wired up.
9993
name = "unknown grpc method"
10094
}
10195
}
10296

103-
traceContext := md[propagationKey]
97+
var (
98+
parentContext trace.SpanContext
99+
traceContext = md[propagationKey]
100+
ok bool
101+
)
104102

105103
if len(traceContext) > 0 {
106104
traceContextBinary := []byte(traceContext[0])
107-
spanContext, ok = propagation.FromBinary(traceContextBinary)
105+
parentContext, ok = propagation.FromBinary(traceContextBinary)
108106
if ok && !cfg.public {
109107
ctx, _ = trace.StartSpanWithRemoteParent(
110108
ctx,
111109
name,
112-
spanContext,
110+
parentContext,
113111
trace.WithSpanKind(trace.SpanKindServer),
114112
trace.WithSampler(cfg.sampler),
115113
)
@@ -125,8 +123,8 @@ func GRPCServerTrace(options ...TracerOption) kitgrpc.ServerOption {
125123
if ok {
126124
span.AddLink(
127125
trace.Link{
128-
TraceID: spanContext.TraceID,
129-
SpanID: spanContext.SpanID,
126+
TraceID: parentContext.TraceID,
127+
SpanID: parentContext.SpanID,
130128
Type: trace.LinkTypeChild,
131129
},
132130
)
@@ -138,8 +136,7 @@ func GRPCServerTrace(options ...TracerOption) kitgrpc.ServerOption {
138136
serverFinalizer := kitgrpc.ServerFinalizer(
139137
func(ctx context.Context, err error) {
140138
if span := trace.FromContext(ctx); span != nil {
141-
s, ok := status.FromError(err)
142-
if ok {
139+
if s, ok := status.FromError(err); ok {
143140
span.SetStatus(trace.Status{Code: int32(s.Code()), Message: s.Message()})
144141
} else {
145142
span.SetStatus(trace.Status{Code: int32(codes.Internal), Message: err.Error()})

tracing/opencensus/http_test.go

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ import (
1919
kithttp "github.com/go-kit/kit/transport/http"
2020
)
2121

22-
func TestHttpClientTrace(t *testing.T) {
22+
func TestHTTPClientTrace(t *testing.T) {
2323
var (
2424
err error
2525
rec = &recordingExporter{}
@@ -38,7 +38,6 @@ func TestHttpClientTrace(t *testing.T) {
3838
}
3939

4040
for _, tr := range traces {
41-
4241
clientTracer := ockit.HTTPClientTrace(ockit.WithName(tr.name))
4342
ep := kithttp.NewClient(
4443
"GET",

0 commit comments

Comments
 (0)