mybatis-plus?實(shí)現(xiàn)查詢表名動(dòng)態(tài)修改的示例代碼

更新時(shí)間：2025年03月17日 10:31:45 作者：foolishflyfox

通過MyBatis-Plus實(shí)現(xiàn)表名的動(dòng)態(tài)替換,根據(jù)配置或入?yún)⑦x擇不同的表,本文主要介紹了mybatis-plus?實(shí)現(xiàn)查詢表名動(dòng)態(tài)修改的示例代碼,具有一定的參考價(jià)值,感興趣的可以了解一下

-- 學(xué)科表
drop table if exists subject;
create table subject(id int primary key , name varchar(64));

-- 學(xué)生表
drop table if exists student;
create table student (id int primary key , name varchar(64));

-- 成績表(學(xué)生-學(xué)科 多對多)
drop table if exists score;
create table score(id int primary key , student_id int, subject_id int, result int);

根據(jù)三張表級聯(lián)查詢成績的查詢語句為:

select subject.name as subject_name, student.name as student_name, score.result as score
    from score, student, subject where score.student_id=student.id and score.subject_id=subject.id;

現(xiàn)在又來了一個(gè)新需求，我們的這套成績查詢系統(tǒng)需要部署在不同學(xué)校的服務(wù)器上，因?yàn)槊總€(gè)學(xué)校的學(xué)生表和成績表都要同步到教育局的服務(wù)器中，因此需要為這兩個(gè)表添加學(xué)校前綴，ER 圖如下所示。

不同學(xué)校的建表語句不同，對于 USTC 學(xué)校而言，建表語句為:

-- 學(xué)科表
drop table if exists subject;
create table subject(id int primary key , name varchar(64));

-- 學(xué)生表
drop table if exists ustc_student;
create table ustc_student (id int primary key , name varchar(64));

-- 成績表(學(xué)生-學(xué)科 多對多)
drop table if exists ustc_score;
create table ustc_score(id int primary key , student_id int, subject_id int, result int);

對于 ZJU 學(xué)校而言，建表語句為:

-- 學(xué)科表
drop table if exists subject;
create table subject(id int primary key , name varchar(64));

-- 學(xué)生表
drop table if exists zju_student;
create table zju_student (id int primary key , name varchar(64));

-- 成績表(學(xué)生-學(xué)科 多對多)
drop table if exists zju_score;
create table zju_score(id int primary key , student_id int, subject_id int, result int);

我們的成績查詢系統(tǒng)會(huì)安裝在不同的學(xué)校，并且校名是動(dòng)態(tài)可配的，因此該成績查詢系統(tǒng)需要根據(jù)配置文件動(dòng)態(tài)修改 sql 語句表名的功能。

實(shí)現(xiàn)

源碼地址: https://github.com/foolishflyfox/blog/tree/main/backend-code/mp-dynamic-tablename

數(shù)據(jù)庫初始化

創(chuàng)建程序需要的表:

create database if not exists mp_dynamic_tablename_test;

use mp_dynamic_tablename_test;

-- 學(xué)科表
drop table if exists subject;
create table subject(id int primary key , name varchar(64));
insert into subject(id, name) values (1, 'Math'), (2, 'English'), (3, 'Chinese');

-- 學(xué)生表
drop table if exists student;
create table student (id int primary key , name varchar(64));
insert into student(id, name) values(1, 'aaa'), (2, 'bbb');

-- 成績表(學(xué)生-學(xué)科 多對多)
drop table if exists score;
create table score(id int primary key , student_id int, subject_id int, result int);
insert into score(id, student_id, subject_id, result) values (1, 1, 1, 74), (2, 1, 2, 83), (3, 1, 3, 69),
    (4, 2, 1, 91), (5, 2, 3, 87);

-- 指定前綴 ustc 的表
-- 學(xué)生表
drop table if exists ustc_student;
create table ustc_student (id int primary key , name varchar(64));
insert into ustc_student(id, name) values(1, 'u_aaa'), (2, 'u_bbb');

-- 成績表(學(xué)生-學(xué)科 多對多)
drop table if exists ustc_score;
create table ustc_score(id int primary key , student_id int, subject_id int, result int);
insert into ustc_score(id, student_id, subject_id, result) values (1, 1, 1, 89), (2, 1, 2, 81), (3, 1, 3, 32),
                                                             (4, 2, 1, 71), (5, 2, 2, 77);

-- 指定前綴 zju 的表
-- 學(xué)生表
drop table if exists zju_student;
create table zju_student (id int primary key , name varchar(64));
insert into zju_student(id, name) values(5, 'z_aaa'), (6, 'z_bbb');

