spring和mybatis集成动态切换数据源配置实现

最新推荐文章于 2022-05-12 07:30:00 发布

依晨恋

最新推荐文章于 2022-05-12 07:30:00 发布

阅读量3.6k

点赞数

分类专栏： mybatis 数据库 spring3

本文链接：https://blog.csdn.net/yichenlian/article/details/37055393

版权

mybatis 同时被 3 个专栏收录

5 篇文章 0 订阅

订阅专栏

spring3

3 篇文章 0 订阅

订阅专栏

数据库

1 篇文章 0 订阅

订阅专栏

由于系统有根据方法名实现动态切换数据源的需求，所以研究了下spring的动态切换数据源方式，然后和mybatis集成。

主要使用到了c3p0连接池、mybatis、spring（aop切面）、自定义拦截器

1.首先定义数据源，我的数据源很多，只拿几个做例子；

<context:property-placeholder ignore-unresolvable="true"
			location="classpath*:/conf/dbpool.properties" />
	    <bean id="dataSourceUserCenter_master" class="com.mchange.v2.c3p0.ComboPooledDataSource" destroy-method="close">
	        <property name="driverClass">
	            <value>${driverClassName}</value>
	        </property>
	        <property name="jdbcUrl">
	            <value>${usercenter_master}</value>
	        </property>
	        <property name="user">
	            <value>${username}</value>
	        </property>
	        <property name="password">
	            <value>${password}</value>
	        </property>
	        <!--连接池中保留的最小连接数。Default: 5 -->
	        <property name="minPoolSize">
	            <value>${minPoolSize}</value>
	        </property>
	        <!--连接池中保留的最大连接数。Default: 15 -->
	        <property name="maxPoolSize">
	            <value>${maxPoolSize}</value>
	        </property>
			<!--初始化时获取的连接数，取值应在minPoolSize与maxPoolSize之间。Default: 10 -->
	        <property name="initialPoolSize">
	            <value>${initialPoolSize}</value>
	        </property>
	 		<!--当连接池中的连接耗尽的时候c3p0一次同时获取的连接数。Default: 3 -->
	        <property name="acquireIncrement">
	            <value>${acquireIncrement}</value>
	        </property>
		 	<!--最大空闲时间,60秒内未使用则连接被丢弃。若为0则永不丢弃。Default: 0 -->
	        <property name="maxIdleTime">
	            <value>${maxIdleTime}</value>
	        </property>
	 		<!--JDBC的标准参数，用以控制数据源内加载的PreparedStatements数量。但由于预缓存的statements
			  属于单个connection而不是整个连接池。所以设置这个参数需要考虑到多方面的因素。
			  如果maxStatements与maxStatementsPerConnection均为0，则缓存被关闭。Default: 0-->
	        <property name="maxStatements">
	            <value>${maxStatements}</value>
	        </property>
	 		<!--每60秒检查所有连接池中的空闲连接。Default: 0 -->
	        <property name="idleConnectionTestPeriod">
	            <value>${idleConnectionTestPeriod}</value>
	        </property>
	 		<!--定义在从数据库获取新连接失败后重复尝试的次数。Default: 30 -->
	        <property name="acquireRetryAttempts">
	            <value>${acquireRetryAttempts}</value>
	        </property>
			<!--获取连接失败将会引起所有等待连接池来获取连接的线程抛出异常。但是数据源仍有效
			  保留，并在下次调用getConnection()的时候继续尝试获取连接。如果设为true，那么在尝试
			  获取连接失败后该数据源将申明已断开并永久关闭。Default: false-->
	        <property name="breakAfterAcquireFailure">
	            <value>${breakAfterAcquireFailure}</value>
	        </property>
			<!--因性能消耗大请只在需要的时候使用它。如果设为true那么在每个connection提交的
			  时候都将校验其有效性。建议使用idleConnectionTestPeriod或automaticTestTable
			  等方法来提升连接测试的性能。Default: false -->
	        <property name="testConnectionOnCheckout">
	            <value>${testConnectionOnCheckout}</value>
	        </property>
		</bean>
		
	    <bean id="dataSourceUserCenter_slave" parent="dataSourceUserCenter_master">
	        <property name="jdbcUrl">
	            <value>${usercenter_slave}</value>
	        </property>
	        <property name="user">
	            <value>${username}</value>
	        </property>
	        <property name="password">
	            <value>${password}</value>
	        </property>
		</bean>
		
	     <bean id="dataSourceTest_slave" parent="dataSourceUserCenter_master">
            <property name="jdbcUrl">
                <value>${test_slave}</value>
            </property>
            <property name="user">
                <value>${username}</value>
            </property>
            <property name="password">
                <value>${password}</value>
            </property>
        </bean>

