You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
46 lines
1.1 KiB
Go
46 lines
1.1 KiB
Go
package uptracehelper
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/uptrace/uptrace-go/uptrace"
|
|
)
|
|
|
|
type Config struct {
|
|
ServiceName string
|
|
ServiceVersion string
|
|
DeploymentEnvironment string
|
|
ShutdownTimeout time.Duration
|
|
}
|
|
|
|
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...)
|
|
}
|
|
|
|
// Bootstrap 初始化并等待 uptrace
|
|
// 配合 [git.blauwelle.com/go/crate/exegroup] 使用
|
|
// env
|
|
// - UPTRACE_DISABLED 存在就不再初始化
|
|
// - UPTRACE_DSN 服务端地址
|
|
func Bootstrap(cfg Config) func(ctx context.Context) error {
|
|
Setup(cfg)
|
|
return func(ctx context.Context) error {
|
|
<-ctx.Done()
|
|
shutdownCtx := context.Background()
|
|
if cfg.ShutdownTimeout >= 0 {
|
|
var cancel func()
|
|
shutdownCtx, cancel = context.WithTimeout(shutdownCtx, cfg.ShutdownTimeout)
|
|
defer cancel()
|
|
}
|
|
return uptrace.Shutdown(shutdownCtx)
|
|
}
|
|
}
|