如何在Golang中使用gRPC拦截器_Golang gRPC拦截器使用方法

gRPC拦截器是Go中用于在请求前后插入通用逻辑的钩子函数，分为服务端和客户端的一元与流式拦截器。通过grpc.UnaryInterceptor注册服务端拦截器可实现日志、错误恢复等，如loggingInterceptor记录请求信息；客户端拦截器如authInterceptor可添加认证头。多个拦截器可通过grpc-middleware.ChainUnaryServer组合使用，便于统一处理认证、监控等横切关注点，提升代码复用性与可维护性。

在Go语言中使用gRPC拦截器可以让你在请求处理前后插入通用逻辑，比如日志记录、认证、监控、错误处理等。gRPC本身不直接提供拦截器功能，但通过 grpc-go 提供的中间件机制（即拦截器）可以轻松实现。

什么是gRPC拦截器

gRPC拦截器是一种钩子函数，可以在gRPC方法执行前或后运行。分为两种类型：

Unary Interceptor：用于处理一元调用（普通请求-响应模式）
Stream Interceptor：用于处理流式调用（客户端流、服务端流、双向流）

你可以为客户端和服务端分别设置拦截器。

服务端一元拦截器使用方法

定义一个服务端一元拦截器函数，其签名必须符合 grpc.UnaryServerInterceptor 类型：

示例：实现一个简单的日志和错误恢复拦截器

func loggingInterceptor(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (resp interface{}, err error) {
    // 请求前
    log.Printf("Received request: %s", info.FullMethod)
    
    // 处理请求
    resp, err = handler(ctx, req)
    
    // 请求后
    if err != nil {
        log.Printf("Error handling request: %v", err)
    } else {
        log.Printf("Request completed successfully")
    }
    
    return resp, err
}

注册到gRPC服务器：

server := grpc.NewServer(grpc.UnaryInterceptor(loggingInterceptor))
pb.RegisterYourServiceServer(server, &yourService{})

客户端一元拦截器使用方法

客户端拦截器可用于添加认证头、重试、超时等逻辑。

定义一个客户端一元拦截器：

func authInterceptor(ctx context.Context, method string, req, reply interface{}, cc *grpc.ClientConn, invoker grpc.UnaryInvoker, opts ...grpc.CallOption) error {
    // 添加认证信息到上下文
    ctx = metadata.AppendToOutgoingContext(ctx, "authorization", "Bearer your-token")
    
    // 调用实际的RPC
    return invoker(ctx, method, req, reply, cc, opts...)
}

创建客户端时使用：

conn, err := grpc.Dial("localhost:50051",
    grpc.WithInsecure(),
    grpc.WithUnaryInterceptor(authInterceptor),
)
if err != nil {
    log.Fatal(err)
}

多个拦截器的组合

如果你需要多个拦截器，可以使用第三方库如 github.com/grpc-ecosystem/go-grpc-middleware 来链式组合。

安装：

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

使用示例：

import "github.com/grpc-ecosystem/go-grpc-middleware"

// 组合一元拦截器
unaryInterceptors := []grpc.UnaryServerInterceptor{
    loggingInterceptor,
    recoveryInterceptor, // 恢复panic
}

server := grpc.NewServer(
    grpc.UnaryInterceptor(grpc_middleware.ChainUnaryServer(unaryInterceptors...)),
)

这个方式让代码更清晰，易于维护。

基本上就这些。掌握拦截器后，你可以统一处理横切关注点，避免在每个方法中重复写日志、认证等逻辑。关键是理解拦截器的函数签名和执行时机，然后根据业务需求封装通用功能。不复杂但容易忽略细节，比如上下文传递和错误处理。

# golang # git # go # github # go语言 # app # ai # stream # 代码复用 # cos # 中间件 # 封装

返回目录在线咨询

上一篇：Android中ProgressDialog的dismiss()与cancel()方法的区别
下一篇：多用户智能建站平台支持哪些类型的网站构建？

您的项目需求