rebase

surbhigarg92 · surbhigarg92 · commit e3e05fb2efbe · 2025-04-15T23:11:27.000+05:30
diff --git a/google-cloud-spanner/src/main/java/com/google/cloud/spanner/spi/v1/GapicSpannerRpc.java b/google-cloud-spanner/src/main/java/com/google/cloud/spanner/spi/v1/GapicSpannerRpc.java
@@ -1994,6 +1994,9 @@ private GrpcCallContext createBaseCallContext() {
     if (endToEndTracingEnabled) {
       context = context.withExtraHeaders(metadataProvider.newEndToEndTracingHeader());
     }
+    if (isEnableAFEServerTiming()) {
+      context = context.withExtraHeaders(metadataProvider.newAfeServerTimingHeader());
+    }
     return context
         .withStreamWaitTimeoutDuration(waitTimeout)
         .withStreamIdleTimeoutDuration(idleTimeout);
@@ -2044,12 +2047,6 @@ <ReqT, RespT> GrpcCallContext newCallContext(
     if (routeToLeader && leaderAwareRoutingEnabled) {
       context = context.withExtraHeaders(metadataProvider.newRouteToLeaderHeader());
     }
-    if (endToEndTracingEnabled) {
-      context = context.withExtraHeaders(metadataProvider.newEndToEndTracingHeader());
-    }
-    if (isEnableAFEServerTiming()) {
-      context = context.withExtraHeaders(metadataProvider.newAfeServerTimingHeader());
-    }
     if (callCredentialsProvider != null) {
       CallCredentials callCredentials = callCredentialsProvider.getCallCredentials();
       if (callCredentials != null) {
