写这个博客,是记录一下自己的学习过程。
因为网上信息杂乱不清,所以此博客只做记录和分享作用。
参考博客链接:https://blog.csdn.net/QiuHaoqian/article/details/120479890
再此基础上修改了一些,使得mybatisplus无mapper.xml文件可以使用此代码。
先上效果图:
可以看出这个数据库没有这个帐号。
然后更换请求头Language参数,即可更换数据源-数据库
登录成功,返回数据。
实现,只需要写四个类,外加配置数据库和导入相关jar包
话不多说,先上pom.xml
<dependency>
<groupId>com.alibaba</groupId>
<artifactId>druid-spring-boot-starter</artifactId>
<version>1.2.3</version>
</dependency>
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-aop</artifactId>
</dependency>
然后配置文件 application.yml
mybatis-plus:
configuration:
map-underscore-to-camel-case: true
spring:
datasource:
primary:
//此处配置自己的第一个数据库
jdbc-url: jdbc:mysql://localhost:3306/XXXX1?useSSL=false&useUnicode=true&characterEncoding=utf-8&autoReconnect=true&serverTimezone=Asia/Shanghai
username: root
password: XXXXX
driver‐class‐name: com.mysql.cj.jdbc.Driver
secondary:
//此处配置自己的第二个数据库
jdbc-url: jdbc:mysql://localhost:3306/XXXX2?useSSL=false&useUnicode=true&characterEncoding=utf-8&autoReconnect=true&serverTimezone=Asia/Shanghai
username: root
password: XXXXX
driver‐class‐name: com.mysql.cj.jdbc.Driver
server:
port:
8081
再上相关类:
DataSourceAspect.java
package com.xxxx.aop;
import com.xxxx.datasource.DataSourceHolder;
import org.apache.commons.lang3.StringUtils;
import org.aspectj.lang.annotation.After;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.annotation.Order;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;
import java.util.Objects;
/**
* 拦截controller方法,从请求头中获取使用的数据库编号
*
* @author qyz
*/
@Aspect
@Order(1)
@Configuration
public class DataSourceAspect {
/**
* 数据库名称在请求头中的key(从header中取)
*/
private static final String LANGUAGE = "Language";
/**
* 切入点,放在controller的每个方法上进行切入,更新数据源
*/
//这里填入自己项目的controller路径
@Pointcut("execution(* com.1这里要改.controller..*.*(..))")
private void anyMethod() {
}
@Before("anyMethod()")
public void dataSourceChange() {
// 请求头head中获取对应数据库编号 name=Language
String Language = ((ServletRequestAttributes) Objects
.requireNonNull(RequestContextHolder.getRequestAttributes()))
.getRequest()
.getHeader(LANGUAGE);
System.out.println("当前数据源: " + Language);
if (StringUtils.isBlank(Language)) {
// TODO 根据业务抛异常
throw new NullPointerException("请求头中没有" + LANGUAGE);
}
// 根据请求头中数据库名称来更改对应的数据源(核心)
DataSourceHolder.setDataSource(Language);
}
@After("anyMethod()")
public void after() {
// 数据源重置(必须在请求完成后清空ThreadLocal线程上下文,否则会内存溢出)
DataSourceHolder.clearDataSource();
}
}
DataSourceHolder.java
package com.xxxx.datasource;
public class DataSourceHolder {
/**
* 线程本地环境 (存储数据库名称)
*/
private static final ThreadLocal<String> DATA_SOURCES = new ThreadLocal<>();
/**
* 设置数据源(动态切换数据源),就是调用这个setDataSource方法
*/
public static void setDataSource(String customerType) {
DATA_SOURCES.set(customerType);
}
/**
* 获取数据源
*/
public static String getDataSource() {
return DATA_SOURCES.get();
}
/**
* 清除数据源
*/
public static void clearDataSource() {
DATA_SOURCES.remove();
}
}
DynamicDataSource.java
package com.xxxx.datasource;
import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
public class DynamicDataSource extends AbstractRoutingDataSource {
/**
* 每次请求动态请求哪一个数据源
*
* @return
*/
@Override
protected Object determineCurrentLookupKey() {
return DataSourceHolder.getDataSource();
}
}
DynamicDataSourceConfig.java
package com.xxxx.datasource;
import com.baomidou.mybatisplus.core.MybatisConfiguration;
import com.baomidou.mybatisplus.core.MybatisXMLLanguageDriver;
import org.apache.ibatis.session.SqlSessionFactory;
import org.apache.ibatis.type.JdbcType;
import org.mybatis.spring.SqlSessionFactoryBean;
import org.mybatis.spring.annotation.MapperScan;
import org.mybatis.spring.transaction.SpringManagedTransactionFactory;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.boot.jdbc.DataSourceBuilder;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import javax.sql.DataSource;
import java.util.HashMap;
import java.util.Map;
@Configuration
//改成你项目的mapper路径
@MapperScan(basePackages = "com.2这里也要改.mapper", sqlSessionFactoryRef = "SqlSessionFactory")
public class DynamicDataSourceConfig {
/**
* 将第1个数据源对象放入Spring容器中
*
* @ConfigurationProperties 读取application.properties中的前缀为spring.datasource.primary的配置参数并映射成为一个对象
*/
@Bean(name = "dateSource1")
@ConfigurationProperties(prefix = "spring.datasource.primary")
public DataSource DateSource1() {
return DataSourceBuilder.create().build();
}
/**
* 将第2个数据源对象放入Spring容器中
*/
@Bean(name = "dateSource2")
@ConfigurationProperties(prefix = "spring.datasource.secondary")
public DataSource DateSource2() {
return DataSourceBuilder.create().build();
}
/**
* 将动态代理数据源对象放入Spring容器中
*/
@Bean(name = "dynamicDataSource")
public DynamicDataSource DynamicDataSource(@Qualifier("dateSource1") DataSource primaryDataSource,
@Qualifier("dateSource2") DataSource secondaryDataSource) {
// 这个地方是比较核心的targetDataSource 集合是我们数据库和名字之间的映射
Map<Object, Object> targetDataSource = new HashMap<>();
targetDataSource.put("ch", primaryDataSource);
targetDataSource.put("en", secondaryDataSource);
DynamicDataSource dataSource = new DynamicDataSource();
// 设置所有的数据源
dataSource.setTargetDataSources(targetDataSource);
// 设置默认使用的数据源对象
dataSource.setDefaultTargetDataSource(primaryDataSource);
return dataSource;
}
@Bean(name = "SqlSessionFactory")
public SqlSessionFactory SqlSessionFactory(@Qualifier("dynamicDataSource") DataSource dynamicDataSource)
throws Exception {
SqlSessionFactoryBean sqlSessionFactoryBean = new SqlSessionFactoryBean();
sqlSessionFactoryBean.setDataSource(dynamicDataSource);
//也是改成pojo路径(pojo实体类)
sqlSessionFactoryBean.setTypeAliasesPackage("com.3这里要改.pojo");
MybatisConfiguration configuration = new MybatisConfiguration();
configuration.setDefaultScriptingLanguage(MybatisXMLLanguageDriver.class);
configuration.setJdbcTypeForNull(JdbcType.NULL);
sqlSessionFactoryBean.setConfiguration(configuration);
sqlSessionFactoryBean.setTransactionFactory(new SpringManagedTransactionFactory());
return sqlSessionFactoryBean.getObject();
}
}
基本的增删改查代码就不贴了。
测试如上方效果图在请求头填入参数即可。
至此项目完成。