/
mware.go
80 lines (70 loc) · 2.35 KB
/
mware.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
package grpc
import (
"context"
"github.com/configcat/configcat-proxy/log"
"google.golang.org/grpc"
"google.golang.org/grpc/codes"
"google.golang.org/grpc/metadata"
"google.golang.org/grpc/peer"
"google.golang.org/grpc/status"
"strings"
"time"
)
func DebugLogUnaryInterceptor(log log.Logger) grpc.UnaryServerInterceptor {
return func(ctx context.Context, req any, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (any, error) {
if shouldIgnore(info.FullMethod) {
return handler(ctx, req)
}
peerCtx, ok := peer.FromContext(ctx)
if !ok {
return nil, status.Errorf(codes.InvalidArgument, "missing peer info")
}
md, ok := metadata.FromIncomingContext(ctx)
if !ok {
return nil, status.Errorf(codes.InvalidArgument, "missing metadata")
}
start := time.Now()
log.Debugf("rpc starting %s [peer: %s] %s", info.FullMethod, peerCtx.Addr, md["user-agent"])
resp, err := handler(ctx, req)
stat, ok := status.FromError(err)
if !ok {
stat = status.FromContextError(err)
}
duration := time.Since(start)
log.Debugf("request finished %s [peer: %s] %s [code: %s] [duration: %dms]",
info.FullMethod, peerCtx.Addr, md["user-agent"], stat.Code().String(), duration.Milliseconds())
return resp, err
}
}
func DebugLogStreamInterceptor(log log.Logger) grpc.StreamServerInterceptor {
return func(srv any, ss grpc.ServerStream, info *grpc.StreamServerInfo, handler grpc.StreamHandler) error {
if shouldIgnore(info.FullMethod) {
return handler(srv, ss)
}
peerCtx, ok := peer.FromContext(ss.Context())
if !ok {
return status.Errorf(codes.InvalidArgument, "missing peer info")
}
md, ok := metadata.FromIncomingContext(ss.Context())
if !ok {
return status.Errorf(codes.InvalidArgument, "missing metadata")
}
start := time.Now()
log.Debugf("rpc starting %s [peer: %s] %s", info.FullMethod, peerCtx.Addr, md["user-agent"])
err := handler(srv, ss)
stat, ok := status.FromError(err)
if !ok {
stat = status.FromContextError(err)
}
duration := time.Since(start)
log.Debugf("request finished %s [peer: %s] %s [code: %s] [duration: %dms]",
info.FullMethod, peerCtx.Addr, md["user-agent"], stat.Code().String(), duration.Milliseconds())
return err
}
}
func shouldIgnore(method string) bool {
if strings.Contains(method, "grpc.health") || strings.Contains(method, "grpc.reflection") {
return true
}
return false
}