基于Spring的aop实现多数据源动态切换

Spring多数据源管理类AbstractRoutingDataSource

实现AbstractRoutingDataSource重写determineCurrentLookupKey

首先配置项目的多数据源实体类交给spring管理

<beans>
        <!-- 加载配置属性文件 -->
	<context:property-placeholder ignore-unresolvable="true" location="classpath:application.properties" />
<!-- 数据源配置, 使用 BoneCP 数据库连接池 -->
	<bean id="dataSourceMaster" class="com.alibaba.druid.pool.DruidDataSource" init-method="init" destroy-method="close">
	    <!-- 数据源驱动类可不写，Druid默认会自动根据URL识别DriverClass -->
	    <property name="driverClassName" value="${default.jdbc.driver}" />

		<!-- 基本属性 url、user、password -->
		<property name="url" value="${default.jdbc.url}" />
		<property name="username" value="${default.jdbc.username}" />
		<property name="password" value="${default.jdbc.password}" />

		<!-- 配置初始化大小、最小、最大 -->
		<property name="initialSize" value="${default.jdbc.pool.init}" />
		<property name="minIdle" value="${default.jdbc.pool.minIdle}" />
		<property name="maxActive" value="${default.jdbc.pool.maxActive}" />

		<!-- 配置获取连接等待超时的时间 -->
		<property name="maxWait" value="60000" />

		<!-- 配置间隔多久才进行一次检测，检测需要关闭的空闲连接，单位是毫秒 -->
		<property name="timeBetweenEvictionRunsMillis" value="60000" />

		<!-- 配置一个连接在池中最小生存的时间，单位是毫秒 -->
		<property name="minEvictableIdleTimeMillis" value="300000" />

		<property name="validationQuery" value="${default.jdbc.testSql}" />
		<property name="testWhileIdle" value="true" />
		<property name="testOnBorrow" value="false" />
		<property name="testOnReturn" value="false" />

		<!-- 配置监控统计拦截的filters -->
	    <property name="filters" value="stat" />
	</bean>

	<!-- 分发库数据库 -->
	<bean id="ddsDataSource" class="com.alibaba.druid.pool.DruidDataSource" init-method="init" destroy-method="close">
		<!-- 数据源驱动类可不写，Druid默认会自动根据URL识别DriverClass -->
		<property name="driverClassName" value="${dds.jdbc.driver}" />

		<!-- 基本属性 url、user、password -->
		<property name="url" value="${dds.jdbc.url}" />
		<property name="username" value="${dds.jdbc.username}" />
		<property name="password" value="${dds.jdbc.password}" />

		<!-- 配置初始化大小、最小、最大 -->
		<property name="initialSize" value="${default.jdbc.pool.init}" />
		<property name="minIdle" value="${default.jdbc.pool.minIdle}" />
		<property name="maxActive" value="${default.jdbc.pool.maxActive}" />

		<!-- 配置获取连接等待超时的时间 -->
		<property name="maxWait" value="60000" />

		<!-- 配置间隔多久才进行一次检测，检测需要关闭的空闲连接，单位是毫秒 -->
		<property name="timeBetweenEvictionRunsMillis" value="60000" />

		<!-- 配置一个连接在池中最小生存的时间，单位是毫秒 -->
		<property name="minEvictableIdleTimeMillis" value="300000" />

		<property name="validationQuery" value="${default.jdbc.testSql}" />
		<property name="testWhileIdle" value="true" />
		<property name="testOnBorrow" value="false" />
		<property name="testOnReturn" value="false" />
	</bean>

	<!-- 可支持多数据源配置 -->
	<bean id="dataSource" class="cn.angyi.jlkd.comm.mybatis.ThreadLocalRountingDataSource">
		<property name="defaultTargetDataSource" ref="dataSourceMaster" />
		<property name="targetDataSources">
			<map key-type="cn.angyi.jlkd.comm.mybatis.DataSources">
				<entry key="MASTER" value-ref="dataSourceMaster" />
				<entry key="SLAVE" value-ref="ddsDataSource" />
			</map>
		</property>
	</bean>

	<aop:aspectj-autoproxy />
