io.micrometer.core.instrument.simple.SimpleMeterRegistry Maven / Gradle / Ivy
Show all versions of micrometer-core Show documentation
/**
* Copyright 2017 VMware, 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
*
* https://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.micrometer.core.instrument.simple;
import io.micrometer.core.instrument.*;
import io.micrometer.core.instrument.cumulative.*;
import io.micrometer.core.instrument.distribution.DistributionStatisticConfig;
import io.micrometer.core.instrument.distribution.HistogramGauges;
import io.micrometer.core.instrument.distribution.pause.PauseDetector;
import io.micrometer.core.instrument.internal.DefaultGauge;
import io.micrometer.core.instrument.internal.DefaultLongTaskTimer;
import io.micrometer.core.instrument.internal.DefaultMeter;
import io.micrometer.core.instrument.step.*;
import io.micrometer.core.lang.Nullable;
import java.util.concurrent.TimeUnit;
import java.util.function.ToDoubleFunction;
import java.util.function.ToLongFunction;
/**
* A minimal meter registry implementation primarily used for tests.
*
* @author Jon Schneider
* @author Johnny Lim
*/
public class SimpleMeterRegistry extends MeterRegistry {
private final SimpleConfig config;
public SimpleMeterRegistry() {
this(SimpleConfig.DEFAULT, Clock.SYSTEM);
}
public SimpleMeterRegistry(SimpleConfig config, Clock clock) {
super(clock);
config.requireValid();
this.config = config;
}
@Override
protected DistributionSummary newDistributionSummary(Meter.Id id, DistributionStatisticConfig distributionStatisticConfig, double scale) {
DistributionStatisticConfig merged = distributionStatisticConfig.merge(DistributionStatisticConfig.builder()
.expiry(config.step())
.build());
DistributionSummary summary;
switch (config.mode()) {
case CUMULATIVE:
summary = new CumulativeDistributionSummary(id, clock, merged, scale, false);
break;
case STEP:
default:
summary = new StepDistributionSummary(id, clock, merged, scale, config.step().toMillis(), false);
break;
}
HistogramGauges.registerWithCommonFormat(summary, this);
return summary;
}
@Override
protected Meter newMeter(Meter.Id id, Meter.Type type, Iterable measurements) {
return new DefaultMeter(id, type, measurements);
}
@Override
protected Timer newTimer(Meter.Id id, DistributionStatisticConfig distributionStatisticConfig, PauseDetector pauseDetector) {
DistributionStatisticConfig merged = distributionStatisticConfig.merge(DistributionStatisticConfig.builder()
.expiry(config.step())
.build());
Timer timer;
switch (config.mode()) {
case CUMULATIVE:
timer = new CumulativeTimer(id, clock, merged, pauseDetector, getBaseTimeUnit(), false);
break;
case STEP:
default:
timer = new StepTimer(id, clock, merged, pauseDetector, getBaseTimeUnit(), config.step().toMillis(), false);
break;
}
HistogramGauges.registerWithCommonFormat(timer, this);
return timer;
}
@Override
protected Gauge newGauge(Meter.Id id, @Nullable T obj, ToDoubleFunction valueFunction) {
return new DefaultGauge<>(id, obj, valueFunction);
}
@Override
protected Counter newCounter(Meter.Id id) {
switch (config.mode()) {
case CUMULATIVE:
return new CumulativeCounter(id);
case STEP:
default:
return new StepCounter(id, clock, config.step().toMillis());
}
}
@Override
protected LongTaskTimer newLongTaskTimer(Meter.Id id, DistributionStatisticConfig distributionStatisticConfig) {
DefaultLongTaskTimer ltt = new DefaultLongTaskTimer(id, clock, getBaseTimeUnit(), distributionStatisticConfig, false);
HistogramGauges.registerWithCommonFormat(ltt, this);
return ltt;
}
@Override
protected FunctionTimer newFunctionTimer(Meter.Id id, T obj, ToLongFunction countFunction, ToDoubleFunction totalTimeFunction, TimeUnit totalTimeFunctionUnit) {
switch (config.mode()) {
case CUMULATIVE:
return new CumulativeFunctionTimer<>(id, obj, countFunction, totalTimeFunction, totalTimeFunctionUnit, getBaseTimeUnit());
case STEP:
default:
return new StepFunctionTimer<>(id, clock, config.step().toMillis(), obj, countFunction, totalTimeFunction, totalTimeFunctionUnit, getBaseTimeUnit());
}
}
@Override
protected FunctionCounter newFunctionCounter(Meter.Id id, T obj, ToDoubleFunction countFunction) {
switch (config.mode()) {
case CUMULATIVE:
return new CumulativeFunctionCounter<>(id, obj, countFunction);
case STEP:
default:
return new StepFunctionCounter<>(id, clock, config.step().toMillis(), obj, countFunction);
}
}
@Override
protected TimeUnit getBaseTimeUnit() {
return TimeUnit.SECONDS;
}
@Override
protected DistributionStatisticConfig defaultHistogramConfig() {
return DistributionStatisticConfig.builder()
.expiry(config.step())
.build()
.merge(DistributionStatisticConfig.DEFAULT);
}
}