|
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