spring整合mybatis动态切换数据源+事务管理_mybatis 在事务中 切换数据源1146-程序员宅基地

技术标签: spring  mybatis  

公司因为业务需要,需要把同一份服务的数据根据不同标识分别存放到两个数据库中(数据结构一样),刚开始为了尽快满足需求,使用了个简版(代码切换数据源),闲下来以后实在觉得太low,查看了一些demo以后,整理出来了适合自己项目的配置。话不多说,上步骤。

1、配置数据源

<!--引入外部资源文件-->
    <context:property-placeholder location="/WEB-INF/properties/db.properties" ignore-unresolvable="true"/>
<!--配置数据源 1-->
    <bean id="dataSource1" class="com.alibaba.druid.pool.DruidDataSource"
          destroy-method="close">
        <property name="url" value="${jdbc.url}"/>
        <property name="username" value="${jdbc.username}"/>
        <property name="password" value="${jdbc.password}"/>
        <property name="driverClassName" value="${jdbc.driver}"/>
        <property name="maxActive" value="10"/>
        <property name="minIdle" value="5"/>
    </bean>
    <!--配置数据源 2-->
    <bean id="dataSource2" class="com.alibaba.druid.pool.DruidDataSource"
          destroy-method="close">
        <property name="url" value="${hongmeng.url}"/>
        <property name="username" value="${hongmeng.username}"/>
        <property name="password" value="${hongmeng.password}"/>
        <property name="driverClassName" value="${jdbc.driver}"/>
        <property name="maxActive" value="10"/>
        <property name="minIdle" value="5"/>
    </bean>
    <!--配置多数据源映射关系-->
    <bean id="dataSource" class="com.hlsoft.heguivedio.datasources.DynamicDataSource">
        <property name="targetDataSources">
            <map key-type="java.lang.String">
                <entry key="dataSource1" value-ref="dataSource1"></entry>
                <entry key="dataSource2" value-ref="dataSource2"></entry>
            </map>
        </property>
        <!--设置主数据库为默认数据源-->
        <property name="defaultTargetDataSource" ref="dataSource1"/>
    </bean>

    <!--配置工厂-->
    <bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
        <property name="dataSource" ref="dataSource"/>
        <property name="typeAliasesPackage" value="com.hlsoft.heguivedio.pojo"/>
        <property name="configLocation" value="/WEB-INF/sqlMapConfig.xml"/>
    </bean>

    <!--配置映射器-->
    <bean class="org.mybatis.spring.mapper.MapperScannerConfigurer">
        <property name="basePackage" value="com.hlsoft.heguivedio.mapper"/>
        <property name="sqlSessionFactoryBeanName" value="sqlSessionFactory"/>
    </bean>

上面的配置都是spring整合mybatis的基本配置,唯一不同的是数据源工厂加载数据源时没有直接使用我们常用的数据库连接池,而是使用了DynamicDataSource,这个类在很多demo中都有写,这正是能帮助我们动态切换数据源的关键一步,下面是这个类的代码

public class DynamicDataSource extends AbstractRoutingDataSource {
    
    @Override
    protected Object determineCurrentLookupKey() {
    
        return DataSourceHolder.getDataSource();
    }
}

从这个类中可以看到,只有一个重写的determineCurrentLookupKey方法,这里不深究它如何帮助我们切换数据源,其实是我他喵的自己没搞懂,不过这不妨碍我们实现功能,接下来继续贴代码

public class DataSourceHolder {
    
