spring多数据源配置

项目中我们经常会遇到多数据源的问题，尤其是数据同步或定时任务等项目更是如此。多数据源让人最头痛的，不是配置多个数据源，而是如何能灵活动态的切换数据源。例如在一个spring和hibernate的框架的项目中，我们在spring配置中往往是配置一个dataSource来连接数据库，然后绑定给sessionFactory，在dao层代码中再指定sessionFactory来进行数据库操作。

正如上图所示，每一块都是指定绑死的，如果是多个数据源，也只能是下图中那种方式。

可看出在Dao层代码中写死了两个SessionFactory，这样日后如果再多一个数据源，还要改代码添加一个SessionFactory，显然这并不符合开闭原则。

那么正确的做法应该是

代码如下：

1. applicationContext.xml

<?xml version="1.0" encoding="UTF-8"?>

<beans xmlns="http://www.springframework.org/schema/beans"

xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xmlns:aop="http://www.springframework.org/schema/aop"

xmlns:cache="http://www.springframework.org/schema/cache"

xmlns:context="http://www.springframework.org/schema/context"

xmlns:jdbc="http://www.springframework.org/schema/jdbc" xmlns:jee="http://www.springframework.org/schema/jee"

xmlns:jms="http://www.springframework.org/schema/jms" xmlns:lang="http://www.springframework.org/schema/lang"

xmlns:mvc="http://www.springframework.org/schema/mvc" xmlns:oxm="http://www.springframework.org/schema/oxm"

xmlns:p="http://www.springframework.org/schema/p" xmlns:task="http://www.springframework.org/schema/task"

xmlns:tx="http://www.springframework.org/schema/tx" xmlns:util="http://www.springframework.org/schema/util"

xsi:schemaLocation="http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans.xsd 

http://www.springframework.org/schema/aop http://www.springframework.org/schema/aop/spring-aop-3.1.xsd 

  http://www.springframework.org/schema/cache http://www.springframework.org/schema/cache/spring-cache-3.1.xsd 

    http://www.springframework.org/schema/context http://www.springframework.org/schema/context/spring-context-3.1.xsd 

    http://www.springframework.org/schema/jdbc http://www.springframework.org/schema/jdbc/spring-jdbc-3.1.xsd 

    http://www.springframework.org/schema/jee http://www.springframework.org/schema/jee/spring-jee-3.1.xsd 

    http://www.springframework.org/schema/jms http://www.springframework.org/schema/jms/spring-jms-3.1.xsd 

  http://www.springframework.org/schema/lang http://www.springframework.org/schema/lang/spring-lang-3.1.xsd 

  http://www.springframework.org/schema/mvc http://www.springframework.org/schema/mvc/spring-mvc-3.1.xsd 

  http://www.springframework.org/schema/oxm http://www.springframework.org/schema/oxm/spring-oxm-3.1.xsd 

http://www.springframework.org/schema/task http://www.springframework.org/schema/task/spring-task-3.1.xsd 

    http://www.springframework.org/schema/tx http://www.springframework.org/schema/tx/spring-tx-3.1.xsd 

    http://www.springframework.org/schema/util http://www.springframework.org/schema/util/spring-util-3.1.xsd">

<context:annotation-config />

<context:component-scan base-package="com"></context:component-scan>

<bean class="org.springframework.beans.factory.config.PropertyPlaceholderConfigurer">

<property name="locations">

<list>

<value>classpath:com/resource/config.properties</value>

</list>

</property>

</bean>

<bean id="dataSourceOne" class="com.mchange.v2.c3p0.ComboPooledDataSource"

destroy-method="close">

<property name="driverClass" value="${dbOne.jdbc.driverClass}" />

<property name="jdbcUrl" value="${dbOne.jdbc.url}" />

<property name="user" value="${dbOne.jdbc.user}" />

<property name="password" value="${dbOne.jdbc.password}" />

<property name="initialPoolSize" value="${dbOne.jdbc.initialPoolSize}" />

