-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
6 changed files
with
322 additions
and
22 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,121 @@ | ||
| package api | ||
|
|
||
| import ( | ||
| "context" | ||
| "errors" | ||
| "time" | ||
|
|
||
| "go.opentelemetry.io/otel" | ||
| "go.opentelemetry.io/otel/exporters/stdout/stdoutlog" | ||
| "go.opentelemetry.io/otel/exporters/stdout/stdoutmetric" | ||
| "go.opentelemetry.io/otel/exporters/stdout/stdouttrace" | ||
| "go.opentelemetry.io/otel/log/global" | ||
| "go.opentelemetry.io/otel/propagation" | ||
| "go.opentelemetry.io/otel/sdk/log" | ||
| "go.opentelemetry.io/otel/sdk/metric" | ||
| "go.opentelemetry.io/otel/sdk/trace" | ||
| ) | ||
|
|
||
| // setupOTelSDK bootstraps the OpenTelemetry pipeline. | ||
| // If it does not return an error, make sure to call shutdown for proper cleanup. | ||
| func SetupOTelSDK(ctx context.Context) (shutdown func(context.Context) error, err error) { | ||
| var shutdownFuncs []func(context.Context) error | ||
|
|
||
| // shutdown calls cleanup functions registered via shutdownFuncs. | ||
| // The errors from the calls are joined. | ||
| // Each registered cleanup will be invoked once. | ||
| shutdown = func(ctx context.Context) error { | ||
| var err error | ||
| for _, fn := range shutdownFuncs { | ||
| err = errors.Join(err, fn(ctx)) | ||
| } | ||
| shutdownFuncs = nil | ||
| return err | ||
| } | ||
|
|
||
| // handleErr calls shutdown for cleanup and makes sure that all errors are returned. | ||
| handleErr := func(inErr error) { | ||
| err = errors.Join(inErr, shutdown(ctx)) | ||
| } | ||
|
|
||
| // Set up propagator. | ||
| prop := newPropagator() | ||
| otel.SetTextMapPropagator(prop) | ||
|
|
||
| // Set up trace provider. | ||
| tracerProvider, err := newTraceProvider() | ||
| if err != nil { | ||
| handleErr(err) | ||
| return | ||
| } | ||
| shutdownFuncs = append(shutdownFuncs, tracerProvider.Shutdown) | ||
| otel.SetTracerProvider(tracerProvider) | ||
|
|
||
| // Set up meter provider. | ||
| meterProvider, err := newMeterProvider() | ||
| if err != nil { | ||
| handleErr(err) | ||
| return | ||
| } | ||
| shutdownFuncs = append(shutdownFuncs, meterProvider.Shutdown) | ||
| otel.SetMeterProvider(meterProvider) | ||
|
|
||
| // Set up logger provider. | ||
| loggerProvider, err := newLoggerProvider() | ||
| if err != nil { | ||
| handleErr(err) | ||
| return | ||
| } | ||
| shutdownFuncs = append(shutdownFuncs, loggerProvider.Shutdown) | ||
| global.SetLoggerProvider(loggerProvider) | ||
|
|
||
| return | ||
| } | ||
|
|
||
| func newPropagator() propagation.TextMapPropagator { | ||
| return propagation.NewCompositeTextMapPropagator( | ||
| propagation.TraceContext{}, | ||
| propagation.Baggage{}, | ||
| ) | ||
| } | ||
|
|
||
| func newTraceProvider() (*trace.TracerProvider, error) { | ||
| traceExporter, err := stdouttrace.New( | ||
| stdouttrace.WithPrettyPrint()) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| traceProvider := trace.NewTracerProvider( | ||
| trace.WithBatcher(traceExporter, | ||
| // Default is 5s. Set to 1s for demonstrative purposes. | ||
| trace.WithBatchTimeout(time.Second)), | ||
| ) | ||
| return traceProvider, nil | ||
| } | ||
|
|
||
| func newMeterProvider() (*metric.MeterProvider, error) { | ||
| metricExporter, err := stdoutmetric.New() | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| meterProvider := metric.NewMeterProvider( | ||
| metric.WithReader(metric.NewPeriodicReader(metricExporter, | ||
| // Default is 1m. Set to 3s for demonstrative purposes. | ||
| metric.WithInterval(3*time.Second))), | ||
| ) | ||
| return meterProvider, nil | ||
| } | ||
|
|
||
| func newLoggerProvider() (*log.LoggerProvider, error) { | ||
| logExporter, err := stdoutlog.New() | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| loggerProvider := log.NewLoggerProvider( | ||
| log.WithProcessor(log.NewBatchProcessor(logExporter)), | ||
| ) | ||
| return loggerProvider, nil | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.