如何在Golang中实现gRPC中间件-Golang-PHP中文网

如何在Golang中实现gRPC中间件

P粉602998670

发布： 2025-10-16 17:47:02

原创

756人浏览过

答案：Golang中通过Interceptor实现gRPC中间件，支持Unary和Streaming两种类型，可用于日志、认证、限流等通用逻辑。1. Unary Interceptor处理一元调用，接收上下文、请求、方法信息及处理器函数，示例中实现请求日志记录；2. Streaming Interceptor用于流式调用，需包装stream对象，示例记录流开始与结束日志；3. 可使用go-grpc-middleware库组合多个拦截器，如日志、认证、恢复等，通过ChainUnaryServer和ChainStreamServer链式注册；4. 常见中间件包括认证、日志、错误恢复、限流和监控，注意避免修改原始请求，推荐通过context或metadata传递数据。

如何在golang中实现grpc中间件

在Golang中实现gRPC中间件，主要是通过拦截请求和响应来添加通用逻辑，比如日志、认证、限流等。gRPC本身不提供“中间件”这一抽象概念，但支持通过 Interceptor 实现类似功能。你可以为 unary（一元）调用和 streaming（流式）调用分别定义拦截器。

1. Unary Interceptor（一元中间件）

Unary 拦截器用于处理普通的 RPC 调用。它是一个函数，接收上下文、方法名、请求、响应、目标客户端或服务端、以及一个执行后续处理的函数（handler）。

示例：实现一个简单的日志中间件

func LoggingInterceptor(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (interface{}, error) {
    log.Printf("Received request for %s", info.FullMethod)
    resp, err := handler(ctx, req)
    if err != nil {
        log.Printf("Error handling request: %v", err)
    } else {
        log.Printf("Successfully processed request")
    }
    return resp, err
}

登录后复制

注册到 gRPC 服务端：

<pre class="brush:php;toolbar:false;">s := grpc.NewServer(grpc.UnaryInterceptor(LoggingInterceptor))

登录后复制

2. Streaming Interceptor（流式中间件）

Streaming 拦截器用于处理流式 RPC。它更复杂一些，因为需要包装 stream 对象。

立即学习“go语言免费学习笔记（深入）”；

示例：流式日志拦截器

<pre class="brush:php;toolbar:false;">func LoggingStreamInterceptor(srv interface{}, ss grpc.ServerStream, info *grpc.StreamServerInfo, handler grpc.StreamHandler) error {
    log.Printf("Starting stream: %s", info.FullMethod)
    err := handler(srv, ss)
    if err != nil {
        log.Printf("Stream error: %v", err)
    }
    log.Printf("Stream finished")
    return err
}

登录后复制

注册到服务端：

百度文心百中

百度大模型语义搜索体验中心

查看详情

<pre class="brush:php;toolbar:false;">s := grpc.NewServer(grpc.StreamInterceptor(LoggingStreamInterceptor))

登录后复制

3. 组合多个中间件

可以使用第三方库如 go-grpc-middleware 来简化多个拦截器的组合。

安装：

<pre class="brush:php;toolbar:false;">go get github.com/grpc-ecosystem/go-grpc-middleware

登录后复制

组合使用：

<code>import "github.com/grpc-ecosystem/go-grpc-middleware"

myInterceptors := []grpc.UnaryServerInterceptor{
    LoggingInterceptor,
    AuthInterceptor,
    RecoveryInterceptor,
}

s := grpc.NewServer(
    grpc.UnaryInterceptor(grpc_middleware.ChainUnaryServer(myInterceptors...)),
    grpc.StreamInterceptor(grpc_middleware.ChainStreamServer(
        LoggingStreamInterceptor,
        AuthStreamInterceptor,
    )),
)

登录后复制