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+注解的方式实现控制，方法多多！

总结

以上就是这篇文章的全部内容了，希望本文的内容对大家的学习或者工作具有一定的参考学习价值，如果有疑问大家可以留言交流，谢谢大家对的支持。