io.tracee.binding.springmvc.async.PreTpicAsyncBeanPostProcessor Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of tracee-springmvc Show documentation

Please refer to https://github.com/tracee/tracee.

The newest version!

package io.tracee.binding.springmvc.async;

import io.tracee.Tracee;
import io.tracee.TraceeConstants;
import org.aopalliance.aop.Advice;
import org.aopalliance.intercept.MethodInvocation;
import org.springframework.aop.framework.ReflectiveMethodInvocation;
import org.springframework.beans.factory.BeanFactory;
import org.springframework.core.Ordered;
import org.springframework.scheduling.annotation.AnnotationAsyncExecutionInterceptor;
import org.springframework.scheduling.annotation.AsyncAnnotationAdvisor;
import org.springframework.scheduling.annotation.AsyncAnnotationBeanPostProcessor;

import java.util.Map;
import java.util.concurrent.Executor;

public class PreTpicAsyncBeanPostProcessor extends AsyncAnnotationBeanPostProcessor {

	public PreTpicAsyncBeanPostProcessor(Executor executor) {
		advisor = new TpicPreAdvisor(executor);
		setBeforeExistingAdvisors(true);
	}

	boolean isBeforeExistingAdvisors() {
		return beforeExistingAdvisors;
	}

	@Override
	public void setBeanFactory(BeanFactory beanFactory) {
	}

	static class TpicPreAdvisor extends AsyncAnnotationAdvisor {

		private final Executor executor;

		TpicPreAdvisor(Executor executor) {
			super();
			this.executor = executor;
			setTaskExecutor(executor); // compatible with spring 4
		}

		// use getAdvice instead of buildAdvice to be compatible with Spring 4
		@Override
		public Advice getAdvice() {
			return new DelegateTpicToAsyncInterceptor(executor);
		}
	}

	static class DelegateTpicToAsyncInterceptor extends AnnotationAsyncExecutionInterceptor {

		DelegateTpicToAsyncInterceptor(Executor defaultExecutor) {
			super(defaultExecutor);
		}

		@Override
		public Object invoke(MethodInvocation invocation) throws Throwable {
			if (invocation instanceof ReflectiveMethodInvocation) {
				final Map tpic = Tracee.getBackend().copyToMap();
				((ReflectiveMethodInvocation) invocation).setUserAttribute(TraceeConstants.TPIC_HEADER, tpic);
			}

			return invocation.proceed();
		}

		@Override
		public int getOrder() {
			return Ordered.HIGHEST_PRECEDENCE;
		}
	}
}