巧用Metadata和RPC实现自定义认证，轻松掌控微服务的安全

2023-10-19 03:54:22

在浩瀚的分布式系统世界中，微服务架构凭借其模块化、可扩展性等优势，正日益成为构建现代应用程序的首选方案。然而，随着微服务数量的不断增加，如何确保它们之间的通信安全也成为了一项亟需解决的挑战。

传统上，我们在 HTTP/1.1 中常常通过直接操纵 Header 来传递数据，而对于 gRPC 来讲，它基于 HTTP/2 协议，本质上也可是通过 Header 来进行传递，但我们不会直接的去操纵它，而是通过更加优雅的方式——Metadata。

Metadata 作为一种元数据，可以承载各种各样的信息，例如用户 ID、角色、请求来源等。而 RPC（Remote Procedure Call）作为一种远程过程调用机制，允许客户端调用位于不同机器上的服务端方法。将 Metadata 与 RPC 相结合，我们可以轻松实现微服务之间的自定义认证。

首先，我们需要在服务端拦截器中提取 Metadata 中的认证信息，并对其进行验证。如果认证通过，则允许请求继续执行，否则拒绝请求。

func AuthInterceptor(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (interface{}, error) {
    // 从 Metadata 中提取认证信息
    md, ok := metadata.FromIncomingContext(ctx)
    if !ok {
        return nil, grpc.Errorf(codes.Unauthenticated, "missing metadata")
    }
    token := md.Get("authorization")
    if token == "" {
        return nil, grpc.Errorf(codes.Unauthenticated, "missing token")
    }

    // 验证认证信息
    claims, err := verifyToken(token)
    if err != nil {
        return nil, grpc.Errorf(codes.Unauthenticated, "invalid token: %v", err)
    }

    // 将认证信息附加到上下文中
    newCtx := context.WithValue(ctx, "claims", claims)

    // 继续执行请求
    return handler(newCtx, req)
}

在客户端，我们需要在发送请求之前将认证信息添加到 Metadata 中。

func SendRequest(ctx context.Context, client MyServiceClient, req *MyRequest) (*MyResponse, error) {
    // 从上下文中提取认证信息
    claims := ctx.Value("claims").(Claims)

    // 将认证信息添加到 Metadata 中
    md := metadata.New(map[string]string{
        "authorization": claims.Token,
    })
    ctx = metadata.NewOutgoingContext(ctx, md)

    // 发送请求
    return client.MyMethod(ctx, req)
}