-- 成績表(學(xué)生-學(xué)科 多對多)
drop table if exists zju_score;
create table zju_score(id int primary key , student_id int, subject_id int, result int);
insert into zju_score(id, student_id, subject_id, result) values (1, 5, 1, 91), (2, 5, 2, 66), (3, 5, 3, 85),
                                                                  (4, 6, 1, 48), (5, 6, 2, 59);

依賴包

需要引入 spring-boot-starter-web、spring-boot-starter-test、spring-boot-configuration-processor、mybatis-plus-boot-starter、mysql-connector-java、lombok 庫。

配置讀取類

我們先定義一個(gè)配置讀取類，用于獲取動(dòng)態(tài)配置的學(xué)校以及需要?jiǎng)討B(tài)添加學(xué)校前綴的表名。

package cn.fff.config.properties;

import lombok.Data;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.stereotype.Component;

import java.util.HashSet;
import java.util.Set;

@Component
@ConfigurationProperties("school")
@Data
public class SchoolProperties {
    /** 學(xué)校名，動(dòng)態(tài)表名會(huì)添加前綴: 學(xué)校名_ */
    private String name;
    /** 需要?jiǎng)討B(tài)添加前綴的表 */
    private Set<String> dynamicTables = new HashSet<>();
}

為 application.yml 添加如下配置:

school:
  name: ustc
  dynamic-tables:
    - student
    - score

表示需要為 student 和 score 動(dòng)態(tài)添加前綴 ustc，即查詢 student 表時(shí)會(huì)動(dòng)態(tài)替換為 ustc_student，查詢 score 表時(shí)會(huì)動(dòng)態(tài)替換為 ustc_score。如果 school.name 修改為 zju，則查詢 student 表時(shí)會(huì)動(dòng)態(tài)替換為 zju_student，查詢 score 表時(shí)會(huì)動(dòng)態(tài)替換為 zju_score。

設(shè)置 mybatis-plus 插件

實(shí)體類、mapper、服務(wù)類的創(chuàng)建比較基礎(chǔ)，此處略過，可直接查看源碼。動(dòng)態(tài)表面主要通過創(chuàng)建一個(gè) mybatis 插件實(shí)現(xiàn):

package cn.fff.config.mp;

import cn.fff.config.properties.SchoolProperties;
import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor;
import com.baomidou.mybatisplus.extension.plugins.inner.DynamicTableNameInnerInterceptor;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.util.StringUtils;

@Configuration
public class DynamicTableNameConfig {
    @Autowired
    private SchoolProperties schoolProperties;

    @Bean
    public MybatisPlusInterceptor mybatisPlusInterceptor() {
        MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
        interceptor.addInnerInterceptor(dynamicTableNameInnerInterceptor());
        return interceptor;
    }

    private DynamicTableNameInnerInterceptor dynamicTableNameInnerInterceptor() {
        DynamicTableNameInnerInterceptor innerInterceptor = new DynamicTableNameInnerInterceptor();
        innerInterceptor.setTableNameHandler((sql, tableName) -> {
            String newTableName = tableName;
            // 配置了學(xué)校名并且當(dāng)前查詢的表名在指定配置中，則添加表名前綴
            if (StringUtils.hasLength(schoolProperties.getName())
                    && schoolProperties.getDynamicTables().contains(tableName)) {
                newTableName = schoolProperties.getName() + "_" + tableName;
            }
            return newTableName;
        });

        return innerInterceptor;
    }
}

測試

在 test 中創(chuàng)建一個(gè)測試類 ScoreServiceTest :

package cn.fff;

import cn.fff.entity.StudentScore;
import cn.fff.service.ScoreService;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;

import java.util.List;

@SpringBootTest
public class ScoreServiceTest {

    @Autowired
    private ScoreService scoreService;

    @Test
    public void testDynameTableName() {
        List<StudentScore> studentScores = scoreService.queryStudentScore();
        studentScores.forEach(e -> System.out.printf("%s %s %d\n", e.getStudentName(), e.getSubjectName(), e.getScore()));
    }
}

當(dāng) school.name 為 ustc 時(shí)，輸出為：

u_aaa Math 89
u_aaa English 81
u_aaa Chinese 32
u_bbb Math 71
u_bbb English 77

當(dāng) school.name 為 zju 時(shí)，輸出為:

z_aaa Math 91
z_aaa English 66
z_aaa Chinese 85
z_bbb Math 48
z_bbb English 59

這樣我們就實(shí)現(xiàn)了根據(jù)配置動(dòng)態(tài)切換操作表名的功能。

到此這篇關(guān)于mybatis-plus 實(shí)現(xiàn)查詢表名動(dòng)態(tài)修改的示例代碼的文章就介紹到這了,更多相關(guān)mybatis-plus 查詢表名動(dòng)態(tài)修改內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: