|
16 | 16 |
|
17 | 17 | package org.springframework.boot.actuate.autoconfigure.tracing.otlp;
|
18 | 18 |
|
| 19 | +import java.time.Duration; |
19 | 20 | import java.util.List;
|
20 | 21 | import java.util.function.Supplier;
|
21 | 22 |
|
@@ -230,6 +231,45 @@ void grpcShouldUseMeterProviderIfSet() {
|
230 | 231 | });
|
231 | 232 | }
|
232 | 233 |
|
| 234 | + @Test |
| 235 | + void shouldCustomizeHttpTransportWithOtlpHttpSpanExporterBuilderCustomizer() { |
| 236 | + Duration connectTimeout = Duration.ofMinutes(20); |
| 237 | + Duration timeout = Duration.ofMinutes(10); |
| 238 | + this.contextRunner |
| 239 | + .withBean("httpCustomizer1", OtlpHttpSpanExporterBuilderCustomizer.class, |
| 240 | + () -> (builder) -> builder.setConnectTimeout(connectTimeout)) |
| 241 | + .withBean("httpCustomizer2", OtlpHttpSpanExporterBuilderCustomizer.class, |
| 242 | + () -> (builder) -> builder.setTimeout(timeout)) |
| 243 | + .withPropertyValues("management.otlp.tracing.endpoint=http://localhost:4317/v1/traces") |
| 244 | + .run((context) -> { |
| 245 | + assertThat(context).hasSingleBean(OtlpHttpSpanExporter.class).hasSingleBean(SpanExporter.class); |
| 246 | + OtlpHttpSpanExporter exporter = context.getBean(OtlpHttpSpanExporter.class); |
| 247 | + assertThat(exporter).extracting("delegate.httpSender.client") |
| 248 | + .hasFieldOrPropertyWithValue("connectTimeoutMillis", (int) connectTimeout.toMillis()) |
| 249 | + .hasFieldOrPropertyWithValue("callTimeoutMillis", (int) timeout.toMillis()); |
| 250 | + }); |
| 251 | + } |
| 252 | + |
| 253 | + @Test |
| 254 | + void shouldCustomizeGrpcTransportWhenEnabledWithOtlpGrpcSpanExporterBuilderCustomizer() { |
| 255 | + Duration timeout = Duration.ofMinutes(10); |
| 256 | + Duration connectTimeout = Duration.ofMinutes(20); |
| 257 | + this.contextRunner |
| 258 | + .withBean("grpcCustomizer1", OtlpGrpcSpanExporterBuilderCustomizer.class, |
| 259 | + () -> (builder) -> builder.setConnectTimeout(connectTimeout)) |
| 260 | + .withBean("grpcCustomizer2", OtlpGrpcSpanExporterBuilderCustomizer.class, |
| 261 | + () -> (builder) -> builder.setTimeout(timeout)) |
| 262 | + .withPropertyValues("management.otlp.tracing.endpoint=http://localhost:4317/v1/traces", |
| 263 | + "management.otlp.tracing.transport=grpc") |
| 264 | + .run((context) -> { |
| 265 | + assertThat(context).hasSingleBean(OtlpGrpcSpanExporter.class).hasSingleBean(SpanExporter.class); |
| 266 | + OtlpGrpcSpanExporter exporter = context.getBean(OtlpGrpcSpanExporter.class); |
| 267 | + assertThat(exporter).extracting("delegate.grpcSender.client") |
| 268 | + .hasFieldOrPropertyWithValue("connectTimeoutMillis", (int) connectTimeout.toMillis()) |
| 269 | + .hasFieldOrPropertyWithValue("callTimeoutMillis", (int) timeout.toMillis()); |
| 270 | + }); |
| 271 | + } |
| 272 | + |
233 | 273 | @Configuration(proxyBeanMethods = false)
|
234 | 274 | private static final class MeterProviderConfiguration {
|
235 | 275 |
|
|
0 commit comments