/*
* Copyright 2013 Proofpoint, Inc.
*
* 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 com.facebook.presto.jdbc.internal.airlift.stats;
import com.facebook.presto.jdbc.internal.jackson.annotation.JsonCreator;
import com.facebook.presto.jdbc.internal.jackson.annotation.JsonProperty;
import com.facebook.presto.jdbc.internal.guava.base.Objects;
import com.facebook.presto.jdbc.internal.guava.base.Ticker;
import com.facebook.presto.jdbc.internal.airlift.stats.TimeDistribution.TimeDistributionSnapshot;
import com.facebook.presto.jdbc.internal.airlift.units.Duration;
import org.weakref.jmx.Managed;
import org.weakref.jmx.Nested;
import java.util.concurrent.Callable;
import java.util.concurrent.TimeUnit;
public class TimeStat
{
private final TimeDistribution oneMinute;
private final TimeDistribution fiveMinutes;
private final TimeDistribution fifteenMinutes;
private final TimeDistribution allTime;
private final Ticker ticker;
public TimeStat()
{
this(Ticker.systemTicker(), TimeUnit.SECONDS);
}
public TimeStat(Ticker ticker)
{
this(ticker, TimeUnit.SECONDS);
}
public TimeStat(TimeUnit unit)
{
this(Ticker.systemTicker(), unit);
}
public TimeStat(Ticker ticker, TimeUnit unit)
{
this.ticker = ticker;
oneMinute = new TimeDistribution(ExponentialDecay.oneMinute(), unit);
fiveMinutes = new TimeDistribution(ExponentialDecay.fiveMinutes(), unit);
fifteenMinutes = new TimeDistribution(ExponentialDecay.fifteenMinutes(), unit);
allTime = new TimeDistribution(unit);
}
public void add(double value, TimeUnit timeUnit)
{
add(new Duration(value, timeUnit));
}
public void add(Duration duration)
{
add((long) duration.getValue(TimeUnit.NANOSECONDS));
}
private void add(long value)
{
oneMinute.add(value);
fiveMinutes.add(value);
fifteenMinutes.add(value);
allTime.add(value);
}
public <T> T time(Callable<T> callable)
throws Exception
{
long start = ticker.read();
T result = callable.call();
add(ticker.read() - start);
return result;
}
public BlockTimer time()
{
return new BlockTimer();
}
public class BlockTimer
implements AutoCloseable
{
private final long start = ticker.read();
@Override
public void close()
{
add(ticker.read() - start);
}
}
@Managed
@Nested
public TimeDistribution getOneMinute()
{
return oneMinute;
}
@Managed
@Nested
public TimeDistribution getFiveMinutes()
{
return fiveMinutes;
}
@Managed
@Nested
public TimeDistribution getFifteenMinutes()
{
return fifteenMinutes;
}
@Managed
@Nested
public TimeDistribution getAllTime()
{
return allTime;
}
public TimeDistributionStatSnapshot snapshot()
{
return new TimeDistributionStatSnapshot(
getOneMinute().snapshot(),
getFiveMinutes().snapshot(),
getFifteenMinutes().snapshot(),
getAllTime().snapshot());
}
public static class TimeDistributionStatSnapshot
{
private final TimeDistributionSnapshot oneMinute;
private final TimeDistributionSnapshot fiveMinute;
private final TimeDistributionSnapshot fifteenMinute;
private final TimeDistributionSnapshot allTime;
@JsonCreator
public TimeDistributionStatSnapshot(
@JsonProperty("oneMinute") TimeDistributionSnapshot oneMinute,
@JsonProperty("fiveMinute") TimeDistributionSnapshot fiveMinute,
@JsonProperty("fifteenMinute") TimeDistributionSnapshot fifteenMinute,
@JsonProperty("allTime") TimeDistributionSnapshot allTime)
{
this.oneMinute = oneMinute;
this.fiveMinute = fiveMinute;
this.fifteenMinute = fifteenMinute;
this.allTime = allTime;
}
@JsonProperty
public TimeDistributionSnapshot getOneMinute()
{
return oneMinute;
}
@JsonProperty
public TimeDistributionSnapshot getFiveMinutes()
{
return fiveMinute;
}
@JsonProperty
public TimeDistributionSnapshot getFifteenMinutes()
{
return fifteenMinute;
}
@JsonProperty
public TimeDistributionSnapshot getAllTime()
{
return allTime;
}
@Override
public String toString()
{
return Objects.toStringHelper(this)
.add("oneMinute", oneMinute)
.add("fiveMinute", fiveMinute)
.add("fifteenMinute", fifteenMinute)
.add("allTime", allTime)
.toString();
}
}
}