应用一
主数据源和应用数据源(代码中手动切换数据源)
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(); } }