generated from pkg/go-template
38 lines
1.3 KiB
Go
38 lines
1.3 KiB
Go
|
package rpc
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"fmt"
|
|||
|
|
"git.icechen.cn/monorepo/backend/pkg/env"
|
|||
|
|
"github.com/gofiber/fiber/v2"
|
|||
|
|
ctxLogger "github.com/luizsuper/ctxLoggers"
|
|||
|
|
"go.uber.org/zap"
|
|||
|
|
"google.golang.org/grpc"
|
|||
|
|
"google.golang.org/grpc/metadata"
|
|||
|
|
"strings"
|
|||
|
|
"time"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// Interceptor 将requestId 存入ctx，记录出入参数，记录时间
|
|||
|
|
func Interceptor(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo,
|
|||
|
|
handler grpc.UnaryHandler) (interface{}, error) {
|
|||
|
|
serverName := env.Namespace + "-" + env.AppName + "-" + "service"
|
|||
|
|
startTime := time.Now()
|
|||
|
|
md, ok := metadata.FromIncomingContext(ctx)
|
|||
|
|
if !ok {
|
|||
|
|
fmt.Printf("get metadata error")
|
|||
|
|
}
|
|||
|
|
if t, ok := md[strings.ToLower(fiber.HeaderXRequestID)]; ok {
|
|||
|
|
ctx = context.WithValue(ctx, fiber.HeaderXRequestID, t[0])
|
|||
|
|
}
|
|||
|
|
ctxLogger.Debug(ctx, serverName, zap.String("gRPC method", fmt.Sprintf(" %s", info.FullMethod)), zap.Any("req", req))
|
|||
|
|
resp, err := handler(ctx, req)
|
|||
|
|
costSeconds := time.Since(startTime).Seconds()
|
|||
|
|
if err != nil {
|
|||
|
|
ctxLogger.Debug(ctx, serverName, zap.String("gRPC method", fmt.Sprintf(" %s", info.FullMethod)), zap.Float64("cost_time", costSeconds), zap.String("errorReason", err.Error()))
|
|||
|
|
} else {
|
|||
|
|
ctxLogger.Debug(ctx, serverName, zap.String("gRPC method", fmt.Sprintf(" %s, %v", info.FullMethod, resp)), zap.Float64("cost_time", costSeconds), zap.Any("resp", resp))
|
|||
|
|
}
|
|||
|
|
return resp, err
|
|||
|
|
}
|