OpenTelemetry Transformation LanguageでGCPとアプリケーションのトレース情報を連携する

package main
 
import (
	"fmt"
 
  	gcppropagator "github.com/GoogleCloudPlatform/opentelemetry-operations-go/propagator"
	"github.com/gofiber/contrib/otelfiber/v2"
	"github.com/gofiber/fiber/v2"
	"go.opentelemetry.io/otel"
	"go.opentelemetry.io/otel/attribute"
	"go.opentelemetry.io/otel/propagation"
)
 
func main() {
 
  // エクスポーターの設定(省略)
 
	app := fiber.New()
	app.Use(traceMiddleware())
 
	// エンドポイントとハンドラーの設定(省略)
 
	if err := app.Listen(":3000"); err != nil {
		fmt.Printf("Error: %v", err)
	}
}
 
// ミドルウェアの設定
func traceMiddleware() fiber.Handler {
 
	otelMiddleware := otelfiber.Middleware(
		otelfiber.WithTracerProvider(otel.GetTracerProvider()),
 
		// middlewareが生成するスパンにattributeを設定
		otelfiber.WithCustomAttributes(func(c *fiber.Ctx) []attribute.KeyValue {
			return []attribute.KeyValue{
				// リクエストのデータをattributeとして追加
				attribute.String("http.remote_ip", c.IP()),
				attribute.String("http.referrer", string(c.Request().Header.Referer())),
				attribute.String("http.user_agent", string(c.Request().Header.UserAgent())),
 
				attribute.Bool("parent_span_is_lb", true), // middlewareで生成するスパンに印をつける
			}
		}),
		// スパンの名前を設定
		otelfiber.WithSpanNameFormatter(func(ctx *fiber.Ctx) string {
			return ctx.Method() + " " + ctx.Route().Path
		}),
 
		// コンテキストのpropagatorを設定
		otelfiber.WithPropagators(propagation.NewCompositeTextMapPropagator(
			gcppropagator.CloudTraceFormatPropagator{},
			propagation.TraceContext{},
			propagation.Baggage{},
		)),
	)
 
	// リクエスト毎にotelMiddlwareを実行するハンドラーを返す
	return func(c *fiber.Ctx) error {
 
		if err := otelMiddleware(c); err != nil {
			return err
		}
 
		// リクエストが既に完了している場合は処理をスキップ
		ctx := c.UserContext()
		if ctx.Err() != nil {
			return nil
		}
 
		// 次のミドルウェアを実行
		return c.Next()
	}
}

receivers:
  otlp:
    protocols:
      grpc:
        endpoint: 0.0.0.0:4317
 
processors:
  # otelFiberのMiddlewareで作成したスパンの親スパンIDを削除する
  transform:
    trace_statements:
      - context: span
        conditions:
          # 条件に一致するスパンの親スパン ID を持つ場合
          - span.attributes["parent_span_is_lb"] == true
        statements:
          # 条件に一致するスパンの親スパン ID を削除
          - set(span.parent_span_id, SpanID(0x0000000000000000))
          # span情報を編集したことを記録
          - set(span.status.message, "span edited by transform processor")
 
exporters:
  # GCP Cloud Trace/Monitoringへのエクスポーター
  googlecloud:
    trace: {}
 
service:
  pipelines:
    traces:
      receivers: [otlp]
      processors: [transform]
      exporters: [googlecloud]