Merge pull request #462 from elyrank/master

issue #461 - cannot change delay of CassandraHostRetryService
hector-client · May 1, 2012 · a498ad2 · a498ad2
2 parents 38bdaa8 + 1e1e21b
commit a498ad2
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 2 deletions.
diff --git a/core/src/main/java/me/prettyprint/cassandra/connection/HConnectionManager.java b/core/src/main/java/me/prettyprint/cassandra/connection/HConnectionManager.java
@@ -463,5 +463,9 @@ public void shutdown() {
     }
   }
 
+  public void setCassandraHostRetryDelay(int retryDelay) {
+    cassandraHostRetryService.setRetryDelayInSeconds(retryDelay);
+    cassandraHostRetryService.applyRetryDelay();
+  }
 
 }
diff --git a/core/src/main/java/me/prettyprint/cassandra/service/CassandraClientMonitor.java b/core/src/main/java/me/prettyprint/cassandra/service/CassandraClientMonitor.java
@@ -231,7 +231,22 @@ public boolean unsuspendCassandraHost(String hostStr) {
     return connectionManager.unsuspendCassandraHost(new CassandraHost(hostStr));
   }
 
-
+  @Override
+  public boolean setCassandraHostRetryDelay(String retryDelay) {
+    int delay;
+    try {
+      delay = Integer.parseInt(retryDelay);
+      if (delay > 0) {
+        connectionManager.setCassandraHostRetryDelay(delay);
+        return true;
+      } else {
+        return false;
+      }
+    } catch (NumberFormatException e) {
+      log.error("Invalid number entered: " + retryDelay);
+      return false;
+    }
+  }
 
 
 

diff --git a/core/src/main/java/me/prettyprint/cassandra/service/CassandraClientMonitorMBean.java b/core/src/main/java/me/prettyprint/cassandra/service/CassandraClientMonitorMBean.java
@@ -149,5 +149,7 @@ public interface CassandraClientMonitorMBean {
    */
   boolean unsuspendCassandraHost(String hostStr);
 
-  Set<String> getSuspendedCassandraHosts(); 
+  Set<String> getSuspendedCassandraHosts();
+
+  boolean setCassandraHostRetryDelay(String retryDelay);
 }