diff --git a/driver/docker-container/driver.go b/driver/docker-container/driver.go index 78538c85..63e97444 100644 --- a/driver/docker-container/driver.go +++ b/driver/docker-container/driver.go @@ -341,11 +341,17 @@ func (d *Driver) Client(ctx context.Context) (*client.Client, error) { return nil, err } - td, _ := exp.(client.TracerDelegate) + clientOpts := []client.ClientOpt{ + client.WithContextDialer(func(context.Context, string) (net.Conn, error) { + return conn, nil + }), + } + + if td, _ := exp.(client.TracerDelegate); td != nil { + clientOpts = append(clientOpts, client.WithTracerDelegate(td)) + } - return client.New(ctx, "", client.WithContextDialer(func(context.Context, string) (net.Conn, error) { - return conn, nil - }), client.WithTracerDelegate(td)) + return client.New(ctx, "", clientOpts...) } func (d *Driver) Factory() driver.Factory { diff --git a/driver/kubernetes/driver.go b/driver/kubernetes/driver.go index 92725dff..60f83da9 100644 --- a/driver/kubernetes/driver.go +++ b/driver/kubernetes/driver.go @@ -211,11 +211,17 @@ func (d *Driver) Client(ctx context.Context) (*client.Client, error) { return nil, err } - td, _ := exp.(client.TracerDelegate) + clientOpts := []client.ClientOpt{ + client.WithContextDialer(func(context.Context, string) (net.Conn, error) { + return conn, nil + }), + } + + if td, _ := exp.(client.TracerDelegate); td != nil { + clientOpts = append(clientOpts, client.WithTracerDelegate(td)) + } - return client.New(ctx, "", client.WithContextDialer(func(context.Context, string) (net.Conn, error) { - return conn, nil - }), client.WithTracerDelegate(td)) + return client.New(ctx, "", clientOpts...) } func (d *Driver) Factory() driver.Factory {