使用Spring多數(shù)據(jù)源配置

更新時間：2024年07月15日 08:50:19 作者：程序三兩行

這篇文章主要介紹了使用Spring多數(shù)據(jù)源配置方式,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教

Spring多數(shù)據(jù)源配置

Spring多數(shù)據(jù)源配置，這里持久層框架使用mybatis，集成Mybatis多數(shù)據(jù)源有兩種方式：

1、創(chuàng)建多個SqlSessionFactory，掃描每個SqlSessionFactoryBean對應(yīng)的包，形成了每個Factory對應(yīng)一個數(shù)據(jù)源。

2、創(chuàng)建一個SqlSessionFactory，通過動態(tài)切換數(shù)據(jù)源對象，達到多數(shù)據(jù)源操作功能。

方式一：創(chuàng)建多個SqlSessionFactory

整合mysql+clickhouse雙數(shù)據(jù)源

<!--mybatis + mysql + druid-->
    <dependency>
      <groupId>org.mybatis</groupId>
      <artifactId>mybatis-spring</artifactId>
      <version>2.0.6</version>
    </dependency>
    <dependency>
      <groupId>com.alibaba</groupId>
      <artifactId>druid</artifactId>
      <version>1.1.23</version>
    </dependency>
    <dependency>
      <groupId>org.mybatis</groupId>
      <artifactId>mybatis</artifactId>
      <version>3.4.5</version>
    </dependency>
    <dependency>
      <groupId>mysql</groupId>
      <artifactId>mysql-connector-java</artifactId>
      <version>5.1.49</version>
    </dependency>
    <!--clickhouse-->
    <dependency>
      <groupId>ru.yandex.clickhouse</groupId>
      <artifactId>clickhouse-jdbc</artifactId>
      <version>0.1.53</version>
    </dependency>

jdbc.properties

jdbc.driverClassName=com.mysql.jdbc.Driver
jdbc.master.url=jdbc:mysql://localhost:3306/test
jdbc.master.username=root
jdbc.master.password=root
jdbc.master.maxActive=10
jdbc.master.minIdle=5
jdbc.master.maxWait=10000

clickhouse.properties

clickhouse.driverClassName=ru.yandex.clickhouse.ClickHouseDriver
clickhouse.url=jdbc:clickhouse://localhost:8123/test
clickhouse.username=root
clickhouse.password=root
clickhouse.maxActive=10
clickhouse.minIdle=5
clickhouse.maxWait=10000

spring配置文件中配置多個數(shù)據(jù)源

