8.【go-kit教程】go-kit链路追踪

• 在Go kit中，可以使用OpenTelemetry来实现链路追踪功能。

• OpenTelemetry是一个云原生的分布式应用追踪系统，可以帮助开发人员跟踪和分析分布式系统的性能和行为。它提供了标准化的API和规范，可以在不同的编程语言和框架中实现链路追踪功能。

• 在Go kit中，可以使用 github.com/open-telemetry/opentelemetry-go 和 github.com/open-telemetry/opentelemetry-go-contrib 包来集成OpenTelemetry。

  import (
      "go.opentelemetry.io/otel"
      "go.opentelemetry.io/otel/propagation"
      "go.opentelemetry.io/otel/exporters/trace/jaeger"
      "go.opentelemetry.io/otel/sdk/trace"
      otelcontrib "go.opentelemetry.io/contrib"
      "github.com/go-kit/kit/endpoint"
  )
  
  func main() {
      // 创建一个tracer
      exporter, err := jaeger.NewRawExporter(jaeger.WithCollectorEndpoint("http://localhost:14268/api/traces"))
      if err != nil {
          log.Fatal(err)
      }
      tp := trace.NewTracerProvider(trace.WithBatcher(exporter))
      otel.SetTracerProvider(tp)
  
      // 注册OpenTelemetry的中间件
      otelEndpoint := otelcontrib.NewOpenTelemetryMiddleware(endpoint, "my-service", otelcontrib.WithPropagators(propagation.NewCompositeTextMapPropagator()))
  
      // 运行OpenTelemetry的endpoint
      // ...
  }
  
  

• 代码示例

  package main
  
  import (
  	"context"
  	"fmt"
  	"net/http"
  	"os"
  
  	"github.com/go-kit/kit/endpoint"
  	httptransport "github.com/go-kit/kit/transport/http"
  	"github.com/gorilla/mux"
  	"go.opentelemetry.io/otel"
  	"go.opentelemetry.io/otel/exporters/trace/jaeger"
  	"go.opentelemetry.io/otel/sdk/trace"
  	"go.opentelemetry.io/otel/propagation"
  	otelcontrib "go.opentelemetry.io/contrib"
  )
  
  func main() {
  	// 创建Jaeger exporter并设置为全局tracer provider
  	exporter, err := jaeger.NewRawExporter(
  		jaeger.WithCollectorEndpoint("http://localhost:14268/api/traces"),
  	)
  	if err != nil {
  		fmt.Fprintf(os.Stderr, "failed to create exporter: %v\n", err)
  		os.Exit(1)
  	}
  	provider := trace.NewTracerProvider(
  		trace.WithBatcher(exporter),
  	)
  	otel.SetTracerProvider(provider)
  
  	// 创建HTTP server
  	r := mux.NewRouter()
  	r.Use(otelcontrib.NewHTTPServerMiddleware("my-service", otelcontrib.WithPropagators(propagation.NewCompositeTextMapPropagator())))
  
  	// 创建endpoint
  	helloEndpoint := func(ctx context.Context, request interface{}) (interface{}, error) {
  		return "Hello, World!", nil
  	}
  	helloHandler := httptransport.NewServer(
  		otelcontrib.NewOpenTelemetryMiddleware(endpoint.Endpoint(helloEndpoint), "my-endpoint"),
  		func(ctx context.Context, _ *http.Request) (interface{}, error) {
  			return nil, nil
  		},
  	)
  	r.Handle("/hello", helloHandler).Methods("GET")
  
  	fmt.Println("listening on :8080...")
  	http.ListenAndServe(":8080", r)
  }
  
  

• 我们首先创建了一个Jaeger exporter，并将其设置为全局tracer provider。然后，我们使用 go-kit 的 httptransport.NewServer() 函数创建了一个HTTP handler，并在handler上应用了 otelcontrib.NewOpenTelemetryMiddleware() 中间件，从而实现了endpoint级别的链路追踪。最后，我们使用 go-kit 的 mux.NewRouter() 函数创建了一个HTTP server，并在server上应用了 otelcontrib.NewHTTPServerMiddleware() 中间件，从而实现了服务级别的链路追踪。