gRPC 中的中间件，你想知道的一切都在这里！

前端

2023-12-08 20:29:57

在过去的几年中，gRPC 已经成为构建现代微服务应用程序的首选框架之一。凭借其出色的性能和可靠性，gRPC 已经赢得了众多开发者的青睐。然而，对于许多开发者来说，gRPC 中间件仍然是一个相对陌生的概念。

什么是 gRPC 中间件？

gRPC 中间件是一个允许您在 gRPC 服务的请求和响应之间插入自定义逻辑的软件层。这使得您可以轻松地添加各种功能，例如身份验证、日志记录、监控和限流。

gRPC 中间件的类型

gRPC 中间件主要分为两大类：

一元拦截器 ：一元拦截器在每个 RPC 请求和响应上运行。这使得它们非常适合于需要在每个请求上执行的通用任务，例如身份验证和日志记录。
服务端/客户端拦截器 ：服务端/客户端拦截器仅在服务端或客户端上运行。这使得它们非常适合于需要在特定环境中执行的任务，例如限流和负载均衡。

如何使用 gRPC 中间件

使用 gRPC 中间件非常简单。您只需在您的 gRPC 服务中注册所需的中间件即可。您可以通过以下方式注册中间件：

在 gRPC 服务的构造函数中注册中间件：

import (
	"context"

	"github.com/grpc-ecosystem/go-grpc-middleware"
	"google.golang.org/grpc"
)

type GreeterService struct {
	// ...
}

func NewGreeterService() *GreeterService {
	return &GreeterService{}
}

func (s *GreeterService) SayHello(ctx context.Context, req *pb.HelloRequest) (*pb.HelloReply, error) {
	// ...
}

func main() {
	// 注册中间件
	opts := []grpc.ServerOption{
		grpc_middleware.WithUnaryServerChain(
			grpc_ctxtags.UnaryServerInterceptor(),
			grpc_prometheus.UnaryServerInterceptor(),
		),
	}
	// ...
}

在 gRPC 服务的 Serve() 方法中注册中间件：

import (
	"context"

	"github.com/grpc-ecosystem/go-grpc-middleware"
	"google.golang.org/grpc"
)

type GreeterService struct {
	// ...
}

func NewGreeterService() *GreeterService {
	return &GreeterService{}
}

func (s *GreeterService) SayHello(ctx context.Context, req *pb.HelloRequest) (*pb.HelloReply, error) {
	// ...
}

func main() {
	s := NewGreeterService()
	// 注册中间件
	opts := []grpc.ServerOption{
		grpc_middleware.WithUnaryServerChain(
			grpc_ctxtags.UnaryServerInterceptor(),
			grpc_prometheus.UnaryServerInterceptor(),
		),
	}
	grpc.EnableTracing = true
	server := grpc.NewServer(opts...)
	pb.RegisterGreeterServer(server, s)
	// ...
}