crate/uptracehelper/helper.go

package uptracehelper

import (
	"context"

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

type Config struct {
	ServiceName           string
	ServiceVersion        string
	DeploymentEnvironment string
}

func Setup(cfg Config) {
	opts := []uptrace.Option{
		uptrace.WithServiceName(cfg.ServiceName),
		uptrace.WithServiceVersion(cfg.ServiceVersion),
	}
	if cfg.DeploymentEnvironment != "" {
		opts = append(opts, uptrace.WithDeploymentEnvironment(cfg.DeploymentEnvironment))
	}
	uptrace.ConfigureOpentelemetry(opts...)
}

// GoStop 初始化并等待 uptrace
// 配合 [git.blauwelle.com/go/crate/exegroup] 使用
// env
//   - UPTRACE_DISABLED 存在就不再初始化
//   - UPTRACE_DSN 服务端地址
func GoStop(cfg Config) (func(ctx context.Context) error, func(ctx context.Context)) {
	Setup(cfg)
	shutdownErr := make(chan error, 1)
	goFunc := func(context.Context) error {
		return <-shutdownErr
	}
	stopFunc := func(ctx context.Context) {
		shutdownErr <- uptrace.Shutdown(ctx)
	}
	return goFunc, stopFunc
}