package com.zdd.data.aspect;import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;import org.apache.commons.lang3.StringUtils;
import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;/*** 获取数据源* */
public class ChooseDataSource extends AbstractRoutingDataSource {public static Map<String, List<String>> METHODTYPE = new HashMap<String, List<String>>();// 获取数据源名称protected Object determineCurrentLookupKey() {System.out.println(HandleDataSource.getDataSource());return HandleDataSource.getDataSource();}// 设置方法名前缀对应的数据源public void setMethodType(Map<String, String> map) {for (String key : map.keySet()) {List<String> v = new ArrayList<String>();String[] types = map.get(key).split(",");for (String type : types) {if (StringUtils.isNotBlank(type)) {v.add(type);}}METHODTYPE.put(key, v);}}
}

package com.zdd.data.aspect;import org.apache.commons.lang3.StringUtils;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.After;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.slf4j.LoggerFactory;
import org.springframework.context.annotation.EnableAspectJAutoProxy;
import org.springframework.core.annotation.Order;
import org.springframework.stereotype.Component;@Aspect
@EnableAspectJAutoProxy(proxyTargetClass = true)
@Component
public class DataSourceAspect {private final org.slf4j.Logger logger = LoggerFactory.getLogger(DataSourceAspect.class);@Pointcut("execution(* com.zdd.data.service.impl.*.*(..))")public void aspect() {}/*** 配置前置通知,使用在方法aspect()上注册的切入点*/@Before("aspect()")public void before(JoinPoint point) {String className = point.getTarget().getClass().getName();String method = point.getSignature().getName();logger.info(className + "." + method + "(" + StringUtils.join(point.getArgs(), ",") + ")");try {L: for (String key : ChooseDataSource.METHODTYPE.keySet()) {for (String type : ChooseDataSource.METHODTYPE.get(key)) {if (method.startsWith(type)) {System.out.println(key);HandleDataSource.putDataSource(key);break L;}}}} catch (Exception e) {logger.error(e.toString());HandleDataSource.putDataSource("write");}}@After("aspect()")public void after(JoinPoint point) {HandleDataSource.clear();}
}

package com.zdd.data.aspect;/***/
public class HandleDataSource {// 数据源名称线程池private static final ThreadLocal<String> holder = new ThreadLocal<String>();public static void putDataSource(String datasource) {holder.set(datasource);}public static String getDataSource() {return holder.get();}public static void clear() {holder.remove();}
}

<bean id="dataSource" class="com.zdd.data.aspect.ChooseDataSource" lazy-init="true"><description>数据源</description><property name="targetDataSources"><map key-type="java.lang.String" value-type="javax.sql.DataSource"><!-- write --><entry key="write" value-ref="writedataSource" /><!-- read --><entry key="read" value-ref="readdataSource" /></map></property><!-- 设置默认的目标数据源为读 --><property name="defaultTargetDataSource" ref="readdataSource" /><property name="methodType"><map key-type="java.lang.String"><!-- read --><entry key="read" value=",get,,find,select,count,list,query," /><!-- write --><entry key="write" value=",add,insert,create,update,delete,remove," /></map></property></bean>