cn.fantasticmao.mundo.data.partition.PartitionDataSourcePostProcessor Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mundo-data Show documentation

Mundo is a tiny and out-of-the-box component for business programming.

There is a newer version: 1.0.10

package cn.fantasticmao.mundo.data.partition;

import org.aopalliance.intercept.MethodInterceptor;
import org.aopalliance.intercept.MethodInvocation;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.aop.framework.ProxyFactory;
import org.springframework.data.repository.core.RepositoryInformation;
import org.springframework.data.repository.core.support.RepositoryProxyPostProcessor;

import javax.annotation.Nonnull;
import java.lang.annotation.Annotation;
import java.lang.reflect.Method;

/**
 * PartitionDataSourcePostProcessor
 *
 * @author fantasticmao
 * @version 1.0
 * @since 2019/1/2
 */
public class PartitionDataSourcePostProcessor implements RepositoryProxyPostProcessor {

    @Override
    public void postProcess(@Nonnull ProxyFactory factory, @Nonnull RepositoryInformation repositoryInformation) {
        factory.addAdvice(PartitionMethodInterceptor.INSTANCE);
    }

    enum PartitionMethodInterceptor implements MethodInterceptor {
        INSTANCE;

        private final Logger LOGGER = LoggerFactory.getLogger(PartitionMethodInterceptor.class);

        @Override
        public Object invoke(MethodInvocation invocation) throws Throwable {
            Method method = invocation.getMethod();
            Annotation[][] annotations = method.getParameterAnnotations();
            int pos = -1;
            posFlag:
            for (int i = 0; i < annotations.length; i++) {
                for (int j = 0; j < annotations[i].length; j++) {
                    Annotation ann = annotations[i][j];
                    if (ann instanceof PartitionParam) {
                        pos = i;
                        break posFlag;
                    }
                }
            }

            if (pos >= 0) {
                Object[] arguments = invocation.getArguments();
                Object partitionSeed = arguments[pos];
                if (LOGGER.isDebugEnabled()) {
                    LOGGER.debug("found @PartitionParam argument \"{}\" at {}", partitionSeed, method.toString());
                }
                PartitionSeedContext.push(partitionSeed);
                if (LOGGER.isDebugEnabled()) {
                    LOGGER.debug("push partition seed object \"{}\" to PartitionSeedContext", partitionSeed);
                }
            }
            Object result = invocation.proceed();
            if (pos >= 0) {
                PartitionSeedContext.clear();
                if (LOGGER.isDebugEnabled()) {
                    LOGGER.debug("clear partition seed object from PartitionSeedContext");
                }
            }
            return result;
        }
    }
}