<!--加載mysql+clickhouse配置文件-->
    <context:property-placeholder location="classpath:jdbc.properties"/>
    <context:property-placeholder location="classpath:clickhouse.properties"/>

    <!--=======================mysql數(shù)據(jù)源配置=======================-->
    <bean id="dataSource" class="com.alibaba.druid.pool.DruidDataSource">
        <property name="driverClassName" value="${jdbc.driverClassName}"/>
        <property name="url" value="${jdbc.master.url}"/>
        <property name="password" value="${jdbc.master.password}"/>
        <property name="username" value="${jdbc.master.username}"/>
        <property name="maxActive" value="${jdbc.master.maxActive}"/>
        <property name="maxWait" value="${jdbc.master.maxWait}"/>
        <property name="minIdle" value="${jdbc.master.minIdle}"/>
    </bean>
    <bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
        <property name="configLocation" value="classpath:mybatis-config.xml"/>
        <property name="dataSource" ref="dataSource"/>
        <property name="typeAliasesPackage" value="com.demo.flash.price.redpot.entity"/>
        <property name="mapperLocations" value="classpath:mysql_mapper/*.xml"/>
        <property name="plugins">
            <set>
                <!--配置pageHelper 分頁插件-->
                <bean class="com.github.pagehelper.PageInterceptor">
                    <property name="properties">
                        <props>
                            <!--方言：-->
                            <prop key="helperDialect">mysql</prop>
                        </props>
                    </property>
                </bean>
            </set>
        </property>
    </bean>
    <bean class="org.mybatis.spring.mapper.MapperScannerConfigurer">
        <property name="basePackage" value="com.demo.flash.price.redpot.mapperMysql"/>
        <property name="sqlSessionFactoryBeanName" value="sqlSessionFactory"/>
    </bean>
    <bean id="transactionManager"
          class="org.springframework.jdbc.datasource.DataSourceTransactionManager">
        <property name="dataSource" ref="dataSource"></property>
    </bean>
    <tx:annotation-driven transaction-manager="transactionManager" />

    <!--=======================clickhouse數(shù)據(jù)源配置=======================-->
    <bean id="dataSourceClickhouse" class="com.alibaba.druid.pool.DruidDataSource">
        <property name="driverClassName" value="${clickhouse.driverClassName}"/>
        <property name="url" value="${clickhouse.url}"/>
        <property name="password" value="${clickhouse.password}"/>
        <property name="username" value="${clickhouse.username}"/>
        <property name="maxActive" value="${clickhouse.maxActive}"/>
        <property name="maxWait" value="${clickhouse.maxWait}"/>
        <property name="minIdle" value="${clickhouse.minIdle}"/>
    </bean>
    <bean id="sqlSessionFactoryClickhouse" class="org.mybatis.spring.SqlSessionFactoryBean">
        <property name="configLocation" value="classpath:mybatis-config.xml"/>
        <property name="dataSource" ref="dataSourceClickhouse"/>
        <property name="typeAliasesPackage" value="com.demo.flash.price.redpot.entity"/>
        <property name="mapperLocations" value="classpath:clickhouse_mapper/*.xml"/>
        <property name="plugins">
            <set>
                <!--配置pageHelper 分頁插件-->
                <bean class="com.github.pagehelper.PageInterceptor">
                    <property name="properties">
                        <props>
                            <!--方言：-->
                            <prop key="helperDialect">mysql</prop>
                        </props>
                    </property>
                </bean>
            </set>
        </property>
    </bean>
    <bean class="org.mybatis.spring.mapper.MapperScannerConfigurer">
        <property name="basePackage" value="com.demo.flash.price.redpot.mapperClickhouse"/>
        <property name="sqlSessionFactoryBeanName" value="sqlSessionFactoryClickhouse"/>
    </bean>
    <bean id="transactionManagerClickhouse"
          class="org.springframework.jdbc.datasource.DataSourceTransactionManager">
        <property name="dataSource" ref="dataSourceClickhouse"></property>
    </bean>
    <tx:annotation-driven transaction-manager="transactionManagerClickhouse" />

方式二：動態(tài)切換數(shù)據(jù)源對象

創(chuàng)建單個SqlSessionFactory，動態(tài)切換數(shù)據(jù)源對象，在Spring的jdbc包下有個支持動態(tài)數(shù)據(jù)源切換的抽象類AbstractRoutingDataSource，其中的抽象方法determineCurrentLookupKey需要我們自己來實現(xiàn)，到底返回哪個數(shù)據(jù)源

public class DynamicDataSource extends AbstractRoutingDataSource {
 
	@Override
	protected Object determineCurrentLookupKey() {
        //在這里做數(shù)據(jù)源切換
		return DataSourceTypeManager.get();
	}
 
}

public class DataSourceTypeManager {
    //數(shù)據(jù)源保存
	private static final ThreadLocal<MybatisDataSource> dataSourceTypes = new ThreadLocal<MybatisDataSource>() {
		 
		@Override
		protected MybatisDataSource initialValue() {
			return MybatisDataSource.JKDSJ;
	}
	};
 
	public static MybatisDataSource get() {
		return dataSourceTypes.get();
	}
 
	public static void set(MybatisDataSource dataSourceType) {
		dataSourceTypes.set(dataSourceType);
	}
 
	public static void reset() {
		dataSourceTypes.set(MybatisDataSource.JKDSJ);
	}
 
}