From 1220b5dc83aa2bc76c0e7500cd3c1e2454d99de7 Mon Sep 17 00:00:00 2001 From: chahatsagarmain Date: Sun, 6 Oct 2024 15:11:54 +0530 Subject: [PATCH] Using component for no op host instead of component test Signed-off-by: chahatsagarmain --- plugin/storage/grpc/factory.go | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/plugin/storage/grpc/factory.go b/plugin/storage/grpc/factory.go index c12cb370c75..9ac64f455a9 100644 --- a/plugin/storage/grpc/factory.go +++ b/plugin/storage/grpc/factory.go @@ -12,7 +12,6 @@ import ( "github.com/spf13/viper" "go.opentelemetry.io/collector/component" - "go.opentelemetry.io/collector/component/componenttest" "go.opentelemetry.io/collector/config/configgrpc" "go.opentelemetry.io/collector/config/configtelemetry" "go.opentelemetry.io/contrib/instrumentation/google.golang.org/grpc/otelgrpc" @@ -49,11 +48,21 @@ type Factory struct { remoteConn *grpc.ClientConn } +type nopHost struct{} + +func NewNopHost() component.Host { + return &nopHost{} +} + // NewFactory creates a new Factory. func NewFactory() *Factory { return &Factory{} } +func (*nopHost) GetExtensions() map[component.ID]component.Component { + return nil +} + // NewFactoryWithConfig is used from jaeger(v2). func NewFactoryWithConfig( cfg Config, @@ -98,7 +107,8 @@ func (f *Factory) Initialize(metricsFactory metrics.Factory, logger *zap.Logger) for _, opt := range opts { clientOpts = append(clientOpts, configgrpc.WithGrpcDialOption(opt)) } - return f.config.ToClientConn(context.Background(), componenttest.NewNopHost(), telset, clientOpts...) + noophost := NewNopHost() + return f.config.ToClientConn(context.Background(), noophost, telset, clientOpts...) } var err error