Golang使用gRPC拦截器处理请求示例-Golang-PHP中文网

Golang使用gRPC拦截器处理请求示例

P粉602998670

发布： 2025-09-13 12:24:01

原创

1023人浏览过

使用gRPC拦截器可统一处理日志、认证等逻辑，无需修改业务代码。2. 一元拦截器通过grpc.UnaryServerInterceptor实现，用于记录请求耗时与日志。3. 流式拦截器通过grpc.StreamServerInfo处理流式RPC调用。4. 在grpc.NewServer时注册拦截器选项。5. 客户端可通过grpc.WithUnaryInterceptor添加认证头信息，提升可观测性与安全性。

golang使用grpc拦截器处理请求示例

在Go语言中使用gRPC拦截器可以统一处理请求前后的逻辑，比如日志记录、认证、错误处理等。通过拦截器，你可以在不修改业务代码的前提下增强服务行为。下面是一个使用gRPC一元拦截器和流式拦截器的简单示例。

1. 定义并实现一元拦截器

一元拦截器用于处理普通的RPC调用（非流式）。你可以用它来记录请求耗时、打印日志或进行身份验证。

注意：拦截器函数需要符合

grpc.UnaryServerInterceptor

登录后复制

类型定义。

示例：实现一个日志和耗时统计的一元拦截器

func loggingUnaryInterceptor(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (interface{}, error) {
    start := time.Now()
    log.Printf("Received unary request: %s", info.FullMethod)

    resp, err := handler(ctx, req)

    log.Printf("Completed unary request: %s, duration: %v, error: %v",
        info.FullMethod, time.Since(start), err)
    return resp, err
}

2. 定义并实现流式拦截器

流式拦截器用于处理客户端流、服务端流或双向流的gRPC方法。

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

芦笋演示

一键出成片的录屏演示软件，专为制作产品演示、教学课程和使用教程而设计。

查看详情

示例：一个简单的流式拦截器

func loggingStreamInterceptor(srv interface{}, ss grpc.ServerStream, info *grpc.StreamServerInfo, handler grpc.StreamHandler) error {
    start := time.Now()
    log.Printf("Received stream request: %s", info.FullMethod)

    err := handler(srv, ss)

    log.Printf("Completed stream request: %s, duration: %v, error: %v",
        info.FullMethod, time.Since(start), err)
    return err
}

3. 注册拦截器到gRPC服务器

在启动gRPC服务器时，通过

grpc.UnaryInterceptor

登录后复制

和

grpc.StreamInterceptor

登录后复制

选项注册拦截器。

func main() {
    lis, err := net.Listen("tcp", ":50051")
    if err != nil {
        log.Fatalf("failed to listen: %v", err)
    }

    opts := []grpc.ServerOption{
        grpc.UnaryInterceptor(loggingUnaryInterceptor),
        grpc.StreamInterceptor(loggingStreamInterceptor),
    }

    s := grpc.NewServer(opts...)
    pb.RegisterYourServiceServer(s, &server{})

    log.Println("gRPC server listening on :50051")
    if err := s.Serve(lis); err != nil {
        log.Fatalf("failed to serve: %v", err)
    }
}

4. 客户端也可使用拦截器

如果你希望在客户端也做类似操作（如添加认证头），可以设置客户端拦截器。

func authClientInterceptor(ctx context.Context, method string, req, reply interface{},
    cc *grpc.ClientConn, invoker grpc.UnaryInvoker, opts ...grpc.CallOption) error {
    newCtx := metadata.AppendToOutgoingContext(ctx, "authorization", "Bearer <token>")
    return invoker(newCtx, method, req, reply, cc, opts...)
}

// 使用
conn, err := grpc.Dial(address, grpc.WithInsecure(),
    grpc.WithUnaryInterceptor(authClientInterceptor)) 基本上就这些。拦截器是gRPC中非常实用的功能，合理使用能极大提升代码可维护性和可观测性。

以上就是Golang使用gRPC拦截器处理请求示例的详细内容，更多请关注php中文网其它相关文章！