官方文档: mybatis-plus 官方文档
一、添加依赖pom.xml
<!-- mysql驱动 -->
<dependency>
<groupId>mysql</groupId>
<artifactId>mysql-connector-java</artifactId>
</dependency>
<!-- druid 连接池 -->
<dependency>
<groupId>com.alibaba</groupId>
<artifactId>druid-spring-boot-starter</artifactId>
<version>1.1.10</version>
</dependency>
<!-- mybatis-plus -->
<dependency>
<groupId>com.baomidou</groupId>
<artifactId>mybatis-plus-boot-starter</artifactId>
<version>3.1.0</version>
</dependency>
<!-- mybatis plus 代码生成器依赖 -->
<dependency>
<groupId>com.baomidou</groupId>
<artifactId>mybatis-plus-generator</artifactId>
<version>3.1.0</version>
</dependency>
<!-- 代码生成器模板 -->
<dependency>
<groupId>org.freemarker</groupId>
<artifactId>freemarker</artifactId>
</dependency>
<!-- 防止打包没有打入mapper.xml文件 -->
<resources>
<resource>
<directory>src/main/resources</directory>
<filtering>true</filtering>
<!-- 打包时所包含得文件 -->
<includes>
<include>application.yml</include>
<include>mapper/*.xml</include>
<include>config/application-${env}.yml</include>
</includes>
</resource>
</resources>
二、application.yml添加配置
# 数据库相关
spring:
#数据库配置
datasource:
url: jdbc:mysql://IP:3306/demo?useUnicode=true&useSSL=false&characterEncoding=utf-8
username: name
password: password
# 使用druid数据源
type: com.alibaba.druid.pool.DruidDataSource
driver-class-name: com.mysql.jdbc.Driver
# 数据源配置
druid:
# druid连接池监控
stat-view-servlet:
enabled: true
url-pattern: /druid/*
login-username: admin
login-password: admin
# 初始化时建立物理连接的个数
initial-size: 5
# 最大连接池数量
max-active: 30
# 最小连接池数量
min-idle: 5
# 获取连接时最大等待时间,单位毫秒
max-wait: 60000
# 配置间隔多久才进行一次检测,检测需要关闭的空闲连接,单位是毫秒
time-between-eviction-runs-millis: 60000
# 连接保持空闲而不被驱逐的最小时间
min-evictable-idle-time-millis: 300000
# 用来检测连接是否有效的sql,要求是一个查询语句
validation-query: select count(*) from dual
# 建议配置为true,不影响性能,并且保证安全性。申请连接的时候检测,如果空闲时间大于timeBetweenEvictionRunsMillis,执行validationQuery检测连接是否有效。
test-while-idle: true
# 申请连接时执行validationQuery检测连接是否有效,做了这个配置会降低性能。
test-on-borrow: false
# 归还连接时执行validationQuery检测连接是否有效,做了这个配置会降低性能。
test-on-return: false
# 是否缓存preparedStatement,也就是PSCache。PSCache对支持游标的数据库性能提升巨大,比如说oracle。在mysql下建议关闭。
pool-prepared-statements: false
# 要启用PSCache,必须配置大于0,当大于0时,poolPreparedStatements自动触发修改为true。
max-pool-prepared-statement-per-connection-size: 50
# 配置监控统计拦截的filters,去掉后监控界面sql无法统计
filters: stat,wall
# 通过connectProperties属性来打开mergeSql功能;慢SQL记录
connection-properties:
druid.stat.mergeSql: true
druid.stat.slowSqlMillis: 500
# 合并多个DruidDataSource的监控数据
use-global-data-source-stat: true
filter:
stat:
log-slow-sql: true
slow-sql-millis: 1000
merge-sql: true
wall:
config:
multi-statement-allow: true
# mybatis-plus 配置
mybatis-plus:
configuration:
map-underscore-to-camel-case: true
auto-mapping-behavior: full
# 日志打印
log-impl: org.apache.ibatis.logging.stdout.StdOutImpl
# 配置 Mapper 对应的 XML 文件路径
mapper-locations: classpath*:mapper/**/*Mapper.xml
# 参考配置
mybatis-plus:
mapper-locations: classpath:/mapping/**/*.xml
global-config:
db-config:
#刷新mapper 调试神器
#主键类型 0:"数据库ID自增", 1:"用户输入ID",2:"全局唯一ID (数字类型唯一ID)", 3:"全局唯一ID UUID";
id-type: auto
#驼峰下划线转换
capital-mode: true
#逻辑删除配置
sql-injector: com.baomidou.mybatisplus.mapper.LogicSqlInjector
logic-delete-value: 1
logic-not-delete-value: 0
# 配置数据源
db-type: mysql
# 操作是否判断null
field-strategy: not-null
configuration:
map-underscore-to-camel-case: true
cache-enabled: false
mybatis-plus:
config-location: xxxxxxx # MyBatis 配置文件位置 ---String 默认:null
mapper-locations: # MyBatis Mapper 所对应的 XML 文件位置 ---String[] 默认: []
- classpath*:/mapper/comm
- classpath*:/mapper/member
type-aliases-package: bertram.wang.domain.entity # 实体包 --- String
type-aliases-super-type: bertram.wang.domain.entity.BaseEntity # 实体的父类类名 Class<?>
type-handlers-package: xxxxxxxx # SqlSessionFactoryBean 会把该包下面的类注册为对应的 TypeHandler ---String 默认:null
typeEnumsPackage: xxxxx # 让实体类字段能够简单快捷的使用枚举属性 ---String 默认:null
check-config-location: false # 检查mybatis xml 文件的存在
executor-type: simple # 知道mybatis的执行器
# configuration-properties: # 指定外部化 MyBatis Properties 配置,通过该配置可以抽离配置,实现不同环境的配置部署
configuration: # 本部分(Configuration)的配置大都为 MyBatis 原生支持的配置
map-underscore-to-camel-case: true # 是否开启自动驼峰命名规则(camel case)映射
default-enum-type-handler: org.apache.ibatis.type.EnumTypeHandler
aggressive-lazy-loading: true #当设置为 true 的时候,懒加载的对象可能被任何懒属性全部加载,否则,每个属性都按需加载。需要和 lazyLoadingEnabled 一起使用
auto-mapping-behavior: partial # 自动映射策略
auto-mapping-unknown-column-behavior: none # 自动映射遇到未知列时的处理策略
cache-enabled: true # 映射器缓存开启
call-setters-on-nulls: false # 空值的列
configuration-factory: xxxxxxx # 指定一个提供 Configuration 实例的工厂类 (从 3.2.3 版本开始)
global-config:
banner: true # 控制台 print mybatis-plus 的 LOGO
sql-parser-cache: true # 开启缓存
worker-id: 1234312 # ID的部分值
datacenter-id: 1313 # 数据标识ID部分
super-mapper-class: com.baomidou.mybatisplus.core.mapper.Mapper.class # 通用Mapper父类
db-config:
db-type: mysql # 数据库类型
id-type: auto # id策略
table-prefix: t_ # 表前缀
table-underline: true # 表明下划线命名
column-like: false # 是否开启 LIKE 查询,即根据 entity 自动生成的 where 条件中 String 类型字段 是否使用 LIKE,默认不开启
capital-mode: false # 不开启大写命名
logic-delete-value: 1 # 逻辑已删除值
logic-not-delete-value: 0 #逻辑未删除值
三、application配置@MapperScan
@MapperScan("com.alibaba.demo.mapper")
@EnableConfigurationProperties({DemoBeanConfig.class})
@SpringBootApplication
public class DemoApplication {
public static void main(String[] args) {
SpringApplication.run(DemoApplication.class, args);
}
}
四、代码生成器
自动生成entity,mapper等文件,使用代码生成器来自动生成对应的文件了。需要修改几个地方:
1、数据库连接
2、文件需要放置的文件夹地址。
具体代码:
package com.alibaba.demo;
import com.baomidou.mybatisplus.core.exceptions.MybatisPlusException;
import com.baomidou.mybatisplus.core.toolkit.StringPool;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.InjectionConfig;
import com.baomidou.mybatisplus.generator.config.*;
import com.baomidou.mybatisplus.generator.config.po.TableInfo;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;
/**
* 代码自动生成工具
* 执行 main 方法控制台输入模块表名回车自动生成对应项目目录中
* @author shf
*/
public class CodeGenerator {
/**
* <p>
* 读取控制台内容
* </p>
*/
public static String scanner(String tip) {
Scanner scanner = new Scanner(System.in);
StringBuilder help = new StringBuilder();
help.append("请输入" + tip + ":");
System.out.println(help.toString());
if (scanner.hasNext()) {
String ipt = scanner.next();
if (StringUtils.isNotEmpty(ipt)) {
return ipt;
}
}
throw new MybatisPlusException("请输入正确的" + tip + "!");
}
public static void main(String[] args) {
// 代码生成器
AutoGenerator mpg = new AutoGenerator();
// 全局配置
GlobalConfig gc = new GlobalConfig();
// TODO 工程目录,需要手动填写
String projectPath = "D:/workSpace/技术分享/6.springboot搭建工程/demo";
gc.setOutputDir(projectPath + "/src/main/java");
gc.setAuthor("史恒飞");
gc.setOpen(false);
// service 命名方式
gc.setServiceName("%sService");
// service impl 命名方式
gc.setServiceImplName("%sServiceImpl");
// mapper文件命名,注意 %s 会自动填充表实体属性!
gc.setMapperName("%sMapper");
// xml文件结尾
gc.setXmlName("%sMapper");
gc.setFileOverride(true);
gc.setActiveRecord(true);
// XML 二级缓存
gc.setEnableCache(false);
// XML ResultMap
gc.setBaseResultMap(true);
// XML columList
gc.setBaseColumnList(false);
// gc.setSwagger2(true); 实体属性 Swagger2 注解
mpg.setGlobalConfig(gc);
// TODO 数据源配置
DataSourceConfig dsc = new DataSourceConfig();
dsc.setUrl("jdbc:mysql://47.112.96.83:3306/demo?useUnicode=true&useSSL=false&characterEncoding=utf-8");
// dsc.setSchemaName("public");
dsc.setDriverName("com.mysql.jdbc.Driver");
dsc.setUsername("root");
dsc.setPassword("123456");
mpg.setDataSource(dsc);
// TODO 包配置,按模板分
PackageConfig pc = new PackageConfig();
//pc.setModuleName(scanner("模块名"));
pc.setParent("com.alibaba.demo");
pc.setEntity("model");
pc.setService("service");
pc.setServiceImpl("service.impl");
pc.setMapper("dao");
mpg.setPackageInfo(pc);
// 自定义配置
InjectionConfig cfg = new InjectionConfig() {
@Override
public void initMap() {
// to do nothing
}
};
// 如果模板引擎是 freemarker
String templatePath = "/templates/mapper.xml.ftl";
// 自定义输出配置
List<FileOutConfig> focList = new ArrayList<>();
// 自定义配置会被优先输出
focList.add(new FileOutConfig(templatePath) {
@Override
public String outputFile(TableInfo tableInfo) {
// 自定义输出文件名 , 如果你 Entity 设置了前后缀、此处注意 xml 的名称会跟着发生变化!!
return projectPath + "/src/main/resources/mapper/" +
tableInfo.getEntityName() + "Mapper" + StringPool.DOT_XML;
}
});
cfg.setFileOutConfigList(focList);
mpg.setCfg(cfg);
// 配置模板
TemplateConfig templateConfig = new TemplateConfig();
// 配置自定义输出模板
// 指定自定义模板路径,注意不要带上.ftl/.vm, 会根据使用的模板引擎自动识别
// templateConfig.setEntity("templates/entity2.java");
// templateConfig.setService();
// templateConfig.setController();
templateConfig.setXml(null);
mpg.setTemplate(templateConfig);
// 策略配置,驼峰式
StrategyConfig strategy = new StrategyConfig();
strategy.setNaming(NamingStrategy.underline_to_camel);
strategy.setColumnNaming(NamingStrategy.underline_to_camel);
strategy.setEntityLombokModel(true);
strategy.setRestControllerStyle(true);
// 公共父类,实体继承自通用父类,例如时间等字段可以提取
// strategy.setSuperControllerClass("com.alibaba.demo.model");
// 写于父类中的公共字段
// strategy.setSuperEntityColumns("id");
strategy.setInclude(scanner("表名,多个英文逗号分割").split(","));
strategy.setControllerMappingHyphenStyle(true);
strategy.setTablePrefix(pc.getModuleName() + "_");
mpg.setStrategy(strategy);
mpg.setTemplateEngine(new FreemarkerTemplateEngine());
mpg.execute();
}
}
五、添加测试
测试Mybatis-Plus的CURD等方法。
@RunWith(SpringRunner.class)
@SpringBootTest
public class SampleTest {
private static Logger log = LoggerFactory.getLogger(SampleTest.class);
@Autowired
private MpUserService mpUserService;
@Test
public void test1() {
// 插入新记录
MpUser mpUser = new MpUser();
//mpUser.setId(1L);
mpUser.setEmail("test66@baomidou.com");
mpUser.setAge(22);
mpUser.setName("David Hong");
mpUserService.save(mpUser);
// 或者
mpUser.insertOrUpdate();
// 更新完成后,mpUser对象的id会被补全
log.info("mpUser={}", mpUser.toString());
}
@Test
public void test2() {
// 通过主键id查询
MpUser mpUser = mpUserService.getById(1);
log.info("mpUser={}", mpUser.toString());
}
@Test
public void test3() {
// 条件查询,下面相当于xml中的 select * from mp_user where name = 'Tom' and age = '28' limit 1
MpUser mpUser = mpUserService.getOne(new QueryWrapper<MpUser>().eq("name", "Tom").eq("age", "28").last("limit 1"));
log.info("mpUser={}", mpUser.toString());
// 批量查询
List<MpUser> mpUserList = mpUserService.list();
System.out.println("------------------------------all");
mpUserList.forEach(System.out::println);
// 分页查询
int pageNum = 1;
int pageSize = 10;
IPage<MpUser> mpUserIPage = mpUserService.page(new Page<>(pageNum, pageSize), new QueryWrapper<MpUser>().gt("age", "20"));
// IPage to List
List<MpUser> mpUserList1 = mpUserIPage.getRecords();
System.out.println("------------------------------page");
mpUserList1.forEach(System.out::println);
// 总页数
long allPageNum = mpUserIPage.getPages();
System.out.println("------------------------------allPageNum");
System.out.println(allPageNum);
}
@Test
public void test4() {
MpUser mpUser = mpUserService.getById(2);
// 修改更新
mpUser.setName("广东广州");
//mpUserService.updateById(mpUser);
// 或者
mpUser.insertOrUpdate();
// 通过主键id删除
mpUserService.removeById(1);
// 或者
//mpUser.deleteById();
}
}
六、数据分页
1、简单分页方法
int pageNum = 1;
int pageSize = 10;
IPage<MpUser> mpUserIPage = mpUserService.page(new Page<>(pageNum, pageSize), new QueryWrapper<MpUser>().gt("age", "20"));
上面的分页其实是调用BaseMapper的selectPage方法,这样的分页返回的数据确实是分页后的数据,但在控制台打印的SQL语句上看到其实并没有真正的物理分页,而是通过缓存来获得全部数据中再进行的分页,这样对于大数据量操作时是不可取的,那么接下来就叙述一下,真正实现物理分页的方法。
2、 物理分页方法
新建一个MybatisPlusConfig配置类文件
//Spring boot方式
@EnableTransactionManagement
@Configuration
@MapperScan("com.baomidou.cloud.service.*.mapper*")
public class MybatisPlusConfig {
@Bean
public PaginationInterceptor paginationInterceptor() {
PaginationInterceptor paginationInterceptor = new PaginationInterceptor();
// 设置请求的页面大于最大页后操作, true调回到首页,false 继续请求 默认false
// paginationInterceptor.setOverflow(false);
// 设置最大单页限制数量,默认 500 条,-1 不受限制
// paginationInterceptor.setLimit(500);
// 设置数据源为mysql,数据源配置使用阿里TDDL或其他数据源时,需要自行设置
paginationInterceptor.setDbType(DbType.MYSQL);
return paginationInterceptor;
}
}
重新调用mpUserService.page
可以看到数据有物理分页
3、XML自定义分页
- UserMapper.java 方法内容
public interface UserMapper{//可以继承或者不继承BaseMapper
/**
* <p>
* 查询 : 根据state状态查询用户列表,分页显示
* 注意!!: 如果入参是有多个,需要加注解指定参数名才能在xml中取值
* </p>
*
* @param page 分页对象,xml中可以从里面进行取值,传递参数 Page 即自动分页,必须放在第一位(你可以继承Page实现自己的分页对象)
* @param state 状态
* @return 分页对象
*/
IPage<User> selectPageVo(Page page, @Param("age") Integer age);
}
- UserMapper.xml 等同于编写一个普通 list 查询,mybatis-plus 自动替你分页
<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="cn.com.bluemoon.demo.mapper.MpUserMapper">
<select id="selectPageVo" resultType="cn.com.bluemoon.demo.entity.MpUser">
SELECT * FROM mp_user WHERE age=#{age}
</select>
</mapper>
- UserServiceImpl.java 调用分页方法
public IPage<User> selectUserPage(Page<User> page, Integer state) {
// 不进行 count sql 优化,解决 MP 无法自动优化 SQL 问题,这时候你需要自己查询 count 部分
// page.setOptimizeCountSql(false);
// 当 total 为小于 0 或者设置 setSearchCount(false) 分页插件不会进行 count 查询
// 要点!! 分页返回的对象与传入的对象是同一个
return baseMapper.selectPageVo(page, state);
}
4、测试自定义方法
@Test
public void test5() {
Page<MpUser> mpUserPage = new Page<>(1,2);
IPage<MpUser> iPage = mpUserService.selectUserPage(mpUserPage,22);
System.out.println("总页数:"+iPage.getPages());
System.out.println("总记录数:"+iPage.getTotal());
List<MpUser> mpUserList1 = iPage.getRecords();
mpUserList1.forEach(System.out::println);
}
七、打印sql日志
为了方便排查错误,很多时候需要打印mybatis生成的sql语句,这时候就需要打印日志了。
在application.yml中添加:
# Logger Config
logging:
level:
cn.com.vicente.demo: debug
或者
mybatis-plus:
configuration:
log-impl: org.apache.ibatis.logging.stdout.StdOutImpl
八、逻辑删除
很多时候需要表的数据虽然删除了,但是还是希望不是真正删除数据,数据还是留在数据库中,只需要使用一个字段来做标志为即可,这时候就需要逻辑删除功能。
SpringBoot 配置方式:
application.yml 加入配置(如果你的默认值和mp默认的一样,该配置可无):
mybatis-plus:
global-config:
db-config:
logic-delete-value: 1 # 逻辑已删除值(默认为 1)
logic-not-delete-value: 0 # 逻辑未删除值(默认为 0)
注册 Bean(3.1.1开始不再需要这一步):
import com.baomidou.mybatisplus.core.injector.ISqlInjector;
import com.baomidou.mybatisplus.extension.injector.LogicSqlInjector;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
@Configuration
public class MyBatisPlusConfiguration {
@Bean
public ISqlInjector sqlInjector() {
return new LogicSqlInjector();
}
}
实体类字段上加上
@TableLogic
注解@TableField(select = false)注解,可以不查询出deleted字段
@TableLogic
//@TableField(select = false)
private Integer deleted;
效果: 使用mp自带方法删除和查找都会附带逻辑删除功能 (自己写的xml不会)
example
删除时 update user set deleted=1 where id =1 and deleted=0
查找时 select * from user where deleted=0
附件说明
逻辑删除是为了方便数据恢复和保护数据本身价值等等的一种方案,但实际就是删除。
- 如果你需要再查出来就不应使用逻辑删除,而是以一个状态去表示。
九、主键策略
mybatis-plus 的主键生成的类型 默认类型 是 IdType.ID_WORKER全局唯一ID,内容为空自动填充(默认配置),雪花算法
1,局部主键策略实现
在实体类中 ID属性加注解
@TableId(type = IdType.AUTO) 主键自增 数据库中需要设置主键自增
private Long id;
@TableId(type = IdType.NONE) 默认跟随全局策略走
private Long id;
@TableId(type = IdType.UUID) UUID类型主键
private Long id;
@TableId(type = IdType.ID_WORKER) 数值类型数据库中也必须是数值类型 否则会报错
private Long id;
@TableId(type = IdType.ID_WORKER_STR) 字符串类型 数据库也要保证一样字符类型
private Long id;
@TableId(type = IdType.INPUT) 用户自定义了 数据类型和数据库保持一致就行
private Long id;
2,全局主键策略实现
需要在application.yml文件中添加
mybatis-plus:
global-config:
db-config:
id-type: uuid/none/input/id_worker/id_worker_str/auto 表示全局主键都采用该策略(如果全局策略和局部策略都有设置,局部策略优先级高)
十、自动填充
很多时候表中都需要添加创建时间,创建人,修改时间,修改人来跟踪数据的来源和变动,但每次插入数据和修改数据的时候都要set这几个字段又感觉很麻烦,这个时候就系统系统能自动填充这几个字段了。
- 注意:自动填充字段类型在拦截器中类型需要一致
字段必须声明
TableField
注解,属性fill
选择对应策略,该申明告知Mybatis-Plus
需要预留注入SQL
字段@TableField(fill = FieldFill.INSERT)
private LocalDateTime createTime;
@TableField(fill = FieldFill.INSERT_UPDATE)
private LocalDateTime updateTime;
属性
fill
有四种对应策略,分别为:public enum FieldFill {
/**
* 默认不处理
*/
DEFAULT,
/**
* 插入填充字段
*/
INSERT,
/**
* 更新填充字段
*/
UPDATE,
/**
* 插入和更新填充字段
*/
INSERT_UPDATE
}
自定义实现类 MyMetaObjectHandler:
@Component
public class MyMetaObjectHandler implements MetaObjectHandler {
private static final Logger LOGGER = LoggerFactory.getLogger(MyMetaObjectHandler.class);
@Override
public void insertFill(MetaObject metaObject) {
LOGGER.info("start insert fill ....");
//this.setFieldValByName("createTime", LocalDateTime.now(), metaObject);
this.setInsertFieldValByName("createTime", LocalDate.now(), metaObject);
this.setInsertFieldValByName("updateTime", LocalDate.now(), metaObject);
}
@Override
public void updateFill(MetaObject metaObject) {
LOGGER.info("start update fill ....");
this.setUpdateFieldValByName("updateTime", LocalDate.now(), metaObject);
}
}
测试使用
@Test
public void testInsert() {
// 插入新记录
MpUser mpUser = new MpUser();
mpUser.setEmail("wm@baomidou.com");
mpUser.setAge(28);
mpUser.setName("王蒙");
mpUserService.save(mpUser);
log.info("mpUser={}", mpUser.toString());
}
@Test
public void testUpdate() {
// 更新记录
MpUser mpUser = new MpUser();
mpUser.setId(1182478087497998337L);
MpUser newUser = mpUser.selectById();
System.out.println(mpUser == newUser);
mpUser.setName("王天");
mpUser.updateById();
log.info("mpUser={}", mpUser.toString());
log.info("newUser={}", newUser.toString());
}
自动填充优化
insertFill方法每次插入的时候都会调用,如果不存在createTime属性的话,每次插入都会白白调用了,浪费资源,所以可以判断是否存在该属性boolean hasCreateTime = metaObject.hasSetter("createTime");
if (hasCreateTime){
this.setInsertFieldValByName("createTime", LocalDateTime.now(), metaObject);
}
希望,当更新时有设定时间,就用更新时设定的时间,当没有设定时就自动填充更新时间,可以这样设置
Object fieldValue = getFieldValByName("updateTime", metaObject);
if (fieldValue == null){
this.setUpdateFieldValByName("updateTime", LocalDateTime.now(), metaObject);/
}
十一、 druid 查看sql执行情况
十二、map自定义驼峰转化
// 配置类注入bean
@Bean
public ConfigurationCustomizer mybatisConfigurationCustomizer(){
return configuration -> configuration.setObjectWrapperFactory(new MybatisMapWrapperFactory());
}
十三、Mybatis-plus多租户应用及避开方式(3.4.1之前版本)
- 后续mybatisplus版本默认支持租户隔离,参考官方文档 https://baomidou.com/pages/2976a3/#innerinterceptor
1、配置类
```java import com.baomidou.mybatisplus.core.parser.ISqlParser; import com.baomidou.mybatisplus.extension.plugins.PaginationInterceptor; import com.baomidou.mybatisplus.extension.plugins.tenant.TenantHandler; import com.baomidou.mybatisplus.extension.plugins.tenant.TenantSqlParser; import net.sf.jsqlparser.expression.Expression; import net.sf.jsqlparser.expression.StringValue; import org.mybatis.spring.annotation.MapperScan; import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Configuration;
import java.util.ArrayList; import java.util.List;
@Configuration @MapperScan(“com.demo.*.mapper”) public class MyBatisPlusConfig {
@Bean
public PaginationInterceptor paginationInterceptor() {
PaginationInterceptor paginationInterceptor = new PaginationInterceptor();
// 创建SQL解析器集合
List<ISqlParser> sqlParserList = new ArrayList<>();
// 创建租户SQL解析器
TenantSqlParser tenantSqlParser = new TenantSqlParser();
// 设置租户处理器
tenantSqlParser.setTenantHandler(new TenantHandler() {
@Override
public Expression getTenantId(boolean where) {
// 设置当前租户标识
return new StringValue(TenantContext.getTenant());//从当前线程中拿租户标识
}
@Override
public String getTenantIdColumn() {
// 对应数据库租户标识的列名
return "tenant_id";
}
@Override
public boolean doTableFilter(String tableName) {
// 过滤哪些表
List<String> tableNameList = Arrays.asList("user");
if (tableNameList.contains(tableName)){
return true;
}
return false;
}
});
sqlParserList.add(tenantSqlParser);
paginationInterceptor.setSqlParserList(sqlParserList);
//需要过滤某一个方法
paginationInterceptor.setSqlParserFilter(new ISqlParserFilter() {
@Override
public boolean doFilter(MetaObject metaObject) {
MappedStatement ms = SqlParserHelper.getMappedStatement(metaObject);
// 对应Mapper、dao中的方法
if("com.demo.mybatis.mapper.MyMapper.getOne".equals(ms.getId())){
return true;
}
return false;
}
});
return paginationInterceptor;
}
}
实现效果:针对没有被过滤的表和方法,SQL后面会增加条件 tenant_id="租户标识"
<a name="BIfPZ"></a>
#### 2、租户获取
```java
public class TenantContext {
private static ThreadLocal<String> currentTenant = new ThreadLocal();
public TenantContext() {
}
//在登录的时候向当前线程中存放租户标识
public static void setTenant(String tenant) {
currentTenant.set(tenant);
}
//需要的时候,从当前线程取租户标识
public static String getTenant() {
return (String)currentTenant.get();
}
public static void clear() {
currentTenant.remove();
}
}
3、忽略租户的方式
方法一:查询数据库前设置租户标识
TenantContext.setTenant("租户标识")
方法二:在MybatisPlusConfig的paginationInterceptor()方法的doTableFilter中设置需要过滤的表名
@Override
public boolean doTableFilter(String tableName) {
// 过滤哪些表
List<String> tableNameList = Arrays.asList("user");
if (tableNameList.contains(tableName)){
return true;
}
return false;
}
方法三:在MybatisPlusConfig的paginationInterceptor()方法中设置setSqlParserFilter,过滤方法
//需要过滤某一个方法
paginationInterceptor.setSqlParserFilter(new ISqlParserFilter() {
@Override
public boolean doFilter(MetaObject metaObject) {
MappedStatement ms = SqlParserHelper.getMappedStatement(metaObject);
// 对应Mapper、dao中的方法
if("com.demo.mybatis.mapper.MyMapper.getOne".equals(ms.getId())){
return true;
}
return false;
}
});