<property name="minPoolSize" value="${dbOne.jdbc.minPoolSize}" />

<property name="maxPoolSize" value="${dbOne.jdbc.maxPoolSize}" />

</bean>

<bean id="dataSourceTwo" class="com.mchange.v2.c3p0.ComboPooledDataSource"

destroy-method="close">

<property name="driverClass" value="${dbTwo.jdbc.driverClass}" />

<property name="jdbcUrl" value="${dbTwo.jdbc.url}" />

<property name="user" value="${dbTwo.jdbc.user}" />

<property name="password" value="${dbTwo.jdbc.password}" />

<property name="initialPoolSize" value="${dbTwo.jdbc.initialPoolSize}" />

<property name="minPoolSize" value="${dbTwo.jdbc.minPoolSize}" />

<property name="maxPoolSize" value="${dbTwo.jdbc.maxPoolSize}" />

</bean>

<bean id="dynamicDataSource" class="com.core.DynamicDataSource">

<property name="targetDataSources">

<map key-type="java.lang.String">

<entry value-ref="dataSourceOne" key="dataSourceOne"></entry>

<entry value-ref="dataSourceTwo" key="dataSourceTwo"></entry>

</map>

</property>

<property name="defaultTargetDataSource" ref="dataSourceOne">

</property>

</bean>

<bean id="sessionFactory" class="org.springframework.orm.hibernate4.LocalSessionFactoryBean">

<property name="dataSource" ref="dynamicDataSource" />

<property name="hibernateProperties">

<props>

<prop key="hibernate.dialect">org.hibernate.dialect.MySQLDialect</prop>

<prop key="hibernate.current_session_context_class">org.springframework.orm.hibernate4.SpringSessionContext</prop>

<prop key="hibernate.show_sql">false</prop>

<prop key="hibernate.format_sql">true</prop>

<prop key="hbm2ddl.auto">create</prop>

</props>

</property>

<property name="packagesToScan">

<list>

<value>com.po</value>

</list>

</property>

</bean>

<bean id="transactionManager" class="org.springframework.orm.hibernate4.HibernateTransactionManager">

<property name="sessionFactory" ref="sessionFactory" />

</bean>

<aop:config>

<aop:pointcut id="transactionPointCut" expression="execution(* com.dao..*.*(..))" />

<aop:advisor advice-ref="txAdvice" pointcut-ref="transactionPointCut" />

</aop:config>

<tx:advice id="txAdvice" transaction-manager="transactionManager">

<tx:attributes>

<tx:method name="add*" propagation="REQUIRED" />

<tx:method name="save*" propagation="REQUIRED" />

<tx:method name="update*" propagation="REQUIRED" />

<tx:method name="delete*" propagation="REQUIRED" />

<tx:method name="*" read-only="true" />

</tx:attributes>

</tx:advice>

<aop:config>

<aop:aspect id="dataSourceAspect" ref="dataSourceInterceptor">

<aop:pointcut id="daoOne" expression="execution(* com.dao.one.*.*(..))" />

<aop:pointcut id="daoTwo" expression="execution(* com.dao.two.*.*(..))" />

<aop:before pointcut-ref="daoOne" method="setdataSourceOne" />

<aop:before pointcut-ref="daoTwo" method="setdataSourceTwo" />

</aop:aspect>

</aop:config>

</beans>

2. DynamicDataSource.class

package com.core;

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;

public class DynamicDataSource extends AbstractRoutingDataSource{

@Override

protected Object determineCurrentLookupKey() {

return DatabaseContextHolder.getCustomerType();

}

}

3. DatabaseContextHolder.class

package com.core;

public class DatabaseContextHolder {

private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();

public static void setCustomerType(String customerType) {

contextHolder.set(customerType);

}

public static String getCustomerType() {

return contextHolder.get();

}

public static void clearCustomerType() {

contextHolder.remove();

}

}

4. DataSourceInterceptor.class

package com.core;

import org.aspectj.lang.JoinPoint;

import org.springframework.stereotype.Component;

@Component

public class DataSourceInterceptor {

public void setdataSourceOne(JoinPoint jp) {

DatabaseContextHolder.setCustomerType("dataSourceOne");

}

public void setdataSourceTwo(JoinPoint jp) {

DatabaseContextHolder.setCustomerType("dataSourceTwo");

}

}

5. po实体类

package com.po;

import javax.persistence.Column;

import javax.persistence.Entity;

import javax.persistence.Id;

import javax.persistence.Table;

@Entity

@Table(name = "BTSF_BRAND", schema = "hotel")

public class Brand {

private String id;

private String names;

private String url;

@Id

@Column(name = "ID", unique = true, nullable = false, length = 10)

public String getId() {

return this.id;

}

public void setId(String id) {

this.id = id;

}

@Column(name = "NAMES", nullable = false, length = 50)

public String getNames() {

return this.names;

}

public void setNames(String names) {

this.names = names;

}

@Column(name = "URL", length = 200)

public String getUrl() {

return this.url;

}

public void setUrl(String url) {

this.url = url;

}

}

package com.po;

import javax.persistence.Column;

import javax.persistence.Entity;

import javax.persistence.Id;

import javax.persistence.Table;

@Entity

@Table(name = "CITY", schema = "car")

public class City {

private Integer id;

private String name;

@Id

@Column(name = "ID", unique = true, nullable = false)

public Integer getId() {

return id;

}

public void setId(Integer id) {

this.id = id;

}

@Column(name = "NAMES", nullable = false, length = 50)

public String getName() {

return name;

}

public void setName(String name) {

this.name = name;

}

}

6. BrandDaoImpl.class

package com.dao.one;

import java.util.List;

import javax.annotation.Resource;

import org.hibernate.Query;

import org.hibernate.SessionFactory;

import org.springframework.stereotype.Repository;

import com.po.Brand;

@Repository

public class BrandDaoImpl implements IBrandDao {

@Resource

protected SessionFactory sessionFactory;

@SuppressWarnings("unchecked")

@Override

public List<Brand> findAll() {

String hql = "from Brand";

Query query = sessionFactory.getCurrentSession().createQuery(hql);

return query.list();

}

}

7. CityDaoImpl.class

package com.dao.two;

import java.util.List;

import javax.annotation.Resource;

import org.hibernate.Query;

import org.hibernate.SessionFactory;

import org.springframework.stereotype.Repository;

import com.po.City;

@Repository

public class CityDaoImpl implements ICityDao {

@Resource

private SessionFactory sessionFactory;

@SuppressWarnings("unchecked")

@Override

public List<City> find() {

String hql = "from City";

Query query = sessionFactory.getCurrentSession().createQuery(hql);

return query.list();

}

}

8. DaoTest.class

package com.test;

import java.util.List;

import javax.annotation.Resource;

import org.junit.Test;

import org.junit.runner.RunWith;

import org.springframework.test.context.ContextConfiguration;

import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;

import org.springframework.test.context.transaction.TransactionConfiguration;

import com.dao.one.IBrandDao;

import com.dao.two.ICityDao;

import com.po.Brand;

import com.po.City;

@RunWith(SpringJUnit4ClassRunner.class)

@ContextConfiguration(locations = "classpath:com/resource/applicationContext.xml")

@TransactionConfiguration(transactionManager = "transactionManager", defaultRollback = false)

public class DaoTest {

@Resource

private IBrandDao brandDao;

@Resource

private ICityDao cityDao;

@Test

public void testList() {

List<Brand> brands = brandDao.findAll();

System.out.println(brands.size());

List<City> cities = cityDao.find();

System.out.println(cities.size());

}

}

利用aop，达到动态更改数据源的目的。当需要增加数据源的时候，我们只需要在applicationContext配置文件中添加aop配置，新建个DataSourceInterceptor即可。而不需要更改任何代码。

写在最后：

码字不易看到最后了，那就点个关注呗，只收藏不点关注的都是在耍流氓！

关注并私信我“架构”，免费送一些Java架构资料，先到先得！

