---

yihanzhen · web-flow · commit 4dc40fb84372 · 2019-03-01T11:25:01.000-08:00
yaml --- r: 20773 b: refs/heads/autosynth-iamcredentials c: 692a530 h: refs/heads/master i: 20771: e999216
diff --git a/[refs] b/[refs]
@@ -162,7 +162,7 @@ refs/heads/pubsub-ordering-keys: 858d4e986a0ba48e08f00d42f51cbdecb175f5d6
 refs/tags/v0.75.0: c3673089ae09a897c1b4cf7dfe167fe4f8ab32fb
 refs/tags/v0.76.0: 395b016826d3ddf9cb8b34919636df15a4dbd032
 refs/tags/v0.77.0: 28a85a77883ccf5d48f297fd0ef3b3dca6ce01f0
-refs/heads/autosynth-iamcredentials: e0d4cdd7e6b914551b9afdabc240a01044bad7b0
+refs/heads/autosynth-iamcredentials: 692a530970f114fd8b2a6976d79af41d4fc8e4dd
 refs/heads/release-google-cloud-java-v0.78.0: fae5e980779cf0173a152636b278015b9f60ee55
 refs/heads/release-google-cloud-java-v0.81.0: 0352cd0dd11f4fd1fbd1ff16e7a96beaccc7b475
 refs/heads/release-google-cloud-java-v0.81.1-SNAPSHOT: 5a74ccb1f12506a3b67b65521881298fde20bd6f
diff --git a/branches/autosynth-iamcredentials/google-cloud-clients/google-cloud-spanner/src/main/java/com/google/cloud/spanner/SpannerExceptionFactory.java b/branches/autosynth-iamcredentials/google-cloud-clients/google-cloud-spanner/src/main/java/com/google/cloud/spanner/SpannerExceptionFactory.java
@@ -18,6 +18,8 @@
 
 import static com.google.cloud.spanner.SpannerException.DoNotConstructDirectly;
 
+import com.google.api.gax.grpc.GrpcStatusCode;
+import com.google.api.gax.rpc.ApiException;
 import com.google.common.base.MoreObjects;
 import com.google.common.base.Predicate;
 import io.grpc.Context;
@@ -48,6 +50,28 @@ public static SpannerException propagateInterrupt(InterruptedException e) {
     return SpannerExceptionFactory.newSpannerException(ErrorCode.CANCELLED, "Interrupted", e);
   }
 
+  /**
+   * Transforms a {@code TimeoutException} to a {@code SpannerException}.
+   *
+   * <pre>
+   * <code>
+   * try {
+   *   Spanner spanner = SpannerOptions.getDefaultInstance();
+   *   spanner
+   *       .getDatabaseAdminClient()
+   *       .createDatabase("[INSTANCE_ID]", "[DATABASE_ID]", [STATEMENTS])
+   *       .get();
+   * } catch (TimeoutException e) {
+   *   propagateTimeout(e);
+   * }
+   * </code>
+   * </pre>
+   */
+  public static SpannerException propagateTimeout(TimeoutException e) {
+    return SpannerExceptionFactory.newSpannerException(
+        ErrorCode.DEADLINE_EXCEEDED, "Operation did not complete in the given time", e);
+  }
+
   /**
    * Creates a new exception based on {@code cause}.
    *
@@ -71,6 +95,8 @@ public static SpannerException newSpannerException(@Nullable Context context, Th
       return newSpannerExceptionPreformatted(e.getErrorCode(), e.getMessage(), e);
     } else if (cause instanceof CancellationException) {
       return newSpannerExceptionForCancellation(context, cause);
+    } else if (cause instanceof ApiException) {
+      return fromApiException((ApiException) cause);
     }
     // Extract gRPC status.  This will produce "UNKNOWN" for non-gRPC exceptions.
     Status status = Status.fromThrowable(cause);
@@ -120,6 +146,17 @@ private static SpannerException newSpannerExceptionPreformatted(
     }
   }
 
+  private static SpannerException fromApiException(ApiException exception) {
+    Status.Code code = ((GrpcStatusCode) exception.getStatusCode()).getTransportCode();
+    ErrorCode errorCode = ErrorCode.fromGrpcStatus(Status.fromCode(code));
+    if (exception.getCause() != null) {
+      return SpannerExceptionFactory.newSpannerException(
+          errorCode, exception.getMessage(), exception.getCause());
+    } else {
+      return SpannerExceptionFactory.newSpannerException(errorCode, exception.getMessage());
+    }
+  }
+
   private static boolean isRetryable(ErrorCode code, @Nullable Throwable cause) {
     switch (code) {
       case INTERNAL:
