org.zalando.tracer.spring.TracerAsyncAutoConfiguration Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of tracer-spring-boot-starter Show documentation

Tracing requests through a distributed system.

There is a newer version: 2.0.0-RC.3

Show newest version

package org.zalando.tracer.spring;

import org.springframework.boot.autoconfigure.AutoConfigureAfter;
import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
import org.springframework.core.task.TaskExecutor;
import org.springframework.scheduling.annotation.Async;
import org.springframework.scheduling.concurrent.ConcurrentTaskExecutor;
import org.zalando.tracer.Tracer;

import java.util.concurrent.ExecutorService;

import static java.util.concurrent.Executors.newCachedThreadPool;
import static org.springframework.aop.interceptor.AsyncExecutionAspectSupport.DEFAULT_TASK_EXECUTOR_BEAN_NAME;
import static org.zalando.tracer.concurrent.TracingExecutors.tryPreserve;

@Configuration
@ConditionalOnClass(Async.class)
@ConditionalOnProperty(name = "tracer.async.enabled", havingValue = "true", matchIfMissing = true)
@ConditionalOnMissingBean(name = DEFAULT_TASK_EXECUTOR_BEAN_NAME)
@AutoConfigureAfter(TracerAutoConfiguration.class)
public class TracerAsyncAutoConfiguration {

    @Bean(destroyMethod = "shutdown")
    @ConditionalOnMissingBean(name = "taskExecutorService")
    public ExecutorService taskExecutorService() {
        return newCachedThreadPool();
    }

    @Primary
    @Bean(name = DEFAULT_TASK_EXECUTOR_BEAN_NAME)
    public TaskExecutor taskExecutor(final ExecutorService taskExecutorService, final Tracer tracer) {
        return new ConcurrentTaskExecutor(tryPreserve(taskExecutorService, tracer));
    }

}