我们在很多的项目中经常会有用到多个数据源。比如数据库读写分离,读操作都去从库里读,写操作都往主库里写。那么这里主库和从库就是两个不同的数据源。再比如要做两个数据库之间的数据转换,从一个数据库读取数据写到另一个数据库中,等等这些情况都需要系统使用两个或多个数据源。那么该如何配置多个数据源呢?我这里使用的是spring管理,数据库连接池使用的是阿里的druid。具体步骤如下: 1.修改spring的配置文件, 配置多个不同的数据源。这里不同的数据源就是由spring维护多个DruidDataSource的实例,每个实例都指定不同的数据库链接信息。配置如下:
<!-- 配置多数据源 ,druid --> <bean id="masterDataSource" class="com.alibaba.druid.pool.DruidDataSource" destroy-method="close"> <property name="url" value="${jdbc.master.url}" /> <property name="username" value="${jdbc.master.username}" /> <property name="password" value="${jdbc.master.password}" /> <property name="driverClassName" value="${jdbc.master.driver}" /> <!-- 最大连接池数量 --> <property name="maxActive" value="${master.maxActive}" /> <!-- 获取连接时最大等待时间,单位毫秒。 配置了maxWait之后,缺省启用公平锁,并发效率会有所下降,如果需要可以通过配置useUnfairLock属性为true使用非公平锁。 --> <property name="maxWait" value="60000" /> <!-- 建议配置为true,不影响性能,并且保证安全性。 申请连接的时候检测,如果空闲时间大于timeBetweenEvictionRunsMillis,执行validationQuery检测连接是否有效。 --> <property name="testWhileIdle" value="true"></property> <property name="filters" value="stat"></property> </bean> <!-- 配置多数据源 ,druid --> <bean id="slaveDataSource" class="com.alibaba.druid.pool.DruidDataSource" destroy-method="close"> <property name="url" value="${jdbc.slave1.url}" /> <property name="username" value="${jdbc.slave1.username}" /> <property name="password" value="${jdbc.slave1.password}" /> <property name="driverClassName" value="${jdbc.slave1.driver}" /> <!-- 最大连接池数量 --> <property name="maxActive" value="${slave1.maxActive}" /> <!-- 获取连接时最大等待时间,单位毫秒。 配置了maxWait之后,缺省启用公平锁,并发效率会有所下降,如果需要可以通过配置useUnfairLock属性为true使用非公平锁。 --> <property name="maxWait" value="60000" /> <!-- 建议配置为true,不影响性能,并且保证安全性。 申请连接的时候检测,如果空闲时间大于timeBetweenEvictionRunsMillis,执行validationQuery检测连接是否有效。 --> <property name="testWhileIdle" value="true"></property> <property name="filters" value="stat"></property> </bean>我这里主要配置了两个数据源,一个是主库的数据源,名字叫masterDataSource,还有一个是从库的数据源,名字叫slaveDataSource
2.选择要使用的数据源。我们有了数据源之后我们要使用哪一个呢,所以这个时候我们需要来定义一个类,指定选择哪一个数据源,这里也很简单,只要继承spring给我们提供的AbstractRoutingDataSource类就可以了,这是一个抽象类,需要重写determineCurrentLookupKey方法,这个方法就是用来选择哪一个数据源的。 我们先给两个数据源分别打个标记,比如,读库的源就叫slave,写库的叫master,在实现这个方法的时候,我们返回一个字符串'slave' 表示要找读库,返回一个master表示要找主库。既然要动态的返回,我们就需要有一个入口能够设置选择的标记。在多线程环境下,为了保证数据安全,可以使用ThreadLocal这个类。代码如下
public class DynamicDataSource extends AbstractRoutingDataSource { @Override protected Object determineCurrentLookupKey() { return DbContextHolder.getDataSource(); } }DbContextHolder的代码如下
public class DbContextHolder { public static final String master = "master"; public static final String slave = "slave"; private static ThreadLocal<String> ds = new ThreadLocal<>(); public static void setDataSource (String dataSource) { ds.set(dataSource); } public static String getDataSource() { return ds.get(); } }3.继续配置文件,不管我们配置了几个数据源,同一时刻生效的只有一个,于是我们要通过DynamicDataSource这个类来告诉spring我们要使用哪个数据源
<bean id="dataSource" class="com.hy.base.db.DynamicDataSource"> <property name="targetDataSources"> <map key-type="java.lang.String"> <!-- write --> <entry key="master" value-ref="masterDataSource" /> <!-- read --> <entry key="slave" value-ref="slaveDataSource" /> </map> </property> <property name="defaultTargetDataSource" ref="masterDataSource" /> </bean>这里要注意,key就是我们上面说到的标记,需要和DbContextHolder里的常量配置的相同。 到这一步,我们就可以说配置完成了,代码中就可以指定数据源,来进行数据库操作。当然我们为了更加方便,我们可以使用注解的方式来选择使用哪一个数据源,首先定义注解类
@Retention(RetentionPolicy.RUNTIME) @Target(ElementType.METHOD) public @interface DataSource { String value(); }我们希望呢,在方法上打上这个注解,当这个方法执行前,我们就调用setDataSource方法来设置注解中指定的值,这样也就不需要手动设置了。下面是定义切面
public class DataSourceAspect { public void before(JoinPoint point) { // System.out.println("--------------------------"); Object target = point.getTarget(); String method = point.getSignature().getName(); Class<?>[] classz = target.getClass().getInterfaces(); Class<?>[] parameterTypes = ((MethodSignature) point.getSignature()) .getMethod().getParameterTypes(); try { Method m = classz[0].getMethod(method, parameterTypes); if (m != null && m.isAnnotationPresent(DataSource.class)) { DataSource data = m .getAnnotation(DataSource.class); DbContextHolder.setDataSource(data.value()); System.out.println(data.value()); } } catch (Exception e) { // e.printStackTrace(); } } }下面对切面进行配置
<aop:aspectj-autoproxy> </aop:aspectj-autoproxy> <bean id="manyDataSourceAspect" class="com.hy.base.db.DataSourceAspect" /> <aop:config> <aop:aspect id="c" ref="manyDataSourceAspect"> <aop:pointcut id="tx1" expression="execution(* *(..))" /> <aop:before pointcut-ref="tx1" method="before" /> </aop:aspect> </aop:config>OK,到现在就大功告成。下面进行测试。我们使用的是mybatis,这里的具体配置就不给大家贴了。执行两个操作,分别针对两个库,如果两个库都能正常的响应,就说明我们的配置成功了。下面的例子就是两个接口,一个去找主库,一个去找从库,分别查询在某个数据库中唯一的一张表,主库的t1表只有主库有,从库的t2表只有从库有,这样如果都能查出来数据,说名两个库都是可以连接的。下面是具体的代码。
1.mapper,两个接口,一个使用主库,一个使用从库
public interface AppMapper { @DataSource("master") List<Map<String, Object>> executeQuery(@Param("sql")String sql); @DataSource("slave") List<Map<String, Object>> executeQuery2(@Param("sql")String sql); }2.测试方法
@Test public void testDatasource() throws Exception { ApplicationContext ac = new ClassPathXmlApplicationContext("classpath:spring/applicationContext-*.xml"); AppMapper mapper = (AppMapper) ac.getBean("appMapper"); String sql = "select * from t1 limit 0,10"; System.out.println(mapper.executeQuery(sql)); sql = "select * from t2"; System.out.println(mapper.executeQuery2(sql)); }这两个表在主库和从库都是唯一的。我们发现都能查出数据,说明我们配置正确了。当然了,这里数据库并没有做主从。 --------------------- 作者:jhappyfly 来源: 原文:https://blog.csdn.net/king_kgh/article/details/75043307 版权声明:本文为博主原创文章,转载请附上博文链接!