</beans>

 在application.properties文件配置数据库参数

default.jdbc.type=
default.jdbc.driver=
default.jdbc.url=
default.jdbc.username=
default.jdbc.password=

default.jdbc.pool.init=1
default.jdbc.pool.minIdle=3
default.jdbc.pool.maxActive=20

default.jdbc.testSql=SELECT 'x' FROM DUAL


dds.jdbc.driver=
dds.jdbc.url=
dds.jdbc.username=
dds.jdbc.password=

数据源枚举类  DataSources

/**
 * 数据源枚举类；
 */
public enum DataSources {
    MASTER, SLAVE
}

自定义数据源管理类  DataSourceTypeManager

/**
 * 数据源类型管理类；
 */
public class DataSourceTypeManager {
    //ThreadLocal一般称为线程本地变量，它是一种特殊的线程绑定机制，将变量与线程绑定在一起，为每一个线程维护一个独立的变量副本 ThreadLocal可以理解为将对象的作用范围限制在一个线程上下文中，使得变量的作用域为“线程级”
    private static final ThreadLocal<DataSources> dataSourceTypes = new ThreadLocal<DataSources>() {
        @Override
        protected DataSources initialValue() {
            return DataSources.MASTER;
        }
    };

    public static DataSources get() {
        return dataSourceTypes.get();
    }

    public static void set(DataSources dataSourceType) {
        dataSourceTypes.set(dataSourceType);
    }

    public static void reset() {
        dataSourceTypes.set(DataSources.MASTER);
    }
}

通过实现AbstractRoutingDataSource重写determineCurrentLookupKey切换数据源

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;

/**
 * 获取本地线程中保留的数据源；
 */
public class ThreadLocalRountingDataSource extends AbstractRoutingDataSource {
    @Override
    protected Object determineCurrentLookupKey() {
        return DataSourceTypeManager.get();
    }
}

最后，通过aop切面动态切换数据源，这里我用的注解的方式，其他的方法请自行百度

注解类 DataSource

import java.lang.annotation.*;

@Documented
@Target({ElementType.METHOD, ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
public @interface DataSource {
    // 这个配置了默认数据源SLAVE
    DataSources targetDataSurce() default DataSources.SLAVE;
}

根据aop动态修改数据源

@Aspect
@Component
@Order(1)
public class DataSourcesChange {
    private static final Logger logger = LoggerFactory.getLogger(DataSource.class);
    @Pointcut("@annotation(这里是上面注解的路径名)")
    public void point() {
    }

    @Before(value = "point()")
    public void before(JoinPoint jp) {
        Class<? extends Object> targetClass = jp.getTarget().getClass();
        String methodName = jp.getSignature().getName(); //方法名称
        Class[] parameterTypes = ((MethodSignature)jp.getSignature()).getParameterTypes(); //方法参数类型列表
        logger.debug("========== "+targetClass.getName()+"."+methodName+" ==========");
        Method method=null;
        try {
            method = targetClass.getMethod(methodName, parameterTypes); //获取方法
        } catch (Exception e) {
            logger.error("========== sourceBefore get method error ==========" ,e);
        }
        if(method!=null && method.isAnnotationPresent(DataSource.class)){
            DataSource da = method.getAnnotation(DataSource.class);
            DataSources dataSurce = da.targetDataSurce();
            DataSourceTypeManager.set(dataSurce);
            logger.debug("========== 	set dataSource "+dataSurce+" ==========");
        }
    }

    @After(value = "point()")
    public void after(JoinPoint jp) {
        DataSourceTypeManager.set(DataSources.MASTER);
    }
}

之后只要在方法中添加注解

@DataSource(targetDataSurce = DataSources.SLAVE)就可以了

也可以使用@DataSource 等价于 @DataSource(targetDataSurce = DataSources.MASTER)