java实现数据源操作的切换 spring 数据源切换

转载

小咪咪 2023-12-01 12:04:40

文章标签 java实现数据源操作的切换 bc 数据源 spring 文章分类 Java 后端开发

应用一

主数据源和应用数据源(代码中手动切换数据源)

1.jdbc.properties（连接两个库）

jdbc.user=root
jdbc.password=123456
jdbc.driver=com.mysql.jdbc.Driver
jdbc.url=jdbc:mysql://localhost:3306/survey?rewriteBatchedStatements=true
log.url=jdbc:mysql://localhost:3306/survey_log?rewriteBatchedStatements=true

2.Spring配置文件的配置

<!-- 引入外部资源文件 -->
 <context:property-placeholder location="classpath:jdbc.properties" />
 <!-- 配置数据源 -->
<bean id="dataSourceTemplate abstract="true"class="com.mchange.v2.c3p0.ComboPooledDataSource">    
     <property name="user" value="${jdbc.user}" />
     <property name="password" value="${jdbc.password}" />
     <property name="driverClass" value="${jdbc.driver}" />
</bean>
<!-- 1配置主数据源 -->
<bean id="dataSourceMain" parent="dataSourceTemplate">
     <property name="jdbcUrl" value="${jdbc.url}" />
</bean>
<!--2 配置日志数据源 -->
<bean id="dataSourceLog" parent="dataSourceTemplate">
     <property name="jdbcUrl" value="${log.url}" />
</bean>
<!--3配置路由器数据源 -->
<bean id="surveyRouterDataSource" class="com.atguigu.survey.router.SurveyRouterDataSource">
     <!-- 配置目标数据源的对应关系  路由器数据源方法返回值：根据返回的key值找对应的数据源 -->
     <property name="targetDataSources">
         <map>
             <entry key="DATASOURCE_MAIN" value-ref="dataSourceMain" />
             <entry key="DATASOURCE_LOG" value-ref="dataSourceLog" />
         </map>
     </property>
        <!-- 默认的数据源 -->
        <property name="defaultTargetDataSource" ref="dataSourceMain" />
    </bean>
<!-- 配置SessionFactory -->
<bean id="sessionFactroy"class="org.springframework.orm.hibernate4.LocalSessionFactoryBean">
     <!-- 装配数据源 -->
     <property name="dataSource" ref="surveyRouterDataSource" />
     <property name="configLocation" value="classpath:mybatis-config.xml"></property>
</bean>

3.定义一个类继承AbstractRoutingDataSource实现determineCurrentLookupKey方法，该方法可以实现数据库的动态切换，如下（方法返回的数据源的key值根据key值去配响应的数据源）

public class SurveyRouterDataSource extends AbstractRoutingDataSource{
    @Override
    protected Object determineCurrentLookupKey() {
        //尝试从当前线程获取令牌
        String token = RouterToken.getToken();
        //检测令牌是否等于日志数据源对应的键
        if(RouterToken.DATASOURCE_LOG.equals(token)){
            
            //当前线程上的令牌，在使用过后需要移除
            //避免影响当前线程内其他的数据库操作
            RouterToken.removeToken();
            
            //如果是日志数据源对应的键，将日志数据源的键返回
            return RouterToken.DATASOURCE_LOG;
        }
        
        //如果不是日志数据源对应的键则返回null采用默认的数据源（主数据源）
        return null;
    }
}

4.将路由器数据源的key值绑定到本地线程中

/**
 * 连接数据源的令牌，返回字符串，根据返回的字符串名称判定连接那个数据源
 * @author dongdong
 */
public class RouterToken {
    // 声明一个本地线程
    private static ThreadLocal<String> local = new ThreadLocal<String>();
    // 将返回的字符串声明为静态全局final形式的
    public static final String DATASOURCE_LOG = "DATASOURCE_LOG";
    // 将字符串绑定到线程上
    public static void bindToken(String token) {
        local.set(token);
    }
    // 获取当前线程的字符串
    public static String getToken() {
        return local.get();
    }
    //将当前线程的字符串移除
    public static void removeToken(){
        local.remove();
    }
}

5.代码中的使用

//将路由器数据源的key绑定到当前线程，必须在Service方法前绑定，因为Service有事物，事物也要连接数据源
RouterToken.bindToken(RouterToken.DATASOURCE_LOG);
Page<Log> page = logService.getPage(pageNoStr, Page.PAGE_SIZE_SMALL);

应用二

主数据源和从数据源(AOP实现自动数据源的切换)

1.jdbc.propertiesss

jdbc.driver=com.mysql.jdbc.Driver

#主数据库
jdbc.master.url=jdbc:mysql://192.168.0.68:3306/master_lxg
jdbc.master.username=root
jdbc.master.password=root

#从数据库
jdbc.slaver.url=jdbc:mysql://192.168.0.12:3306/slave_lxg
jdbc.slaver.username=root
jdbc.slaver.password=root

2.Spring配置文件的配置

<!-- 配置主数据源  -->
<bean id="dataSourceMaster" class="org.apache.commons.dbcp.BasicDataSource">
    <property name="driverClassName" value="${jdbc.driver}" />
    <property name="url" value="${jdbc.master.url}" />
    <property name="username" value="${jdbc.master.username}" />
    <property name="password" value="${jdbc.master.password}" />
