feat: use robot to migrate code

Signed-off-by: kubbot & kubecub <3293172751ysy@gmail.com>
This commit is contained in:
kubbot & kubecub
2023-06-30 09:45:02 +08:00
parent 2d41819008
commit 539e0fdfb6
529 changed files with 64588 additions and 54413 deletions
+27
View File
@@ -0,0 +1,27 @@
package mw
import (
"context"
"google.golang.org/grpc"
)
func InterceptChain(intercepts ...grpc.UnaryServerInterceptor) grpc.UnaryServerInterceptor {
l := len(intercepts)
return func(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (resp interface{}, err error) {
chain := func(currentInter grpc.UnaryServerInterceptor, currentHandler grpc.UnaryHandler) grpc.UnaryHandler {
return func(currentCtx context.Context, currentReq interface{}) (interface{}, error) {
return currentInter(
currentCtx,
currentReq,
info,
currentHandler)
}
}
chainHandler := handler
for i := l - 1; i >= 0; i-- {
chainHandler = chain(intercepts[i], chainHandler)
}
return chainHandler(ctx, req)
}
}