Compare commits
No commits in common. "main" and "0.2.0" have entirely different histories.
@ -1,8 +1,4 @@
|
|||||||
include:
|
include:
|
||||||
- project: 'gitlab/gitlab'
|
- project: 'gitlab/gitlab'
|
||||||
ref: 'main'
|
ref: 'main'
|
||||||
file: 'ci-templates/maven_deps.yml'
|
file: 'ci-templates/maven_deps.yml'
|
||||||
stages:
|
|
||||||
- test
|
|
||||||
- versioning
|
|
||||||
- deploy
|
|
||||||
@ -1 +0,0 @@
|
|||||||
`docker-compose up -d` to start a local opentelemetry collector + zipkin for development purposes
|
|
||||||
@ -1,32 +0,0 @@
|
|||||||
receivers:
|
|
||||||
otlp:
|
|
||||||
protocols:
|
|
||||||
grpc:
|
|
||||||
http:
|
|
||||||
cors:
|
|
||||||
allowed_origins:
|
|
||||||
- http://*
|
|
||||||
- https://*
|
|
||||||
|
|
||||||
exporters:
|
|
||||||
zipkin:
|
|
||||||
endpoint: "http://zipkin-all-in-one:9411/api/v2/spans"
|
|
||||||
prometheus:
|
|
||||||
endpoint: "0.0.0.0:9464"
|
|
||||||
|
|
||||||
processors:
|
|
||||||
batch:
|
|
||||||
|
|
||||||
service:
|
|
||||||
telemetry:
|
|
||||||
logs:
|
|
||||||
level: "debug"
|
|
||||||
pipelines:
|
|
||||||
traces:
|
|
||||||
receivers: [otlp]
|
|
||||||
exporters: [zipkin]
|
|
||||||
processors: [batch]
|
|
||||||
metrics:
|
|
||||||
receivers: [otlp]
|
|
||||||
exporters: [prometheus]
|
|
||||||
processors: [batch]
|
|
||||||
@ -1,18 +0,0 @@
|
|||||||
version: "3"
|
|
||||||
services:
|
|
||||||
collector:
|
|
||||||
image: otel/opentelemetry-collector-contrib:0.53.0
|
|
||||||
command: ["--config=/conf/collector-config.yaml"]
|
|
||||||
volumes:
|
|
||||||
- ./collector-config.yaml:/conf/collector-config.yaml
|
|
||||||
ports:
|
|
||||||
- "9464:9464"
|
|
||||||
- "4317:4317"
|
|
||||||
- "4318:4318"
|
|
||||||
depends_on:
|
|
||||||
- zipkin-all-in-one
|
|
||||||
|
|
||||||
zipkin-all-in-one:
|
|
||||||
image: openzipkin/zipkin:latest
|
|
||||||
ports:
|
|
||||||
- "9411:9411"
|
|
||||||
16
pom.xml
16
pom.xml
@ -5,7 +5,7 @@
|
|||||||
<parent>
|
<parent>
|
||||||
<groupId>org.springframework.boot</groupId>
|
<groupId>org.springframework.boot</groupId>
|
||||||
<artifactId>spring-boot-starter-parent</artifactId>
|
<artifactId>spring-boot-starter-parent</artifactId>
|
||||||
<version>3.2.3</version>
|
<version>3.1.5</version>
|
||||||
<relativePath/>
|
<relativePath/>
|
||||||
</parent>
|
</parent>
|
||||||
|
|
||||||
@ -18,7 +18,7 @@
|
|||||||
<properties>
|
<properties>
|
||||||
<java.version>17</java.version>
|
<java.version>17</java.version>
|
||||||
<spring-cloud.version>2022.0.4</spring-cloud.version>
|
<spring-cloud.version>2022.0.4</spring-cloud.version>
|
||||||
<tenant-commons.version>0.20.0</tenant-commons.version>
|
<tenant-commons.version>0.19.0</tenant-commons.version>
|
||||||
</properties>
|
</properties>
|
||||||
|
|
||||||
|
|
||||||
@ -44,12 +44,6 @@
|
|||||||
<artifactId>spring-boot-configuration-processor</artifactId>
|
<artifactId>spring-boot-configuration-processor</artifactId>
|
||||||
<optional>true</optional>
|
<optional>true</optional>
|
||||||
</dependency>
|
</dependency>
|
||||||
<dependency>
|
|
||||||
<groupId>org.springframework.boot</groupId>
|
|
||||||
<artifactId>spring-boot-starter-web</artifactId>
|
|
||||||
<optional>true</optional>
|
|
||||||
<scope>provided</scope>
|
|
||||||
</dependency>
|
|
||||||
|
|
||||||
<!-- Miscellaneous -->
|
<!-- Miscellaneous -->
|
||||||
<dependency>
|
<dependency>
|
||||||
@ -87,12 +81,6 @@
|
|||||||
<artifactId>spring-rabbit-test</artifactId>
|
<artifactId>spring-rabbit-test</artifactId>
|
||||||
<scope>test</scope>
|
<scope>test</scope>
|
||||||
</dependency>
|
</dependency>
|
||||||
<dependency>
|
|
||||||
<groupId>io.micrometer</groupId>
|
|
||||||
<artifactId>micrometer-tracing-test</artifactId>
|
|
||||||
<scope>test</scope>
|
|
||||||
</dependency>
|
|
||||||
|
|
||||||
</dependencies>
|
</dependencies>
|
||||||
<dependencyManagement>
|
<dependencyManagement>
|
||||||
<dependencies>
|
<dependencies>
|
||||||
|
|||||||
@ -1,60 +1,17 @@
|
|||||||
package com.knecon.fforesight.tracing;
|
package com.knecon.fforesight.tracing;
|
||||||
|
|
||||||
import org.springframework.beans.factory.annotation.Autowired;
|
|
||||||
import org.springframework.boot.actuate.autoconfigure.tracing.ConditionalOnEnabledTracing;
|
|
||||||
import org.springframework.boot.autoconfigure.AutoConfiguration;
|
|
||||||
import org.springframework.context.annotation.Bean;
|
|
||||||
import org.springframework.context.annotation.Import;
|
|
||||||
import org.springframework.context.annotation.PropertySource;
|
|
||||||
import org.springframework.http.server.observation.ServerRequestObservationContext;
|
|
||||||
|
|
||||||
import io.micrometer.context.ContextRegistry;
|
|
||||||
import io.micrometer.observation.ObservationPredicate;
|
|
||||||
import io.micrometer.tracing.Tracer;
|
|
||||||
import io.micrometer.tracing.contextpropagation.ObservationAwareSpanThreadLocalAccessor;
|
|
||||||
import jakarta.annotation.PostConstruct;
|
import jakarta.annotation.PostConstruct;
|
||||||
import lombok.extern.slf4j.Slf4j;
|
import lombok.extern.slf4j.Slf4j;
|
||||||
|
import org.springframework.boot.autoconfigure.AutoConfiguration;
|
||||||
|
import org.springframework.context.annotation.ComponentScan;
|
||||||
|
|
||||||
@AutoConfiguration
|
|
||||||
@ConditionalOnEnabledTracing
|
|
||||||
@Import({
|
|
||||||
TaskExecutionConfiguration.class,
|
|
||||||
TracingRabbitConfiguration.class,
|
|
||||||
WebMvcConfiguration.class,
|
|
||||||
TracingSafetyCheck.class
|
|
||||||
})
|
|
||||||
@PropertySource("classpath:tracing-task.properties")
|
|
||||||
@Slf4j
|
@Slf4j
|
||||||
|
@ComponentScan
|
||||||
|
@AutoConfiguration
|
||||||
public class DefaultTracingAutoConfiguration {
|
public class DefaultTracingAutoConfiguration {
|
||||||
|
|
||||||
@Autowired
|
|
||||||
Tracer tracer;
|
|
||||||
|
|
||||||
@Autowired
|
|
||||||
TracingSafetyCheck tracingSafetyCheck;
|
|
||||||
|
|
||||||
@PostConstruct
|
@PostConstruct
|
||||||
public void postConstruct() {
|
public void postConstruct() {
|
||||||
|
log.info("Tracing AutoConfiguration Loaded!");
|
||||||
if (tracingSafetyCheck.isEndpointResolvable()) {
|
|
||||||
ContextRegistry.getInstance().registerThreadLocalAccessor(new ObservationAwareSpanThreadLocalAccessor(tracer));
|
|
||||||
log.info("Tracing AutoConfiguration Loaded!");
|
|
||||||
} else {
|
|
||||||
log.info("Hostname is not resolvable, tracing is disabled.");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
@Bean
|
|
||||||
ObservationPredicate actuatorServerContextPredicate() {
|
|
||||||
|
|
||||||
// There might be a better solution soon: https://github.com/spring-projects/spring-boot/issues/34801
|
|
||||||
return (name, context) -> {
|
|
||||||
if (name.equals("http.server.requests") && context instanceof ServerRequestObservationContext serverContext) {
|
|
||||||
return !serverContext.getCarrier().getRequestURI().startsWith("/actuator");
|
|
||||||
}
|
|
||||||
return true;
|
|
||||||
};
|
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|||||||
@ -1,22 +0,0 @@
|
|||||||
package com.knecon.fforesight.tracing;
|
|
||||||
|
|
||||||
import org.springframework.beans.factory.annotation.Value;
|
|
||||||
import org.springframework.context.annotation.Configuration;
|
|
||||||
|
|
||||||
import lombok.AllArgsConstructor;
|
|
||||||
import lombok.Data;
|
|
||||||
import lombok.NoArgsConstructor;
|
|
||||||
|
|
||||||
@Configuration
|
|
||||||
@Data
|
|
||||||
@AllArgsConstructor
|
|
||||||
@NoArgsConstructor
|
|
||||||
public class OpenTelemetryConfig {
|
|
||||||
|
|
||||||
@Value("${management.otlp.tracing.endpoint}")
|
|
||||||
private String otlpEndpoint;
|
|
||||||
|
|
||||||
@Value("${management.tracing.enabled}")
|
|
||||||
private boolean tracingEnabled;
|
|
||||||
|
|
||||||
}
|
|
||||||
@ -1,44 +0,0 @@
|
|||||||
package com.knecon.fforesight.tracing;
|
|
||||||
|
|
||||||
import com.knecon.fforesight.tenantcommons.task.KneconTaskDecorator;
|
|
||||||
import io.micrometer.context.ContextSnapshot;
|
|
||||||
import lombok.extern.slf4j.Slf4j;
|
|
||||||
import lombok.val;
|
|
||||||
import org.springframework.beans.factory.ObjectProvider;
|
|
||||||
import org.springframework.beans.factory.annotation.Value;
|
|
||||||
import org.springframework.boot.task.TaskExecutorCustomizer;
|
|
||||||
import org.springframework.context.annotation.Bean;
|
|
||||||
import org.springframework.context.annotation.Configuration;
|
|
||||||
import org.springframework.scheduling.annotation.EnableAsync;
|
|
||||||
|
|
||||||
import java.util.function.Supplier;
|
|
||||||
|
|
||||||
@Configuration(proxyBeanMethods = false)
|
|
||||||
@EnableAsync
|
|
||||||
@Slf4j
|
|
||||||
public class TaskExecutionConfiguration {
|
|
||||||
|
|
||||||
private final static Supplier<KneconTaskDecorator> IDENTITY_KNECON_TASK_DECORATOR_SUPPLIER = () -> runna -> runna;
|
|
||||||
|
|
||||||
@Value("${spring.task.execution.pool.warnPercentageThreshold:80}")
|
|
||||||
private int executorWarnPercentageThreshold;
|
|
||||||
|
|
||||||
@Bean
|
|
||||||
public TaskExecutorCustomizer tracingTaskExecutorCustomizer(ObjectProvider<KneconTaskDecorator> taskDecorator) {
|
|
||||||
return taskExecutor -> {
|
|
||||||
taskExecutor.setTaskDecorator((KneconTaskDecorator) runnable -> {
|
|
||||||
val taskDecoratorUsed = taskDecorator.getIfUnique(IDENTITY_KNECON_TASK_DECORATOR_SUPPLIER);
|
|
||||||
val decoratedRunnable = taskDecoratorUsed.decorate(runnable);
|
|
||||||
val actualQueueSize = taskExecutor.getQueueSize() + 1;
|
|
||||||
val queueOccupancyPercentage = actualQueueSize * 100.f / taskExecutor.getMaxPoolSize();
|
|
||||||
|
|
||||||
if (queueOccupancyPercentage >= executorWarnPercentageThreshold) {
|
|
||||||
log.warn("Executor pool [ " + taskExecutor + " ] queue size reached " + actualQueueSize + "/" + taskExecutor.getMaxPoolSize() +
|
|
||||||
" entries awaiting execution triggering the warn level set for " + executorWarnPercentageThreshold +"% occupancy.");
|
|
||||||
}
|
|
||||||
|
|
||||||
return ContextSnapshot.captureAll(new Object[0]).wrap(decoratedRunnable);
|
|
||||||
});
|
|
||||||
};
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@ -1,15 +1,22 @@
|
|||||||
package com.knecon.fforesight.tracing;
|
package com.knecon.fforesight.tracing;
|
||||||
|
|
||||||
import com.knecon.fforesight.tenantcommons.RabbitTemplateMultiCustomizer;
|
import org.springframework.amqp.rabbit.config.ContainerCustomizer;
|
||||||
import com.knecon.fforesight.tenantcommons.SimpleMessageListenerContainerCustomizer;
|
import org.springframework.amqp.rabbit.core.RabbitTemplate;
|
||||||
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.amqp.rabbit.listener.SimpleMessageListenerContainer;
|
||||||
|
import org.springframework.boot.actuate.autoconfigure.tracing.ConditionalOnEnabledTracing;
|
||||||
import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
|
import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
|
||||||
import org.springframework.context.annotation.Bean;
|
import org.springframework.context.annotation.Bean;
|
||||||
import org.springframework.context.annotation.Configuration;
|
import org.springframework.context.annotation.Configuration;
|
||||||
|
|
||||||
|
import com.knecon.fforesight.tenantcommons.RabbitTemplateMultiCustomizer;
|
||||||
|
import com.knecon.fforesight.tenantcommons.SimpleMessageListenerContainerCustomizer;
|
||||||
|
|
||||||
|
import lombok.extern.slf4j.Slf4j;
|
||||||
|
|
||||||
@Slf4j
|
@Slf4j
|
||||||
@Configuration(proxyBeanMethods = false)
|
@Configuration
|
||||||
@ConditionalOnClass({ SimpleMessageListenerContainerCustomizer.class, RabbitTemplateMultiCustomizer.class })
|
@ConditionalOnEnabledTracing
|
||||||
|
@ConditionalOnClass({RabbitTemplate.class, SimpleMessageListenerContainer.class, ContainerCustomizer.class})
|
||||||
public class TracingRabbitConfiguration {
|
public class TracingRabbitConfiguration {
|
||||||
|
|
||||||
@Bean
|
@Bean
|
||||||
|
|||||||
@ -1,45 +0,0 @@
|
|||||||
package com.knecon.fforesight.tracing;
|
|
||||||
|
|
||||||
import java.net.HttpURLConnection;
|
|
||||||
import java.net.URL;
|
|
||||||
|
|
||||||
import org.springframework.beans.factory.annotation.Autowired;
|
|
||||||
import org.springframework.stereotype.Service;
|
|
||||||
|
|
||||||
import lombok.Builder;
|
|
||||||
import lombok.Getter;
|
|
||||||
import lombok.extern.slf4j.Slf4j;
|
|
||||||
|
|
||||||
@Service
|
|
||||||
@Builder
|
|
||||||
@Getter
|
|
||||||
@Slf4j
|
|
||||||
public class TracingSafetyCheck {
|
|
||||||
|
|
||||||
@Autowired
|
|
||||||
private OpenTelemetryConfig openTelemetryConfig;
|
|
||||||
|
|
||||||
|
|
||||||
public boolean isEndpointResolvable() {
|
|
||||||
|
|
||||||
String endpoint = openTelemetryConfig.getOtlpEndpoint();
|
|
||||||
boolean resolvable = endpoint != null && canResolveEndpoint(endpoint);
|
|
||||||
log.info("Endpoint {} is resolvable: {}", endpoint, resolvable);
|
|
||||||
return resolvable;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
private boolean canResolveEndpoint(String endpoint) {
|
|
||||||
|
|
||||||
try {
|
|
||||||
URL url = new URL(endpoint);
|
|
||||||
HttpURLConnection connection = (HttpURLConnection) url.openConnection();
|
|
||||||
connection.setRequestMethod("HEAD");
|
|
||||||
int responseCode = connection.getResponseCode();
|
|
||||||
return (200 <= responseCode && responseCode <= 399);
|
|
||||||
} catch (Exception e) {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
@ -1,17 +0,0 @@
|
|||||||
package com.knecon.fforesight.tracing;
|
|
||||||
|
|
||||||
import io.micrometer.context.ContextSnapshot;
|
|
||||||
import org.springframework.context.annotation.Configuration;
|
|
||||||
import org.springframework.core.task.SimpleAsyncTaskExecutor;
|
|
||||||
import org.springframework.web.servlet.config.annotation.AsyncSupportConfigurer;
|
|
||||||
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
|
|
||||||
|
|
||||||
@Configuration(proxyBeanMethods = false)
|
|
||||||
public class WebMvcConfiguration implements WebMvcConfigurer {
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void configureAsyncSupport(AsyncSupportConfigurer configurer) {
|
|
||||||
|
|
||||||
configurer.setTaskExecutor(new SimpleAsyncTaskExecutor(r -> new Thread(ContextSnapshot.captureAll().wrap(r))));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@ -1,10 +0,0 @@
|
|||||||
#
|
|
||||||
# Configure task executor specs
|
|
||||||
#
|
|
||||||
# For more available props,
|
|
||||||
# see: https://github.com/spring-projects/spring-boot/blob/main/spring-boot-project/spring-boot-autoconfigure/src/main/java/org/springframework/boot/autoconfigure/task/TaskExecutionProperties.java
|
|
||||||
#
|
|
||||||
spring.task.execution.pool.coreSize=4
|
|
||||||
spring.task.execution.pool.maxSize=4
|
|
||||||
spring.task.execution.pool.queueCapacity=10000
|
|
||||||
spring.task.execution.threadNamePrefix=TracingAwareTaskExecutor-
|
|
||||||
@ -1,63 +0,0 @@
|
|||||||
package com.knecon.fforesight.tracing;
|
|
||||||
|
|
||||||
import static org.assertj.core.api.Assertions.assertThat;
|
|
||||||
import static org.mockito.Mockito.mock;
|
|
||||||
import static org.mockito.Mockito.verify;
|
|
||||||
|
|
||||||
import org.junit.jupiter.api.Test;
|
|
||||||
import org.junit.jupiter.api.extension.ExtendWith;
|
|
||||||
import org.springframework.amqp.rabbit.core.RabbitTemplate;
|
|
||||||
import org.springframework.amqp.rabbit.listener.SimpleMessageListenerContainer;
|
|
||||||
import org.springframework.boot.test.context.runner.ApplicationContextRunner;
|
|
||||||
import org.springframework.boot.test.system.CapturedOutput;
|
|
||||||
import org.springframework.boot.test.system.OutputCaptureExtension;
|
|
||||||
|
|
||||||
import com.knecon.fforesight.tenantcommons.RabbitTemplateMultiCustomizer;
|
|
||||||
import com.knecon.fforesight.tenantcommons.SimpleMessageListenerContainerCustomizer;
|
|
||||||
|
|
||||||
import lombok.extern.slf4j.Slf4j;
|
|
||||||
|
|
||||||
@ExtendWith(OutputCaptureExtension.class)
|
|
||||||
@Slf4j
|
|
||||||
public class ObservationEnabledTest {
|
|
||||||
|
|
||||||
private final ApplicationContextRunner contextRunner = new ApplicationContextRunner().withPropertyValues("management.tracing.enabled=true")
|
|
||||||
.withPropertyValues("spring.profiles.active=test")
|
|
||||||
.withUserConfiguration(SharedTestConfiguration.class, DefaultTracingAutoConfiguration.class, OpenTelemetryConfig.class);
|
|
||||||
|
|
||||||
|
|
||||||
@Test
|
|
||||||
public void testTracingAutoConfigurationLoaded(CapturedOutput output) {
|
|
||||||
|
|
||||||
this.contextRunner.run(context -> {
|
|
||||||
assertThat(output.getOut()).contains("Tracing AutoConfiguration Loaded!");
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
@Test
|
|
||||||
public void testRabbitTracingEnabled() {
|
|
||||||
|
|
||||||
this.contextRunner.run(context -> {
|
|
||||||
var rabbitTemplateCustomizer = context.getBean(RabbitTemplateMultiCustomizer.class);
|
|
||||||
var rabbitTemplate = mock(RabbitTemplate.class);
|
|
||||||
|
|
||||||
rabbitTemplateCustomizer.customize(rabbitTemplate);
|
|
||||||
verify(rabbitTemplate).setObservationEnabled(true);
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
@Test
|
|
||||||
public void testMessageListenerContainerTracingEnabled() {
|
|
||||||
|
|
||||||
this.contextRunner.run(context -> {
|
|
||||||
var containerConfigurer = context.getBean(SimpleMessageListenerContainerCustomizer.class);
|
|
||||||
var container = mock(SimpleMessageListenerContainer.class);
|
|
||||||
|
|
||||||
containerConfigurer.customize(container);
|
|
||||||
verify(container).setObservationEnabled(true);
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
@ -1,31 +0,0 @@
|
|||||||
package com.knecon.fforesight.tracing;
|
|
||||||
|
|
||||||
import org.mockito.Mockito;
|
|
||||||
import org.springframework.boot.autoconfigure.AutoConfiguration;
|
|
||||||
import org.springframework.context.annotation.Bean;
|
|
||||||
import org.springframework.context.annotation.Profile;
|
|
||||||
|
|
||||||
import io.micrometer.tracing.Tracer;
|
|
||||||
import io.micrometer.tracing.test.simple.SimpleTracer;
|
|
||||||
|
|
||||||
@AutoConfiguration
|
|
||||||
@Profile("test")
|
|
||||||
public class SharedTestConfiguration {
|
|
||||||
|
|
||||||
|
|
||||||
@Bean
|
|
||||||
Tracer simpleTracer() {
|
|
||||||
|
|
||||||
return new SimpleTracer();
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
@Bean
|
|
||||||
public TracingSafetyCheck tracingSafetyCheck() {
|
|
||||||
|
|
||||||
TracingSafetyCheck mock = Mockito.mock(TracingSafetyCheck.class);
|
|
||||||
Mockito.when(mock.isEndpointResolvable()).thenReturn(true);
|
|
||||||
return mock;
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
@ -1,98 +0,0 @@
|
|||||||
package com.knecon.fforesight.tracing;
|
|
||||||
|
|
||||||
import static org.assertj.core.api.AssertionsForClassTypes.assertThat;
|
|
||||||
|
|
||||||
import java.util.concurrent.CompletableFuture;
|
|
||||||
import java.util.concurrent.Future;
|
|
||||||
import java.util.concurrent.TimeUnit;
|
|
||||||
|
|
||||||
import org.assertj.core.api.BDDAssertions;
|
|
||||||
import org.junit.jupiter.api.Test;
|
|
||||||
import org.junit.jupiter.api.extension.ExtendWith;
|
|
||||||
import org.springframework.beans.factory.annotation.Autowired;
|
|
||||||
import org.springframework.boot.autoconfigure.AutoConfigurations;
|
|
||||||
import org.springframework.boot.autoconfigure.task.TaskExecutionAutoConfiguration;
|
|
||||||
import org.springframework.boot.test.context.runner.ApplicationContextRunner;
|
|
||||||
import org.springframework.boot.test.system.CapturedOutput;
|
|
||||||
import org.springframework.boot.test.system.OutputCaptureExtension;
|
|
||||||
import org.springframework.scheduling.annotation.Async;
|
|
||||||
import org.springframework.stereotype.Component;
|
|
||||||
|
|
||||||
import io.micrometer.observation.Observation;
|
|
||||||
import io.micrometer.observation.ObservationRegistry;
|
|
||||||
import io.micrometer.tracing.Span;
|
|
||||||
import io.micrometer.tracing.Tracer;
|
|
||||||
import io.micrometer.tracing.test.simple.SimpleTracer;
|
|
||||||
import lombok.extern.slf4j.Slf4j;
|
|
||||||
|
|
||||||
@ExtendWith(OutputCaptureExtension.class)
|
|
||||||
@Slf4j
|
|
||||||
public class TaskExecutionTracingTest {
|
|
||||||
|
|
||||||
private static final String ASYNC_SPAN_NAME = "Async Span";
|
|
||||||
|
|
||||||
private final ApplicationContextRunner contextRunner = new ApplicationContextRunner().withPropertyValues("management.tracing.enabled=true")
|
|
||||||
.withConfiguration(AutoConfigurations.of(TaskExecutionAutoConfiguration.class))
|
|
||||||
.withPropertyValues("spring.profiles.active=test")
|
|
||||||
.withUserConfiguration(SharedTestConfiguration.class, AsyncLogic.class, DefaultTracingAutoConfiguration.class, OpenTelemetryConfig.class);
|
|
||||||
|
|
||||||
private final ObservationRegistry observationRegistry = ObservationRegistry.create();
|
|
||||||
|
|
||||||
|
|
||||||
@Test
|
|
||||||
public void asyncMethodTraceTest(CapturedOutput output) {
|
|
||||||
|
|
||||||
this.contextRunner.run(context -> {
|
|
||||||
SimpleTracer tracer = (SimpleTracer) context.getBean(Tracer.class);
|
|
||||||
AsyncLogic asyncLogic = context.getBean(AsyncLogic.class);
|
|
||||||
|
|
||||||
Observation firstSpan = Observation.createNotStarted("First span", observationRegistry).highCardinalityKeyValue("test", "test 1");
|
|
||||||
try (Observation.Scope scope = firstSpan.start().openScope()) {
|
|
||||||
log.info("Async in test with observation - before call");
|
|
||||||
|
|
||||||
Span secondSpan = tracer.nextSpan().name("Second span").tag("test", "test 2");
|
|
||||||
try (Tracer.SpanInScope scope2 = tracer.withSpan(secondSpan.start())) {
|
|
||||||
log.info("Async in test with span - before call");
|
|
||||||
Future<String> future = asyncLogic.asyncCall();
|
|
||||||
String spanIdFromFuture = future.get(1, TimeUnit.SECONDS);
|
|
||||||
log.info("Async in test with span - after call");
|
|
||||||
BDDAssertions.then(spanIdFromFuture).isEqualTo(secondSpan.context().spanId());
|
|
||||||
} finally {
|
|
||||||
secondSpan.end();
|
|
||||||
}
|
|
||||||
|
|
||||||
log.info("Async in test with observation - after call");
|
|
||||||
} finally {
|
|
||||||
firstSpan.stop();
|
|
||||||
}
|
|
||||||
|
|
||||||
var tracerSpans = tracer.getSpans()
|
|
||||||
.stream()
|
|
||||||
.toList();
|
|
||||||
|
|
||||||
assertThat(tracerSpans.size()).isEqualTo(2);
|
|
||||||
assertThat(tracerSpans.get(1).getName()).isEqualTo(ASYNC_SPAN_NAME);
|
|
||||||
assertThat(tracerSpans.get(1).getParentId()).isEqualTo(tracerSpans.get(0).getSpanId());
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
@Component
|
|
||||||
static class AsyncLogic {
|
|
||||||
|
|
||||||
@Autowired
|
|
||||||
Tracer tracer;
|
|
||||||
|
|
||||||
|
|
||||||
@Async("applicationTaskExecutor")
|
|
||||||
public Future<String> asyncCall() {
|
|
||||||
|
|
||||||
log.info("TASK EXECUTOR");
|
|
||||||
tracer.nextSpan().name(ASYNC_SPAN_NAME).tag("test", "test 3").start().end();
|
|
||||||
String spanId = tracer.currentSpan().context().spanId();
|
|
||||||
return CompletableFuture.supplyAsync(() -> spanId);
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
@ -0,0 +1,43 @@
|
|||||||
|
package com.knecon.fforesight.tracing;
|
||||||
|
|
||||||
|
import static org.mockito.Mockito.mock;
|
||||||
|
import static org.mockito.Mockito.verify;
|
||||||
|
|
||||||
|
import org.junit.jupiter.api.Test;
|
||||||
|
import org.springframework.amqp.rabbit.config.ContainerCustomizer;
|
||||||
|
import org.springframework.amqp.rabbit.core.RabbitTemplate;
|
||||||
|
import org.springframework.amqp.rabbit.listener.SimpleMessageListenerContainer;
|
||||||
|
import org.springframework.boot.autoconfigure.AutoConfigurations;
|
||||||
|
import org.springframework.boot.autoconfigure.amqp.RabbitTemplateCustomizer;
|
||||||
|
import org.springframework.boot.test.context.runner.ApplicationContextRunner;
|
||||||
|
|
||||||
|
import com.knecon.fforesight.tenantcommons.MultiTenancyAutoConfiguration;
|
||||||
|
|
||||||
|
import lombok.extern.slf4j.Slf4j;
|
||||||
|
|
||||||
|
@Slf4j
|
||||||
|
public class TracingTests {
|
||||||
|
|
||||||
|
private final ApplicationContextRunner contextRunner = new ApplicationContextRunner().withConfiguration(AutoConfigurations.of(TracingRabbitConfiguration.class,
|
||||||
|
MultiTenancyAutoConfiguration.class));
|
||||||
|
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testTracingEnabled() {
|
||||||
|
|
||||||
|
this.contextRunner.withPropertyValues("management.tracing.enabled=true").run(context -> {
|
||||||
|
var rabbitTemplateCustomizer = context.getBean(RabbitTemplateCustomizer.class);
|
||||||
|
var rabbitTemplate = mock(RabbitTemplate.class);
|
||||||
|
|
||||||
|
rabbitTemplateCustomizer.customize(rabbitTemplate);
|
||||||
|
verify(rabbitTemplate).setObservationEnabled(true);
|
||||||
|
}).run(context -> {
|
||||||
|
var containerConfigurer = context.getBean(ContainerCustomizer.class);
|
||||||
|
var container = mock(SimpleMessageListenerContainer.class);
|
||||||
|
|
||||||
|
containerConfigurer.configure(container);
|
||||||
|
verify(container).setObservationEnabled(true);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
Loading…
x
Reference in New Issue
Block a user