-
-
Notifications
You must be signed in to change notification settings - Fork 0
/
errorintercept.go
54 lines (42 loc) · 1.51 KB
/
errorintercept.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
package errorintercept
import (
"context"
"github.com/uber-go/tally/v4"
"go.uber.org/zap"
"google.golang.org/grpc"
"google.golang.org/protobuf/types/known/anypb"
"go.datalift.io/admiral/server/middleware"
)
const Name = "admiral.middleware.errorintercept"
type Interceptor interface {
InterceptError(error) error
}
type errorInterceptorFunc func(error) error
func New(cfg *anypb.Any, logger *zap.Logger, scope tally.Scope) (middleware.Middleware, error) {
return NewMiddleware(cfg, logger, scope)
}
func NewMiddleware(cfg *anypb.Any, logger *zap.Logger, scope tally.Scope) (*Middleware, error) {
return &Middleware{}, nil
}
type Middleware struct {
interceptors []errorInterceptorFunc
}
// AddInterceptor is used during gateway start to register any service that implements the Interceptor interface.
func (m *Middleware) AddInterceptor(fn errorInterceptorFunc) {
m.interceptors = append(m.interceptors, fn)
}
func (m *Middleware) UnaryInterceptor() grpc.UnaryServerInterceptor {
return func(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (interface{}, error) {
// Invoke handler.
resp, err := handler(ctx, req)
// Attempt to transform error if there was one.
if err != nil {
// Iterate in reverse order over each interceptor so the 'significant' foundational service's interceptors get applied last.
for i := len(m.interceptors) - 1; i >= 0; i-- {
// Apply interceptor and overwrite error.
err = m.interceptors[i](err)
}
}
return resp, err
}
}