Creating RetryPeriodicDelay

Author: LikeTheSalad

opamp-client/src/main/java/io/opentelemetry/opamp/client/internal/request/delay/RetryPeriodicDelay.java

@@ -0,0 +1,36 @@
+package io.opentelemetry.opamp.client.internal.request.delay;
+
+import java.time.Duration;
+import java.util.Objects;
+import java.util.concurrent.atomic.AtomicReference;
+
+/** Defaults to an exponential backoff strategy, unless a delay is suggested. */
+public final class RetryPeriodicDelay implements PeriodicDelay, AcceptsDelaySuggestion {
+  private final ExponentialBackoffPeriodicDelay exponentialBackoff;
+  private final AtomicReference<PeriodicDelay> currentDelay;
+
+  public static RetryPeriodicDelay create(Duration initialDelay) {
+    return new RetryPeriodicDelay(new ExponentialBackoffPeriodicDelay(initialDelay));
+  }
+
+  private RetryPeriodicDelay(ExponentialBackoffPeriodicDelay exponentialBackoff) {
+    this.exponentialBackoff = exponentialBackoff;
+    currentDelay = new AtomicReference<>(exponentialBackoff);
+  }
+
+  @Override
+  public void suggestDelay(Duration delay) {
+    currentDelay.set(PeriodicDelay.ofFixedDuration(delay));
+  }
+
+  @Override
+  public Duration getNextDelay() {
+    return Objects.requireNonNull(currentDelay.get()).getNextDelay();
+  }
+
+  @Override
+  public void reset() {
+    exponentialBackoff.reset();
+    currentDelay.set(exponentialBackoff);
+  }
+}

opamp-client/src/test/java/io/opentelemetry/opamp/client/internal/request/delay/RetryPeriodicDelayTest.java

@@ -0,0 +1,26 @@
+package io.opentelemetry.opamp.client.internal.request.delay;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+import java.time.Duration;
+import org.junit.jupiter.api.Test;
+
+class RetryPeriodicDelayTest {
+  @Test
+  public void verifyDelayBehavior() {
+    RetryPeriodicDelay retryPeriodicDelay = RetryPeriodicDelay.create(Duration.ofSeconds(1));
+
+    // Without suggested delay
+    assertThat(retryPeriodicDelay.getNextDelay()).isEqualTo(Duration.ofSeconds(1));
+    assertThat(retryPeriodicDelay.getNextDelay()).isEqualTo(Duration.ofSeconds(2));
+    assertThat(retryPeriodicDelay.getNextDelay()).isEqualTo(Duration.ofSeconds(4));
+    retryPeriodicDelay.reset();
+    assertThat(retryPeriodicDelay.getNextDelay()).isEqualTo(Duration.ofSeconds(1));
+
+    // With suggested delay
+    retryPeriodicDelay.suggestDelay(Duration.ofSeconds(5));
+    assertThat(retryPeriodicDelay.getNextDelay()).isEqualTo(Duration.ofSeconds(5));
+    retryPeriodicDelay.reset();
+    assertThat(retryPeriodicDelay.getNextDelay()).isEqualTo(Duration.ofSeconds(1));
+  }
+}