©著作权归作者所有,转载或内容合作请联系作者
  • 人面猴
    序言：七十年代末，一起剥皮案震惊了整个滨河市，随后出现的几起案子，更是在滨河造成了极大的恐慌，老刑警刘岩，带你破解...
    沈念sama阅读 203,547评论 6赞 477
  • 死咒
    序言：滨河连续发生了三起死亡事件，死亡现场离奇诡异，居然都是意外死亡，警方通过查阅死者的电脑和手机，发现死者居然都...
    沈念sama阅读 85,399评论 2赞 381
  • 救了他两次的神仙让他今天三更去死
    文/潘晓璐 我一进店门，熙熙楼的掌柜王于贵愁眉苦脸地迎上来，“玉大人，你说我怎么就摊上这事。” “怎么了？”我有些...
    开封第一讲书人阅读 150,428评论 0赞 337
  • 道士缉凶录：失踪的卖姜人
    文/不坏的土叔 我叫张陵，是天一观的道长。 经常有香客问我，道长，这世上最难降的妖魔是什么？ 我笑而不...
    开封第一讲书人阅读 54,599评论 1赞 274
  • 港岛之恋（遗憾婚礼）
    正文 为了忘掉前任，我火速办了婚礼，结果婚礼上，老公的妹妹穿的比我还像新娘。我一直安慰自己，他们只是感情好，可当我...
    茶点故事阅读 63,612评论 5赞 365
  • 恶毒庶女顶嫁案：这布局不是一般人想出来的
    文/花漫 我一把揭开白布。 她就那样静静地躺着，像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上，一...
    开封第一讲书人阅读 48,577评论 1赞 281
  • 城市分裂传说
    那天，我揣着相机与录音，去河边找鬼。 笑死，一个胖子当着我的面吹牛，可吹牛的内容都是我干的。 我是一名探鬼主播，决...
    沈念sama阅读 37,941评论 3赞 395
  • 双鸳鸯连环套：你想象不到人心有多黑
    文/苍兰香墨 我猛地睁开眼，长吁一口气：“原来是场噩梦啊……” “哼！你这毒妇竟也来了？” 一声冷哼从身侧响起，我...
    开封第一讲书人阅读 36,603评论 0赞 258
  • 万荣杀人案实录
    序言：老挝万荣一对情侣失踪，失踪者是张志新（化名）和其女友刘颖，没想到半个月后，有当地人在树林里发现了一具尸体，经...
    沈念sama阅读 40,852评论 1赞 297
  • 护林员之死
    正文 独居荒郊野岭守林人离奇死亡，尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,605评论 2赞 321
  • 白月光启示录
    正文 我和宋清朗相恋三年，在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,693评论 1赞 329
  • 活死人
    序言：一个原本活蹦乱跳的男人离奇死亡，死状恐怖，灵堂内的尸体忽然破棺而出，到底是诈尸还是另有隐情，我是刑警宁泽，带...
    沈念sama阅读 33,375评论 4赞 318
  • 日本核电站爆炸内幕
    正文 年R本政府宣布，位于F岛的核电站，受9级特大地震影响，放射性物质发生泄漏。R本人自食恶果不足惜，却给世界环境...
    茶点故事阅读 38,955评论 3赞 307
  • 男人毒药：我在死后第九天来索命
    文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹，春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,936评论 0赞 19
  • 一桩弑父案，背后竟有这般阴谋
    文/苍兰香墨 我抬头看了看天上的太阳。三九已至，却和暖如春，着一层夹袄步出监牢的瞬间，已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,172评论 1赞 259
  • 情欲美人皮
    我被黑心中介骗来泰国打工， 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留，地道东北人。 一个月前我还...
    沈念sama阅读 43,970评论 2赞 349
  • 代替公主和亲
    正文 我出身青楼，却偏偏与公主长得像，于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子，可洞房花烛夜当晚...
    茶点故事阅读 42,414评论 2赞 342

推荐阅读更多精彩内容