Use Option pattern for componenthelper

Signed-off-by: Bogdan Drutu <[email protected]>

component/componenthelper/component.go

@@ -26,17 +26,28 @@ type Start func(context.Context, component.Host) error
 // Shutdown specifies the function invoked when the exporter is being shutdown.
 type Shutdown func(context.Context) error
 
-// ComponentSettings represents a settings struct to create components.
-type ComponentSettings struct {
+// Settings represents a settings struct to create components.
+type baseSettings struct {
 	Start
 	Shutdown
 }
 
-// DefaultComponentSettings returns the default settings for a component. The Start and Shutdown are no-op.
-func DefaultComponentSettings() *ComponentSettings {
-	return &ComponentSettings{
-		Start:    func(ctx context.Context, host component.Host) error { return nil },
-		Shutdown: func(ctx context.Context) error { return nil },
+// Option apply changes to internalOptions.
+type Option func(*baseSettings)
+
+// WithStart overrides the default Start function for an processor.
+// The default shutdown function does nothing and always returns nil.
+func WithStart(start Start) Option {
+	return func(o *baseSettings) {
+		o.Start = start
+	}
+}
+
+// WithShutdown overrides the default Shutdown function for an processor.
+// The default shutdown function does nothing and always returns nil.
+func WithShutdown(shutdown Shutdown) Option {
+	return func(o *baseSettings) {
+		o.Shutdown = shutdown
 	}
 }
 
@@ -55,10 +66,25 @@ func (be *baseComponent) Shutdown(ctx context.Context) error {
 	return be.shutdown(ctx)
 }
 
-// NewComponent returns a component.Component that calls the given Start and Shutdown.
-func NewComponent(s *ComponentSettings) component.Component {
+// fromOptions returns the internal settings starting from the default and applying all options.
+func fromOptions(options []Option) *baseSettings {
+	opts := &baseSettings{
+		Start:    func(ctx context.Context, host component.Host) error { return nil },
+		Shutdown: func(ctx context.Context) error { return nil },
+	}
+
+	for _, op := range options {
+		op(opts)
+	}
+
+	return opts
+}
+
+// New returns a component.Component that calls the given Start and Shutdown.
+func New(options ...Option) component.Component {
+	bs := fromOptions(options)
 	return &baseComponent{
-		start:    s.Start,
-		shutdown: s.Shutdown,
+		start:    bs.Start,
+		shutdown: bs.Shutdown,
 	}
 }

component/componenthelper/component_test.go

@@ -28,43 +28,37 @@ import (
 )
 
 func TestDefaultSettings(t *testing.T) {
-	st := componenthelper.DefaultComponentSettings()
-	require.NotNil(t, st)
-	cp := componenthelper.NewComponent(st)
+	cp := componenthelper.New()
 	require.NoError(t, cp.Start(context.Background(), componenttest.NewNopHost()))
 	require.NoError(t, cp.Shutdown(context.Background()))
 }
 
 func TestWithStart(t *testing.T) {
 	startCalled := false
-	st := componenthelper.DefaultComponentSettings()
-	st.Start = func(context.Context, component.Host) error { startCalled = true; return nil }
-	cp := componenthelper.NewComponent(st)
+	start := func(context.Context, component.Host) error { startCalled = true; return nil }
+	cp := componenthelper.New(componenthelper.WithStart(start))
 	assert.NoError(t, cp.Start(context.Background(), componenttest.NewNopHost()))
 	assert.True(t, startCalled)
 }
 
 func TestWithStart_ReturnError(t *testing.T) {
 	want := errors.New("my_error")
-	st := componenthelper.DefaultComponentSettings()
-	st.Start = func(context.Context, component.Host) error { return want }
-	cp := componenthelper.NewComponent(st)
+	start := func(context.Context, component.Host) error { return want }
+	cp := componenthelper.New(componenthelper.WithStart(start))
 	assert.Equal(t, want, cp.Start(context.Background(), componenttest.NewNopHost()))
 }
 
 func TestWithShutdown(t *testing.T) {
 	shutdownCalled := false
-	st := componenthelper.DefaultComponentSettings()
-	st.Shutdown = func(context.Context) error { shutdownCalled = true; return nil }
-	cp := componenthelper.NewComponent(st)
+	shutdown := func(context.Context) error { shutdownCalled = true; return nil }
+	cp := componenthelper.New(componenthelper.WithShutdown(shutdown))
 	assert.NoError(t, cp.Shutdown(context.Background()))
 	assert.True(t, shutdownCalled)
 }
 
 func TestWithShutdown_ReturnError(t *testing.T) {
 	want := errors.New("my_error")
-	st := componenthelper.DefaultComponentSettings()
-	st.Shutdown = func(context.Context) error { return want }
-	cp := componenthelper.NewComponent(st)
+	shutdown := func(context.Context) error { return want }
+	cp := componenthelper.New(componenthelper.WithShutdown(shutdown))
 	assert.Equal(t, want, cp.Shutdown(context.Background()))
 }

component/componenttest/nop_exporter.go

@@ -74,7 +74,7 @@ func (f *nopExporterFactory) CreateLogsExporter(
 }
 
 var nopExporterInstance = &nopExporter{
-	Component: componenthelper.NewComponent(componenthelper.DefaultComponentSettings()),
+	Component: componenthelper.New(),
 	Traces:    consumertest.NewTracesNop(),
 	Metrics:   consumertest.NewMetricsNop(),
 	Logs:      consumertest.NewLogsNop(),

component/componenttest/nop_extension.go

@@ -54,7 +54,7 @@ func (f *nopExtensionFactory) CreateExtension(
 }
 
 var nopExtensionInstance = &nopExtension{
-	Component: componenthelper.NewComponent(componenthelper.DefaultComponentSettings()),
+	Component: componenthelper.New(),
 }
 
 // nopExtension stores consumed traces and metrics for testing purposes.

component/componenttest/nop_processor.go

@@ -77,7 +77,7 @@ func (f *nopProcessorFactory) CreateLogsProcessor(
 }
 
 var nopProcessorInstance = &nopProcessor{
-	Component: componenthelper.NewComponent(componenthelper.DefaultComponentSettings()),
+	Component: componenthelper.New(),
 	Traces:    consumertest.NewTracesNop(),
 	Metrics:   consumertest.NewMetricsNop(),
 	Logs:      consumertest.NewLogsNop(),

component/componenttest/nop_receiver.go

@@ -76,7 +76,7 @@ func (f *nopReceiverFactory) CreateLogsReceiver(
 }
 
 var nopReceiverInstance = &nopReceiver{
-	Component: componenthelper.NewComponent(componenthelper.DefaultComponentSettings()),
+	Component: componenthelper.New(),
 }
 
 // nopReceiver stores consumed traces and metrics for testing purposes.

exporter/exporterhelper/common.go

@@ -26,7 +26,7 @@ import (
 	"go.opentelemetry.io/collector/consumer/consumererror"
 )
 
-// ComponentSettings for timeout. The timeout applies to individual attempts to send data to the backend.
+// Settings for timeout. The timeout applies to individual attempts to send data to the backend.
 type TimeoutSettings struct {
 	// Timeout is the timeout for every attempt to send data to the backend.
 	Timeout time.Duration `mapstructure:"timeout"`
@@ -72,7 +72,7 @@ func (req *baseRequest) setContext(ctx context.Context) {
 
 // baseSettings represents all the options that users can configure.
 type baseSettings struct {
-	*componenthelper.ComponentSettings
+	componentOptions []componenthelper.Option
 	TimeoutSettings
 	QueueSettings
 	RetrySettings
@@ -83,8 +83,7 @@ type baseSettings struct {
 func fromOptions(options []Option) *baseSettings {
 	// Start from the default options:
 	opts := &baseSettings{
-		ComponentSettings: componenthelper.DefaultComponentSettings(),
-		TimeoutSettings:   DefaultTimeoutSettings(),
+		TimeoutSettings: DefaultTimeoutSettings(),
 		// TODO: Enable queuing by default (call DefaultQueueSettings)
 		QueueSettings: QueueSettings{Enabled: false},
 		// TODO: Enable retry by default (call DefaultRetrySettings)
@@ -102,19 +101,19 @@ func fromOptions(options []Option) *baseSettings {
 // Option apply changes to baseSettings.
 type Option func(*baseSettings)
 
-// WithShutdown overrides the default Shutdown function for an exporter.
+// WithStart overrides the default Start function for an exporter.
 // The default shutdown function does nothing and always returns nil.
-func WithShutdown(shutdown componenthelper.Shutdown) Option {
+func WithStart(start componenthelper.Start) Option {
 	return func(o *baseSettings) {
-		o.Shutdown = shutdown
+		o.componentOptions = append(o.componentOptions, componenthelper.WithStart(start))
 	}
 }
 
-// WithStart overrides the default Start function for an exporter.
+// WithShutdown overrides the default Shutdown function for an exporter.
 // The default shutdown function does nothing and always returns nil.
-func WithStart(start componenthelper.Start) Option {
+func WithShutdown(shutdown componenthelper.Shutdown) Option {
 	return func(o *baseSettings) {
-		o.Start = start
+		o.componentOptions = append(o.componentOptions, componenthelper.WithShutdown(shutdown))
 	}
 }
 
@@ -162,7 +161,7 @@ type baseExporter struct {
 func newBaseExporter(cfg configmodels.Exporter, logger *zap.Logger, options ...Option) *baseExporter {
 	bs := fromOptions(options)
 	be := &baseExporter{
-		Component:                  componenthelper.NewComponent(bs.ComponentSettings),
+		Component:                  componenthelper.New(bs.componentOptions...),
 		cfg:                        cfg,
 		convertResourceToTelemetry: bs.ResourceToTelemetrySettings.Enabled,
 	}

internal/testcomponents/example_extension.go

@@ -52,5 +52,5 @@ func createExtensionDefaultConfig() configmodels.Extension {
 
 // CreateExtension creates an Extension based on this config.
 func createExtension(context.Context, component.ExtensionCreateParams, configmodels.Extension) (component.Extension, error) {
-	return componenthelper.NewComponent(componenthelper.DefaultComponentSettings()), nil
+	return componenthelper.New(), nil
 }

processor/processorhelper/processor.go

@@ -62,15 +62,15 @@ type Option func(*baseSettings)
 // The default shutdown function does nothing and always returns nil.
 func WithStart(start componenthelper.Start) Option {
 	return func(o *baseSettings) {
-		o.Start = start
+		o.componentOptions = append(o.componentOptions, componenthelper.WithStart(start))
 	}
 }
 
 // WithShutdown overrides the default Shutdown function for an processor.
 // The default shutdown function does nothing and always returns nil.
 func WithShutdown(shutdown componenthelper.Shutdown) Option {
 	return func(o *baseSettings) {
-		o.Shutdown = shutdown
+		o.componentOptions = append(o.componentOptions, componenthelper.WithShutdown(shutdown))
 	}
 }
 
@@ -83,16 +83,15 @@ func WithCapabilities(capabilities component.ProcessorCapabilities) Option {
 }
 
 type baseSettings struct {
-	*componenthelper.ComponentSettings
-	capabilities component.ProcessorCapabilities
+	componentOptions []componenthelper.Option
+	capabilities     component.ProcessorCapabilities
 }
 
 // fromOptions returns the internal settings starting from the default and applying all options.
 func fromOptions(options []Option) *baseSettings {
 	// Start from the default options:
 	opts := &baseSettings{
-		ComponentSettings: componenthelper.DefaultComponentSettings(),
-		capabilities:      component.ProcessorCapabilities{MutatesConsumedData: true},
+		capabilities: component.ProcessorCapabilities{MutatesConsumedData: true},
 	}
 
 	for _, op := range options {
@@ -114,7 +113,7 @@ type baseProcessor struct {
 func newBaseProcessor(fullName string, options ...Option) baseProcessor {
 	bs := fromOptions(options)
 	be := baseProcessor{
-		Component:    componenthelper.NewComponent(bs.ComponentSettings),
+		Component:    componenthelper.New(bs.componentOptions...),
 		fullName:     fullName,
 		capabilities: bs.capabilities,
 		traceAttributes: []trace.Attribute{

receiver/scraperhelper/scraper.go

@@ -29,8 +29,12 @@ type ScrapeMetrics func(context.Context) (pdata.MetricSlice, error)
 // Scrape resource metrics.
 type ScrapeResourceMetrics func(context.Context) (pdata.ResourceMetricsSlice, error)
 
+type baseSettings struct {
+	componentOptions []componenthelper.Option
+}
+
 // ScraperOption apply changes to internal options.
-type ScraperOption func(*componenthelper.ComponentSettings)
+type ScraperOption func(*baseSettings)
 
 type BaseScraper interface {
 	component.Component
@@ -64,15 +68,15 @@ func (b baseScraper) Name() string {
 
 // WithStart sets the function that will be called on startup.
 func WithStart(start componenthelper.Start) ScraperOption {
-	return func(s *componenthelper.ComponentSettings) {
-		s.Start = start
+	return func(o *baseSettings) {
+		o.componentOptions = append(o.componentOptions, componenthelper.WithStart(start))
 	}
 }
 
 // WithShutdown sets the function that will be called on shutdown.
 func WithShutdown(shutdown componenthelper.Shutdown) ScraperOption {
-	return func(s *componenthelper.ComponentSettings) {
-		s.Shutdown = shutdown
+	return func(o *baseSettings) {
+		o.componentOptions = append(o.componentOptions, componenthelper.WithShutdown(shutdown))
 	}
 }
 
@@ -91,14 +95,14 @@ func NewMetricsScraper(
 	scrape ScrapeMetrics,
 	options ...ScraperOption,
 ) MetricsScraper {
-	set := componenthelper.DefaultComponentSettings()
+	set := &baseSettings{}
 	for _, op := range options {
 		op(set)
 	}
 
 	ms := &metricsScraper{
 		baseScraper: baseScraper{
-			Component: componenthelper.NewComponent(set),
+			Component: componenthelper.New(set.componentOptions...),
 			name:      name,
 		},
 		ScrapeMetrics: scrape,
@@ -130,14 +134,14 @@ func NewResourceMetricsScraper(
 	scrape ScrapeResourceMetrics,
 	options ...ScraperOption,
 ) ResourceMetricsScraper {
-	set := componenthelper.DefaultComponentSettings()
+	set := &baseSettings{}
 	for _, op := range options {
 		op(set)
 	}
 
 	rms := &resourceMetricsScraper{
 		baseScraper: baseScraper{
-			Component: componenthelper.NewComponent(set),
+			Component: componenthelper.New(set.componentOptions...),
 			name:      name,
 		},
 		ScrapeResourceMetrics: scrape,