import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.extension.parsers.DynamicTableNameParser;
import com.baomidou.mybatisplus.extension.parsers.ITableNameHandler;
import com.baomidou.mybatisplus.extension.plugins.PaginationInterceptor;
import org.apache.ibatis.reflection.MetaObject;
import org.mybatis.spring.annotation.MapperScan;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import java.util.Collections;
import java.util.HashMap;
import java.util.List;

/**
 * @Author: xuzz
 * @Description:
 * @Date: Created in 上午11:22 19-10-12
 * @Modified By:
 */
@Configuration
@MapperScan("com.dao")
public class MPConfig {

  @Autowired
  private Tables tableNames;

  /**
   * @Author: xuzz
   * @Date: 上午11:46 19-10-12
   * 動態(tài)表名
   */
  @Bean
  public PaginationInterceptor paginationInterceptor(){
    PaginationInterceptor paginationInterceptor = new PaginationInterceptor();
    DynamicTableNameParser dynamicTableNameParser = new DynamicTableNameParser();
    dynamicTableNameParser.setTableNameHandlerMap(new HashMap<String, ITableNameHandler>(2){{
      //涉及表集合
      List<String> tables = tableNames.getTableNames();
      //動態(tài)表規(guī)則 初始表名+_+code
      tables.forEach(tableTitle -> put(tableTitle,(metaObject, sql, tableName) -> tableName + "_" + String.valueOf(getParamValue("關(guān)鍵字段名",metaObject))));
    }});
    paginationInterceptor.setSqlParserList(Collections.singletonList(dynamicTableNameParser));
    return paginationInterceptor;
  }

  /**
   * @Author: xuzz
   * @Date: 下午2:55 19-10-12
   * 獲取參數(shù)值
   */
  private Object getParamValue(String title, MetaObject metaObject){
    //獲取參數(shù)
    Object originalObject = metaObject.getOriginalObject();
    JSONObject originalObjectJSON = JSON.parseObject(JSON.toJSONString(originalObject));
    JSONObject boundSql = originalObjectJSON.getJSONObject("boundSql");
    JSONObject parameterObject = boundSql.getJSONObject("parameterObject");
     return parameterObject.get(title);
  }

@Configuration
@ConfigurationProperties("mp")
public class Tables {

  private List<String> tableNames;

  public List<String> getTableNames() {
    return tableNames;
  }

  public void setTableNames(List<String> tableNames) {
    this.tableNames = tableNames;
  }
}

mp:
  tableNames: #表頭列表
   - table1
   - table2
   - table3