This repository has been archived by the owner on Jan 7, 2023. It is now read-only.
/
client_interceptors.go
61 lines (53 loc) · 2.19 KB
/
client_interceptors.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
package grpc_slog
import (
"context"
"path"
"time"
"google.golang.org/grpc"
"cdr.dev/slog"
)
var (
// ClientField is used in every client-side log statement made through grpc_slog. Can be overwritten before initialization.
ClientField = slog.Field{"span.kind", "client"}
)
// UnaryClientInterceptor returns a new unary client interceptor that optionally logs the execution of external gRPC calls.
func UnaryClientInterceptor(logger slog.Logger, opts ...Option) grpc.UnaryClientInterceptor {
o := evaluateClientOpt(opts)
return func(ctx context.Context, method string, req, reply interface{}, cc *grpc.ClientConn, invoker grpc.UnaryInvoker, opts ...grpc.CallOption) error {
fields := newClientLoggerFields(ctx, method)
startTime := time.Now()
err := invoker(ctx, method, req, reply, cc, opts...)
logFinalClientLine(ctx, o, logger.With(fields...), startTime, err, "finished client unary call")
return err
}
}
// StreamClientInterceptor returns a new streaming client interceptor that optionally logs the execution of external gRPC calls.
func StreamClientInterceptor(logger slog.Logger, opts ...Option) grpc.StreamClientInterceptor {
o := evaluateClientOpt(opts)
return func(ctx context.Context, desc *grpc.StreamDesc, cc *grpc.ClientConn, method string, streamer grpc.Streamer, opts ...grpc.CallOption) (grpc.ClientStream, error) {
fields := newClientLoggerFields(ctx, method)
startTime := time.Now()
clientStream, err := streamer(ctx, desc, cc, method, opts...)
logFinalClientLine(ctx, o, logger.With(fields...), startTime, err, "finished client streaming call")
return clientStream, err
}
}
func logFinalClientLine(ctx context.Context, o *options, logger slog.Logger, startTime time.Time, err error, msg string) {
code := o.codeFunc(err)
level := o.levelFunc(code)
log(ctx, logger, level, msg,
slog.Error(err),
slog.F("grpc.code", code.String()),
o.durationFunc(time.Now().Sub(startTime)),
)
}
func newClientLoggerFields(ctx context.Context, fullMethodString string) []slog.Field {
service := path.Dir(fullMethodString)[1:]
method := path.Base(fullMethodString)
return []slog.Field{
SystemField,
ClientField,
slog.F("grpc.service", service),
slog.F("grpc.method", method),
}
}