Configure Micrometer distribution buckets for OpenTelemetry

Co-authored-by: Jon Chambers <63609320+jon-signal@users.noreply.github.com>
This commit is contained in:
Jonathan Klabunde Tomer
2025-08-11 11:45:33 -07:00
committed by GitHub
parent 1429efd573
commit 36d0c4422e
19 changed files with 155 additions and 67 deletions

View File

@@ -14,6 +14,7 @@ import static org.mockito.Mockito.when;
import io.micrometer.core.instrument.Counter;
import io.micrometer.core.instrument.Meter;
import io.micrometer.core.instrument.simple.SimpleMeterRegistry;
import java.time.Duration;
import java.util.Optional;
import java.util.UUID;
import org.junit.jupiter.api.BeforeEach;
@@ -42,7 +43,7 @@ class MessageMetricsTest {
when(account.isIdentifiedBy(new AciServiceIdentifier(aci))).thenReturn(true);
when(account.isIdentifiedBy(new PniServiceIdentifier(pni))).thenReturn(true);
simpleMeterRegistry = new SimpleMeterRegistry();
messageMetrics = new MessageMetrics(simpleMeterRegistry);
messageMetrics = new MessageMetrics(simpleMeterRegistry, Duration.ofDays(30));
}
@Test

View File

@@ -10,14 +10,26 @@ import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;
import io.micrometer.core.instrument.Clock;
import io.micrometer.core.instrument.DistributionSummary;
import io.micrometer.core.instrument.Meter;
import io.micrometer.core.instrument.MeterRegistry;
import io.micrometer.core.instrument.cumulative.CumulativeDistributionSummary;
import io.micrometer.core.instrument.distribution.CountAtBucket;
import io.micrometer.core.instrument.distribution.DistributionStatisticConfig;
import io.micrometer.core.instrument.simple.SimpleMeterRegistry;
import io.micrometer.registry.otlp.OtlpMeterRegistry;
import java.time.Duration;
import java.util.List;
import java.util.Map;
import java.util.stream.IntStream;
import org.assertj.core.api.AbstractStringAssert;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.params.ParameterizedTest;
import org.junit.jupiter.params.provider.ValueSource;
import org.whispersystems.textsecuregcm.configuration.OpenTelemetryConfiguration;
import org.whispersystems.textsecuregcm.configuration.dynamic.DynamicConfiguration;
import org.whispersystems.textsecuregcm.configuration.dynamic.DynamicMetricsConfiguration;
import org.whispersystems.textsecuregcm.storage.DynamicConfigurationManager;
@@ -81,4 +93,34 @@ class MetricsUtilTest {
assertThat(registry.getMeters()).hasSize(enableAwsSdkMetrics ? 1 : 0);
}
@Test
void exponentialHistogramBuckets() {
final OpenTelemetryConfiguration openTelemetryConfig =
new OpenTelemetryConfiguration(true, Duration.ZERO, 8, Map.of(), Map.of());
// Yes, this has to be OtlpMeterRegistry—its histograms behave differently than SimpleMeterRegistry's 😭
final MeterRegistry registry = new OtlpMeterRegistry(openTelemetryConfig, Clock.SYSTEM);
MetricsUtil.configureHistogramFilters(registry.config(), openTelemetryConfig);
final DistributionSummary summary = DistributionSummary.builder("test.distribution")
.publishPercentileHistogram()
.minimumExpectedValue(1.0)
.maximumExpectedValue(64.0)
.register(registry);
IntStream.range(1, 128).forEach(summary::record);
final CountAtBucket[] counts = summary.takeSnapshot().histogramCounts();
assertThat(counts).containsExactly(
new CountAtBucket(1.0, 1),
new CountAtBucket(2.0, 1),
new CountAtBucket(4.0, 2),
new CountAtBucket(8.0, 4),
new CountAtBucket(16.0, 8),
new CountAtBucket(32.0, 16),
new CountAtBucket(64.0, 32),
new CountAtBucket(Double.POSITIVE_INFINITY, 63));
}
}