Skip to content

Instantly share code, notes, and snippets.

@jhalterman
Last active January 6, 2019 12:12
Show Gist options
  • Save jhalterman/f7b18b30160ae7817bb93894056eb380 to your computer and use it in GitHub Desktop.
Save jhalterman/f7b18b30160ae7817bb93894056eb380 to your computer and use it in GitHub Desktop.
An exponentially weighted moving average implementation that decays based on the elapsed time since the last update, approximating a time windowed moving average.
/**
* An exponentially weighted moving average implementation that decays based on the elapsed time since the last update,
* approximating a time windowed moving average.
*/
public class MovingAverage {
private final long windowNanos;
// Mutable state
private volatile long lastNanos;
private volatile double average;
/**
* Creates a moving average of samples over a {@code window} for the {@code timeUnit}.
*/
public MovingAverage(long window, TimeUnit timeUnit) {
this.windowNanos = timeUnit.toNanos(window);
}
/**
* Updates the average with the {@code sample}.
*/
public synchronized void update(double sample) {
long now = System.nanoTime();
if (lastNanos == 0) {
average = sample;
lastNanos = now;
return;
}
long elapsedNanos = now - lastNanos;
double coeff = Math.exp(-1.0 * ((double) elapsedNanos / windowNanos));
average = (1.0 - coeff) * sample + coeff * average;
lastNanos = now;
}
/**
* Returns the moving average.
*/
public double get() {
return average;
}
/**
* Tick the internal moving average clock back by the {@code duration}. Useful for testing.
*/
void tick(long duration, TimeUnit timeUnit) {
if (lastNanos != 0)
lastNanos -= timeUnit.toNanos(duration);
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment