feat(telemetry): add capacity of add new resource attributes

formancehq · Nov 18, 2022 · 5504055 · 5504055
1 parent d871466
commit 5504055
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/sharedotlp/pkg/sharedotlptraces/traces.go b/sharedotlp/pkg/sharedotlptraces/traces.go
@@ -41,6 +41,7 @@ type ModuleConfig struct {
 	Batch        bool
 	JaegerConfig *JaegerConfig
 	OTLPConfig   *OTLPConfig
+	Resource     *resource.Resource
 }
 
 func ProvideTracerProviderOption(v any, annotations ...fx.Annotation) fx.Option {
@@ -51,6 +52,13 @@ func ProvideTracerProviderOption(v any, annotations ...fx.Annotation) fx.Option
 func TracesModule(cfg ModuleConfig) fx.Option {
 	options := make([]fx.Option, 0)
 	options = append(options,
+		fx.Provide(func() (*resource.Resource, error) {
+			defaultResource := resource.Default()
+			if cfg.Resource == nil {
+				return defaultResource, nil
+			}
+			return resource.Merge(defaultResource, cfg.Resource)
+		}),
 		fx.Supply(resource.Default()),
 		fx.Provide(func(tp *tracesdk.TracerProvider) trace.TracerProvider { return tp }),
 		fx.Provide(fx.Annotate(func(options ...tracesdk.TracerProviderOption) *tracesdk.TracerProvider {