mybatis-plus 3.4.2版本注解@TableName中动态schema实现

已于 2024-02-23 12:12:13 修改
阅读量3.7k 收藏 5
点赞数
文章标签： mybatis spring boot
于 2023-10-17 17:32:59 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/xie_apple/article/details/133888297
版权

记录springBoot、MybatisPlus框架，
应用场景：
项目数据库test_db和app_db使用同一个mysql实例，分不同的库(表模型)，为了降低复杂度，给db_data_link的用户赋予了db_ds数据库的只读权限，可以跨库查询

而注解@TableName里有属性schema，可以设置表空间，但是因为是写死，不够灵活，当测试环境和生产环境的表空间不一致，而对应的表结构一致，或者表空间修改时，那么就需要改动大批量的实体类代码，后期维护太过繁琐，增加工作量。

以下就是解决以上问题的方法：

当我们需要动态表模式名称获取在@TableName注解里的schema属性里使用的时候，
需要注意的是：我的mybatis-plus版本的是：3.4.2

<!-- mybatis-plus -->
<dependency>
    <groupId>com.baomidou</groupId>
    <artifactId>mybatis-plus-boot-starter</artifactId>
    <version>3.4.2</version>
</dependency>

因为该版本的表名生成拦截器不符合我们动态表空间动态替换，去获取.properties里配置的属性值。所以，我们需要重写DynamicTableNameInnerInterceptor类，根据实际需求，来动态获取配置文件的属性替换表空间。
DynamicTableNameInnerInterceptor类

package com.business.common.config;

import com.baomidou.mybatisplus.core.plugins.InterceptorIgnoreHelper;
import com.baomidou.mybatisplus.core.toolkit.PluginUtils;
import com.baomidou.mybatisplus.core.toolkit.TableNameParser;
import com.baomidou.mybatisplus.extension.plugins.handler.TableNameHandler;
import com.baomidou.mybatisplus.extension.plugins.inner.DynamicTableNameInnerInterceptor;
import com.business.common.util.StringHelper;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
import org.apache.ibatis.executor.Executor;
import org.apache.ibatis.executor.statement.StatementHandler;
import org.apache.ibatis.mapping.BoundSql;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.mapping.SqlCommandType;
import org.apache.ibatis.session.ResultHandler;
import org.apache.ibatis.session.RowBounds;

import java.sql.Connection;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

/**
 * 重写实现动态表空间的方法
 * @author XYP
 * @version 1.0
 * @date 2023/10/17
 */
@Data
@NoArgsConstructor
@AllArgsConstructor
@SuppressWarnings({"rawtypes"})
public class DynamicTableNameInterceptor extends DynamicTableNameInnerInterceptor {

    private Map<String, TableNameHandler> tableNameHandlerMap;

    @Override
    public void beforeQuery(Executor executor, MappedStatement ms, Object parameter, RowBounds rowBounds, ResultHandler resultHandler, BoundSql boundSql) throws SQLException {
        PluginUtils.MPBoundSql mpBs = PluginUtils.mpBoundSql(boundSql);
        if (InterceptorIgnoreHelper.willIgnoreDynamicTableName(ms.getId())) return;
        mpBs.sql(this.changeTable(mpBs.sql()));
    }

    @Override
    public void beforePrepare(StatementHandler sh, Connection connection, Integer transactionTimeout) {
        PluginUtils.MPStatementHandler mpSh = PluginUtils.mpStatementHandler(sh);
        MappedStatement ms = mpSh.mappedStatement();
        SqlCommandType sct = ms.getSqlCommandType();
        if (sct == SqlCommandType.INSERT || sct == SqlCommandType.UPDATE || sct == SqlCommandType.DELETE) {
            if (InterceptorIgnoreHelper.willIgnoreDynamicTableName(ms.getId())) return;
            PluginUtils.MPBoundSql mpBs = mpSh.mPBoundSql();
            mpBs.sql(this.changeTable(mpBs.sql()));
        }
    }

    protected String changeTable(String sql) {
        TableNameParser parser = new TableNameParser(sql);
        List<TableNameParser.SqlToken> names = new ArrayList<>();
        parser.accept(names::add);
        StringBuilder builder = new StringBuilder();
        int last = 0;
        for (TableNameParser.SqlToken name : names) {
            int start = name.getStart();
            if (start != last) {
                builder.append(sql, last, start);
                String value = name.getValue();
                String value2 = "";
                //含有"$"符号的，进行替换
                if (name.getValue().indexOf("$") >= 0){
                    int dNum = name.getValue().lastIndexOf(".");
                    value = name.getValue().substring(0,dNum);
                    if (dNum >= 0 && name.getValue().length() > dNum){
                        value2 = name.getValue().substring(dNum + 1);
                    }
                }
                TableNameHandler handler = tableNameHandlerMap.get(value);
                if (handler != null) {
                    builder.append(handler.dynamicTableName(sql, value) + (StringHelper.isBlank(value2)?"":("." + value2)));
                } else {
                    builder.append(value);
                }
            }
            last = name.getEnd();
        }
        if (last != sql.length()) {
            builder.append(sql.substring(last));
        }
        return builder.toString();
    }
}

接下来就是配置MybatisPlusConfig.java

package com.business.common.config;

import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor;
import com.baomidou.mybatisplus.extension.plugins.handler.TableNameHandler;
import com.baomidou.mybatisplus.extension.plugins.inner.PaginationInnerInterceptor;
import com.zaxxer.hikari.HikariDataSource;
import lombok.Data;
import org.mybatis.spring.annotation.MapperScan;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import java.util.HashMap;
import java.util.Map;

@Data
@Configuration
@MapperScan(basePackages = {"com.business.**.dao"})
public class MybatisPlusConfig {
    @Autowired
    HikariProperties hikariProperties;

    /**
     * 默认的数据源
     */
    private HikariDataSource dataSourceOral() {
        HikariDataSource dataSource = new HikariDataSource();
        hikariProperties.config(dataSource);
        return dataSource;
    }

    /**
     * 单数据源连接池配置
     */
    @Bean
    public HikariDataSource singleDatasource() {
        return dataSourceOral();
    }

    @Value("${test_db_schema}")
    private String test_db_schema;
    @Value("${app_db_schema}")
    private String app_db_schema;
    @Value("${sys_db_schema}")
    private String sys_db_schema;
    @Value("${log_db_schema}")
    private String log_db_schema;

    // 最新版
    @Bean
    public MybatisPlusInterceptor mybatisPlusInterceptor() {
        MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
        interceptor.addInnerInterceptor(new PaginationInnerInterceptor(DbType.POSTGRE_SQL));

        DynamicTableNameInterceptor dynamicTableNameInnerInterceptor = new DynamicTableNameInterceptor();
        Map<String,TableNameHandler> map = new HashMap<>();
        map.put("${test_db_schema}", new TableNameHandler() {
            @Override
            public String dynamicTableName(String sql, String tableName) {
                return test_db_schema;
            }
        });
        map.put("${app_db_schema}",new TableNameHandler() {
            @Override
            public String dynamicTableName(String sql, String tableName) {
                return app_db_schema;
            }
        });
        map.put("${sys_db_schema}",new TableNameHandler() {
            @Override
            public String dynamicTableName(String sql, String tableName) {
                return sys_db_schema;
            }
        });
        map.put("${log_db_schema}",new TableNameHandler() {
            @Override
            public String dynamicTableName(String sql, String tableName) {
                return log_db_schema;
            }
        });
        dynamicTableNameInnerInterceptor.setTableNameHandlerMap(map);
        interceptor.addInnerInterceptor(dynamicTableNameInnerInterceptor);
        return interceptor;
    }
}

那么.properties文件的配置如下：

test_db_schema=test_db
app_db_schema=app_db
sys_db_schema=sys_db
log_db_schema=log_db

如上配置完成后，那么实体类的使用方式如下：
在注解@TableName中，配置schema的属性值为："${test_db_schema}"

package com.business.demo.domain;

import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableName;
import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;

import java.util.List;

/**
 * @Author: xyp
 * @CreateTime: 2023/8/26
 */
@Data
@TableName(value = "t_demo",schema = "${test_db_schema}")
@ApiModel(value = "com.business.demo.domain.Demo", description = "名称参数")
public class Demo {
    /** 主键 */
    @TableId(value = "obj_id",type = IdType.ASSIGN_UUID)
    @ApiModelProperty(hidden = true)
    private String objId;

    @TableField(value = "texas_id")
    @ApiModelProperty(value = "texasId")
    private String texasId;

    /** 名称 */
    @TableField(value = "name")
    @ApiModelProperty(value = "名称")
    private String name;

    @TableField(exist = false)
    @ApiModelProperty(value = "详情数据")
    private List<DemoDetail> data;
}
xie_apple
关注
  • 0
    点赞
  • 踩
  • 5
    收藏
    觉得还不错? 一键收藏
  • 5
    评论
MybatisPlus(3.4.2)简易学习笔记
a_letter的博客
03-11 1358
文章目录1、快速开始2、CRUD操作2.1 添加操作 int insert(T entity);2.2 更新操作2.2.1 根据id更新2.2.2 根据条件更新2.3 删除操作2.3.1 int deleteById(Serializable id);2.3.2 int deleteByMap(@Param(Constants.COLUMN_MAP) Map1、快速开始 Springboot+Mybatis+MP ​ 1.创建工程 ​ 2.导入依赖：mysql驱动 mybatis-plus-boot
SpringBoot+Mybatis实现数据源动态切换
12-27
springboot实现数据源动态切换 注意事项： 1. 该demo采用yml配置数据库信息，注意url标签为jdbc-url 2.项目中加了日志输出，可看到完整执行过程 3.在Service中应用事务时，自定义的注解将失效，解决办法：可将注解配置到Controller方法中
mybatis-plus动态设置 @TableName的schema属性
lgq2016的博客
06-04 374
做ep登录的长时间未登录禁止功能，运维提供的开发环境和生产环境 schema名称不一样，需要基于mybatis-plus动态修改@TableName的schema属性。
mybatis-plus代码生成器、物理分页、动态表名、逻辑删除、自动填充等功能实现
ClearZero的博客
11-26 974
文章目录开篇代码生成器1、依赖2、配置3、生成的代码物理分页逻辑分页物理分页对比适用场景mybatis-plus物理分页插件示例代码动态表名配置逻辑删除配置自动填充mp实现其他使用 开篇 上一篇记录了mybatis-plus入门简单使用, 这一篇记录mybatis-plus代码生成器、动态表名、逻辑删除、自动填充、物理分页等高级功能在实际项目中的使用。初始依赖下面没有写, 在第一篇的基础上进行。(下述配置大部分都是根据官网来的, 依赖版本也是官网copy的, 但是可能由于官网未更新, 有一些过时的类, 但是
MybatisPlus设置动态表名
最新发布
cnzzs的博客
07-09 708
对于一些数据量比较大的表，为了提高查询性能，我们一般将表拆分成多张表，常见的是根据数据量，按年分表或者按月分表；分表虽然太高了查询性能，但是在查询的时候，如何才能查询执行分表数据呢，这里整理了一下，使用MybatisPlus设置动态表名的功能实现增加MyBatisPlus配置类登录后复制 import com.baomi...
mybatis-plus 基于 schema 多租户
qq_32948623的博客
12-20 5803
mybatis-plus 基于 schema 多租户
excle工具类
chizi7811的博客
05-27 95
链接：https://my.oschina.net/u/2968198/blog/1633657 <?xml version="1.0" encoding="UTF-8"?> <project xmlns="http://maven.apache.org/POM/4.0....
Java培训：mybatisplus快速实现动态数据源切换
zy1992As的博客
11-21 1166
或者不同的数据源混合使用。举个简单的例子某企业要做订单网上订单系统这里面就可以涉及到多个子系统的连接，比如：产品主数据的数据源，项目管理系统的数据源（项目可以产品订单）等多个不同数据库类似的数据源，他们可能是ORACLE，SQL SERVER，MYSQL等多种混合数据源。1.本框架只做**切换数据源**这件核心的事情，并**不限制你的具体操作**，切换了数据源可以做任何CRUD。-支持**多层数据源嵌套切换**。**DS**可以注解在方法上或类上，**同时存在就近原则方法上注解优先于类上注解**。
Mybatis -plus 快速入门 修复了图片不正常显示,重写代码构造模板新版,更新了一些问题和新增了代码生成器的内容
weixin_50224527的博客
04-20 269
Mybatis -plus 简介 特性 无侵入：只做增强不做改变，引入它不会对现有工程产生影响，如丝般顺滑 损耗小：启动即会自动注入基本 CURD，性能基本无损耗，直接面向对象操作 强大的 CRUD 操作：内置通用 Mapper、通用 Service，仅仅通过少量配置即可实现单表大部分 CRUD 操作，更有强大的条件构造器，满足各类使用需求 支持 Lambda 形式调用：通过 Lambda 表达式，方便的编写各类查询条件，无需再担心字段写错 支持主键自动生成：支持多达 4 种主键策略（内含分布式唯一
mybatisplus动态schema配置
lengxia0的博客
04-21 3290
查询测试环境数据库与生产环境数据库时，sql中的schema一致。
mybatis-plus (3.4.2)使用
xiuyuandashen的博客
02-09 1765
快速入门 官方文档快速入门案例 配置日志 # 配置日志 mybatis-plus: configuration: # 配置 mybatis-plus执行的日志类型(可以看到执行过程) 下面是使用了控制台输出 sl4j log4j 等等都可以 log-impl: org.apache.ibatis.logging.stdout.StdOutImpl 效果 CRUD扩展 数据库中未指定id自增方式 我们可以通过mybatis-plus 来生成 插入一条数据 User user
MyBatis-Plus实现动态表名
D_A_I_H_A_O的博客
05-20 4031
1、添加MyBatis-Plus依赖；2、添加MyBatis-Plus配置，利用拦截器获取到表名给替换；3、创建请求参数传递辅助类；4、在程序中使用，注意如果实际表名与实体类与不同，可先在实体类类注明表名@TableName("TestUser")
mybatis-3.4.2.jar
08-28
mybatis-3.4.2.jar
MyBatis-PlusDemo
10-01
Maven-MyBatis_PLUS+Spring
MyBatis-Plus入门+MyBatis-Plus文档手册 中文pdf高清版.rar
11-09
mybatis在持久层框架中还是比较火的，一般项目都是基于ssm。虽然mybatis可以直接在xml中...《MyBatis-Plus入门文档》主要介绍了MyBatis-Plus入门使用，以及关于mybatis-plus的更多介绍及特性，感兴趣的可以下载学习一下
mybatis-plus使用@EnumValue处理枚举类型的示例代码
09-07
在MyBatis-Plus中，`@EnumValue`是一个用于处理枚举类型的注解，它使得枚举可以与数据库中的字段进行映射。这个注解主要用于解决枚举类型在序列化和反序列化过程中的问题，确保数据在数据库和Java对象之间能够正确地...
mybatis-plus源码（mybatis-plus-3.5.1.zip）
03-03
此源码包`mybatis-plus-3.5.1.zip`包含了MyBatis-Plus 3.5.1版本的全部源代码，有助于我们深入理解其内部实现机制和工作原理。 MyBatis-Plus的主要功能包括： 1. **CRUD操作**：MyBatis-Plus提供了简化版的CRUD...
Mybatis-Plus中乐观锁@version注解的问题与解决方案.docx
10-26
Mybatis-Plus 中乐观锁 @version 注解的问题与解决方案 Mybatis-Plus 是一个基于 Mybatis 的增强工具，它提供了许多实用的功能来简化数据库交互操作。在实际开发中，我们经常会遇到乐观锁的问题，特别是在使用 @...
mybatis-plus-boot-starter-3.1.0.jar
04-13
mybatis-plus-boot-starter.jar 各个版本下载， SpringBoot 集成 MybatisPlus jar 包下载， Mybatis-Plus（简称MP）是一个基于MyBatis的增强工具库，它简化了与数据库的交互操作并提供了一系列增强功能，使开发者...
mybatis-plus updatebyid前的@路径怎么调用
05-30
在MyBatis-Plus中，`@`符号被用来表示注解，而不是路径。如果您想要使用MyBatis-Plus的`updateById`方法来更新实体对象的数据，您需要按照以下步骤进行操作： 1. 确保您的实体类继承了MyBatis-Plus提供的`BaseEntity`类，这个类中包含了一些通用的字段和方法，例如`id`字段和`updateTime`字段。 2. 在您的Mapper接口中定义一个`updateById`方法，例如： ``` @Mapper public interface UserMapper extends BaseMapper<User> { int updateById(User user); } ``` 3. 在您的Service类中注入该Mapper，并调用`updateById`方法。例如： ``` @Service public class UserServiceImpl implements UserService { @Autowired private UserMapper userMapper; @Override public boolean updateUser(User user) { int result = userMapper.updateById(user); return result > 0; } } ``` 在以上代码中，我们注入了`UserMapper`接口，并在`updateUser`方法中调用了`updateById`方法来更新用户信息。注意，`updateById`方法的参数是一个实体对象，该实体对象的`id`字段必须设置为要更新的记录的ID值。
评论 5
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值