MyBatis拦截器实现原理

天空～之城

已于 2023-10-16 15:51:02 修改

阅读量84

点赞数

分类专栏：源码解析技术问答 Mybatis 文章标签： mybatis java 开发语言

于 2023-10-06 14:48:33 首次发布

技术问答同时被 3 个专栏收录

123 篇文章 4 订阅

订阅专栏

源码解析

22 篇文章 1 订阅

订阅专栏

Mybatis

14 篇文章 0 订阅

订阅专栏

拦截：https://blog.csdn.net/CaptHua/article/details/126417608

在这里插入图片描述

@Component
@Order(1)
@Intercepts({@Signature(type = StatementHandler.class, method = "update", args = {Statement.class}),})
public class SqlValidateMybatisInterceptor extends PRSMybatisInterceptor {
 
    @Override
    protected Object before(Invocation invocation) throws Throwable {
        String sql="";
        Statement statement=(Statement) invocation.getArgs()[0];
        if(Proxy.isProxyClass(statement.getClass())){
            MetaObject metaObject= SystemMetaObject.forObject(statement);
            Object h=metaObject.getValue("h");
            if(h instanceof StatementLogger){
                RoutingStatementHandler rsh=(RoutingStatementHandler) invocation.getTarget();
                sql=rsh.getBoundSql().getSql();
            }else {
                PreparedStatementLogger psl=(PreparedStatementLogger) h;
                sql=psl.getPreparedStatement().toString();
            }
        }else{
            sql=statement.toString();
        }
        if(containsDelete(sql)&&!containsWhere(sql)){
            throw new SQLException("不能删除整张表,sql:"+sql);
        }
        return null;
    }
 
    private boolean containsDelete(String sql){
        return sql.contains("delete")||sql.contains("DELETE");
    }
 
    private boolean containsWhere(String sql){
        return sql.contains("where")||sql.contains("WHERE");
    }
}
public class PRSMybatisInterceptor implements Interceptor {
 
    Boolean needBreak=false;
 
    @Override
    public Object intercept(Invocation invocation) throws Throwable {
        Object result= before(invocation);
        if(needBreak){
            return result;
        }
        result= invocation.proceed();
        result=after(result,invocation);
        return result;
    }
 
    protected Object before(Invocation invocation) throws Throwable{
        return null;
    }
    protected Object after(Object result,Invocation invocation) throws Throwable{
        return result;
    }
 
    @Override
    public Object plugin(Object o) {
        return Plugin.wrap(o, this);
    }
 
    @Override
    public void setProperties(Properties properties) {
    }
}

在这里插入图片描述

private SqlSession openSessionFromConnection(ExecutorType execType, Connection connection) {
    ...
    final Executor executor = configuration.newExecutor(tx, execType);
    return new DefaultSqlSession(configuration, executor, autoCommit);
}
public Executor newExecutor(Transaction transaction, ExecutorType executorType) {
  executorType = executorType == null ? defaultExecutorType : executorType;
  executorType = executorType == null ? ExecutorType.SIMPLE : executorType;
  Executor executor;
  if (ExecutorType.BATCH == executorType) {
    executor = new BatchExecutor(this, transaction);
  } else if (ExecutorType.REUSE == executorType) {
    executor = new ReuseExecutor(this, transaction);
  } else {
    executor = new SimpleExecutor(this, transaction);
  }
  if (cacheEnabled) {
    executor = new CachingExecutor(executor);
  }
  executor = (Executor) interceptorChain.pluginAll(executor);
  return executor;
}

在这里插入图片描述

StatementHandler handler = configuration.newStatementHandler(wrapper, ms, parameterObject, rowBounds, resultHandler, boundSql);
public StatementHandler newStatementHandler(Executor executor, MappedStatement mappedStatement, Object parameterObject, RowBounds rowBounds, ResultHandler resultHandler, BoundSql boundSql) {
  StatementHandler statementHandler = new RoutingStatementHandler(executor, mappedStatement, parameterObject, rowBounds, resultHandler, boundSql);
  statementHandler = (StatementHandler) interceptorChain.pluginAll(statementHandler);
  return statementHandler;
}

在这里插入图片描述

this.parameterHandler = configuration.newParameterHandler(mappedStatement, parameterObject, boundSql);
public ParameterHandler newParameterHandler(MappedStatement mappedStatement, Object parameterObject, BoundSql boundSql) {
  ParameterHandler parameterHandler = mappedStatement.getLang().createParameterHandler(mappedStatement, parameterObject, boundSql);
  parameterHandler = (ParameterHandler) interceptorChain.pluginAll(parameterHandler);
  return parameterHandler;
}

this.resultSetHandler = configuration.newResultSetHandler(executor, mappedStatement, rowBounds, parameterHandler, resultHandler, boundSql);
public ResultSetHandler newResultSetHandler(Executor executor, MappedStatement mappedStatement, RowBounds rowBounds, ParameterHandler parameterHandler,
    ResultHandler resultHandler, BoundSql boundSql) {
  ResultSetHandler resultSetHandler = new DefaultResultSetHandler(executor, mappedStatement, parameterHandler, resultHandler, boundSql, rowBounds);
  resultSetHandler = (ResultSetHandler) interceptorChain.pluginAll(resultSetHandler);
  return resultSetHandler;
}

在这里插入图片描述

Interceptor
@Override
public Object plugin(Object o) {
    return Plugin.wrap(o, this);
}

在这里插入图片描述

public class PRSMybatisInterceptor implements Interceptor {
    Boolean needBreak = false;
    @Override
    public Object intercept(Invocation invocation) throws Throwable {
        Object result = before(invocation);
        if (needBreak) {
            return result;
        }
        result = invocation.proceed();
        result = after(result, invocation);
        return result;
    }
    protected Object before(Invocation invocation) throws Throwable {
        return null;
    }
    protected Object after(Object result, Invocation invocation) throws Throwable {
        return result;
    }
    @Override
    public Object plugin(Object o) {
        return Plugin.wrap(o, this);
    }
    @Override
    public void setProperties(Properties properties) {
    }
}

/**
 * @author CaptHua
 */
@Component
@Order(1)
@Intercepts({@Signature(type = StatementHandler.class, method = "update", args = {Statement.class}),})
public class SqlValidateMybatisInterceptor extends PRSMybatisInterceptor {
    @Override
    protected Object before(Invocation invocation) throws Throwable {
        String sql = "";
        Statement statement = (Statement) invocation.getArgs()[0];
        if (Proxy.isProxyClass(statement.getClass())) {
            MetaObject metaObject = SystemMetaObject.forObject(statement);
            Object h = metaObject.getValue("h");
            if (h instanceof StatementLogger) {
                RoutingStatementHandler rsh = (RoutingStatementHandler) invocation.getTarget();
                sql = rsh.getBoundSql().getSql();
            } else {
                PreparedStatementLogger psl = (PreparedStatementLogger) h;
                sql = psl.getPreparedStatement().toString();
            }
        } else {
            sql = statement.toString();
        }
        if (containsDelete(sql) && !containsWhere(sql)) {
            throw new SQLException("不能删除整张表,sql:" + sql);
        }
        return null;
    }
    private boolean containsDelete(String sql) {
        return sql.contains("delete") || sql.contains("DELETE");
    }
    private boolean containsWhere(String sql) {
        return sql.contains("where") || sql.contains("WHERE");
    }
}