/**
 * 配置的動態(tài)表名白名單
 *
 * @author 老馬
 */
@Data
@ConfigurationProperties(prefix = "ums.database.dynamic-table")
public class DynamicTableProperties {

    /**
     * 允許使用動態(tài)表名的表（邏輯表名）
     */
    private Set<String> tables = new HashSet<>();
}

ums:
  database:
    dynamic-table:
      tables:
        - t_xx_order
        - t_xx_sn

/**
 * 動態(tài)表名白名單
 *
 * @author 老馬
 */
public class DynamicTables {

    private static Set<String> TABLES = Collections.emptySet();

    private DynamicTables() {
    }

    static void init(Set<String> tables) {
        // 創(chuàng)建不可更改的Set
        TABLES = Collections.unmodifiableSet(tables);
    }

    /**
     * 是否允許使用動態(tài)表名
     */
    public static boolean isDynamic(String tableName) {
        if (!StringUtils.hasText(tableName)) {
            return false;
        }
        return TABLES.stream().anyMatch(tableName::startsWith);
    }
}

/**
 * mybatis-plus配置類
 *
 * @author 老馬
 */
@Configuration
@RequiredArgsConstructor
@EnableConfigurationProperties(DynamicTableProperties.class)
public class MybatisPlusConfig {

    private final DynamicTableProperties dynamicTableProperties;

    @Bean
    public MybatisPlusInterceptor mybatisPlusInterceptor() {
        MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
        // 動態(tài)表名插件
        interceptor.addInnerInterceptor(dynamicTableNameInnerInterceptor());
        // 其他插件
        return interceptor;
    }

    /**
     * 動態(tài)表名插件
     */
    private DynamicTableNameInnerInterceptor dynamicTableNameInnerInterceptor() {
        TableNameHandler tableNameHandler = (sql, tableName) -> {
            // 不在白名單，直接返回原表名
            if (!DynamicTables.isDynamic(tableName)) {
                return tableName;
            }
            // 取當前線程綁定的動態(tài)表名
            String dynamicTableName = DynamicTableNameHelper.get();

            //  沒有設(shè)置動態(tài)表名，兜底返回原表名
            return StringUtils.hasText(dynamicTableName)
                    ? dynamicTableName
                    : tableName;
        };
        return new DynamicTableNameInnerInterceptor(tableNameHandler);

    }

    /**
     * 初始化動態(tài)表名白名單
     */
    @PostConstruct
    public void initDynamicTables() {
        DynamicTables.init(dynamicTableProperties.getTables());
    }
}

/**
 * 動態(tài)表名輔助類
 *
 * @author 銀商北分-老馬
 * @since 1.0.0
 */
public final class DynamicTableNameHelper {

    /**
     * 表名正則
     */
    public static final String TABLE_NAME_REGEX = "[a-zA-Z0-9_]+";

    private static final ThreadLocal<String> HOLDER = new ThreadLocal<>();

    private DynamicTableNameHelper() {}

    /**
     * 使用動態(tài)表名
     *
     * @param tableName 表名
     * @return 作用域
     */
    public static Scope use(String tableName) {
        validate(tableName);
        if (!DynamicTables.isDynamic(tableName)) {
            throw new RuntimeException("表 [" + tableName + "] 未配置為允許動態(tài)表名");
        }
        String old = HOLDER.get();
        HOLDER.set(tableName);
        return () -> {
            if (old == null) {
                HOLDER.remove();
            } else {
                HOLDER.set(old);
            }
        };
    }

    /**
     * 在指定的動態(tài)表名作用域內(nèi)執(zhí)行操作
     *
     * @param table    表名
     * @param supplier 執(zhí)行邏輯
     * @param <T>      返回值類型
     * @return 返回值
     */
    public static <T> T withTable(String table, Supplier<T> supplier) {
        try (Scope ignored = use(table)) {
            return supplier.get();
        }
    }

    /**
     * 在指定的動態(tài)表名作用域內(nèi)執(zhí)行操作（無返回值）
     *
     * @param table    表名
     * @param runnable 執(zhí)行邏輯
     */
    public static void withTable(String table, Runnable runnable) {
        try (Scope ignored = use(table)) {
            runnable.run();
        }
    }

    /**
     * 獲取當前作用域的表名
     *
     * @return 表名
     */
    public static String get() {
        return HOLDER.get();
    }

    private static void validate(String tableName) {
        if (tableName == null || tableName.isBlank()) {
            throw new RuntimeException("tableName 不能為空");
        }
        if (!tableName.matches(TABLE_NAME_REGEX)) {
            throw new RuntimeException("非法表名：" + tableName);
        }
    }

    @FunctionalInterface
    public interface Scope extends AutoCloseable {
        /**
         * 關(guān)閉作用域
         */
        @Override
        void close();
    }
}

// 嵌套調(diào)用示例
try (Scope s1 = use("t_xx_order_2025")) {
    // 查詢 2025
    try (Scope s2 = use("t_xx_order_2024")) {
        // 查詢 2024
    }
    // 自動恢復為 2025
}

@Data
@TableName("t_xx_order")
public class OrderEntity implements Serializable {
    /**
     * 主鍵
     */
    @TableId
    private Long id;

    /**
     * 下單日期,格式：yyyy-MM-dd
     */
    private String orderCreateDate;

    /**
     * 訂單號
     */
    private String orderno;
    
    // ...省略其他屬性
}

DynamicTableNameHelper.withTable("t_xx_order_2025", () -> {
    asyncService.doAsyncQuery(); // @Async 方法
});

@Async
public void doAsyncQuery() {
    DynamicTableNameHelper.withTable("t_xx_order_2025", () -> {
        orderMapper.selectById(1L);
    });
}