Golang RPC拦截器使用与调用链追踪示例-Golang-PHP中文网

Golang RPC拦截器使用与调用链追踪示例

P粉602998670

发布： 2025-09-25 10:33:01

原创

230人浏览过

Go语言gRPC中通过拦截器结合OpenTelemetry实现调用链追踪，1. 使用otelgrpc内置拦截器自动创建Span并传递上下文；2. 自定义拦截器添加业务标签如用户ID；3. 初始化TracerProvider并配置Jaeger等导出器，完成链路追踪。

golang rpc拦截器使用与调用链追踪示例

在 Go 语言的 RPC 系统中，拦截器（Interceptor）是一种强大的机制，用于在请求处理前后插入通用逻辑，比如日志记录、认证、超时控制以及调用链追踪。结合 OpenTelemetry 或 Jaeger 等可观测性工具，可以实现分布式系统中的调用链路追踪。

拦截器的基本概念

Go 标准库本身不直接提供拦截器机制，但在使用 gRPC（Google 的高性能 RPC 框架）时，可以通过 UnaryInterceptor 和 StreamInterceptor 实现拦截功能。这里以 gRPC 为例说明如何使用拦截器进行调用链追踪。

拦截器本质上是一个中间函数，它在客户端发送请求前或服务端接收请求后执行额外操作。对于调用链追踪，我们可以在请求开始时创建 Span，在结束时结束 Span，并将上下文传递给下游服务。

集成 OpenTelemetry 实现调用链追踪

OpenTelemetry 是目前主流的可观测性框架，支持自动和手动埋点。下面是一个使用 OpenTelemetry 手动注入追踪信息并通过拦截器传递的例子。

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

1. 安装依赖：

需要引入 gRPC 和 OpenTelemetry 相关包：

go get google.golang.org/grpc
go get go.opentelemetry.io/otel
go get go.opentelemetry.io/contrib/instrumentation/google.golang.org/grpc/otelgrpc

登录后复制

2. 初始化 TracerProvider：

在程序启动时配置 exporter（如 Jaeger、OTLP）和 tracer provider。

Reclaim.ai

为优先事项创建完美的时间表

查看详情

import (
    "context"
    "go.opentelemetry.io/otel"
    "go.opentelemetry.io/otel/exporters/jager"
    "go.opentelemetry.io/otel/sdk/resource"
    sdktrace "go.opentelemetry.io/otel/sdk/trace"
    semconv "go.opentelemetry.io/otel/semconv/v1.17.0"
)

func initTracer() (*sdktrace.TracerProvider, error) {
    exporter, err := jager.New(jager.WithCollectorEndpoint())
    if err != nil {
        return nil, err
    }

    tp := sdktrace.NewTracerProvider(
        sdktrace.WithBatcher(exporter),
        sdktrace.WithResource(resource.NewWithAttributes(
            semconv.SchemaURL,
            semconv.ServiceNameKey.String("my-rpc-service"),
        )),
        sdktrace.WithSampler(sdktrace.AlwaysSample()),
    )

    otel.SetTracerProvider(tp)
    return tp, nil
}

登录后复制

在 gRPC 中注册拦截器

使用 otelgrpc 提供的拦截器自动完成 span 的创建与上下文传播。

1. 服务端拦截器：

import "go.opentelemetry.io/contrib/instrumentation/google.golang.org/grpc/otelgrpc"

server := grpc.NewServer(
    grpc.UnaryInterceptor(otelgrpc.UnaryServerInterceptor()),
    grpc.StreamInterceptor(otelgrpc.StreamServerInterceptor()),
)

登录后复制

2. 客户端拦截器：

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

登录后复制

这样，每次调用都会自动生成 span，并通过 HTTP 头（如 W3C TraceContext）传递 trace-id 和 span-id，实现跨服务链路追踪。

自定义拦截器添加业务标签

如果需要在 span 上附加业务相关的信息（如用户 ID、请求参数），可以编写自定义拦截器。

func tracingUnaryServerInterceptor() grpc.UnaryServerInterceptor {
    return func(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (interface{}, error) {
        // 获取当前 span
        span := otel.GetTracerProvider().Tracer("custom").Start(ctx, info.FullMethod)
        ctx, _ = span.Start(ctx, "business-logic")

        // 示例：添加自定义属性
        span.SetAttributes(attribute.String("user.id", extractUserIDFromRequest(req)))

        // 调用实际处理函数
        resp, err := handler(ctx, req)

        // 结束 span
        span.End()
        return resp, err
    }
}

// 注册时叠加拦截器
server := grpc.NewServer(
    grpc.UnaryInterceptor(func(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (interface{}, error) {
        return tracingUnaryServerInterceptor()(ctx, req, info, handler)
    }),
)

登录后复制

注意：多个拦截器可使用 grpc-middleware 库进行组合，避免嵌套过深。

基本上就这些。通过合理使用拦截器配合 OpenTelemetry，能轻松实现 Go 服务间的调用链追踪，提升系统可观测性。

以上就是Golang RPC拦截器使用与调用链追踪示例的详细内容，更多请关注php中文网其它相关文章！