com.webapp.utils.datasource.DataSourceAspecj Maven / Gradle / Ivy

Go to download
package com.webapp.utils.datasource;

import java.lang.reflect.Method;

import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.stereotype.Component;

/**
* 动态检测和切换数据源
*/
@Component
@Aspect
public class DataSourceAspecj {

	@Around("within(*..*Dao+) || within(*..*DAO+) || within(*..DataSourceSwitch+)")
	public Object myAroundAdvice(ProceedingJoinPoint pjp) throws Throwable {
		Class dsClz = pjp.getTarget().getClass().getInterfaces()[0];
		Method dsMethod = ((MethodSignature)pjp.getSignature()).getMethod();

		DataSource dsAnno = null;
		if(dsMethod.isAnnotationPresent(DataSource.class)){
			dsAnno = dsMethod.getAnnotation(DataSource.class);
		}else if (dsClz.isAnnotationPresent(DataSource.class)) {
			dsAnno = dsClz.getAnnotation(DataSource.class);
		}

		if(dsAnno != null && !dsAnno.value().equals("")){
			MultiDataSource.setDataSourceKey(dsAnno.value());
		}

		Object result = pjp.proceed();

		MultiDataSource.setDefDataSource();
		return result;
	}


}