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();
    }

}