spring配置双数据源

前段时间有个需求，需要将将数据存到两个数据库中，一个库中存放主信息，一个库中存放特殊信息，看来是要使用双数据源了，来搞起来吧

既然是双数据源，先不管怎么切换，配置得先搞起来

数据源配置

<bean name="dataSource" class="com.alibaba.druid.pool.DruidDataSource" init-method="init" destroy-method="close">
    <property name="url" value="${url}"/>
    <property name="username" value="${username}"/>
    <property name="password" value="${password}"/>

    <!-- 初始化连接大小 -->
    <property name="initialSize" value="5"/>
    <!-- 连接池最大使用连接数量 -->
    <property name="maxActive" value="10"/>
    <!-- 连接池最小空闲 -->
    <property name="minIdle" value="1"/>
    <!-- 获取连接最大等待时间 -->
    <property name="maxWait" value="3000"/>

    <property name="poolPreparedStatements" value="true"/>
    <property name="maxPoolPreparedStatementPerConnectionSize" value="33"/>

    <!-- 检测连接是否有效的sql -->
    <property name="validationQuery" value="SELECT 1"/>
    <!-- 申请连接时执行validationQuery检测连接是否有效 -->
    <property name="testOnBorrow" value="false"/>
    <!-- 归还连接时执行validationQuery检测连接是否有效 -->
    <property name="testOnReturn" value="false"/>
    <!-- 申请连接时检测，如果空闲时间大于timeBetweenEvictionRunsMillis，执行validationQuery检测连接是否有效 -->
    <property name="testWhileIdle" value="true"/>

    <!-- 配置间隔多久才进行一次检测，检测需要关闭的空闲连接，单位是毫秒 -->
    <property name="timeBetweenEvictionRunsMillis" value="60000"/>
    <!-- 配置一个连接在池中最小生存的时间，单位是毫秒 -->
    <property name="minEvictableIdleTimeMillis" value="300000"/>

    <!-- 打开removeAbandoned功能 -->
    <property name="removeAbandoned" value="false"/>
    
</bean>

<!-- 数据源 -->
<bean name="adDataSource" class="com.alibaba.druid.pool.DruidDataSource" init-method="init" destroy-method="close">
    <property name="url" value="${adurl}"/>
    <property name="username" value="${adusername}"/>
    <property name="password" value="${adpassword}"/>

    <!-- 初始化连接大小 -->
    <property name="initialSize" value="5"/>
    <!-- 连接池最大使用连接数量 -->
    <property name="maxActive" value="10"/>
    <!-- 连接池最小空闲 -->
    <property name="minIdle" value="1"/>
    <!-- 获取连接最大等待时间 -->
    <property name="maxWait" value="3000"/>

    <property name="poolPreparedStatements" value="true"/>
    <property name="maxPoolPreparedStatementPerConnectionSize" value="33"/>

    <!-- 检测连接是否有效的sql -->
    <property name="validationQuery" value="SELECT 1"/>
    <!-- 申请连接时执行validationQuery检测连接是否有效 -->
    <property name="testOnBorrow" value="false"/>
    <!-- 归还连接时执行validationQuery检测连接是否有效 -->
    <property name="testOnReturn" value="false"/>
    <!-- 申请连接时检测，如果空闲时间大于timeBetweenEvictionRunsMillis，执行validationQuery检测连接是否有效 -->
    <property name="testWhileIdle" value="true"/>

    <!-- 配置间隔多久才进行一次检测，检测需要关闭的空闲连接，单位是毫秒 -->
    <property name="timeBetweenEvictionRunsMillis" value="60000"/>
    <!-- 配置一个连接在池中最小生存的时间，单位是毫秒 -->
    <property name="minEvictableIdleTimeMillis" value="300000"/>

    <!-- 打开removeAbandoned功能 -->
    <property name="removeAbandoned" value="false"/>
    
</bean>

