小蚂蚁 发表于 2021-10-7 17:04:06

spring boot + mybatis实现动态切换数据源实例代码

这篇文章主要给大家介绍了关于spring boot + mybatis实现动态切换数据源的相关资料,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
前言
前几天有个需求,需要使用不同的数据源,例如某业务要用a数据源,另一个业务要用b数据源。我上网收集了一些资料整合了一下,虽然最后这个需求不了了之了,但是多数据源动态切换还是蛮好用的,所以记录一下,或许以后有用呢?或者自己感兴趣又想玩呢!
下面话不多说了,随着小编来一起看看详细的介绍吧
方法如下:
1.加个依赖


<dependency>
<groupid>org.mybatis.spring.boot</groupid>
<artifactid>mybatis-spring-boot-starter</artifactid>
<version>1.3.1</version>
</dependency>
2.application.properties配置文件


#主从数据库
master.db.driverclassname=com.mysql.jdbc.driver
master.db.url=jdbc:mysql://localhost:3306/cbd?characterencoding=utf-8&useunicode=true&usessl=false
master.db.username=root
master.db.password=admin
slave.db.driverclassname=com.mysql.jdbc.driver
slave.db.url=jdbc:mysql://localhost:3306/cbd_test?characterencoding=utf-8&useunicode=true&usessl=false
slave.db.username=root
slave.db.password=admin

mybatis.config-location= classpath:config/mybatis-config.xml
mybatis.mapper-locations=classpath:mapper/**/*.xml
3.禁用springboot默认加载数据源配置


@springbootapplication(exclude = {datasourceautoconfiguration.class})
public class application {

public static void main(string[] args) throws exception {
springapplication.run(application.class, args);
}
}
4.数据源配置类


/**
* 主数据源
*/
@configuration
@configurationproperties(prefix = "master.db")
public class masterdatasourceconfig {
private string url;
private string username;
private string password;
private string driverclassname;
}


/**
* 从数据源配置
*/
@configuration
@configurationproperties(prefix = "slave.db")
public class slavedatasourceconfig {
private string url;
private string username;
private string password;
private string driverclassname;
}


/**
* 数据源配置类
*/
@configuration
public class datasourcecomponent {

@resource
private masterdatasourceconfig masterdatasourceconfig;

@resource
private slavedatasourceconfig slavedatasourceconfig;


@bean(name = "master")
public datasource masterdatasource() {
datasource datasource = new datasource();
datasource.seturl(masterdatasourceconfig.geturl());
datasource.setusername(masterdatasourceconfig.getusername());
datasource.setpassword(masterdatasourceconfig.getpassword());
datasource.setdriverclassname(masterdatasourceconfig.getdriverclassname());
return datasource;
}

@bean(name = "slave")
public datasource slavedatasource() {
datasource datasource = new datasource();
datasource.seturl(slavedatasourceconfig.geturl());
datasource.setusername(slavedatasourceconfig.getusername());
datasource.setpassword(slavedatasourceconfig.getpassword());
datasource.setdriverclassname(slavedatasourceconfig.getdriverclassname());
return datasource;
}

@primary//不加这个会报错。
@bean(name = "multidatasource")
public multiroutedatasource exampleroutedatasource() {
multiroutedatasource multidatasource = new multiroutedatasource();
map<object, object> targetdatasources = new hashmap<>();
targetdatasources.put("master", masterdatasource());
targetdatasources.put("slave", slavedatasource());
multidatasource.settargetdatasources(targetdatasources);
multidatasource.setdefaulttargetdatasource(masterdatasource());
return multidatasource;
}
}
5.数据源上下文


/**
* 数据源上下文
*/
public class datasourcecontext {
private static final threadlocal<string> contextholder = new threadlocal<>();

public static void setdatasource(string value) {
contextholder.set(value);
}

public static string getdatasource() {
return contextholder.get();
}

public static void cleardatasource() {
contextholder.remove();
}
}
6.datasource路由类


/*
* 重写的函数决定了最后选择的datasource
*/
public class multiroutedatasource extends abstractroutingdatasource {

@override
protected object determinecurrentlookupkey() {
//通过绑定线程的数据源上下文实现多数据源的动态切换,有兴趣的可以去查阅资料或源码
return datasourcecontext.getdatasource();
}

}
7.使用,修改上下文中的数据源就可以切换自己想要使用的数据源了。


public uservo finduser(string username) {
datasourcecontext.setdatasource("slave");
uservo uservo = usermapper.findbyvo(username);
system.out.println(uservo.getname()+"=====================");
return null;
}
这种是在业务中使用代码设置数据源的方式,也可以使用aop+注解的方式实现控制,方法多多!
总结
以上就是这篇文章的全部内容了,希望本文的内容对大家的学习或者工作具有一定的参考学习价值,如果有疑问大家可以留言交流,谢谢大家对CodeAE代码之家的支持。
原文链接:http://www.cnblogs.com/tinyj/p/9864128.html

http://www.zzvips.com/article/169717.html
页: [1]
查看完整版本: spring boot + mybatis实现动态切换数据源实例代码