Mybatis-Plus实现动态表名sql解析器实战

Mybatis-Plus实现动态表名sql解析器实战

Mybatis-Plus实现动态表名sql解析器

动态表名sql解析
添加依赖

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

<dependency>
    <groupId>mysql</groupId>
    <artifactId>mysql-connector-java</artifactId>
</dependency>

配置yml

# Mybatis配置
mybatis-plus:
  global-config:
    db-config:
      logic-delete-field: deleted
      logic-delete-value: 1
      logic-not-delete-value: 0
      update-strategy: ignored
  mapper-locations: classpath*:mapper/**/*Mapper.xml
  configuration:
    cache-enabled: true
    use-generated-keys: true
    log-impl: org.apache.ibatis.logging.slf4j.Slf4jImpl
    default-executor-type: SIMPLE
    map-underscore-to-camel-case: true
  type-aliases-package:《包名》

编写sql解析器

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.DynamicTableNameInnerInterceptor;
import com.baomidou.mybatisplus.extension.plugins.inner.PaginationInnerInterceptor;
import com.qiangshi.common.utils.StringUtils;
import org.mybatis.spring.annotation.MapperScan;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import java.util.Objects;

/**
 * Mybatis-plus 配置类
 *
 * @author Nottyjay
 * @since 2020-10-10
 */
@Configuration
@MapperScan("com.dome.*.mapper")
public class MybatisConfig {


    public static ThreadLocal<String> TABLE_NAME = new ThreadLocal<String>();

    /**
     * 新的分页插件,一缓和二缓遵循mybatis的规则,需要设置 MybatisConfiguration#useDeprecatedExecutor = false 避免缓存出现问题(该属性会在旧插件移除后一同移除)
     */
    @Bean
    public MybatisPlusInterceptor mybatisPlusInterceptor() {
        MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
        PaginationInnerInterceptor paginationInnerInterceptor = new PaginationInnerInterceptor(DbType.MYSQL);
        DynamicTableNameInnerInterceptor dynamicTableNameInnerInterceptor = null;
        TableNameHandler tableNameHandler = null;
        interceptor.addInnerInterceptor(paginationInnerInterceptor);
        dynamicTableNameInnerInterceptor = new DynamicTableNameInnerInterceptor();
        tableNameHandler = new TableNameHandler() {
            @Override
            public String dynamicTableName(String sql, String tableName) {
                if (StringUtils.isNotEmpty(TABLE_NAME.get())) {
                    return TABLE_NAME.get();
                } else {
                    return tableName;
                }
            }
        };
        dynamicTableNameInnerInterceptor.setTableNameHandler(tableNameHandler);
        interceptor.addInnerInterceptor(dynamicTableNameInnerInterceptor);
        return interceptor;
    }


}

测试

package com.demo.studynew;

import com.demo.studynew.entity.SysTest;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;

import java.time.LocalDateTime;
import java.util.List;


@RestController
@Api(tags = "动态表名测试")
public class SysTestController {

    @Autowired
    private SysTestMapper sysTestMapper;

    @GetMapping("list")
    @ApiOperation("列表")
    public List<SysTest> list() {
        MybatisConfig .TABLE_NAME.set("sys_test_2020");
        return sysTestMapper.selectList(null);
    }

    @GetMapping("add")
    @ApiOperation("新增")
    public String add(){
        MybatisConfig .TABLE_NAME.set("sys_test_2020");
        SysTest sysTest = new SysTest();
        sysTest.setName("test1");
        sysTest.setGmtCreate(LocalDateTime.now());
        sysTestMapper.insert(sysTest);
        return "ok";
    }
}

至此动态表名sql解析完成。