    //线程本地环境
    private static final ThreadLocal<String> dataSources = new ThreadLocal<String>();
    //设置数据源
    public static void setDataSource(String customerType) {
    
        dataSources.set(customerType);
    }
    //获取数据源
    public static String getDataSource() {
    
        return (String) dataSources.get();
    }
    //清除数据源
    public static void clearDataSource() {
    
        dataSources.remove();
    }

这个类中能看到我们熟悉的ThreadLocal类,上一个类中正是调用了getDataSource方法获取线程中设置好的数据源来指定我们接下来要使用的数据源,关于数据源切换工作已准备就绪,接下来配置如何指定线程中的数据源

2、设置数据源

其实设置数据源的方式很简单,只要调用DataSourceHolder中的setDataSource方法就能成功,如:

DataSourceHolder.setDataSource("dataSource1");
//这里设置的“dataSource1”是在数据源配置文件中已经配好的第一个数据源,同理,设置第二个数据源时传入dataSource2就可以

不过,采用这种方法,未免有些emm…,所以,这个时候我们会想到注解,那玩意多好用,逼格也显然比在代码中重复加上面这一句高了不少,不多逼逼,搞定注解

@Target({
    ElementType.METHOD, ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
public @interface DataSource {
    
    String name() default DataSource.master;

    String master = "dataSource1";

    String slave = "dataSource2";
}

定义一个注解,用来切换数据源,注解中我已经设置好了配置文件中配好的两个数据源字符串,在这里根据自己需求设置就行,当然,也可以不写,使用注解时直接往里填字符串就行,纯看个人喜好。到了这里,只剩下一件事,怎么获取注解中的数据源并配置到线程(也就是我们定义好的DataSourceHolder 的ThreadLocal中),这个过程呢,绝逼又是一段重复代码,这个时候,要用到spring的AOP了,让它在合适的时机,正确的节点帮咱把这事干了

AOP配置

aop帮我们往代码里加一段重复代码,帮咱实现一些必要的功能,不过首先,咱得先写好这段代码,别急,下面这段就是

public class DataSourceAspect {
    

    public void before(JoinPoint point) {
    
        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);
                DataSourceHolder.setDataSource(data.name());
                System.out.println(DataSourceHolder.getDataSource());
            }
        } catch (Exception e) {
    
            e.printStackTrace();
        }
    }

    //清除数据源,使用默认数据源
    public void after() {
    
        DataSourceHolder.clearDataSource();
    }
}

这里定义了两端两个方法,before是aop切入方法之前要执行的,after是方法结束之后要执行的,before的功能就是上面我叨叨了半天可以帮我们获取注解数据源并配置到线程的,after呢,不用说了吧,懂的都懂(切记,一定要写,否则会因为线程缓存出现一些奇奇怪怪的问题,本人已经深受其害)

代码准备就绪,接下来开始配置

<!-- 开启aop,对类代理 -->
    <aop:aspectj-autoproxy proxy-target-class="true" />
    <!-- 配置数据库注解aop -->
    <bean id="dataSourceAspect" class="com.hlsoft.heguivedio.datasources.DataSourceAspect" />
    <aop:config>
        <aop:aspect id="c" ref="dataSourceAspect">
            <aop:pointcut id="tx" expression="execution(* com.hlsoft.heguivedio.service..*.*(..))"/>
            <aop:before pointcut-ref="tx" method="before"/>
            <aop:after pointcut-ref="tx" method="after"/>
        </aop:aspect>
    </aop:config>

这里配置的service包下放置的就是我需要切入的方法,数据源切换到这里就完成了,接下来展示一段调用的方法

public interface cwx_channelService {
    
	@DataSource(name = DataSource.slave)
    void channelsTest();
}

public class cwx_channelServiceImpl implements cwx_channelService {
    
	@Override
    @Transactional("jzHm")
    public void channelsTest() {
    
        List<Reviewer> reviewers = audioFrequencyMapper.channelsTest();
        for (Reviewer reviewer : reviewers) {
    
            System.out.println(reviewer);
        }
    }
}

代码中能看到DataSource注解加到了接口方法上,这是为了事务考虑,设置事务前必须先设置好数据源,否则事务使用默认数据源后,再切换数据源就不再生效

3、事务控制

数据源切换已经搞定,接下来搞定事务,下面是常规的spring事务管理配置

<!--配置事务-->
    <bean id="transactionManager" class="org.springframework.jdbc.datasource.DataSourceTransactionManager">
        <property name="dataSource" ref="dataSource"/>
    </bean>

    <!--开启事务注解-->
    <tx:annotation-driven transaction-manager="transactionManager"/>

