Spring MVC框架搭建扩展--多数据源配置(一)
程序员文章站
2022-07-15 13:26:49
...
本节主要介绍spring mvc+mybatis多数据源配置
1.首先配置2个数据源
在spring-mybatis.xml里
<!-- 配置数据源1 使用的是Druid数据源 -->
<bean id="fdataSource" class="com.alibaba.druid.pool.DruidDataSource"
init-method="init" destroy-method="close">
<property name="url" value="${dataSource.url}" />
<property name="username" value="${dataSource.username}" />
<property name="password" value="${dataSource.password}" />
<!-- 初始化连接大小 -->
<property name="initialSize" value="0" />
<!-- 连接池最大使用连接数量 -->
<property name="maxActive" value="20" />
<!-- 连接池最小空闲 -->
<property name="minIdle" value="0" />
<!-- 获取连接最大等待时间 -->
<property name="maxWait" value="60000" />
<property name="poolPreparedStatements" value="true" />
<property name="maxPoolPreparedStatementPerConnectionSize"
value="33" />
<!-- 用来检测有效sql -->
<property name="validationQuery" value="${validationQuery}" />
<property name="testOnBorrow" value="false" />
<property name="testOnReturn" value="false" />
<property name="testWhileIdle" value="true" />
<!-- 配置间隔多久才进行一次检测,检测需要关闭的空闲连接,单位是毫秒 -->
<property name="timeBetweenEvictionRunsMillis" value="60000" />
<!-- 配置一个连接在池中最小生存的时间,单位是毫秒 -->
<property name="minEvictableIdleTimeMillis" value="25200000" />
<!-- 打开removeAbandoned功能 -->
<property name="removeAbandoned" value="true" />
<!-- 1800秒,也就是30分钟 -->
<property name="removeAbandonedTimeout" value="1800" />
<!-- 关闭abanded连接时输出错误日志 -->
<property name="logAbandoned" value="true" />
<!-- 监控数据库 -->
<property name="filters" value="mergeStat" />
</bean>
<!-- 数据源2 -->
<bean id="sdataSource" class="com.alibaba.druid.pool.DruidDataSource"
init-method="init" destroy-method="close">
<property name="url" value="${jdbc.url}" />
<property name="username" value="${jdbc.username}" />
<property name="password" value="${jdbc.password}" />
<!-- 初始化连接大小 -->
<property name="initialSize" value="0" />
<!-- 连接池最大使用连接数量 -->
<property name="maxActive" value="20" />
<!-- 连接池最小空闲 -->
<property name="minIdle" value="0" />
<!-- 获取连接最大等待时间 -->
<property name="maxWait" value="60000" />
<property name="poolPreparedStatements" value="true" />
<property name="maxPoolPreparedStatementPerConnectionSize"
value="33" />
<!-- 用来检测有效sql -->
<property name="validationQuery" value="${validationQuery}" />
<property name="testOnBorrow" value="false" />
<property name="testOnReturn" value="false" />
<property name="testWhileIdle" value="true" />
<!-- 配置间隔多久才进行一次检测,检测需要关闭的空闲连接,单位是毫秒 -->
<property name="timeBetweenEvictionRunsMillis" value="60000" />
<!-- 配置一个连接在池中最小生存的时间,单位是毫秒 -->
<property name="minEvictableIdleTimeMillis" value="25200000" />
<!-- 打开removeAbandoned功能 -->
<property name="removeAbandoned" value="true" />
<!-- 1800秒,也就是30分钟 -->
<property name="removeAbandonedTimeout" value="1800" />
<!-- 关闭abanded连接时输出错误日志 -->
<property name="logAbandoned" value="true" />
<!-- 监控数据库 -->
<property name="filters" value="mergeStat" />
</bean>
2.准备一个类(DynamicDataSource)继承AbstractRoutingDataSource实现determineCurrentLookupKey方法,该方法可以实现数据库的动态切换;一个设置和获取数据源的类(DataSourceContextHolder)
public class DynamicDataSource extends AbstractRoutingDataSource{
@Override
protected Object determineCurrentLookupKey() {
return DataSourceContextHolder.getDataSource();
}
}
public class DataSourceContextHolder {
private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();
/**
*设置数据源
*/
public static void setDataSource(String dataSource){
contextHolder.set(dataSource);
}
/**
*获取数据源
*/
public static String getDataSource(){
return contextHolder.get();
}
/**
* 清除数据源名字
*/
public static void clearDataSource(){
contextHolder.remove();
}
}
3.spring-mybatis.xml里配置多数据源
<!-- 多数据源配置 -->
<bean id="dataSource" class="com.ghca.util.db.DynamicDataSource">
<property name="defaultTargetDataSource" ref="fdataSource"></property>
<property name="targetDataSources">
<map key-type ="java.lang.String">
<entry key="fdataSource" value-ref="fdataSource" />
<entry key="sdataSource" value-ref="sdataSource" />
</map>
</property>
</bean>
<!-- myBatis文件 -->
<bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
<property name="dataSource" ref="dataSource" />
<!-- 自动扫描entity目录, 省掉Configuration.xml里的手工配置 -->
<property name="mapperLocations" value="classpath:mapper/*.xml" />
</bean>
<bean class="org.mybatis.spring.mapper.MapperScannerConfigurer">
<property name="basePackage" value="com.ghca.dao" />
<property name="sqlSessionFactoryBeanName" value="sqlSessionFactory" />
</bean>
<!-- 配置事务管理器 -->
<bean id="transactionManager"
class="org.springframework.jdbc.datasource.DataSourceTransactionManager">
<property name="dataSource" ref="dataSource" />
</bean>
4.通过aop配置来实现自动数据源切换
本文先通过xml方式配置
(1)准备一个Annotation,用于记录数据源名:
@Target({ElementType.METHOD,ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface DataSource {
String name() default DataSource.fdataSource;
public static String fdataSource = "fdataSource";
public static String sdataSource = "sdataSource";
}
(2)准备一个切面:
public class AspectDataSourceExchange{
public void afterReturning(){
DataSourceContextHolder.clearDataSource();
}
public void before(JoinPoint joinPoint){
// 从切点上获取目标方法
MethodSignature methodSignature = (MethodSignature) joinPoint.getSignature();
Method method = methodSignature.getMethod();
if (method.isAnnotationPresent(DataSource.class)) {
DataSource datasource = method.getAnnotation(DataSource.class);
if (null == datasource || (null != datasource && null == datasource.name())) {
DataSourceContextHolder.setDataSource(DataSource.fdataSource);
System.out.println("当前数据源为:" + DataSourceContextHolder.getDataSource());
}
else {
DataSourceContextHolder.setDataSource(datasource.name());
System.out.println("当前数据源为:" + DataSourceContextHolder.getDataSource());
}
}
}
}
(3)aop配置
spring-mybatis.xml里
<!-- 注册切换数据源bean (切面) -->
<bean id="aspectDataSourceExchange" class="com.ghca.util.db.AspectDataSourceExchange" />
<!-- Spring aop事务管理 -->
<aop:config>
<aop:aspect id="aspect" ref="aspectDataSourceExchange">
<aop:pointcut expression="execution(* com.ghca.service.impl.LoginServiceImpl.*(..))"
id="transactionPointcut" />
<aop:before method="before" pointcut-ref="transactionPointcut"/>
<aop:after-returning method="afterReturning"
pointcut-ref="transactionPointcut"/>
</aop:aspect>
</aop:config>
(4)**代理
在spring-servlet.xml里
<!-- **自动代理功能 -->
<aop:aspectj-autoproxy />
(ps:
1. 配置的位置不对容易导致报错!尽量放在DispatcherServlet对应的xml文件里
2. proxy-target-class默认为false,如果配成true,则需导入CGLIB包
)
5.代码测试
LoginService.java
public interface LoginService {
//设置数据源
@DataSource(name = DataSource.sdataSource)
User findUser(String username);
}
LoginServiceImpl .java
@Service
public class LoginServiceImpl implements LoginService{
@Autowired
private UserMapper userMapper;
public User findUser(String username) {
User user = userMapper.findUser(username);
return user;
}
}
Controller
@Controller
public class LoginController {
@Autowired
private LoginService loginService;
@RequestMapping("/login")
@ResponseBody
@DataSource(name = DataSource.sdataSource)
public Object login(String username){
User user = loginService.findUser(username);
if (user == null) {
return "该用户未注册!";
}
return user;
}
}
测试结果: