com.github.jun1st.datasource.aop.AbstractMasterSlaveDataSourceInterceptor Maven / Gradle / Ivy

Go to download
package com.github.jun1st.datasource.aop;

import com.github.jun1st.datasource.MSDataSourceContextHolder;
import org.springframework.aop.AfterReturningAdvice;
import org.springframework.aop.MethodBeforeAdvice;

import java.lang.reflect.Method;

/**
 * User AOP to intercept @MS annotation,
 * determine the lookup key, use master or slave database
 * @author fengde
 */
public abstract class AbstractMasterSlaveDataSourceInterceptor implements MethodBeforeAdvice, AfterReturningAdvice {

    @Override
    public void before(Method method, Object[] objects, Object o) throws Throwable {
        String datasource = determinDataSource(method, objects, o);
        MSDataSourceContextHolder.setDataSourceLookupKey(datasource);
    }


    protected abstract String determinDataSource(Method method, Object[] args, Object target);

    @Override
    public void afterReturning(Object o, Method method, Object[] objects, Object o1) throws Throwable {
        MSDataSourceContextHolder.clearDataSourceLookupKey();
    }
}