//package com.gkhy.hazmat.common.config;
|
//
|
//import com.baomidou.mybatisplus.extension.plugins.handler.TableNameHandler;
|
//import com.gkhy.hazmat.common.utils.StringUtils;
|
//
|
//import java.util.HashMap;
|
//import java.util.Map;
|
//
|
///**
|
// * 自定义动态表名处理器，基于id的方式进行分表
|
// */
|
//public class IdTableNameHandler implements TableNameHandler {
|
// private static final Map<String,Integer> CONFIG_TABLE_INFO_MAP=new HashMap<>();
|
//
|
// static {
|
// CONFIG_TABLE_INFO_MAP.put("hz_hazmat",20);
|
// CONFIG_TABLE_INFO_MAP.put("hz_hazmat_flow",20);
|
// CONFIG_TABLE_INFO_MAP.put("hz_product",20);
|
// CONFIG_TABLE_INFO_MAP.put("hz_product_flow",20);
|
// }
|
//
|
// private static final ThreadLocal<Long> ID_DATA=new ThreadLocal<>();
|
// @Override
|
// public String dynamicTableName(String sql, String tableName) {
|
// if(StringUtils.isBlank(tableName)||!CONFIG_TABLE_INFO_MAP.containsKey(tableName)){
|
// return tableName;
|
// }
|
// //分表个数
|
// int tableSize=CONFIG_TABLE_INFO_MAP.get(tableName);
|
// //当前分表的id
|
// Long currentId=getCurrentId();
|
// if(currentId==null){
|
// throw new RuntimeException("请设置分表id值");
|
// }
|
// int tableIndex= (int) (currentId%tableSize);
|
//
|
// //清理当前id
|
// // removeCurrentId();
|
//
|
// return tableName+"_"+tableIndex;
|
// }
|
//
|
// /**
|
// * 获取当前分表id
|
// * @param id
|
// */
|
// public static void setCurrentId(Long id){
|
// ID_DATA.set(id);
|
// }
|
//
|
// public static Long getCurrentId(){
|
// return ID_DATA.get();
|
// }
|
//
|
// public static void removeCurrentId(){
|
// ID_DATA.remove();
|
// }
|
//
|
//}
|
