基于MyBatis插件实现动态表名解决多环境单一数据库问题

业务场景

在为某新能源汽车厂商进行我司系统私有化部署时,在预演环境和生产环境中,客户仅提供了一个 MySQL 数据库实例。为了确保数据隔离并避免不同环境之间的数据冲突,常规做法是为每个环境创建独立的表(如通过添加环境前缀或后缀)。

然而,如果每次切换环境都需要手动修改 SQL 或配置文件中的表名,不仅效率低下,而且容易出错。为此,小编利用 MyBatis 插件机制 实现了动态表名替换,从而优雅地解决了多环境共用数据库的问题。

解决方案

java 复制代码
/**
 * 动态表名工具参数
 *
 * @author 言安
 * @date 2025/3/20 14:37
 */
@Data
@Component
@ConfigurationProperties(prefix = SpongeDynamicTableNameProperties.PREFIX)
public class SpongeDynamicTableNameProperties {

    public static final String PREFIX = "sponge.dynamic";

    /**
     * 是否开启
     */
    private Boolean enable = Boolean.FALSE;
    /**
     * 默认表名后缀
     */
    private String defaultTableNameSuffix;
    /**
     * 表名后缀映射(key:表名, value: 后缀)
     * 优先取表名后缀映射,取不到取 defaultTableNameSuffix
     */
    private Map<String, String> tableNameSuffixMap;
}
java 复制代码
/**
 * 动态表名处理器
 *
 * @author 言安
 * @date 2025/3/20 13:41
 */
@Component
public class SpongeDynamicTableNameHandler implements TableNameHandler {


    // 默认后缀
    private final static String DEFAULT_SUFFIX = "_test";

    @Resource
    private SpongeDynamicTableNameProperties spongeDynamicTableNameProperties;

    @Override
    public String dynamicTableName(String sql, String tableName) {
        String suffix = DEFAULT_SUFFIX;
        if (spongeDynamicTableNameProperties != null) {
            if (spongeDynamicTableNameProperties.getDefaultTableNameSuffix() != null) {
                suffix = spongeDynamicTableNameProperties.getDefaultTableNameSuffix();
            }
            if (spongeDynamicTableNameProperties.getTableNameSuffixMap() != null && spongeDynamicTableNameProperties.getTableNameSuffixMap().containsKey(tableName)) {
                suffix = spongeDynamicTableNameProperties.getTableNameSuffixMap().get(tableName);
            }
        }
        return StringUtil.isBlank(suffix) ? tableName : tableName + suffix;
    }
}
java 复制代码
/**
 * MybatisPlus配置类
 *
 * @author 言安
 * @date 2025/3/20 13:41
 */
@Configuration
@EnableTransactionManagement
public class MybatisPlusConfiguration {

    @Resource
    private SpongeDynamicTableNameProperties spongeDynamicTableNameProperties;

    @Resource
    private SpongeDynamicTableNameHandler myTableNameHandler;

    @Bean
    public MybatisPlusInterceptor mybatisPlusInterceptor() {
        MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
        if (spongeDynamicTableNameProperties != null && Boolean.TRUE.equals(spongeDynamicTableNameProperties.getEnable())) {
            // 动态表名插件
            DynamicTableNameInnerInterceptor dynamicTableNameInnerInterceptor = new DynamicTableNameInnerInterceptor();
            dynamicTableNameInnerInterceptor.setTableNameHandler(myTableNameHandler);
            interceptor.addInnerInterceptor(dynamicTableNameInnerInterceptor);
        }
        interceptor.addInnerInterceptor(new PaginationInnerInterceptor(DbType.MYSQL));
        return interceptor;
    }
}
相关推荐
m0_7202450120 分钟前
QT(四)基本组件
数据库·qt·microsoft
Databend36 分钟前
使用 Databend Cloud 归档 OceanBase 数据数据库
数据库
fei飛fei飞1 小时前
数据库事务中的陷阱:脏读、幻读与不可重复读
数据库
FINE!(正在努力!)1 小时前
关于sql面试积累
数据库·sql
看天走路吃雪糕1 小时前
墨者:SQL过滤字符后手工绕过漏洞测试(万能口令)
数据库·sql·sql注入·墨者学院·万能口令
阿里云大数据AI技术2 小时前
【跨国数仓迁移最佳实践3】资源消耗减少50%!解析跨国数仓迁移至MaxCompute背后的性能优化技术
数据库·数据分析·云计算
GBASE2 小时前
“G”术时刻:如何用Perl DBD-ODBC成功连接南大通用GBase 8a数据库(一)
数据库
Yu_Lijing2 小时前
MySQL进阶学习与初阶复习第二天
数据库·c++·学习·mysql
孫治AllenSun2 小时前
【JSqlParser】sql解析器使用案例
数据库·windows·sql
Vinkey_Z3 小时前
MongoDB
数据库