[grpc][v2] Use v2 gRPC factory in storage extension (jaegertracing#6969)

## Which problem is this PR solving?
- Resolves jaegertracing#6965

## Description of the changes
- This PR updates the jaegerstorage extension to use the new v2 gRPC
storage factory

## How was this change tested?
- CI

## Checklist
- [x] I have read
https://github.com/jaegertracing/jaeger/blob/master/CONTRIBUTING_GUIDELINES.md
- [x] I have signed all commits
- [x] I have added unit tests for the new functionality
- [x] I have run lint and test steps successfully
  - for `jaeger`: `make lint test`
  - for `jaeger-ui`: `npm run lint` and `npm run test`

---------

Signed-off-by: Mahad Zaryab <[email protected]>

Author: mahadzaryab1

cmd/jaeger/internal/extension/jaegerstorage/config.go

@@ -19,8 +19,8 @@ import (
 	"github.com/jaegertracing/jaeger/internal/storage/v1/badger"
 	"github.com/jaegertracing/jaeger/internal/storage/v1/cassandra"
 	es "github.com/jaegertracing/jaeger/internal/storage/v1/elasticsearch"
-	"github.com/jaegertracing/jaeger/internal/storage/v1/grpc"
 	"github.com/jaegertracing/jaeger/internal/storage/v1/memory"
+	"github.com/jaegertracing/jaeger/internal/storage/v2/grpc"
 )
 
 var (

cmd/jaeger/internal/extension/jaegerstorage/extension.go

@@ -18,9 +18,9 @@ import (
 	"github.com/jaegertracing/jaeger/internal/storage/v1/badger"
 	"github.com/jaegertracing/jaeger/internal/storage/v1/cassandra"
 	es "github.com/jaegertracing/jaeger/internal/storage/v1/elasticsearch"
-	"github.com/jaegertracing/jaeger/internal/storage/v1/grpc"
 	"github.com/jaegertracing/jaeger/internal/storage/v1/memory"
 	"github.com/jaegertracing/jaeger/internal/storage/v2/api/tracestore"
+	"github.com/jaegertracing/jaeger/internal/storage/v2/grpc"
 	"github.com/jaegertracing/jaeger/internal/storage/v2/v1adapter"
 	"github.com/jaegertracing/jaeger/internal/telemetry"
 )
@@ -142,7 +142,7 @@ func newStorageExt(config *Config, telset component.TelemetrySettings) *storageE
 	}
 }
 
-func (s *storageExt) Start(_ context.Context, host component.Host) error {
+func (s *storageExt) Start(ctx context.Context, host component.Host) error {
 	telset := telemetry.FromOtelComponent(s.telset, host)
 	telset.Metrics = telset.Metrics.Namespace(metrics.NSOptions{Name: "jaeger"})
 	scopedMetricsFactory := func(name, kind, role string) metrics.Factory {
@@ -175,8 +175,7 @@ func (s *storageExt) Start(_ context.Context, host component.Host) error {
 		case cfg.GRPC != nil:
 			grpcTelset := telset
 			grpcTelset.Metrics = scopedMetricsFactory(storageName, "grpc", "tracestore")
-			//nolint: contextcheck
-			v1Factory, err = grpc.NewFactoryWithConfig(*cfg.GRPC, grpcTelset)
+			factory, err = grpc.NewFactory(ctx, *cfg.GRPC, grpcTelset)
 		case cfg.Cassandra != nil:
 			v1Factory, err = cassandra.NewFactoryWithConfig(
 				*cfg.Cassandra,

cmd/jaeger/internal/extension/jaegerstorage/extension_test.go

@@ -25,9 +25,9 @@ import (
 	esCfg "github.com/jaegertracing/jaeger/internal/storage/elasticsearch/config"
 	"github.com/jaegertracing/jaeger/internal/storage/v1/badger"
 	"github.com/jaegertracing/jaeger/internal/storage/v1/cassandra"
-	"github.com/jaegertracing/jaeger/internal/storage/v1/grpc"
 	"github.com/jaegertracing/jaeger/internal/storage/v1/memory"
 	"github.com/jaegertracing/jaeger/internal/storage/v2/api/tracestore"
+	"github.com/jaegertracing/jaeger/internal/storage/v2/grpc"
 )
 
 type errorFactory struct {

internal/storage/v2/grpc/factory.go

@@ -43,6 +43,7 @@ type Factory struct {
 
 // NewFactory initializes a new gRPC (remote) storage backend.
 func NewFactory(
+	ctx context.Context,
 	cfg Config,
 	telset telemetry.Settings,
 ) (*Factory, error) {
@@ -58,7 +59,7 @@ func NewFactory(
 		for _, opt := range opts {
 			clientOpts = append(clientOpts, configgrpc.WithGrpcDialOption(opt))
 		}
-		return f.config.ToClientConn(context.Background(), f.telset.Host, telset, clientOpts...)
+		return f.config.ToClientConn(ctx, f.telset.Host, telset, clientOpts...)
 	}
 
 	if err := f.initializeConnections(readerTelset, writerTelset, newClientFn); err != nil {

internal/storage/v2/grpc/factory_test.go

@@ -4,6 +4,7 @@
 package grpc
 
 import (
+	"context"
 	"net"
 	"testing"
 	"time"
@@ -26,7 +27,7 @@ func TestNewFactory_NonEmptyAuthenticator(t *testing.T) {
 			Auth: &configauth.Authentication{},
 		},
 	}
-	_, err := NewFactory(*cfg, telemetry.NoopSettings())
+	_, err := NewFactory(context.Background(), *cfg, telemetry.NoopSettings())
 	require.ErrorContains(t, err, "authenticator is not supported")
 }
 
@@ -50,7 +51,7 @@ func TestNewFactory(t *testing.T) {
 		},
 	}
 	telset := telemetry.NoopSettings()
-	f, err := NewFactory(cfg, telset)
+	f, err := NewFactory(context.Background(), cfg, telset)
 	require.NoError(t, err)
 	require.NoError(t, f.Close())
 }
@@ -86,11 +87,13 @@ func TestFactory(t *testing.T) {
 }
 
 func TestInitializeConnections_ClientError(t *testing.T) {
-	f, err := NewFactory(Config{
-		ClientConfig: configgrpc.ClientConfig{
-			Endpoint: ":0",
-		},
-	}, telemetry.NoopSettings())
+	f, err := NewFactory(
+		context.Background(),
+		Config{
+			ClientConfig: configgrpc.ClientConfig{
+				Endpoint: ":0",
+			},
+		}, telemetry.NoopSettings())
 	require.NoError(t, err)
 	t.Cleanup(func() { require.NoError(t, f.Close()) })
 	newClientFn := func(_ component.TelemetrySettings, _ ...grpc.DialOption) (conn *grpc.ClientConn, err error) {