Merge branch 'partitioned-query' of github.com:pratickchokhani/java-spanner into partitioned-query

Author: pratickchokhani

google-cloud-spanner/src/main/java/com/google/cloud/spanner/DatabaseClientImpl.java

@@ -317,7 +317,7 @@ public long executePartitionedUpdate(final Statement stmt, final UpdateOption...
     if (useMultiplexedSessionPartitionedOps) {
       return getMultiplexedSession().executePartitionedUpdate(stmt, options);
     }
-    return executePartitionedUpdateSession(stmt, options);
+    return executePartitionedUpdateWithPooledSession(stmt, options);
   }
 
   private long executePartitionedUpdateWithPooledSession(

google-cloud-spanner/src/main/java/com/google/cloud/spanner/DelayedMultiplexedSessionTransaction.java

@@ -232,15 +232,9 @@ private SessionReference getSessionReference() {
    */
   @Override
   public long executePartitionedUpdate(Statement stmt, UpdateOption... options) {
-    try {
-      SessionReference sessionReference = getSessionReference();
-      return new MultiplexedSessionTransaction(
-              client, span, sessionReference, NO_CHANNEL_HINT, /* singleUse = */ true)
-          .executePartitionedUpdate(stmt, options);
-    } catch (InterruptedException e) {
-      throw new RuntimeException(e);
-    } catch (ExecutionException e) {
-      throw new RuntimeException(e);
-    }
+    SessionReference sessionReference = getSessionReference();
+    return new MultiplexedSessionTransaction(
+            client, span, sessionReference, NO_CHANNEL_HINT, /* singleUse = */ true)
+        .executePartitionedUpdate(stmt, options);
   }
 }

google-cloud-spanner/src/main/java/com/google/cloud/spanner/MultiplexedSessionDatabaseClient.java

@@ -556,7 +556,8 @@ public AsyncTransactionManager transactionManagerAsync(TransactionOption... opti
 
   @Override
   public long executePartitionedUpdate(Statement stmt, UpdateOption... options) {
-    return createMultiplexedSessionTransaction(/* singleUse = */ true).executePartitionedUpdate(stmt, options);
+    return createMultiplexedSessionTransaction(/* singleUse = */ true)
+        .executePartitionedUpdate(stmt, options);
   }
 
   /**