feat(spanner): Removed env variable for Partitioned Ops ensuring that…

… Multiplexed Session for Partitioned Ops is not available to customers.
googleapis · Dec 9, 2024 · 263901f · 263901f
1 parent 2db1cc6
commit 263901f
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 2 deletions.
diff --git a/...oud-spanner-executor/src/main/java/com/google/cloud/spanner/SessionPoolOptionsHelper.java b/...oud-spanner-executor/src/main/java/com/google/cloud/spanner/SessionPoolOptionsHelper.java
@@ -44,4 +44,12 @@ public static SessionPoolOptions.Builder setUseMultiplexedSessionForRW(
       SessionPoolOptions.Builder sessionPoolOptionsBuilder, boolean useMultiplexedSessionForRW) {
     return sessionPoolOptionsBuilder.setUseMultiplexedSessionForRW(useMultiplexedSessionForRW);
   }
+
+  // TODO: Remove when multiplexed session for partitioned operations are released.
+  public static SessionPoolOptions.Builder setUseMultiplexedSessionForPartitionedOperations(
+      SessionPoolOptions.Builder sessionPoolOptionsBuilder,
+      boolean useMultiplexedSessionForPartitionedOps) {
+    return sessionPoolOptionsBuilder.setUseMultiplexedSessionPartitionedOps(
+        useMultiplexedSessionForPartitionedOps);
+  }
 }
diff --git a/google-cloud-spanner/src/main/java/com/google/cloud/spanner/SessionPoolOptions.java b/google-cloud-spanner/src/main/java/com/google/cloud/spanner/SessionPoolOptions.java
@@ -373,7 +373,9 @@ private static Boolean getUseMultiplexedSessionFromEnvVariable() {
   @VisibleForTesting
   @InternalApi
   protected static Boolean getUseMultiplexedSessionFromEnvVariablePartitionedOps() {
-    return parseBooleanEnvVariable("GOOGLE_CLOUD_SPANNER_MULTIPLEXED_SESSIONS_PARTITIONED_OPS");
+    // Checks the value of env, GOOGLE_CLOUD_SPANNER_MULTIPLEXED_SESSIONS_PARTITIONED_OPS
+    // This returns null until Partitioned Operations is supported.
+    return null;
   }
 
   private static Boolean parseBooleanEnvVariable(String variableName) {

diff --git a/google-cloud-spanner/src/test/java/com/google/cloud/spanner/BatchClientImplTest.java b/google-cloud-spanner/src/test/java/com/google/cloud/spanner/BatchClientImplTest.java
@@ -36,7 +36,6 @@
 import com.google.spanner.v1.Session;
 import com.google.spanner.v1.Transaction;
 import io.opentelemetry.api.OpenTelemetry;
-
 import java.time.Duration;
 import java.util.Collections;
 import java.util.Map;