Update the cli to manage own handler

Author: MrAlias

cli/main.go

@@ -14,7 +14,10 @@ import (
 	"path/filepath"
 	"syscall"
 
+	semconv "go.opentelemetry.io/otel/semconv/v1.26.0"
+
 	"go.opentelemetry.io/auto"
+	"go.opentelemetry.io/auto/export/otelsdk"
 	"go.opentelemetry.io/auto/internal/pkg/process"
 )
 
@@ -106,9 +109,22 @@ func main() {
 		"version", newVersion(),
 	)
 
+	v := semconv.TelemetryDistroVersionKey.String(auto.Version())
+	h, err := otelsdk.New(
+		ctx,
+		otelsdk.WithEnv(),
+		otelsdk.WithLogger(logger),
+		otelsdk.WithResourceAttributes(v),
+	)
+	if err != nil {
+		logger.Error("failed to create OTel SDK handler", "error", err)
+		return
+	}
+
 	instOptions := []auto.InstrumentationOption{
 		auto.WithEnv(),
 		auto.WithLogger(logger),
+		auto.WithHandler(h),
 	}
 	if globalImpl {
 		instOptions = append(instOptions, auto.WithGlobal())
@@ -131,4 +147,14 @@ func main() {
 	if err = inst.Run(ctx); err != nil && !errors.Is(err, process.ErrInterrupted) {
 		logger.Error("instrumentation crashed", "error", err)
 	}
+
+	logger.Info("shutting down")
+
+	ctx, cancel = signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)
+	defer cancel()
+
+	err = h.Shutdown(ctx)
+	if err != nil {
+		logger.Error("failed to flush handler", "error", err)
+	}
 }