commit 12b2d7430f51b3e0195e5a7320d27429c1b39edf
author Dionysios Logothetis <dlogothetis@gmail.com> Mon May 13 17:32:13 2019 -0700
committer Dionysios Logothetis <dionysios@fb.com> Mon May 13 17:32:13 2019 -0700
tree 79514f9e3d77d60e0f6e046dc41121ab27675846
parent e827a5668a94d13800201191ebf7d34642728771

GIRAPH-1219

closes #102

giraph-core/src/main/java/org/apache/giraph/graph/RetryableJobProgressTrackerClient.java

diff --git a/giraph-core/src/main/java/org/apache/giraph/graph/RetryableJobProgressTrackerClient.java b/giraph-core/src/main/java/org/apache/giraph/graph/RetryableJobProgressTrackerClient.java
index 369feec..2fa4446 100644
--- a/giraph-core/src/main/java/org/apache/giraph/graph/RetryableJobProgressTrackerClient.java
+++ b/giraph-core/src/main/java/org/apache/giraph/graph/RetryableJobProgressTrackerClient.java
@@ -201,14 +201,22 @@
           retry(runnable);
           break; // If the retry succeeded, we simply break from the loop
 
-          // CHECKSTYLE: stop IllegalCatch
-        } catch (Exception e) {
-          // CHECKSTYLE: resume IllegalCatch
+        } catch (RuntimeTTransportException | RejectedExecutionException e) {
+          // If a RuntimeTTTransportException happened, then we will retry
           if (LOG.isInfoEnabled()) {
             LOG.info("Exception occurred while talking to " +
               "JobProgressTracker server after retry " + i +
               " of " + numRetries, e);
           }
+          // CHECKSTYLE: stop IllegalCatch
+        } catch (Exception e) {
+          // CHECKSTYLE: resume IllegalCatch
+          // If any other exception happened (e.g. application-specific),
+          // then we stop.
+          LOG.info("Exception occurred while talking to " +
+            "JobProgressTracker server after retry " + i +
+            " of " + numRetries + ", giving up", e);
+          break;
         }
       }
       // CHECKSTYLE: stop IllegalCatch