commit c8670b0fdcfebd0edbf9e43ef3cc6d46cde36b0f
author Yifan Cai <52585731+yifan-c@users.noreply.github.com> Mon Apr 13 23:30:56 2020 -0700
committer GitHub <noreply@github.com> Tue Apr 14 08:30:56 2020 +0200
tree 860fd968d525677bf2605b5887e8dda9e29159ae
parent 8090d55a4e6740c57cb9d3d97b6370c53b9a71cd

Enrich error message on request exceptions (#7)

spark-job/src/main/java/org/apache/cassandra/diff/DiffCluster.java

diff --git a/spark-job/src/main/java/org/apache/cassandra/diff/DiffCluster.java b/spark-job/src/main/java/org/apache/cassandra/diff/DiffCluster.java
index 4e108e3..4bbd81c 100644
--- a/spark-job/src/main/java/org/apache/cassandra/diff/DiffCluster.java
+++ b/spark-job/src/main/java/org/apache/cassandra/diff/DiffCluster.java
@@ -86,8 +86,10 @@
         try {
             return Uninterruptibles.getUninterruptibly(fetchPartitionKeys(table, prevToken, token));
         }
-        catch (ExecutionException e) {
-            throw new RuntimeException(e);
+        catch (ExecutionException ex) {
+            throw new RuntimeException(String.format("Unable to get partition keys (%s, %s] in table (%s) from cluster (%s)",
+                                                     prevToken, token, table, clusterId.name()),
+                                       ex);
         }
     }
 
@@ -118,9 +120,16 @@
     }
 
     public Iterator<Row> getPartition(TableSpec table, PartitionKey key, boolean shouldReverse) {
-        return readPartition(table.getTable(), key, shouldReverse)
-               .getUninterruptibly()
-               .iterator();
+        try {
+            return readPartition(table.getTable(), key, shouldReverse)
+                       .getUninterruptibly()
+                       .iterator();
+        }
+        catch (Exception ex) {
+            throw new RuntimeException(String.format("Unable to get partition (%s) in table (%s) from cluster (%s)",
+                                                     key.getTokenAsBigInteger(), table, clusterId.name()),
+                                       ex);
+        }
     }
 
     private ResultSetFuture readPartition(String table, final PartitionKey key, boolean shouldReverse) {