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>