From 12a6a7a7536b063f923d53eb9f6c244849b69fc5 Mon Sep 17 00:00:00 2001 From: buchgr Date: Fri, 7 Jul 2017 04:43:58 -0400 Subject: remote: Make RetryException aware of gRPC's StatusException. RELNOTES: None. PiperOrigin-RevId: 161179228 --- .../com/google/devtools/build/lib/remote/RetryException.java | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) (limited to 'src/main/java/com/google/devtools/build/lib/remote') 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 6e9890afd6..24ddd365bd 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 final class RetryException extends IOException { } 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 -- cgit v1.2.3