net.snowflake.client.util.DecorrelatedJitterBackoff Maven / Gradle / Ivy

Go to download
package net.snowflake.client.util;

import java.util.concurrent.ThreadLocalRandom;

/**
 * Decorrelated Jitter backoff
 *
 * https://aws.amazon.com/blogs/architecture/exponential-backoff-and-jitter/
 */
public class DecorrelatedJitterBackoff {
  private final long base;
  private final long cap;

  public DecorrelatedJitterBackoff(long base, long cap) {
    this.base = base;
    this.cap = cap;
  }

  public long nextSleepTime(long sleep) {
    return Math.min(cap, ThreadLocalRandom.current().nextLong(base, sleep * 3));
  }

  public long getJitterForLogin(long currentTime) {
    double multiplicationFactor = chooseRandom(-1, 1);
    long jitter = (long) (multiplicationFactor * currentTime * 0.5);
    return jitter;
  }

  public double chooseRandom(double min, double max) {
    return min + (Math.random() * (max - min));
  }
}