add method to retry as many time for a duration

Author: DomGarguilo

core/src/main/java/org/apache/accumulo/core/util/Retry.java

@@ -268,6 +268,12 @@ public interface NeedsRetries {
      * @return this builder with the maximum number of retries set to the provided value
      */
     NeedsRetryDelay maxRetries(long max);
+
+    /**
+     * @return this builder with the maximum number of retries set to the number of retries that can
+     *         occur within the given duration
+     */
+    NeedsRetryDelay retriesForDuration(long duration, TimeUnit unit);
   }
 
   public interface NeedsRetryDelay {
@@ -380,6 +386,28 @@ public NeedsRetryDelay maxRetries(long max) {
       return this;
     }
 
+    @Override
+    public NeedsRetryDelay retriesForDuration(long duration, TimeUnit unit) {
+      checkState();
+      long durationInMillis = unit.toMillis(duration);
+      long totalWait = 0;
+      long retries = 0;
+      long currentWait = initialWait;
+      while (totalWait < durationInMillis && (maxRetries == -1 || retries < maxRetries)) {
+        totalWait += currentWait;
+        if (totalWait < durationInMillis) {
+          retries++;
+          if (backOffFactor > 1) {
+            currentWait = Math.min(maxWait, (long) (currentWait * backOffFactor));
+          } else {
+            currentWait = Math.min(maxWait, currentWait + waitIncrement);
+          }
+        }
+      }
+      this.maxRetries = retries;
+      return this;
+    }
+
     @Override
     public NeedsTimeIncrement retryAfter(long duration, TimeUnit unit) {
       checkState();

core/src/test/java/org/apache/accumulo/core/util/RetryTest.java

@@ -344,4 +344,37 @@ public void testInfiniteRetryWithBackoff() throws InterruptedException {
       }
     }
   }
+
+  @Test
+  public void testRetriesForDuration() throws InterruptedException {
+    final TimeUnit unit = MILLISECONDS;
+    final long duration = 10_000;
+
+    Retry retry = Retry.builder().retriesForDuration(duration, unit).retryAfter(100, unit)
+        .incrementBy(100, unit).maxWait(500, unit).backOffFactor(1.5).logInterval(3, MINUTES)
+        .createRetry();
+
+    long totalTime = 0;
+    long expectedRetries = 0;
+
+    // While the total wait time is less than the duration and there are retries left
+    while (totalTime <= duration && retry.canRetry()) {
+      totalTime += retry.getCurrentWait();
+      if (totalTime <= duration) {
+        retry.useRetry();
+        expectedRetries++;
+        if (retry.canRetry()) {
+          try {
+            retry.waitForNextAttempt(log, "Iteration " + expectedRetries);
+          } catch (IllegalArgumentException | InterruptedException e) {
+            log.error("Failed on iteration: {}", expectedRetries, e);
+            throw e;
+          }
+        }
+      }
+    }
+
+    assertEquals(expectedRetries, retry.retriesCompleted());
+  }
+
 }