如何在Golang中实现gRPC中间件

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

在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 服务端：

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

2. Streaming Interceptor（流式中间件）

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

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

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
}

注册到服务端：

自动隐藏的QQ在线客服代码

自动隐藏的QQ在线客服代码是一款基于jQuery实现的默认展开可自动隐藏的QQ在线客服代码，界面非常简洁，此款JS代码特效可按客服功能放置不同的客服QQ号码，鼠标移上去即展开客服信息，移走自动隐藏，如果需要修改其中客服信息，修改service.js文件即可。

下载

s := grpc.NewServer(grpc.StreamInterceptor(LoggingStreamInterceptor))

3. 组合多个中间件

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

安装：

go get github.com/grpc-ecosystem/go-grpc-middleware

组合使用：

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,
    )),
)