2.对数据源进行整合包装，便于统一管理。

           <bean id="multipleDataSource" class="com.test.interceptor.datasource.MultipleDataSource">
	    	<property name="defaultTargetDataSource" ref="dataSourceUserCenter_master"/>
	    	<property name="targetDataSources">
	    		<map>
	    			<!-- 需要注意这里的key值与实现类里面的常量值要一样 -->
	    			<entry value-ref="dataSourceUserCenter_master" key="UserCenter_master"/>
	    			<entry value-ref="dataSourceUserCenter_slave" key="UserCenter_slave"/>
	    			<entry value-ref="dataSourceTest_slave" key="Test_slave"/>
	    		</map>
	    	</property>
	    </bean>

其中

com.test.interceptor.datasource.MultipleDataSource

实现了spring的AbstractRoutingDataSource类，故需用切换数据源的能力。defaultTargetDataSource是默认数据源、targetDataSources是所有数据源集合，默认数据源也要放置进来。其中每个entry的key要与java类中的常量相同。

3.和mybatis集成

<bean id="sqlSessionFactory"
          class="org.mybatis.spring.SqlSessionFactoryBean"
          p:dataSource-ref="multipleDataSource"
          p:configLocation="classpath:conf/mybatis-config.xml"
          p:mapperLocations="classpath*:mapper/*.xml"/>

4.aop指向进行数据源切换的处理过滤器

<!-- 启用aop注解 -->
	<aop:aspectj-autoproxy></aop:aspectj-autoproxy>

    <!-- 配置一个拦截器对象，处理具体的切换数据源的业务 -->
    <bean id="dataSourceMethodInterceptor" class="com.test.interceptor.DataSourceMethodInterceptor"/>
    
    <!-- 参与动态切换数据源的切入点对象 (切入点对象，确定何时何地调用拦截器) -->
    <bean id="methodCachePointCut" class="org.springframework.aop.support.RegexpMethodPointcutAdvisor">
        <!-- 配置缓存aop切面 -->
        <property name="advice" ref="dataSourceMethodInterceptor" />
        <!-- .*表示前面的前缀(包括包名) 表示print方法-->
        <property name="patterns">
            <list>
           	<value>com.test.*.service.impl.*Service*\.*.*</value>
            </list>
        </property>
    </bean>

如上配置，扫描所有Service类中的方法，然后根据一定的规范来决定切换成哪个数据源。

5.MultipleDataSource.java

public class MultipleDataSource extends AbstractRoutingDataSource{

	@Override
	protected Object determineCurrentLookupKey() {
		return DataSourceContextHandler.getDataSourceContext();
	}

}

6.DataSourceContextHandler.java

/**
	 * 线程局部变量，只为该线程服务
	 */
	private final static ThreadLocal<String> contextHandler = new ThreadLocal<String>();
	
	/**
	 * 设置该线程数据源id
	 * @param dataSourceType
	 */
	public static void setDataSourceContext(String dataSourceType){
		contextHandler.set(dataSourceType);
	}
	/**
	 * 得到该线程数据源id
	 * @return
	 */
	public static String getDataSourceContext(){
		return contextHandler.get();
	}
	/**
	 * 清除该线程的数据源变量值
	 */
	public static void clearDataSourceContext(){
		contextHandler.remove();
	}

7.DataSourceMethodInterceptor.java

        @Override
	public void afterPropertiesSet() throws Exception {
		logger.info("after properties set ...");
	}

	@Override
	public Object invoke(MethodInvocation invocation) throws Throwable {
		String dbname=....
        if (StringUtils.isNotBlank(dbname)) {
        	if(isSlave){
        		DataSourceContextHandler.setDataSourceContext(dbname+"_slave");
        	}else{
        		DataSourceContextHandler.setDataSourceContext(dbname+"_master");
        	}
        }else{
        	DataSourceContextHandler.clearDataSourceContext();
        }
        Object result = invocation.proceed();
        return result;
	}

在invoke方法中根据自定义的规则完成了数据源的选择。

暂时就写这些吧...