fix(spanner): Updated the error message validation for Multiplexed Session fallback in Partitioned Operations.

pratickchokhani · pratickchokhani · commit 2130f5235296 · 2025-02-18T09:07:11.000+05:30
diff --git a/google-cloud-spanner/src/main/java/com/google/cloud/spanner/BatchClientImpl.java b/google-cloud-spanner/src/main/java/com/google/cloud/spanner/BatchClientImpl.java
@@ -297,8 +297,7 @@ public List<Partition> partitionQuery(
     }
 
     void maybeMarkUnimplementedForPartitionedOps(SpannerException spannerException) {
-      if (spannerException.getErrorCode() == ErrorCode.INVALID_ARGUMENT
-          && MultiplexedSessionDatabaseClient.verifyErrorMessage(
+      if (MultiplexedSessionDatabaseClient.verifyErrorMessage(
               spannerException,
               "Partitioned operations are not supported with multiplexed sessions")) {
         unimplementedForPartitionedOps.set(true);

Original file line number	Diff line number	Diff line change
`@@ -297,8 +297,7 @@ public List<Partition> partitionQuery(`
`297`	`297`	`}`
`298`	`298`
`299`	`299`	`void maybeMarkUnimplementedForPartitionedOps(SpannerException spannerException) {`
`300`		`- if (spannerException.getErrorCode() == ErrorCode.INVALID_ARGUMENT`
`301`		`- && MultiplexedSessionDatabaseClient.verifyErrorMessage(`
	`300`	`+ if (MultiplexedSessionDatabaseClient.verifyErrorMessage(`
`302`	`301`	`spannerException,`
`303`	`302`	`"Partitioned operations are not supported with multiplexed sessions")) {`
`304`	`303`	`unimplementedForPartitionedOps.set(true);`