<!-- 动态数据源 -->
    <bean id="dynamicDataSource" class="com.zhanghe.webconfig.datasource.DynamicDataSource">
        <property name="targetDataSources">
            <map key-type="java.lang.String">
                <entry key="video" value-ref="dataSource"/>
                <entry key="ad" value-ref="adDataSource"/>
            </map>
        </property>
        <!-- 默认数据源 -->
        <property name="defaultTargetDataSource" ref="dataSource"/>
    </bean>

动态数据源

配置好了数据源之后，需要进行定义动态数据源，继承AbstractRoutingDataSource，AbstractRoutingDataSource是基于特定的查找key路由到特定的数据源。它内部维护了一组目标数据源，并且做了路由key与目标数据源之间的映射，提供基于key查找数据源的方法。

public class DynamicDataSource extends AbstractRoutingDataSource {
    private final Logger LOGGER = LoggerFactory.getLogger(DynamicDataSource.class);

    @Override
    protected Object determineCurrentLookupKey() {
      	// 获取当前数据源key
        String key = DataSourceHolder.getCurDataSource();
        LOGGER.info("{}线程 获取到的数据源key--->{}",Thread.currentThread().getName(),key);
				// 如果没有的话，则使用默认的key
        if(StringUtils.isBlank(key)){
            key = DataSourceHolder.getDefaultDataSource();
        }
        LOGGER.info("{}线程 数据源选择--->{}",Thread.currentThread().getName(),key);
        return key;
    }
}

// 存储当前数据源的key
public class DataSourceHolder {
    private static final ThreadLocal<String> CUR_DATA_SOURCE = new ThreadLocal<>();

    private static final String DEFAULT_DATA_SOURCE = "video";

    public static String getCurDataSource(){
        return CUR_DATA_SOURCE.get();
    }

    public static String getDefaultDataSource(){
        return DEFAULT_DATA_SOURCE;
    }

    public static void setCurDataSource(String dataSource){
        if(StringUtils.isNotBlank(dataSource)){
            CUR_DATA_SOURCE.set(dataSource);
        }
    }

    public static void clearDataSource(){
        CUR_DATA_SOURCE.remove();
    }
}

配置是都搞定了，那么怎么切换呢，可以看到在动态数据源中其实是根据key来进行路由获取数据源的，那么其实就是怎么改变这个key了，而且是动态改变，那么就用spring aop来进行解决吧

数据源切换

首先定义一个注解@DataSource，来标识当前方法要使用的数据源

@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
public @interface DataSource {

    String name() default "video";
}

然后来进行aop的逻辑来根据注解的name属性来存储key

@Order(1)  // 这里要注意一下，由于spring中的@Transactional也是使用的aop来开启事务的，而切换数据源要在开启事务之前，所以我将@Order设置为了1
@Aspect
@Component
public class DataSourceAspect {
    private final Logger LOGGER = LoggerFactory.getLogger(DynamicDataSource.class);

    @Pointcut(value="@annotation(com.zhanghe.webconfig.datasource.DataSource)")
    public void pointcut(){

    }

    @Before(value = "pointcut()")
    public void before(JoinPoint joinPoint){
        String name = getDataSourceName(joinPoint);
        LOGGER.info("{}线程拦截切换数据源{}",Thread.currentThread().getName(),name);
        DataSourceHolder.setCurDataSource(name);
    }

    @After(value = "pointcut()")
    public void after(){
        DataSourceHolder.clearDataSource();
    }

    /**
     * 获取数据源lookupkey
     * @param joinPoint
     * @return
     */
    public String getDataSourceName(JoinPoint joinPoint) {
        MethodSignature signature = (MethodSignature) joinPoint.getSignature();
        Method method = signature.getMethod();
        if (method != null) {
            DataSource dataSource = method.getAnnotation(DataSource.class);

            return dataSource.name();
        } else {
            return null;
        }
    }
}

这就大功告成了，双数据源的配置就搞定了