go-libs/pkg/fx/observefx/resource.go

package observefx

import (
	"github.com/spf13/cobra"
	"go.opentelemetry.io/otel/sdk/resource"
	"go.uber.org/fx"

	"github.com/hanzo-fi/go-libs/v5/pkg/observe"
)

func ResourceModule(cfg observe.Config) fx.Option {
	return fx.Options(
		fx.Provide(func() (*resource.Resource, error) {
			return observe.BuildResource(cfg.ServiceName, cfg.ResourceAttributes, cfg.ServiceVersion)
		}),
	)
}

func ResourceModuleFromFlags(cmd *cobra.Command, opts ...observe.Option) fx.Option {
	otelServiceName, _ := cmd.Flags().GetString(observe.OtelServiceNameFlag)
	otelResourceAttributes, _ := cmd.Flags().GetStringSlice(observe.OtelResourceAttributesFlag)

	cfg := observe.NewConfig(opts...)
	cfg.ServiceName = otelServiceName
	cfg.ResourceAttributes = otelResourceAttributes
	return ResourceModule(cfg)
}