org.fissore.slf4j.LoggerStats Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of slf4j-fluent Show documentation

A fluent API for SLF4J

There is a newer version: 0.14.0

package org.fissore.slf4j;

import java.time.Instant;
import java.time.temporal.ChronoUnit;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.atomic.AtomicBoolean;

class LoggerStats {

  private final ConcurrentHashMap timestamps;
  private final ConcurrentHashMap counts;

  LoggerStats() {
    timestamps = new ConcurrentHashMap<>();
    counts = new ConcurrentHashMap<>();
  }

  boolean recordCallAndCheckIfEnoughTimePassed(String caller, long amount, ChronoUnit unit) {
    Instant now = Instant.now();

    AtomicBoolean result = new AtomicBoolean(true);
    timestamps.merge(caller, now, (previousTimestamp, _v) -> {
      if (previousTimestamp.plus(amount, unit).isBefore(now)) {
        return now;
      }
      result.set(false);
      return previousTimestamp;
    });

    return result.get();
  }

  boolean recordCallThenCheckIfNumberOfCallsMatchesAmount(String caller, int amount) {
    Long countedCalls = counts.merge(caller, 1L, (currentCounts, _v) -> currentCounts + 1);
    return countedCalls == 1 || countedCalls % amount == 0;
  }

}