有了这段配置之后,使用transactionManager注解控制事务即可,但在多数据源项目中,需要配置对应数据源的事务管理器,否则spring只识别默认数据源事务管理器,然后导致数据源切换也不能成功,更别提事务管理了,增加数据源事务管理器配置如下

 <!--主数据库事务管理器类 -->
    <bean id="txManagerHl" class="org.springframework.jdbc.datasource.DataSourceTransactionManager">
        <property name="dataSource" ref="dataSource1" />
        <qualifier value="jzHl" />
    </bean>
    <!--从库事务管理器类 -->
    <bean id="txManagerHm" class="org.springframework.jdbc.datasource.DataSourceTransactionManager">
        <property name="dataSource" ref="dataSource2" />
        <qualifier value="jzHm" />
    </bean>

上面增加了我项目中的两个数据源的事务管理器(分别对应不同的数据源),qualifier配置的内容主要用来识别要使用哪个事务管理器(可随意配置),事务管理时填入配置的两个字符串即可切换事务管理,配置完成后同时将spring事务管理配置进行修改,看图

<!--配置事务-->
    <bean id="transactionManager" class="org.springframework.jdbc.datasource.DataSourceTransactionManager">
        <property name="dataSource" ref="dataSource"/>
        <qualifier value="mybatis"/>
    </bean>

    <!--开启事务注解-->
    <tx:annotation-driven transaction-manager="transactionManager"/>

这里和默认配置唯一不同的是加上了qualifier ,mybatis是固定配置,不能修改,完成这些配置之后,多数据源的事务管理就能成功使用了,使用方法看上方第二部分完成后的调用方法,使用哪个数据源的事务就写上事务管理器中qualifier 的值

数据源切换及事务控制配置大致是以上的步骤,有说明不清楚或者有误的地方欢迎大家指正。

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/qq_42160456/article/details/109584644

智能推荐

oracle 12c 集群安装后的检查_12c查看crs状态-程序员宅基地

文章浏览阅读1.6k次。安装配置gi、安装数据库软件、dbca建库见下:http://blog.csdn.net/kadwf123/article/details/784299611、检查集群节点及状态:[root@rac2 ~]# olsnodes -srac1 Activerac2 Activerac3 Activerac4 Active[root@rac2 ~]_12c查看crs状态

解决jupyter notebook无法找到虚拟环境的问题_jupyter没有pytorch环境-程序员宅基地

文章浏览阅读1.3w次,点赞45次,收藏99次。我个人用的是anaconda3的一个python集成环境,自带jupyter notebook,但在我打开jupyter notebook界面后,却找不到对应的虚拟环境,原来是jupyter notebook只是通用于下载anaconda时自带的环境,其他环境要想使用必须手动下载一些库:1.首先进入到自己创建的虚拟环境(pytorch是虚拟环境的名字)activate pytorch2.在该环境下下载这个库conda install ipykernelconda install nb__jupyter没有pytorch环境

国内安装scoop的保姆教程_scoop-cn-程序员宅基地

文章浏览阅读5.2k次,点赞19次,收藏28次。选择scoop纯属意外,也是无奈,因为电脑用户被锁了管理员权限,所有exe安装程序都无法安装,只可以用绿色软件,最后被我发现scoop,省去了到处下载XXX绿色版的烦恼,当然scoop里需要管理员权限的软件也跟我无缘了(譬如everything)。推荐添加dorado这个bucket镜像,里面很多中文软件,但是部分国外的软件下载地址在github,可能无法下载。以上两个是官方bucket的国内镜像,所有软件建议优先从这里下载。上面可以看到很多bucket以及软件数。如果官网登陆不了可以试一下以下方式。_scoop-cn

Element ui colorpicker在Vue中的使用_vue el-color-picker-程序员宅基地

