mybatis拦截器实现动态表名

### 使用 MyBatis 拦截器实现动态表名功能 为了实现在多租户环境中根据 `tenantId` 动态更改表名的需求，可以利用 MyBatis 的拦截器机制来修改 SQL 语句中的表名。下面是一个完整的解决方案。 #### 创建自定义拦截器类 创建一个新的 Java 类继承于 `Interceptor` 接口并重写其方法： ```java import org.apache.ibatis.executor.statement.StatementHandler; import org.apache.ibatis.plugin.*; import java.sql.Connection; @Intercepts({ @Signature(type = StatementHandler.class, method = "prepare", args = {Connection.class, Integer.class}) }) public class TenantTableInterceptor implements Interceptor { private static final String TENANT_ID_KEY = "TENANT_ID"; @Override public Object intercept(Invocation invocation) throws Throwable { // 获取原始SQL语句 StatementHandler statementHandler = (StatementHandler)invocation.getTarget(); MetaObject metaObject = SystemMetaObject.forObject(statementHandler); MappedStatement mappedStatement = (MappedStatement)metaObject.getValue("delegate.mappedStatement"); BoundSql boundSql = statementHandler.getBoundSql(); // 修改SQL语句中的表名为带有租户ID前缀的形式 String originalSql = boundSql.getSql(); String tenantId = getTenantIdFromContext(); // 假设这里是从上下文中获取到的租户ID if(tenantId != null && !"".equals(tenantId.trim())) { String newSql = replaceTableNameWithTenantPrefix(originalSql, tenantId); metaObject.setValue("delegate.boundSql.sql", newSql); } return invocation.proceed(); } /** * 替换原SQL中的表名为带租户前缀的新名称 */ private String replaceTableNameWithTenantPrefix(String sql, String tenantId){ StringBuilder resultBuilder = new StringBuilder(sql.length()); int index = 0; while(index < sql.length()){ char c = sql.charAt(index++); if(c == '`' || c=='['){ // 处理反引号或方括号包围的情况 int endIndex = sql.indexOf(']',index)>-1 ? sql.indexOf(']',index):sql.indexOf('`',index); if(endIndex>-1){ String tableName = sql.substring(index,endIndex).trim(); // 添加租户前缀 resultBuilder.append("`").append(tenantId.toLowerCase()).append("_").append(tableName).append("`"); index=endIndex+1; continue; } } resultBuilder.append(c); } return resultBuilder.toString(); } /** * 从线程局部变量或其他地方取得当前请求对应的租户ID */ protected abstract String getTenantIdFromContext(); } ``` 此代码片段展示了如何编写一个简单的 MyBatis 插件用于替换查询中的表名[^1]。注意这里的 `getTenantIdFromContext()` 方法需要依据实际应用场景自行实现逻辑以获得当前操作所属的租户标识。 #### 注册插件配置 为了让上述拦截器生效，在 Spring Boot 应用程序中注册该插件即可： ```yaml mybatis: configuration: plugins: - com.example.TenantTableInterceptor # 替换成真实的包路径加类名 ``` 或者也可以通过编程方式添加至 MyBatis 配置对象内： ```java @Configuration public class MyBatisConfig { @Bean public ConfigurationCustomizer myBatisConfigurationCustomizer() { return configuration -> { configuration.addInterceptor(new TenantTableInterceptor()); }; } } ``` 以上就是基于 MyBatis 拦截器技术实现动态调整表名的一个实例教程以及最佳实践建议[^2]。