chore(spanner): update exception block

Author: harshachinta

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

@@ -20,6 +20,7 @@
 import com.google.api.core.ApiFutures;
 import com.google.cloud.Timestamp;
 import com.google.common.util.concurrent.MoreExecutors;
+import java.util.concurrent.ExecutionException;
 import java.util.concurrent.Executor;
 
 /**
@@ -43,7 +44,10 @@ ApiFuture<AsyncRunner> getAsyncRunner() {
           if (exception instanceof InterruptedException) {
             throw SpannerExceptionFactory.propagateInterrupt((InterruptedException) exception);
           }
-          throw SpannerExceptionFactory.asSpannerException(exception.getCause());
+          if (exception instanceof ExecutionException) {
+            throw SpannerExceptionFactory.causeAsRunTimeException((ExecutionException) exception);
+          }
+          throw exception;
         },
         MoreExecutors.directExecutor());
   }

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

@@ -39,12 +39,7 @@ AsyncTransactionManager getAsyncTransactionManager() {
     try {
       return this.asyncTransactionManagerApiFuture.get();
     } catch (ExecutionException executionException) {
-      // Propagate the underlying exception as a RuntimeException (SpannerException is also a
-      // RuntimeException).
-      if (executionException.getCause() instanceof RuntimeException) {
-        throw (RuntimeException) executionException.getCause();
-      }
-      throw SpannerExceptionFactory.asSpannerException(executionException.getCause());
+      throw SpannerExceptionFactory.causeAsRunTimeException(executionException);
     } catch (InterruptedException interruptedException) {
       throw SpannerExceptionFactory.propagateInterrupt(interruptedException);
     }

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

@@ -219,12 +219,7 @@ private SessionReference getSessionReference() {
     try {
       return this.sessionFuture.get();
     } catch (ExecutionException executionException) {
-      // Propagate the underlying exception as a RuntimeException (SpannerException is also a
-      // RuntimeException).
-      if (executionException.getCause() instanceof RuntimeException) {
-        throw (RuntimeException) executionException.getCause();
-      }
-      throw SpannerExceptionFactory.asSpannerException(executionException.getCause());
+      throw SpannerExceptionFactory.causeAsRunTimeException(executionException);
     } catch (InterruptedException interruptedException) {
       throw SpannerExceptionFactory.propagateInterrupt(interruptedException);
     }

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

@@ -44,12 +44,7 @@ T getReadContext() {
     try {
       return this.readContextFuture.get();
     } catch (ExecutionException executionException) {
-      // Propagate the underlying exception as a RuntimeException (SpannerException is also a
-      // RuntimeException).
-      if (executionException.getCause() instanceof RuntimeException) {
-        throw (RuntimeException) executionException.getCause();
-      }
-      throw SpannerExceptionFactory.asSpannerException(executionException.getCause());
+      throw SpannerExceptionFactory.causeAsRunTimeException(executionException);
     } catch (InterruptedException interruptedException) {
       throw SpannerExceptionFactory.propagateInterrupt(interruptedException);
     }

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

@@ -38,12 +38,7 @@ TransactionManager getTransactionManager() {
     try {
       return this.transactionManagerFuture.get();
     } catch (ExecutionException executionException) {
-      // Propagate the underlying exception as a RuntimeException (SpannerException is also a
-      // RuntimeException).
-      if (executionException.getCause() instanceof RuntimeException) {
-        throw (RuntimeException) executionException.getCause();
-      }
-      throw SpannerExceptionFactory.asSpannerException(executionException.getCause());
+      throw SpannerExceptionFactory.causeAsRunTimeException(executionException);
     } catch (InterruptedException interruptedException) {
       throw SpannerExceptionFactory.propagateInterrupt(interruptedException);
     }

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

@@ -38,12 +38,7 @@ TransactionRunner getTransactionRunner() {
     try {
       return this.transactionRunnerFuture.get();
     } catch (ExecutionException executionException) {
-      // Propagate the underlying exception as a RuntimeException (SpannerException is also a
-      // RuntimeException).
-      if (executionException.getCause() instanceof RuntimeException) {
-        throw (RuntimeException) executionException.getCause();
-      }
-      throw SpannerExceptionFactory.asSpannerException(executionException.getCause());
+      throw SpannerExceptionFactory.causeAsRunTimeException(executionException);
     } catch (InterruptedException interruptedException) {
       throw SpannerExceptionFactory.propagateInterrupt(interruptedException);
     }

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

@@ -31,6 +31,7 @@
 import io.grpc.StatusRuntimeException;
 import io.grpc.protobuf.ProtoUtils;
 import java.util.concurrent.CancellationException;
+import java.util.concurrent.ExecutionException;
 import java.util.concurrent.TimeoutException;
 import javax.annotation.Nullable;
 
@@ -181,6 +182,15 @@ public static SpannerException newSpannerException(@Nullable Context context, Th
     return newSpannerException(ErrorCode.fromGrpcStatus(status), cause.getMessage(), cause);
   }
 
+  public static RuntimeException causeAsRunTimeException(ExecutionException executionException) {
+    // Propagate the underlying exception as a RuntimeException (SpannerException is also a
+    // RuntimeException).
+    if (executionException.getCause() instanceof RuntimeException) {
+      throw (RuntimeException) executionException.getCause();
+    }
+    throw asSpannerException(executionException.getCause());
+  }
+
   /**
    * Creates a new SpannerException that indicates that the RPC or transaction should be retried on
    * a different gRPC channel. This is an experimental feature that can be removed in the future.