commit 12a6a7a7536b063f923d53eb9f6c244849b69fc5
author buchgr <buchgr@google.com> Fri Jul 07 04:43:58 2017 -0400
committer John Cater <jcater@google.com> Fri Jul 07 07:09:01 2017 -0400
tree 9bde923a1634c5b6ab62a720ece30f41d3a439e4
parent 87250d979d3622bc3fdb6095799a668841a2b152

remote: Make RetryException aware of gRPC's StatusException.

RELNOTES: None.
PiperOrigin-RevId: 161179228

src/main/java/com/google/devtools/build/lib/remote/RetryException.java

diff --git a/src/main/java/com/google/devtools/build/lib/remote/RetryException.java b/src/main/java/com/google/devtools/build/lib/remote/RetryException.java
index 6e9890a..24ddd36 100644
--- a/src/main/java/com/google/devtools/build/lib/remote/RetryException.java
+++ b/src/main/java/com/google/devtools/build/lib/remote/RetryException.java
@@ -15,6 +15,7 @@
 package com.google.devtools.build.lib.remote;
 
 import io.grpc.Status.Code;
+import io.grpc.StatusException;
 import io.grpc.StatusRuntimeException;
 import java.io.IOException;
 
@@ -32,8 +33,12 @@
   }
 
   public boolean causedByStatusCode(Code code) {
-    return getCause() instanceof StatusRuntimeException
-          && ((StatusRuntimeException) getCause()).getStatus().getCode() == code;
+    if (getCause() instanceof StatusRuntimeException) {
+      return ((StatusRuntimeException) getCause()).getStatus().getCode() == code;
+    } else if (getCause() instanceof StatusException) {
+      return ((StatusException) getCause()).getStatus().getCode() == code;
+    }
+    return false;
   }
 
   @Override