文章说明:本文为懒汉式教程,只需要粘贴复制就可以实现。
废话不多说,直接代码走起
这里已ssm框架为例,项目所需的相关依赖jar都已经集成在框架中了。。。
1、数据源连接参数
jdbc.properties
jdbc.driverClassName=com.mysql.jdbc.Driver
#mysql----1
jdbc.url=jdbc\:mysql\://47.96.67.113\:3306/redpacketforwarding?useUnicode\=true&characterEncoding\=UTF-8
jdbc.username=root
jdbc.password=A028c78165fb18d40165fb18d4c30D0f
#mysql----2
jdbc.url2=jdbc\:mysql\://47.96.67.113\:3306/jiatui?useUnicode\=true&characterEncoding\=UTF-8
jdbc.username2=root
jdbc.password2=A028c78165fb18d40165fb18d4c30D0f
2、配置数据源
spring-mvc-context.xml
<!-- 配置DataSource数据源 -->
<bean id="dataSource" class="org.apache.commons.dbcp.BasicDataSource" destroy-method="close">
<!-- 数据库连接 -->
<property name="driverClassName" value="${jdbc.driverClassName}" />
<property name="url" value="${jdbc.url}" />
<property name="username" value="${jdbc.username}" />
<property name="password" value="${jdbc.password}" />
<!-- 连接池设置 -->
<property name="initialSize" value="5" />
<property name="maxActive" value="100" />
<property name="maxIdle" value="30" />
<property name="maxWait" value="500" />
<property name="poolPreparedStatements" value="false" />
<property name="defaultAutoCommit" value="false" />
</bean>
<bean id="dataSource2" class="org.apache.commons.dbcp.BasicDataSource" destroy-method="close">
<!-- 数据库连接 -->
<property name="driverClassName" value="${jdbc.driverClassName}" />
<property name="url" value="${jdbc.url2}" />
<property name="username" value="${jdbc.username2}" />
<property name="password" value="${jdbc.password2}" />
<!-- 连接池设置 -->
<property name="initialSize" value="5" />
<property name="maxActive" value="100" />
<property name="maxIdle" value="30" />
<property name="maxWait" value="500" />
<property name="poolPreparedStatements" value="false" />
<property name="defaultAutoCommit" value="false" />
</bean>
<!-- 配置多数源映射关系 -->
<bean id="dynamicDataSource" class="com.ssay.util.dynamicDataSource.DynamicDataSource">
<property name="targetDataSources">
<map key-type="java.lang.String">
<entry value-ref="dataSource" key="dataSource"></entry>
<entry value-ref="dataSource2" key="dataSource2"></entry>
</map>
</property>
<!-- 默认数据源1 -->
<property name="defaultTargetDataSource" ref="dataSource"></property>
</bean>
<!-- 开启aop注解方式 -->
<aop:aspectj-autoproxy proxy-target-class="true"></aop:aspectj-autoproxy>
3、自定义数据源
package com.ssay.util.dynamicDataSource;
import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
/**
* 动态数据源
* */
public class DynamicDataSource extends AbstractRoutingDataSource{
@Override
protected Object determineCurrentLookupKey() {
return DBContextHolder.getDbType();
}
}
package com.ssay.util.dynamicDataSource;
/**
* 数据源类型
*/
public class DBContextHolder {
/**
* 注意:数据源标识保存在线程变量中,避免多线程操作数据源时互相干扰
*/
private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();
// 设置当前数据源
public static void setDbType(String dbType) {
contextHolder.set(dbType);
}
// 获取当前数据源
public static String getDbType() {
return ((String) contextHolder.get());
}
// 清空上下文数据源
public static void clearDbType() {
contextHolder.remove();
}
}
4、自定义@DataSource注解
package com.ssay.util.dynamicDataSource;
import java.lang.annotation.Retention;
import java.lang.annotation.Target;
import java.lang.annotation.ElementType;
import java.lang.annotation.RetentionPolicy;
/**自定义@DataSource注解 */
@Target({ ElementType.TYPE,ElementType.METHOD })
@Retention(RetentionPolicy.RUNTIME)
public @interface DataSource {
String value();
}
5、定义DataSourceAspect切面
package com.ssay.util.dynamicDataSource;
import java.lang.reflect.Method;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.After;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;
import com.ssay.util.dynamicDataSource.DBContextHolder;
import com.ssay.util.dynamicDataSource.DataSource;
/**
* DataSource切面
*/
@Component
@Aspect
public class DataSourceAspect {
private static final Logger logger = LoggerFactory.getLogger(DataSourceAspect.class);
// 指定切入点表单式
@Pointcut("execution(* com.ssay.*.*.dao.mapper.*.*(..))")
public void pointCut_() {
}
// 前置通知 : 在执行目标方法之前执行
@Before("pointCut_()")
public void begin(JoinPoint point) {
System.err.println("正在设置数据源。。。");
//获取由@DataSource指定的数据源标识,设置到线程存储中以便切换数据源
Class<?> target = point.getTarget().getClass();
MethodSignature signature = (MethodSignature) point.getSignature();
// 默认使用目标类型的注解,如果没有则使用其实现接口的注解
for (Class<?> clazz : target.getInterfaces()) {
resolveDataSource(clazz,signature.getMethod());
}
resolveDataSource(target,signature.getMethod());
}
// 后置/最终通知:在执行目标方法之后执行 【无论是否出现异常最终都会执行】
@After("pointCut_()")
public void after(JoinPoint point) {
DBContextHolder.clearDbType();
System.out.println("切换默认数据源成功");
}
/**
* 提取目标对象方法注解和类型注解中的数据源标识
* @param clazz
* @param method
*/
private void resolveDataSource(Class<?> clazz, Method method) {
try {
Class<?>[] types = method.getParameterTypes();
// 默认使用类型注解
if (clazz.isAnnotationPresent(DataSource.class)) {
DataSource source = clazz.getAnnotation(DataSource.class);
DBContextHolder.setDbType(source.value());
}
// 方法注解可以覆盖类型注解
Method m = clazz.getMethod(method.getName(),types);
if (m != null && m.isAnnotationPresent(DataSource.class)) {
DataSource source = m.getAnnotation(DataSource.class);
DBContextHolder.setDbType(source.value());
logger.debug("当前数据源:"+source.value());
}
} catch (Exception e) {
logger.error(clazz +":" + e.getMessage());
}
}
}
代码截图
6.在dao层加@DataSource("dataSource2")注解
这个注解可以在类上加,也可以在方法上加
public interface TestAreaInfoMapper {
@DataSource("dataSource2")
AreaInfoTest selectAreaInfoByCode(@Param("areaCode") String areaCode);
}
调用成功后的截图如下