Created new grpc wrapper

Author: surbhigarg92

google-cloud-spanner/src/main/java/com/google/cloud/spanner/spi/v1/GapicSpannerRpc.java

@@ -81,6 +81,7 @@
 import com.google.cloud.spanner.admin.instance.v1.stub.InstanceAdminStubSettings;
 import com.google.cloud.spanner.encryption.EncryptionConfigProtoMapper;
 import com.google.cloud.spanner.v1.stub.GrpcSpannerStub;
+import com.google.cloud.spanner.v1.stub.GrpcSpannerStubWrapper;
 import com.google.cloud.spanner.v1.stub.SpannerStub;
 import com.google.cloud.spanner.v1.stub.SpannerStubSettings;
 import com.google.common.annotations.VisibleForTesting;
@@ -410,7 +411,7 @@ public GapicSpannerRpc(final SpannerOptions options) {
                         /* isAdminClient = */ false, isEmulatorEnabled(options, emulatorHost)))
                 .build();
         ClientContext clientContext = ClientContext.create(spannerStubSettings);
-        this.spannerStub = GrpcSpannerStub.create(spannerStubSettings, clientContext);
+        this.spannerStub = GrpcSpannerStubWrapper.create(spannerStubSettings, clientContext);
         BuiltInMetricsConstant.DIRECT_PATH_ENABLED =
             ((GrpcTransportChannel) clientContext.getTransportChannel()).isDirectPath()
                 && isAttemptDirectPathXds;
@@ -461,7 +462,7 @@ public GapicSpannerRpc(final SpannerOptions options) {
                   .getStreamWatchdogProvider()
                   .withCheckInterval(pdmlSettings.getStreamWatchdogCheckInterval()));
         }
-        this.partitionedDmlStub = GrpcSpannerStub.create(pdmlSettings.build());
+        this.partitionedDmlStub = GrpcSpannerStubWrapper.create(pdmlSettings.build());
         this.instanceAdminStubSettings =
             options
                 .getInstanceAdminStubSettings()

google-cloud-spanner/src/main/java/com/google/cloud/spanner/v1/stub/GrpcSpannerStub.java

@@ -247,11 +247,6 @@ public static final GrpcSpannerStub create(ClientContext clientContext) throws I
     return new GrpcSpannerStub(SpannerStubSettings.newBuilder().build(), clientContext);
   }
 
-  public static final GrpcSpannerStub create(
-      SpannerStubSettings settings, ClientContext clientContext) throws IOException {
-    return new GrpcSpannerStub(settings, clientContext);
-  }
-
   public static final GrpcSpannerStub create(
       ClientContext clientContext, GrpcStubCallableFactory callableFactory) throws IOException {
     return new GrpcSpannerStub(

google-cloud-spanner/src/main/java/com/google/cloud/spanner/v1/stub/GrpcSpannerStubWrapper.java

@@ -0,0 +1,15 @@
+package com.google.cloud.spanner.v1.stub;
+
+import com.google.api.gax.rpc.ClientContext;
+import java.io.IOException;
+
+public class GrpcSpannerStubWrapper extends GrpcSpannerStub {
+
+  public static final GrpcSpannerStubWrapper create(SpannerStubSettings settings, ClientContext clientContext) throws IOException {
+    return new GrpcSpannerStubWrapper(settings, clientContext);
+  }
+  protected GrpcSpannerStubWrapper(SpannerStubSettings settings, ClientContext clientContext)
+      throws IOException {
+    super(settings, clientContext);
+  }
+}