import "github.com/uptrace/uptrace-go/uptrace"

uptrace.ConfigureOpentelemetry(
    // copy your project DSN here or use UPTRACE_DSN env var
    uptrace.WithDSN("https://<key>@uptrace.dev/<project_id>"),
    uptrace.WithServiceName("myservice"),
    uptrace.WithServiceVersion("1.0.0"),
)

import "github.com/go-kratos/kratos/v2/middleware/tracing"

func NewServer() {
	var opts = []grpc.ServerOption{
		grpc.Middleware(
			tracing.Server(),
		),
	}
}

func NewClient() {
	return grpc.DialInsecure(
		context.Background(),
		grpc.WithMiddleware(
			tracing.Client(),
		),
	)
}

import "github.com/go-redis/redis/extra/redisotel/v8"

func NewRDB() {
	rdb.AddHook(redisotel.NewTracingHook())
}

import "github.com/uptrace/opentelemetry-go-extra/otelsql"

func NewDB() {
	sqlDB, err := otelsql.Open(
		dbcf.Driver, source,
		otelsql.WithDBName(dbcf.Database),
		otelsql.WithDBSystem(dbcf.Driver),
	)
}

import (
	"go.opentelemetry.io/otel"

	"go.opentelemetry.io/otel/trace"
)

func Demo(ctx context.Context) {
	// 使用全局 TracerProvider 创建一个新的 Tracer.
	tracer := otel.Tracer("tracer-name")
	ctx, span := tracer.Start(ctx, "span-name")
	defer span.End()

	// 设置 span status
	if err := do(); err != nil {
		span.SetStatus(codes.Error, err.Error())
		// RecordError() 不会设置 span status, 只是简单的记录一下错误.
		span.RecordError(err)
	}

	// 当你不想创建一个 span 时, 你可以从 ctx 中获取当前 span.
	span := trace.SpanFromContext(ctx)
	span.SetAttributes(attribute.Bool("is_true", true))
}

import (
	"github.com/sirupsen/logrus"
	"github.com/uptrace/opentelemetry-go-extra/otellogrus"
)

func Demo(ctx context.Context) {
	/// logrus
	logrus.AddHook(otellogrus.NewHook(otellogrus.WithLevels(
		logrus.PanicLevel,
		logrus.FatalLevel,
		logrus.ErrorLevel,
		logrus.WarnLevel,
	)))
	logrus.WithContext(ctx).Log()
}

// SpanFromContext returns the current Span from ctx.
//
// If no Span is currently set in ctx an implementation of a Span that
// performs no operations is returned.
func SpanFromContext(ctx context.Context) Span {
	if ctx == nil {
		return noopSpan{}
	}
	if span, ok := ctx.Value(currentSpanKey).(Span); ok {
		return span
	}
	return noopSpan{}
}

// TextMapCarrier is the storage medium used by a TextMapPropagator.
type TextMapCarrier interface {
	// Get returns the value associated with the passed key.
	Get(key string) string

	// Set stores the key-value pair.
	Set(key string, value string)

	// Keys lists the keys stored in this carrier.
	Keys() []string
}

// TextMapPropagator propagates cross-cutting concerns as key-value text
// pairs within a carrier that travels in-band across process boundaries.
type TextMapPropagator interface {
	// Inject set cross-cutting concerns from the Context into the carrier.
	Inject(ctx context.Context, carrier TextMapCarrier)

	// Extract reads cross-cutting concerns from the carrier into a Context.
	Extract(ctx context.Context, carrier TextMapCarrier) context.Context

	// Fields returns the keys whose values are set with Inject.
	Fields() []string
}