使用 mybatis-plus 拦截器对项目的所有执行的sql做监控和修改。

出现的问题

项目中使用了别人封装好的jar包,使用的是逻辑删除,但是在我们的数据库中由于唯一索引的存在,必须使用物理删除,因此需要对使用到逻辑删除的sql做监控,并且求改这些sql。

代码 

/**
 * mybatis拦截器
 * 对所有sql语句进行监视,修改
 */
@Slf4j
@Component
@Intercepts({@Signature(type = StatementHandler.class, method = "prepare", args = {Connection.class, Integer.class})})
public class SqlignoreDeletedInterceptor implements Interceptor {
    @Autowired
    private lignoreDeleteTablesBeen lignoreDeleteTables;

    @Override
    public Object intercept(Invocation invocation) throws Throwable {
        StatementHandler statementHandler = (StatementHandler) invocation.getTarget();
        //通过MetaObject优雅访问对象的属性,这里是访问statementHandler的属性;:MetaObject是Mybatis提供的一个用于方便、
        //优雅访问对象属性的对象,通过它可以简化代码、不需要try/catch各种reflect异常,同时它支持对JavaBean、Collection、Map三种类型对象的操作。
        
         MetaObject metaObject = MetaObject
                .forObject(statementHandler, SystemMetaObject.DEFAULT_OBJECT_FACTORY, SystemMetaObject.DEFAULT_OBJECT_WRAPPER_FACTORY,  new DefaultReflectorFactory());
         //先拦截到RoutingStatementHandler,里面有个StatementHandler类型的delegate变量,其实现类是BaseStatementHandler,然后就到BaseStatementHandler的成员变量mappedStatement
                MappedStatement mappedStatement = (MappedStatement) metaObject.getValue("delegate.mappedStatement");
          //id为执行的mapper方法的全路径名,如com.uv.dao.UserMapper.insertUser
           String id = mappedStatement.getId();
        // log.info("id为执行的mapper方法的全路径名:" + id);
        //sql语句类型 select、delete、insert、update

         String sqlCommandType = mappedStatement.getSqlCommandType().toString();

         BoundSql boundSql = statementHandler.getBoundSql();
        //获取到原始sql语句
        String sql = boundSql.getSql();
        // log.info("拦截的sql语句:" + sql);
        //可以先打印出原始的sql语句,然后根据实际情况修改,我个人建议是下面这样修改
        //只在原sql上添加条件而不是删除条件
         sql = lignore(sql);
        log.info("修改之后的sql:" + sql);
        //通过反射修改sql语句
        //下面类似于替换sql
        Field field = boundSql.getClass().getDeclaredField("sql");
        field.setAccessible(true);
        field.set(boundSql, sql);
        return invocation.proceed();
    }

@Override
    public Object plugin(Object target) {
        return Interceptor.super.plugin(target);
    }
@Override
    public void setProperties(Properties properties) {
        Interceptor.super.setProperties(properties);
    }

/**    自己需要修改原sql的规则
     * 校验表是否需要去掉deleted=0
     *
     * @param
     */
  private String lignore(String sql) {
        List<String> tables = lignoreDeleteTables.getLignoreDeleteTables();
         if (!tables.isEmpty()) {
            for (String table : tables) {
                if (sql.contains(table)) {
                    sql = sql.replaceAll("\n", " ");//过滤到换行符。
                    sql = sql.replaceAll("\\s+", " "); //多个空格变一个空格。
                    log.info("格式化后的sql:" + sql);
                    if (sql.contains("SET deleted=1")) {
                        sql = sql.replaceAll("SET deleted=1", "");
                          sql = sql.replaceAll("UPDATE", "DELETE FROM");
                        return sql.replaceAll("AND deleted=0", "");
                        //TODO 这个写法有问题,连表查询会导致sql异常,暂时增加了判断条件
                         } else if (!sql.contains(".deleted") && sql.contains("deleted = 0 AND")) {
                        return sql.replaceAll("deleted = 0 AND", "");
                        } else if (!sql.contains(".deleted") && sql.contains("deleted=0 AND")) {
                        return sql.replaceAll("deleted=0 AND", "");
                     } else if (!sql.contains(".deleted") && sql.contains("AND deleted = 0")) {
                        return sql.replaceAll("AND deleted = 0", "");
                     } else if (!sql.contains(".deleted") && sql.contains("AND deleted=0")) {
                        return sql.replaceAll("AND deleted=0", "");
                     } else if (sql.contains("WHERE deleted = 0")) {
                        return sql.replaceAll("WHERE deleted = 0", "");
                     } else if (sql.contains("WHERE deleted=0")) {
                        return sql.replaceAll("WHERE deleted=0", "");
                      } else {
                        return sql;
                    }
              }
            }
        }
        return sql;
    }


}

 结果

最终运行起来还是满足了对sql的监控和修改功能。

上一篇:18. 【Android教程】图片控件 ImageView


下一篇:网络安全(黑客)—2024自学手册