MyBatis-Plus动态表名的使用

2023-05-14 20:05:32 mybatis

MyBatis-Plus实现动态表名

MyBatis实现方法如下现在要用MyBatis-Plus 实现

   <select id="getList" resultType="com.wys.entity.User">
        SELECT *
        FROM ${tableName}
    </select>

MyBatis-Plus官网说明

MyBatis-Plus版本

1、添加MyBatis-Plus依赖

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

MyBatis-Plus配置

2、添加MyBatis-Plus配置,利用拦截器获取到表名给替换

@Configuration
public class MybatisPlusConfig {
    static List<String> tableList(){
        List<String> tables = new ArrayList<>();
        //伪表名  可以为任意字符串 建议设置复杂度 避免重复    tables.add("C55EA8171877E962E08DFF63AA3678841");
        tables.add("TestUser");
        return tables;
    }

	//拦截器,获取到表名给替换
    @Bean
    public MybatisPlusInterceptor dynamicTableNameInnerInterceptor() {
        MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
        DynamicTableNameInnerInterceptor dynamicTableNameInnerInterceptor = new DynamicTableNameInnerInterceptor();
        dynamicTableNameInnerInterceptor.setTableNameHandler((sql, tableName) -> {
            String newTable = null;
            for (String table : tableList()) {
                newTable = RequestDataHelper.getRequestData(table);
                if (table.equals(tableName) && newTable!=null){
                    tableName = newTable;
                    break;
                }
            }
            return tableName;
        });
        interceptor.addInnerInterceptor(dynamicTableNameInnerInterceptor);
        return interceptor;
    }
}

如果上面的拦截器不生效可以使用下面这个https://www.jb51.net/article/280321.htm

@Configuration
@AutoConfigureAfter(PageHelperAutoConfiguration.class)
public class MybatisPlusConfig {
    static List<String> tableList(){
        List<String> tables = new ArrayList<>();
        //表名
        tables.add("C55EA8171877E962E08DFF63AA3678841");
        return tables;
    }

    //拦截器,获取到表名给替换
//    @Bean
//    public MybatisPlusInterceptor dynamicTableNameInnerInterceptor() {
//        MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
//        DynamicTableNameInnerInterceptor dynamicTableNameInnerInterceptor = new DynamicTableNameInnerInterceptor();
//        dynamicTableNameInnerInterceptor.setTableNameHandler((sql, tableName) -> {
//            String newTable = null;
//            for (String table : tableList()) {
//                newTable = RequestDataHelper.getRequestData(table);
//                if (table.equals(tableName) && newTable!=null){
//                    tableName = newTable;
//                    break;
//                }
//            }
//            return tableName;
//        });
//        interceptor.addInnerInterceptor(dynamicTableNameInnerInterceptor);
//        return interceptor;
//    }

    @Autowired
    private List<SqlSessionFactory> sqlSessionFactoryList;

    @PostConstruct
    public void addMyInterceptor() {
        MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
        DynamicTableNameInnerInterceptor dynamicTableNameInnerInterceptor = new DynamicTableNameInnerInterceptor();
        dynamicTableNameInnerInterceptor.setTableNameHandler((sql, tableName) -> {
            String newTable = null;
            for (String table : tableList()) {
                newTable = RequestDataHelper.getRequestData(table);
                if (table.equals(tableName) && newTable!=null){
                    tableName = newTable;
                    break;
                }
            }
            return tableName;
        });
        interceptor.addInnerInterceptor(dynamicTableNameInnerInterceptor);
        for (SqlSessionFactory sqlSessionFactory : sqlSessionFactoryList) {
            sqlSessionFactory.getConfiguration().addInterceptor(interceptor);
        }
    }
}

请求参数传递辅助类

3、创建请求参数传递辅助类

public class RequestDataHelper {
    
    private static final ThreadLocal<Map<String, Object>> REQUEST_DATA = new ThreadLocal<>();

    
    public static void setRequestData(Map<String, Object> requestData) {
        REQUEST_DATA.set(requestData);
    }

    
    public static <T> T getRequestData(String param) {
        Map<String, Object> dataMap = getRequestData();
        if (CollectionUtils.isNotEmpty(dataMap)) {
            return (T) dataMap.get(param);
        }
        return null;
    }

    
    public static Map<String, Object> getRequestData() {
        return REQUEST_DATA.get();
    }
}

使用

4、在程序中使用,注意如果实际表名与实体类与不同,可先在实体类类注明表名@TableName(“TestUser”)

@GetMapping("/listUser")
    public void listUser(){

        RequestDataHelper.setRequestData(new HashMap<String, Object>() {{
            put("kfafkasfaskfasjfkasf", "user_2018");

        }});
        Integer age=2018;

       User user=new User();
        List list = userMapper.getList(user);
     //  User user_2019 = userMapper.findById("user_2019", 2019);

        System.out.println(list);
        System.out.println("-------------");
       // System.out.println(user_2019);
        RequestDataHelper.setRequestData(new HashMap<String, Object>() {{
            put("kfafkasfaskfasjfkasf", "user_2019");

        }});
        List lis2 = userMapper.getList(user);
        System.out.println(lis2);
        System.out.println("-------------");

    }

结果:

相关文章