cn.mybatis.mp.routing.datasource.RoutingDataSourceAbstractPointcutAdvisor Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mybatis-mp-routing-datasource Show documentation

There is a newer version: 1.7.2

Show newest version

package cn.mybatis.mp.routing.datasource;

import org.aopalliance.aop.Advice;
import org.springframework.aop.Pointcut;
import org.springframework.aop.support.AbstractPointcutAdvisor;
import org.springframework.aop.support.AopUtils;
import org.springframework.aop.support.StaticMethodMatcherPointcut;
import org.springframework.core.annotation.AnnotatedElementUtils;

import java.lang.reflect.Method;
import java.lang.reflect.Proxy;

public class RoutingDataSourceAbstractPointcutAdvisor extends AbstractPointcutAdvisor {

    private final Pointcut pointcut = new StaticMethodMatcherPointcut() {
        @Override
        public boolean matches(Method method, Class targetClass) {
            if (Proxy.isProxyClass(targetClass)) {
                return false;
            }
            if (methodMatch(method)) {
                return true;
            }
            Method specificMethod = AopUtils.getMostSpecificMethod(method, targetClass);
            if (methodMatch(specificMethod)) {
                return true;
            }

            return method.getDeclaringClass().isAnnotationPresent(DS.class);
        }
    };
    private final RoutingDataSourceSpringInterceptor interceptor;

    public RoutingDataSourceAbstractPointcutAdvisor(RoutingDataSourceSpringInterceptor routingDataSourceSpringInterceptor) {
        this.interceptor = routingDataSourceSpringInterceptor;
    }

    private boolean methodMatch(Method method) {
        return AnnotatedElementUtils.hasAnnotation(method, DS.class);
    }

    @Override
    public Pointcut getPointcut() {
        return this.pointcut;
    }

    @Override
    public Advice getAdvice() {
        return this.interceptor;
    }
}