/*
* Copyright (c) 2007, 2008, 2009, 2010, 2011, 2012, 2013, 2014, 2015, 2016, 2017 David Berkman
*
* This file is part of the SmallMind Code Project.
*
* The SmallMind Code Project is free software, you can redistribute
* it and/or modify it under either, at your discretion...
*
* 1) The terms of GNU Affero General Public License as published by the
* Free Software Foundation, either version 3 of the License, or (at
* your option) any later version.
*
* ...or...
*
* 2) The terms of the Apache License, Version 2.0.
*
* The SmallMind Code Project is distributed in the hope that it will
* be useful, but WITHOUT ANY WARRANTY; without even the implied warranty
* of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* General Public License or Apache License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* and the Apache License along with the SmallMind Code Project. If not, see
* <http://www.gnu.org/licenses/> or <http://www.apache.org/licenses/LICENSE-2.0>.
*
* Additional permission under the GNU Affero GPL version 3 section 7
* ------------------------------------------------------------------
* If you modify this Program, or any covered work, by linking or
* combining it with other code, such other code is not for that reason
* alone subject to any of the requirements of the GNU Affero GPL
* version 3.
*/
package org.smallmind.instrument;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.ScheduledFuture;
import java.util.concurrent.ThreadFactory;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicLong;
import org.smallmind.instrument.context.MetricFact;
import org.smallmind.instrument.context.MetricItem;
import org.smallmind.instrument.context.MetricSnapshot;
import org.smallmind.nutsnbolts.time.TimeUtility;
public class GaugeImpl extends MetricImpl<Gauge> implements Gauge {
private static final ScheduledExecutorService SCHEDULED_EXECUTOR = Executors.newScheduledThreadPool(2, new ThreadFactory() {
@Override
public Thread newThread (final Runnable runnable) {
Thread thread = new Thread(runnable);
thread.setDaemon(true);
return thread;
}
});
private final ExponentiallyWeightedMovingAverage m1Average;
private final ExponentiallyWeightedMovingAverage m5Average;
private final ExponentiallyWeightedMovingAverage m15Average;
private final ScheduledFuture<?> future;
private final AtomicLong startTime;
private final AtomicLong count = new AtomicLong(0);
private final Clock clock;
private final TimeUnit tickTimeUnit;
public GaugeImpl () {
this(5, TimeUnit.SECONDS, Clocks.EPOCH.getClock());
}
public GaugeImpl (long tickInterval, TimeUnit tickTimeUnit) {
this(tickInterval, tickTimeUnit, Clocks.EPOCH.getClock());
}
public GaugeImpl (long tickInterval, TimeUnit tickTimeUnit, Clock clock) {
this.tickTimeUnit = tickTimeUnit;
this.clock = clock;
startTime = new AtomicLong(clock.getTimeMilliseconds());
m1Average = ExponentiallyWeightedMovingAverage.lastOneMinute(tickInterval, tickTimeUnit);
m5Average = ExponentiallyWeightedMovingAverage.lastFiveMinutes(tickInterval, tickTimeUnit);
m15Average = ExponentiallyWeightedMovingAverage.lastFifteenMinutes(tickInterval, tickTimeUnit);
future = SCHEDULED_EXECUTOR.scheduleAtFixedRate(new Runnable() {
@Override
public void run () {
m1Average.tick();
m5Average.tick();
m15Average.tick();
}
}, tickInterval, tickInterval, tickTimeUnit);
}
@Override
public Class<Gauge> getMetricClass () {
return Gauge.class;
}
@Override
public void clear () {
MetricSnapshot metricSnapshot;
startTime.set(clock.getTimeMilliseconds());
count.set(0);
m15Average.clear();
m15Average.clear();
m15Average.clear();
if ((metricSnapshot = getMetricSnapshot()) != null) {
if (metricSnapshot.willTrace(MetricFact.COUNT)) {
metricSnapshot.addItem(new MetricItem<>("count", 0L));
}
if (metricSnapshot.willTrace(MetricFact.M1_AVG)) {
metricSnapshot.addItem(new MetricItem<>("1 min avg", 0.0));
}
if (metricSnapshot.willTrace(MetricFact.M5_AVG)) {
metricSnapshot.addItem(new MetricItem<>("5 min avg", 0.0));
}
if (metricSnapshot.willTrace(MetricFact.M15_AVG)) {
metricSnapshot.addItem(new MetricItem<>("15 min avg", 0.0));
}
}
}
@Override
public void mark () {
mark(1);
}
@Override
public void mark (long n) {
MetricSnapshot metricSnapshot;
long current;
current = count.addAndGet(n);
m1Average.update(n);
m5Average.update(n);
m15Average.update(n);
if ((metricSnapshot = getMetricSnapshot()) != null) {
if (metricSnapshot.willTrace(MetricFact.COUNT)) {
metricSnapshot.addItem(new MetricItem<>("count", current));
}
if (metricSnapshot.willTrace(MetricFact.M1_AVG)) {
metricSnapshot.addItem(new MetricItem<>("1 min avg", m1Average.getMovingAverage(tickTimeUnit)));
}
if (metricSnapshot.willTrace(MetricFact.M5_AVG)) {
metricSnapshot.addItem(new MetricItem<>("5 min avg", m5Average.getMovingAverage(tickTimeUnit)));
}
if (metricSnapshot.willTrace(MetricFact.M15_AVG)) {
metricSnapshot.addItem(new MetricItem<>("15 min avg", m15Average.getMovingAverage(tickTimeUnit)));
}
}
}
@Override
public Clock getClock () {
return clock;
}
@Override
public TimeUnit getRateTimeUnit () {
return tickTimeUnit;
}
@Override
public long getCount () {
return count.get();
}
@Override
public double getOneMinuteAvgRate () {
return m1Average.getMovingAverage(tickTimeUnit);
}
@Override
public double getFiveMinuteAvgRate () {
return m5Average.getMovingAverage(tickTimeUnit);
}
@Override
public double getFifteenMinuteAvgRate () {
return m15Average.getMovingAverage(tickTimeUnit);
}
@Override
public double getAverageRate () {
long currentCount = count.get();
if (currentCount == 0) {
return 0.0;
} else {
return (((double)currentCount) / (clock.getTimeMilliseconds() - startTime.get())) * TimeUtility.convertToDouble(1, tickTimeUnit, TimeUnit.MILLISECONDS);
}
}
@Override
public void stop () {
future.cancel(false);
}
}