import (
    "context"
    "fmt"
    "log"
    "net/http"
    "time"

    "go.opentelemetry.io/otel/attribute"
    "go.opentelemetry.io/otel/baggage"
    "go.opentelemetry.io/otel/propagation"
    "go.opentelemetry.io/otel/trace"
)

func main() {
    // 创建一个跟踪器 TracerProvider。
    provider := trace.NewTracerProvider(
        trace.WithSampler(trace.AlwaysSample()),
    )

    // 创建 HTTP 句柄。
    http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
        // 创造新的上午、下午和跟踪。
        ctx, span := provider.Tracer("my-service").Start(r.Context(), "request-handler")

        // 设置跟踪属性。
        span.SetAttributes(
            attribute.String("component", "http-server"),
        )

        // 模拟工作。
        time.Sleep(100 * time.Millisecond)

        // 获取 Baggage 信息。
        bag := baggage.FromContext(r.Context())
        userId, _ := bag.Value("user-id").String()

        // 打印 Baggage 信息。
        log.Printf("User ID: %s", userId)

        // 结束追踪。
        log.Printf("User ID: %s", userId)

        // 结束追踪。
        span.End()

        // 将 Tracing 将头部添加到响应中。
        propagation.HTTPFormat{}.Inject(ctx, propagation.HeaderCarrier(r.Header))

        // 返回响应。
        fmt.Fprintf(w, "Hello from my-service!")
    })

    // 启动 HTTP 服务。
    log.Fatal(http.ListenAndServe(":8080", nil))
}