使用 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的监控和修改功能。