欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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;
    }
}

测试结果:
Spring MVC框架搭建扩展--多数据源配置(一)
Spring MVC框架搭建扩展--多数据源配置(一)