|
6 | 6 | package io.opentelemetry.instrumentation.spring.autoconfigure.internal.instrumentation.runtimemetrics;
|
7 | 7 |
|
8 | 8 | import io.opentelemetry.api.OpenTelemetry;
|
9 |
| -import io.opentelemetry.instrumentation.api.incubator.config.internal.InstrumentationConfig; |
10 | 9 | import io.opentelemetry.instrumentation.runtimemetrics.java8.RuntimeMetrics;
|
11 |
| -import java.util.function.Consumer; |
| 10 | +import io.opentelemetry.instrumentation.spring.autoconfigure.internal.ConditionalOnEnabledInstrumentation; |
| 11 | +import io.opentelemetry.instrumentation.spring.autoconfigure.internal.properties.ConfigPropertiesBridge; |
| 12 | +import io.opentelemetry.sdk.autoconfigure.spi.ConfigProperties; |
| 13 | +import javax.annotation.PreDestroy; |
12 | 14 | import org.slf4j.Logger;
|
13 | 15 | import org.slf4j.LoggerFactory;
|
| 16 | +import org.springframework.boot.CommandLineRunner; |
14 | 17 | import org.springframework.boot.autoconfigure.condition.AnyNestedCondition;
|
15 | 18 | import org.springframework.boot.autoconfigure.condition.ConditionalOnJava;
|
16 | 19 | import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
|
17 | 20 | import org.springframework.boot.system.JavaVersion;
|
| 21 | +import org.springframework.context.annotation.Bean; |
| 22 | +import org.springframework.context.annotation.Conditional; |
| 23 | +import org.springframework.context.annotation.Configuration; |
18 | 24 |
|
19 | 25 | /**
|
20 | 26 | * Configures runtime metrics collection for Java 8.
|
21 | 27 | *
|
22 | 28 | * <p>This class is internal and is hence not for public use. Its APIs are unstable and can change
|
23 | 29 | * at any time.
|
24 | 30 | */
|
25 |
| -@ConditionalOnJava(value = JavaVersion.SEVENTEEN, range = ConditionalOnJava.Range.OLDER_THAN) |
26 |
| -public class Java8RuntimeMetricsAutoConfiguration implements RuntimeMetricsProvider { |
27 |
| - private static final Logger logger = LoggerFactory.getLogger(Java8RuntimeMetricsAutoConfiguration.class); |
28 |
| - |
29 |
| - @Override |
30 |
| - public void start( |
31 |
| - OpenTelemetry openTelemetry, Consumer<Runnable> shutdownHook, InstrumentationConfig config) { |
32 |
| - logger.debug("Use runtime metrics instrumentation for Java 8"); |
33 |
| - RuntimeMetrics.builder(openTelemetry) |
34 |
| - .setShutdownHook(shutdownHook) |
35 |
| - .startFromInstrumentationConfig(config); |
| 31 | +@ConditionalOnEnabledInstrumentation(module = "runtime-telemetry") |
| 32 | +@Configuration |
| 33 | +@Conditional(Java8RuntimeMetricsAutoConfiguration.Java8MetricsCondition.class) |
| 34 | +public class Java8RuntimeMetricsAutoConfiguration { |
| 35 | + private static final Logger logger = |
| 36 | + LoggerFactory.getLogger(Java8RuntimeMetricsAutoConfiguration.class); |
| 37 | + |
| 38 | + private Runnable shutdownHook; |
| 39 | + |
| 40 | + @PreDestroy |
| 41 | + public void stopMetrics() { |
| 42 | + if (shutdownHook != null) { |
| 43 | + shutdownHook.run(); |
| 44 | + } |
| 45 | + } |
| 46 | + |
| 47 | + @Bean |
| 48 | + CommandLineRunner startMetrics(OpenTelemetry openTelemetry, ConfigProperties configProperties) { |
| 49 | + return (args) -> { |
| 50 | + logger.debug("Use runtime metrics instrumentation for Java 8"); |
| 51 | + RuntimeMetrics.builder(openTelemetry) |
| 52 | + .setShutdownHook(runnable -> shutdownHook = runnable) |
| 53 | + .startFromInstrumentationConfig(new ConfigPropertiesBridge(configProperties)); |
| 54 | + }; |
36 | 55 | }
|
37 | 56 |
|
38 |
| - public class Java8MetricsCondition extends AnyNestedCondition { |
| 57 | + static class Java8MetricsCondition extends AnyNestedCondition { |
39 | 58 | public Java8MetricsCondition() {
|
40 | 59 | super(ConfigurationPhase.PARSE_CONFIGURATION);
|
41 | 60 | }
|
42 | 61 |
|
43 | 62 | @ConditionalOnJava(value = JavaVersion.SEVENTEEN, range = ConditionalOnJava.Range.OLDER_THAN)
|
44 |
| - class OlderThanJava17 {} |
| 63 | + static class OlderThanJava17 {} |
45 | 64 |
|
46 | 65 | // GraalVM native image does not support Java 17 metrics yet
|
47 | 66 | @ConditionalOnProperty("org.graalvm.nativeimage.imagecode")
|
48 |
| - class GraalVM {} |
| 67 | + static class GraalVm {} |
49 | 68 | }
|
50 | 69 | }
|
0 commit comments