SpringBoot集成mybatis拦截器修改表名
背景
公司的框架是基于mysql5.7开发的,最近有一个应用项目部署在linux系统上,使用的是mysql8.0,安装时未开启大小写敏感忽略,客户又不允许重装mysql环境,导致一些框架代码和业务代码中表名使用大写的地方会出现表名找不不到的情况,所以需要进行统一处理
自定义SQLAST适配器
自定义ASTVisitorAdapter对表名进行修改
public class MySqlExportTableAliasVisitor extends MySqlASTVisitorAdapter {
@Override
public boolean visit(SQLExprTableSource x) {
SystemConfig systemConfig = SpringBootBeanUtil.getBean(SystemConfig.class);
if(systemConfig.getDbTableNameProxy().equals("lowcase")){
x.setExpr(x.getTableName().toLowerCase());
}else if(systemConfig.getDbTableNameProxy().equals("upcase")){
x.setExpr(x.getTableName().toUpperCase());
}
return true;
}
}
自定义Mybatis拦截器
通过BoundSql获取sql语句,使用Druid的SQLUtils对sql语句进行结构化分析,表名修改完成后再重新复制sql语句
@Intercepts({@Signature(type = StatementHandler.class, method = "prepare", args = {Connection.class, Integer.class})})
public class SQLTableNameHandleInterceptor implements Interceptor {
@Override
public Object intercept(Invocation invocation) throws Throwable {
DataBaseInfoUtil dataBaseInfoUtil = SpringBootBeanUtil.getBean(DataBaseInfoUtil.class);
if(dataBaseInfoUtil.checkOpenTableNameHandle()){
StatementHandler statementHandler = (StatementHandler) invocation.getTarget();
MetaObject metaStatementHandler = MetaObject.forObject(statementHandler, new DefaultObjectFactory(), new DefaultObjectWrapperFactory(), new DefaultReflectorFactory());
BoundSql boundSql = (BoundSql) metaStatementHandler.getValue("delegate.boundSql");
String sql = boundSql.getSql();
List<SQLStatement> stmtList = SQLUtils.parseStatements(sql, JdbcConstants.MYSQL);
MySqlExportTableAliasVisitor visitor = new MySqlExportTableAliasVisitor();
for (SQLStatement stmt : stmtList) {
stmt.accept(visitor);
}
String handleSQL = SQLUtils.toSQLString(stmtList, JdbcConstants.MYSQL);
metaStatementHandler.setValue("delegate.boundSql.sql", handleSQL);
}
return invocation.proceed();
}
@Override
public Object plugin(Object target) {
return Plugin.wrap(target, this);
}
@Override
public void setProperties(Properties properties) {
}
}
注册自定义Myabits拦截器
@Configuration
public class FrameMyBatisPluginConfig {
@Bean
@Conditional({SQLTableNameHandleInterceptorCondition.class})
public String SQLTableNameHandleInterceptor(SqlSessionFactory sqlSessionFactory) {
//实例化插件
SQLTableNameHandleInterceptor sqlTableNameHandleInterceptor = new SQLTableNameHandleInterceptor();
//创建属性值
Properties properties = new Properties();
properties.setProperty("prop1","value1");
//将属性值设置到插件中
sqlTableNameHandleInterceptor.setProperties(properties);
//将插件添加到SqlSessionFactory工厂
sqlSessionFactory.getConfiguration().addInterceptor(sqlTableNameHandleInterceptor);
return "interceptor";
}
}
分类:
JAVA
标签:
mybatis-plus
, springboot
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】博客园社区专享云产品让利特惠,阿里云新客6.5折上折
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· DeepSeek “源神”启动!「GitHub 热点速览」
· 我与微信审核的“相爱相杀”看个人小程序副业
· 微软正式发布.NET 10 Preview 1:开启下一代开发框架新篇章
· 如何使用 Uni-app 实现视频聊天(源码,支持安卓、iOS)
· C# 集成 DeepSeek 模型实现 AI 私有化(本地部署与 API 调用教程)