chore: create and reuse a base CallContext (#3816)

Every RPC that is executed by the client needs to create a new CallContext. Some
of the settings in this CallContext is the same for all calls, and there is no
need to create a new one completely from scratch for the properties that are the
same for all calls. This change creates a base CallContext that is reused as the
base for all RPCs.

Author: olavloite

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

@@ -280,6 +280,8 @@ public class GapicSpannerRpc implements SpannerRpc {
 
   private Supplier<Boolean> directPathEnabledSupplier = () -> false;
 
+  private final GrpcCallContext baseGrpcCallContext;
+
   public static GapicSpannerRpc create(SpannerOptions options) {
     return new GapicSpannerRpc(options);
   }
@@ -333,6 +335,7 @@ public GapicSpannerRpc(final SpannerOptions options) {
     this.endToEndTracingEnabled = options.isEndToEndTracingEnabled();
     this.numChannels = options.getNumChannels();
     this.isGrpcGcpExtensionEnabled = options.isGrpcGcpExtensionEnabled();
+    this.baseGrpcCallContext = createBaseCallContext();
 
     if (initializeStubs) {
       // First check if SpannerOptions provides a TransportChannelProvider. Create one
@@ -1978,6 +1981,20 @@ private static <T> T get(final Future<T> future) throws SpannerException {
     }
   }
 
+  private GrpcCallContext createBaseCallContext() {
+    GrpcCallContext context = GrpcCallContext.createDefault();
+    if (compressorName != null) {
+      // This sets the compressor for Client -> Server.
+      context = context.withCallOptions(context.getCallOptions().withCompression(compressorName));
+    }
+    if (endToEndTracingEnabled) {
+      context = context.withExtraHeaders(metadataProvider.newEndToEndTracingHeader());
+    }
+    return context
+        .withStreamWaitTimeoutDuration(waitTimeout)
+        .withStreamIdleTimeoutDuration(idleTimeout);
+  }
+
   // Before removing this method, please verify with a code owner that it is not used
   // in any internal testing infrastructure.
   @VisibleForTesting
@@ -2002,7 +2019,7 @@ <ReqT, RespT> GrpcCallContext newCallContext(
       ReqT request,
       MethodDescriptor<ReqT, RespT> method,
       boolean routeToLeader) {
-    GrpcCallContext context = GrpcCallContext.createDefault();
+    GrpcCallContext context = this.baseGrpcCallContext;
     if (options != null) {
       if (this.isGrpcGcpExtensionEnabled) {
         // Set channel affinity in gRPC-GCP.
@@ -2019,28 +2036,17 @@ <ReqT, RespT> GrpcCallContext newCallContext(
         context = context.withChannelAffinity(Option.CHANNEL_HINT.getLong(options).intValue());
       }
     }
-    if (compressorName != null) {
-      // This sets the compressor for Client -> Server.
-      context = context.withCallOptions(context.getCallOptions().withCompression(compressorName));
-    }
     context = context.withExtraHeaders(metadataProvider.newExtraHeaders(resource, projectName));
     if (routeToLeader && leaderAwareRoutingEnabled) {
       context = context.withExtraHeaders(metadataProvider.newRouteToLeaderHeader());
     }
-    if (endToEndTracingEnabled) {
-      context = context.withExtraHeaders(metadataProvider.newEndToEndTracingHeader());
-    }
     if (callCredentialsProvider != null) {
       CallCredentials callCredentials = callCredentialsProvider.getCallCredentials();
       if (callCredentials != null) {
         context =
             context.withCallOptions(context.getCallOptions().withCallCredentials(callCredentials));
       }
     }
-    context =
-        context
-            .withStreamWaitTimeoutDuration(waitTimeout)
-            .withStreamIdleTimeoutDuration(idleTimeout);
     CallContextConfigurator configurator = SpannerOptions.CALL_CONTEXT_CONFIGURATOR_KEY.get();
     ApiCallContext apiCallContextFromContext = null;
     if (configurator != null) {