<dependency>
    <groupId>com.baomidou</groupId>
    <artifactId>mybatis-plus-boot-starter</artifactId>
    <version>最新版本號</version>
</dependency>

@Component
@Intercepts({@Signature(type = StatementHandler.class, method = "prepare", args = {Connection.class, Integer.class})})
public class DynamicTableInterceptor implements Interceptor {
  	/**
     * 使用ThreadLocal來存儲線程特有的數(shù)據(jù)，這里用于存儲動態(tài)表名
     */
    private static final ThreadLocal<String> TABLE_NAME_HOLDER = new ThreadLocal<>();
    /**
     * 設(shè)置動態(tài)表名
     *
     * @param tableName 需要設(shè)置的表名，由調(diào)用者指定
     *                  此方法允許在運行時動態(tài)地設(shè)置數(shù)據(jù)庫表名，以便在多數(shù)據(jù)源或動態(tài)表名的場景下靈活地切換表
     */
    public static void setDynamicTableName(String tableName) {
        TABLE_NAME_HOLDER.set(tableName);
    }
    /**
     * 清除當(dāng)前線程中的動態(tài)表名
     * 此方法用于清除ThreadLocal中存儲的表名信息，以避免內(nèi)存泄漏
     * 它應(yīng)在每次使用動態(tài)表名后調(diào)用，確保不會對后續(xù)的請求產(chǎn)生影響
     */
    public static void clearDynamicTableName() {
        TABLE_NAME_HOLDER.remove();
    }
    /**
     * 攔截器
     */
    @Override
    public Object intercept(Invocation invocation) throws Throwable {
        StatementHandler handler = (StatementHandler) invocation.getTarget();
        BoundSql boundSql = handler.getBoundSql();
        String tableName = TABLE_NAME_HOLDER.get();
        if (tableName != null) {
            /**
             * 從實體類的@tableName中獲取表名，必須在實體類上面添加@tableName注解
             * 實體類中的@tableName注解中的值，必須是數(shù)據(jù)庫中的表名，否則會報錯
             * 例如 ：@TableName("t_user_")，數(shù)據(jù)庫是根據(jù)月份來的，在這里替換則需要根據(jù)當(dāng)前月份進行拼接表名
             */
            Class<?> entityType = boundSql.getParameterObject().getClass();
            String oldTableName = entityType.getAnnotation(TableName.class).value();
            String newSql = boundSql.getSql().replace(oldTableName, tableName);
            Field sqlField = boundSql.getClass().getDeclaredField("sql");
            sqlField.setAccessible(true);
            sqlField.set(boundSql, newSql);
        }
        return invocation.proceed();
    }
}

DynamicTableInterceptor.setDynamicTableName("your_dynamic_table_name");
try {
    myService.saveOrUpdateBatch(entities); // MyBatis-Plus 操作
} finally {
    DynamicTableInterceptor.clearDynamicTableName();
}

@Override
public Object intercept(Invocation invocation) throws Throwable {
    StatementHandler handler = (StatementHandler) invocation.getTarget();
    BoundSql boundSql = handler.getBoundSql();
    long startTime = System.currentTimeMillis();
    Object result = invocation.proceed();
    long endTime = System.currentTimeMillis();
    System.out.println("SQL: " + boundSql.getSql() + " Execution Time: " + (endTime - startTime) + "ms");
    return result;
}

@Override
public Object intercept(Invocation invocation) throws Throwable {
    StatementHandler handler = (StatementHandler) invocation.getTarget();
    BoundSql boundSql = handler.getBoundSql();
    String originalSql = boundSql.getSql();
    String tenantId = TenantContext.getCurrentTenantId();
    String modifiedSql = originalSql + " WHERE tenant_id = " + tenantId;
    Field sqlField = boundSql.getClass().getDeclaredField("sql");
    sqlField.setAccessible(true);
    sqlField.set(boundSql, modifiedSql);
    return invocation.proceed();
}

@Override
public Object intercept(Invocation invocation) throws Throwable {
    StatementHandler handler = (StatementHandler) invocation.getTarget();
    BoundSql boundSql = handler.getBoundSql();
    String sql = boundSql.getSql();
    // 自定義加密邏輯
    // 加密處理參數(shù)
    return invocation.proceed();
}

@Override
public Object intercept(Invocation invocation) throws Throwable {
    StatementHandler handler = (StatementHandler) invocation.getTarget();
    BoundSql boundSql = handler.getBoundSql();
    String originalSql = boundSql.getSql();
    String paginatedSql = originalSql + " LIMIT " + offset + ", " + limit;
    Field sqlField = boundSql.getClass().getDeclaredField("sql");
    sqlField.setAccessible(true);
    sqlField.set(boundSql, paginatedSql);
    return invocation.proceed();
}

@Override
public Object intercept(Invocation invocation) throws Throwable {
    StatementHandler handler = (StatementHandler) invocation.getTarget();
    BoundSql boundSql = handler.getBoundSql();
    String originalSql = boundSql.getSql();
    String userRole = SecurityContext.getCurrentUserRole();
    String restrictedSql = originalSql + " AND role = '" + userRole + "'";
    // 動態(tài)修改 SQL
    return invocation.proceed();
}

@Override
public Object intercept(Invocation invocation) throws Throwable {
    // 檢查緩存
    Object cachedResult = CacheManager.get(boundSql.getSql());
    if (cachedResult != null) {
        return cachedResult;
    }
    // 如果沒有命中，執(zhí)行查詢
    Object result = invocation.proceed();
    // 存入緩存
    return result;
}

@Override
public Object intercept(Invocation invocation) throws Throwable {
    String dataSourceKey = DataSourceContextHolder.getDataSourceKey();
    DynamicDataSource.setDataSourceKey(dataSourceKey);
    return invocation.proceed();
}