อินเตอร์เซปเตอร์และข้อมูลเมทาดาทาของ gRPC

อินเตอร์เซปเตอร์คืออะไร

อินเตอร์เซปเตอร์ของเซิร์ฟเวอร์แบบเอกภาค

func LoggingInterceptor(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (interface{}, error) {
    log.Printf("RPC: %s", info.FullMethod)
    resp, err := handler(ctx, req)
    log.Printf("done: %v", err)
    return resp, err
}

grpcSrv := grpc.NewServer(grpc.UnaryInterceptor(LoggingInterceptor))

อินเตอร์เซปเตอร์ของเซิร์ฟเวอร์แบบสตรีม

func StreamLogging(srv interface{}, ss grpc.ServerStream, info *grpc.StreamServerInfo, handler grpc.StreamHandler) error {
    log.Printf("stream: %s", info.FullMethod)
    return handler(srv, ss)
}

การเชื่อมอินเตอร์เซปเตอร์

grpcSrv := grpc.NewServer(
    grpc.ChainUnaryInterceptor(LoggingInterceptor, AuthInterceptor, MetricsInterceptor),
)

อินเตอร์เซปเตอร์ของไคลเอนต์

conn, _ := grpc.NewClient(addr,
    grpc.WithUnaryInterceptor(ClientAuthInterceptor),
    grpc.WithTransportCredentials(insecure.NewCredentials()),
)

ข้อมูลเมตาของ gRPC

md := metadata.New(map[string]string{"authorization": "Bearer " + token})
ctx := metadata.NewOutgoingContext(context.Background(), md)

การอ่านข้อมูลเมตาบนเซิร์ฟเวอร์

md, ok := metadata.FromIncomingContext(ctx)
if !ok { return nil, status.Error(codes.Unauthenticated, "no metadata") }
token := md["authorization"]

การส่งข้อมูลเมตาจากเซิร์ฟเวอร์

grpc.SetHeader(ctx, metadata.Pairs("x-request-id", requestID))

go-grpc-middleware

รูปแบบอินเตอร์เซปเตอร์สำหรับการยืนยันตัวตน

func AuthInterceptor(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (interface{}, error) {
    md, _ := metadata.FromIncomingContext(ctx)
    if !validateToken(md["authorization"]) {
        return nil, status.Error(codes.Unauthenticated, "invalid token")
    }
    return handler(ctx, req)
}

อินเตอร์เซปเตอร์สำหรับกู้คืนจากแพนิก

defer func() {
    if r := recover(); r != nil {
        err = status.Errorf(codes.Internal, "panic: %v", r)
    }
}()

ตรวจสอบความเข้าใจอย่างรวดเร็ว

สรุป: อินเตอร์เซปเตอร์และข้อมูลเมตาของ gRPC