文章浏览阅读4.5k次,点赞2次,收藏3次。首先要有一个color-picker组件 <el-color-picker v-model="headcolor"></el-color-picker>在data里面data() { return {headcolor: ’ #278add ’ //这里可以选择一个默认的颜色} }然后在你想要改变颜色的地方用v-bind绑定就好了,例如:这里的:sty..._vue el-color-picker

迅为iTOP-4412精英版之烧写内核移植后的镜像_exynos 4412 刷机-程序员宅基地

文章浏览阅读640次。基于芯片日益增长的问题,所以内核开发者们引入了新的方法,就是在内核中只保留函数,而数据则不包含,由用户(应用程序员)自己把数据按照规定的格式编写,并放在约定的地方,为了不占用过多的内存,还要求数据以根精简的方式编写。boot启动时,传参给内核,告诉内核设备树文件和kernel的位置,内核启动时根据地址去找到设备树文件,再利用专用的编译器去反编译dtb文件,将dtb还原成数据结构,以供驱动的函数去调用。firmware是三星的一个固件的设备信息,因为找不到固件,所以内核启动不成功。_exynos 4412 刷机

Linux系统配置jdk_linux配置jdk-程序员宅基地

文章浏览阅读2w次,点赞24次,收藏42次。Linux系统配置jdkLinux学习教程,Linux入门教程(超详细)_linux配置jdk

随便推点

matlab(4):特殊符号的输入_matlab微米怎么输入-程序员宅基地

文章浏览阅读3.3k次,点赞5次,收藏19次。xlabel('\delta');ylabel('AUC');具体符号的对照表参照下图:_matlab微米怎么输入

C语言程序设计-文件(打开与关闭、顺序、二进制读写)-程序员宅基地

文章浏览阅读119次。顺序读写指的是按照文件中数据的顺序进行读取或写入。对于文本文件,可以使用fgets、fputs、fscanf、fprintf等函数进行顺序读写。在C语言中,对文件的操作通常涉及文件的打开、读写以及关闭。文件的打开使用fopen函数,而关闭则使用fclose函数。在C语言中,可以使用fread和fwrite函数进行二进制读写。‍ Biaoge 于2024-03-09 23:51发布 阅读量:7 ️文章类型:【 C语言程序设计 】在C语言中,用于打开文件的函数是____,用于关闭文件的函数是____。

Touchdesigner自学笔记之三_touchdesigner怎么让一个模型跟着鼠标移动-程序员宅基地

文章浏览阅读3.4k次,点赞2次,收藏13次。跟随鼠标移动的粒子以grid(SOP)为partical(SOP)的资源模板,调整后连接【Geo组合+point spirit(MAT)】,在连接【feedback组合】适当调整。影响粒子动态的节点【metaball(SOP)+force(SOP)】添加mouse in(CHOP)鼠标位置到metaball的坐标,实现鼠标影响。..._touchdesigner怎么让一个模型跟着鼠标移动

【附源码】基于java的校园停车场管理系统的设计与实现61m0e9计算机毕设SSM_基于java技术的停车场管理系统实现与设计-程序员宅基地

文章浏览阅读178次。项目运行环境配置:Jdk1.8 + Tomcat7.0 + Mysql + HBuilderX(Webstorm也行)+ Eclispe(IntelliJ IDEA,Eclispe,MyEclispe,Sts都支持)。项目技术:Springboot + mybatis + Maven +mysql5.7或8.0+html+css+js等等组成,B/S模式 + Maven管理等等。环境需要1.运行环境:最好是java jdk 1.8,我们在这个平台上运行的。其他版本理论上也可以。_基于java技术的停车场管理系统实现与设计

Android系统播放器MediaPlayer源码分析_android多媒体播放源码分析 时序图-程序员宅基地

文章浏览阅读3.5k次。前言对于MediaPlayer播放器的源码分析内容相对来说比较多,会从Java-&amp;amp;gt;Jni-&amp;amp;gt;C/C++慢慢分析,后面会慢慢更新。另外,博客只作为自己学习记录的一种方式,对于其他的不过多的评论。MediaPlayerDemopublic class MainActivity extends AppCompatActivity implements SurfaceHolder.Cal..._android多媒体播放源码分析 时序图

java 数据结构与算法 ——快速排序法-程序员宅基地

文章浏览阅读2.4k次,点赞41次,收藏13次。java 数据结构与算法 ——快速排序法_快速排序法