Add MustInitTracerProvider

otto-de · Dec 19, 2023 · 9b5b94b · 9b5b94b
1 parent d35c8b4
commit 9b5b94b
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/pkg/gcp/trace.go b/pkg/gcp/trace.go
@@ -0,0 +1,17 @@
+package gcp
+
+import (
+	texporter "github.com/GoogleCloudPlatform/opentelemetry-operations-go/exporter/trace"
+	"go.opentelemetry.io/otel"
+	sdktrace "go.opentelemetry.io/otel/sdk/trace"
+)
+
+func MustInitTracerProvider(project string, opts ...sdktrace.TracerProviderOption) *sdktrace.TracerProvider {
+	exporter, err := texporter.New(texporter.WithProjectID(project))
+	if err != nil {
+		panic(err)
+	}
+	tp := sdktrace.NewTracerProvider(append(opts, sdktrace.WithBatcher(exporter))...)
+	otel.SetTracerProvider(tp)
+	return tp
+}