feat: refactor.

tracer/tracer.go

@@ -0,0 +1,128 @@
+package tracer
+
+import (
+	"context"
+	"errors"
+
+	"go.opentelemetry.io/otel/exporters/otlp/otlptrace"
+	"go.opentelemetry.io/otel/exporters/otlp/otlptrace/otlptracegrpc"
+	"go.opentelemetry.io/otel/exporters/otlp/otlptrace/otlptracehttp"
+	"go.opentelemetry.io/otel/exporters/stdout/stdouttrace"
+	"go.opentelemetry.io/otel/exporters/zipkin"
+
+	"go.opentelemetry.io/otel"
+	"go.opentelemetry.io/otel/attribute"
+	"go.opentelemetry.io/otel/sdk/resource"
+	traceSdk "go.opentelemetry.io/otel/sdk/trace"
+	semConv "go.opentelemetry.io/otel/semconv/v1.4.0"
+
+	conf "github.com/tx7do/kratos-bootstrap/api/gen/go/conf/v1"
+	"github.com/tx7do/kratos-bootstrap/config"
+)
+
+// NewTracerExporter 创建一个导出器，支持：zipkin、otlp-http、otlp-grpc
+func NewTracerExporter(exporterName, endpoint string, insecure bool) (traceSdk.SpanExporter, error) {
+	ctx := context.Background()
+
+	switch exporterName {
+	case "zipkin":
+		return NewZipkinExporter(ctx, endpoint)
+	case "jaeger":
+		return nil, errors.New("jaeger exporter is no longer supported, please use otlp-http or otlp-grpc replace it")
+	case "otlp-http":
+		return NewOtlpHttpExporter(ctx, endpoint, insecure)
+	case "otlp-grpc":
+		return NewOtlpGrpcExporter(ctx, endpoint, insecure)
+	default:
+		fallthrough
+	case "stdout":
+		return stdouttrace.New()
+	}
+}
+
+// NewTracerProvider 创建一个链路追踪器
+func NewTracerProvider(cfg *conf.Tracer, serviceInfo *config.ServiceInfo) error {
+	if cfg == nil {
+		return errors.New("tracer config is nil")
+	}
+
+	if cfg.Sampler == 0 {
+		cfg.Sampler = 1.0
+	}
+
+	if cfg.Env == "" {
+		cfg.Env = "dev"
+	}
+
+	opts := []traceSdk.TracerProviderOption{
+		traceSdk.WithSampler(traceSdk.ParentBased(traceSdk.TraceIDRatioBased(cfg.GetSampler()))),
+		traceSdk.WithResource(resource.NewSchemaless(
+			semConv.ServiceNameKey.String(serviceInfo.Name),
+			semConv.ServiceVersionKey.String(serviceInfo.Version),
+			semConv.ServiceInstanceIDKey.String(serviceInfo.Id),
+			attribute.String("env", cfg.GetEnv()),
+		)),
+	}
+
+	if len(cfg.GetEndpoint()) > 0 {
+		exp, err := NewTracerExporter(cfg.GetBatcher(), cfg.GetEndpoint(), cfg.GetInsecure())
+		if err != nil {
+			panic(err)
+		}
+
+		opts = append(opts, traceSdk.WithBatcher(exp))
+	}
+
+	tp := traceSdk.NewTracerProvider(opts...)
+	if tp == nil {
+		return errors.New("create tracer provider failed")
+	}
+
+	otel.SetTracerProvider(tp)
+
+	return nil
+}
+
+// NewZipkinExporter 创建一个zipkin导出器，默认对端地址：http://localhost:9411/api/v2/spans
+func NewZipkinExporter(_ context.Context, endpoint string) (traceSdk.SpanExporter, error) {
+	return zipkin.New(endpoint)
+}
+
+//// NewJaegerExporter 创建一个jaeger导出器，默认对端地址：http://localhost:14268/api/traces
+//func NewJaegerExporter(_ context.Context, endpoint string) (traceSdk.SpanExporter, error) {
+//	return jaeger.New(jaeger.WithCollectorEndpoint(jaeger.WithEndpoint(endpoint)))
+//}
+
+// NewOtlpHttpExporter 创建OTLP/HTTP导出器，默认端口：4318
+func NewOtlpHttpExporter(ctx context.Context, endpoint string, insecure bool, options ...otlptracehttp.Option) (traceSdk.SpanExporter, error) {
+	var opts []otlptracehttp.Option
+	opts = append(opts, otlptracehttp.WithEndpoint(endpoint))
+
+	if insecure {
+		opts = append(opts, otlptracehttp.WithInsecure())
+	}
+
+	opts = append(opts, options...)
+
+	return otlptrace.New(
+		ctx,
+		otlptracehttp.NewClient(opts...),
+	)
+}
+
+// NewOtlpGrpcExporter 创建OTLP/gRPC导出器，默认端口：4317
+func NewOtlpGrpcExporter(ctx context.Context, endpoint string, insecure bool, options ...otlptracegrpc.Option) (traceSdk.SpanExporter, error) {
+	var opts []otlptracegrpc.Option
+	opts = append(opts, otlptracegrpc.WithEndpoint(endpoint))
+
+	if insecure {
+		opts = append(opts, otlptracegrpc.WithInsecure())
+	}
+
+	opts = append(opts, options...)
+
+	return otlptrace.New(
+		ctx,
+		otlptracegrpc.NewClient(opts...),
+	)
+}