</bean>
<!-- 配置从数据源 -->
<bean id="dataSourceSlaver" class="org.apache.commons.dbcp.BasicDataSource">
     <property name="driverClassName" value="${jdbc.driver}" />
     <property name="url" value="${jdbc.slaver.url}" />
     <property name="username" value="${jdbc.slaver.username}" />
     <property name="password" value="${jdbc.slaver.password}" />
</bean>
<bean id="dataSource" class="com.lxg.springdatesource.DynamicDataSource">
    <!-- 通过key-value的形式来关联数据源 -->
    <property name="targetDataSources">
        <map>
            <entry key="Master" value-ref="dataSourceMaster"></entry>
            <entry key="Slaver" value-ref="dataSourceSlaver"></entry>
        </map>
    </property>
    <property name="defaultTargetDataSource" ref="dataSourceMaster" />
</bean>

<!-- MyBatis配置 -->
<bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
    <property name="dataSource" ref="dataSource" />
    <!-- 显式指定Mapper文件位置 -->
    <property name="mapperLocations" value="classpath*:/mybatis/*Mapper.xml" />
    <!-- mybatis配置文件路径 -->
    <property name="configLocation" value="classpath:/mybatis/config.xml" />
</bean>
<bean id="sqlSessionTemplate" class="org.mybatis.spring.SqlSessionTemplate">
    <constructor-arg index="0" ref="sqlSessionFactory" />
    <!-- 这个执行器会批量执行更新语句, 还有SIMPLE 和 REUSE -->
    <constructor-arg index="1" value="BATCH" />
</bean>

<!-- 扫描basePackage接口 -->
<bean class="org.mybatis.spring.mapper.MapperScannerConfigurer">
    <!-- 映射器接口文件的包路径， -->
    <property name="basePackage" value="com.lxg.dao" />
    <property name="sqlSessionFactoryBeanName" value="sqlSessionFactory" />
</bean>

<!-- 开启注解式事务控制 -->
<tx:annotation-driven transaction-manager="transactionManager" />

<!-- 配置事务管理器 -->
<bean id="transactionManager" class="org.springframework.jdbc.datasource.DataSourceTransactionManager">
    <property name="dataSource" ref="dataSource"></property>
</bean>

<!-- 定义事务策略 -->
<tx:advice id="txAdvice" transaction-manager="transactionManager"/>

<!-- 定义AOP切面处理器 -->
<bean class="com.lxg.springdatesource.DataSourceAspect" id="dataSourceAspect" />
    <aop:config>
        <!--pointcut元素定义一个切入点，execution中的第一个星号 用以匹配方法的返回类型，这里星号表明匹配所有返回类型。com.lxg.service.*.*(..)表明匹配com.lxg.service包下的所有类的所有方法 -->
        <aop:pointcut id="myPointcut" expression="execution(* com.lxg.service.impl.*.*(..))" />
        <!--将定义好的事务处理策略应用到上述的切入点 -->
        <aop:advisor advice-ref="txAdvice" pointcut-ref="myPointcut" />
        <!-- 将切面应用到自定义的切面处理器上，-9999保证该切面优先级最高执行 -->
        <aop:aspect ref="dataSourceAspect" order="-9999">
            <aop:before method="before" pointcut-ref="myPointcut" />
        </aop:aspect>
    </aop:config>
</bean>

3.定义数据源的AOP切面，通过该Service的方法名判断是应该走读库还是写库,并且设置数据源

public class DataSourceAspect {
    /**
     * 在进入Service方法之前执行
     */
    public void before(JoinPoint point) {
        // 获取到当前执行的方法名
        String methodName = point.getSignature().getName();
        if (isSlave(methodName)) { 
            // 标记为从库 应当先clear一下  DBContextHolder.clearDBType ();
            DBContextHolder.setDBType(DBContextHolder.DATA_SOURCE_SLAVER);
        } else {
            // 标记为主库  应当先clear一下  DBContextHolder.clearDBType ();
            DBContextHolder.setDBType(DBContextHolder.DATA_SOURCE_MASTER);
        }
    }
    /**
     * 判断是否为读库
     */
    private Boolean isSlave(String methodName) {
        // 方法名以query、find、get开头的方法名走从库
        return StringUtils.startsWithAny(methodName, "query", "find", "get");
    }
}

4.定义一个类继承AbstractRoutingDataSource实现determineCurrentLookupKey方法，该方法可以实现数据库的动态切换，如下（方法返回的数据源的key值根据key值去配响应的数据源）

public class DynamicDataSource extends AbstractRoutingDataSource{ 
   @Override 
   protected Object determineCurrentLookupKey() { 
       return DBContextHolder.getDBType(); 
   } 
}

5.将路由器数据源的key值绑定到本地线程中

public class DBContextHolder{ 
    public static final String DATA_SOURCE_MASTER = "Master"; 
    public static final String DATA_SOURCE_SLAVER = "Slaver"; 
    private static final ThreadLocal<String> THREAD_LOCAL = new ThreadLocal<String>(); 
    public static void setDBType(String dbType) { 
        THREAD_LOCAL.set(dbType); 
    } 
    public static String getDBType() { 
        return THREAD_LOCAL.get(); 
    } 
    public static void clearDBType() { 
        THREAD_LOCAL.remove(); 
    } 
}

本文章为转载内容，我们尊重原作者对文章享有的著作权。如有内容错误或侵权问题，欢迎原作者联系我们进行内容更正或删除文章。