io.rainfall.statistics.Statistics Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of rainfall-core Show documentation
Rainfall-core is the core of the Rainfall framework
There is a newer version: 1.6.10
/*
 * Copyright (c) 2014-2019 Aurélien Broszniowski
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package io.rainfall.statistics;


import java.util.concurrent.ConcurrentHashMap;

/**
 * A {@link Statistics} instance holds the statistics of all results at a given point in time
 *
 * @author Aurelien Broszniowski
 */

public class Statistics> {

  private final String name;
  private Enum[] results;
  private final ConcurrentHashMap periodicCounters = new ConcurrentHashMap();
  private final ConcurrentHashMap periodicTotalLatenciesInNs = new ConcurrentHashMap();
  private final ConcurrentHashMap cumulativeCounters = new ConcurrentHashMap();   //TODO replace with max, average
  private final ConcurrentHashMap cumulativeTotalLatenciesInNs = new ConcurrentHashMap();
  private Long periodicStartTime;
  private Long cumulativeStartTime;

  public Statistics(String name, Enum[] results) {
    this.name = name;
    this.results = results;
    for (Enum result : results) {
      this.periodicCounters.put(result, new LongAdder());
      this.periodicTotalLatenciesInNs.put(result, new LongAdder());
      this.cumulativeCounters.put(result, new LongAdder());
      this.cumulativeTotalLatenciesInNs.put(result, new LongAdder());
    }
    this.periodicStartTime = getTimeInNs();
    this.cumulativeStartTime = this.periodicStartTime;
  }

  Statistics(String name, Enum[] results, long startTime) {
    this.results = results;
    this.name = name;
    for (Enum result : results) {
      this.periodicCounters.put(result, new LongAdder());
      this.periodicTotalLatenciesInNs.put(result, new LongAdder());
      this.cumulativeCounters.put(result, new LongAdder());
      this.cumulativeTotalLatenciesInNs.put(result, new LongAdder());
    }
    this.periodicStartTime = startTime;
    this.cumulativeStartTime = this.periodicStartTime;
  }

  LongAdder getPeriodicCounters(Enum result) {
    return periodicCounters.get(result);
  }

  LongAdder getPeriodicTotalLatenciesInNs(Enum result) {
    return periodicTotalLatenciesInNs.get(result);
  }

  LongAdder getCumulativeCounters(Enum result) {
    return cumulativeCounters.get(result);
  }

  LongAdder getCumulativeTotalLatencies(Enum result) {
    return cumulativeTotalLatenciesInNs.get(result);
  }

  public void increaseCounterAndSetLatencyInNs(final Enum result, final long latency) {
    periodicCounters.get(result).increment();
    periodicTotalLatenciesInNs.get(result).add(latency);
  }

  public String getName() {
    return name;
  }

  protected long getTimeInNs() {
    return System.nanoTime();
  }

  public synchronized long getCurrentTps(Enum result) {
    long time = getTimeInNs() - periodicStartTime;
    return time < 1000000L ? periodicCounters.get(result).longValue() :
        periodicCounters.get(result).longValue() * 1000L * 1000000L / time;
  }

  public synchronized StatisticsPeek peek(final long timestamp) {
    StatisticsPeek statisticsPeek = new StatisticsPeek(this.name, this.results, timestamp);
    long now = getTimeInNs();
    statisticsPeek.setPeriodicValues(now - periodicStartTime, periodicCounters, periodicTotalLatenciesInNs);
    for (Enum key : results) {
      this.cumulativeCounters.get(key).add(this.periodicCounters.get(key).sumThenReset());
      this.cumulativeTotalLatenciesInNs.get(key).add(this.periodicTotalLatenciesInNs.get(key).sumThenReset());
    }
    statisticsPeek.setCumulativeValues(now - cumulativeStartTime, cumulativeCounters, cumulativeTotalLatenciesInNs);
    this.periodicStartTime = getTimeInNs();
    return statisticsPeek;
  }

  public void reset() {
    for (Enum key : results) {
      this.periodicCounters.get(key).reset();
      this.periodicTotalLatenciesInNs.get(key).reset();
      this.cumulativeCounters.get(key).reset();
      this.cumulativeTotalLatenciesInNs.get(key).reset();
    }
    this.periodicStartTime = getTimeInNs();
    this.cumulativeStartTime = this.periodicStartTime;
  }
}