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

智能推荐

c# 调用c++ lib静态库_c#调用lib-程序员宅基地

文章浏览阅读2w次,点赞7次,收藏51次。四个步骤1.创建C++ Win32项目动态库dll 2.在Win32项目动态库中添加 外部依赖项 lib头文件和lib库3.导出C接口4.c#调用c++动态库开始你的表演...①创建一个空白的解决方案,在解决方案中添加 Visual C++ , Win32 项目空白解决方案的创建:添加Visual C++ , Win32 项目这......_c#调用lib

deepin/ubuntu安装苹方字体-程序员宅基地

文章浏览阅读4.6k次。苹方字体是苹果系统上的黑体,挺好看的。注重颜值的网站都会使用,例如知乎:font-family: -apple-system, BlinkMacSystemFont, Helvetica Neue, PingFang SC, Microsoft YaHei, Source Han Sans SC, Noto Sans CJK SC, W..._ubuntu pingfang

html表单常见操作汇总_html表单的处理程序有那些-程序员宅基地

文章浏览阅读159次。表单表单概述表单标签表单域按钮控件demo表单标签表单标签基本语法结构<form action="处理数据程序的url地址“ method=”get|post“ name="表单名称”></form><!--action,当提交表单时,向何处发送表单中的数据,地址可以是相对地址也可以是绝对地址--><!--method将表单中的数据传送给服务器处理,get方式直接显示在url地址中,数据可以被缓存,且长度有限制;而post方式数据隐藏传输,_html表单的处理程序有那些

PHP设置谷歌验证器(Google Authenticator)实现操作二步验证_php otp 验证器-程序员宅基地

文章浏览阅读1.2k次。使用说明:开启Google的登陆二步验证(即Google Authenticator服务)后用户登陆时需要输入额外由手机客户端生成的一次性密码。实现Google Authenticator功能需要服务器端和客户端的支持。服务器端负责密钥的生成、验证一次性密码是否正确。客户端记录密钥后生成一次性密码。下载谷歌验证类库文件放到项目合适位置(我这边放在项目Vender下面)https://github.com/PHPGangsta/GoogleAuthenticatorPHP代码示例://引入谷_php otp 验证器

【Python】matplotlib.plot画图横坐标混乱及间隔处理_matplotlib更改横轴间距-程序员宅基地

文章浏览阅读4.3k次,点赞5次,收藏11次。matplotlib.plot画图横坐标混乱及间隔处理_matplotlib更改横轴间距

docker — 容器存储_docker 保存容器-程序员宅基地

文章浏览阅读2.2k次。①Storage driver 处理各镜像层及容器层的处理细节,实现了多层数据的堆叠,为用户 提供了多层数据合并后的统一视图②所有 Storage driver 都使用可堆叠图像层和写时复制(CoW)策略③docker info 命令可查看当系统上的 storage driver主要用于测试目的,不建议用于生成环境。_docker 保存容器

随便推点

网络拓扑结构_网络拓扑csdn-程序员宅基地

文章浏览阅读834次,点赞27次,收藏13次。网络拓扑结构是指计算机网络中各组件(如计算机、服务器、打印机、路由器、交换机等设备)及其连接线路在物理布局或逻辑构型上的排列形式。这种布局不仅描述了设备间的实际物理连接方式,也决定了数据在网络中流动的路径和方式。不同的网络拓扑结构影响着网络的性能、可靠性、可扩展性及管理维护的难易程度。_网络拓扑csdn

JS重写Date函数,兼容IOS系统_date.prototype 将所有 ios-程序员宅基地

文章浏览阅读1.8k次,点赞5次,收藏8次。IOS系统Date的坑要创建一个指定时间的new Date对象时,通常的做法是:new Date("2020-09-21 11:11:00")这行代码在 PC 端和安卓端都是正常的,而在 iOS 端则会提示 Invalid Date 无效日期。在IOS年月日中间的横岗许换成斜杠,也就是new Date("2020/09/21 11:11:00")通常为了兼容IOS的这个坑,需要做一些额外的特殊处理,笔者在开发的时候经常会忘了兼容IOS系统。所以就想试着重写Date函数,一劳永逸,避免每次ne_date.prototype 将所有 ios

如何将EXCEL表导入plsql数据库中-程序员宅基地

文章浏览阅读5.3k次。方法一:用PLSQL Developer工具。 1 在PLSQL Developer的sql window里输入select * from test for update; 2 按F8执行 3 打开锁, 再按一下加号. 鼠标点到第一列的列头,使全列成选中状态,然后粘贴,最后commit提交即可。(前提..._excel导入pl/sql

Git常用命令速查手册-程序员宅基地

文章浏览阅读83次。Git常用命令速查手册1、初始化仓库git init2、将文件添加到仓库git add 文件名 # 将工作区的某个文件添加到暂存区 git add -u # 添加所有被tracked文件中被修改或删除的文件信息到暂存区,不处理untracked的文件git add -A # 添加所有被tracked文件中被修改或删除的文件信息到暂存区,包括untracked的文件...

分享119个ASP.NET源码总有一个是你想要的_千博二手车源码v2023 build 1120-程序员宅基地

文章浏览阅读202次。分享119个ASP.NET源码总有一个是你想要的_千博二手车源码v2023 build 1120

【C++缺省函数】 空类默认产生的6个类成员函数_空类默认产生哪些类成员函数-程序员宅基地

文章浏览阅读1.8k次。版权声明:转载请注明出处 http://blog.csdn.net/irean_lau。目录(?)[+]1、缺省构造函数。2、缺省拷贝构造函数。3、 缺省析构函数。4、缺省赋值运算符。5、缺省取址运算符。6、 缺省取址运算符 const。[cpp] view plain copy_空类默认产生哪些类成员函数

推荐文章

热门文章

相关标签