spring boot 实现mybatis拦截器

spring boot 实现mybatis拦截器

项目是个报表系统，服务端是简单的Java web架构，直接在请求参数里面加了个query id参数，就是mybatis mapper的query id，刚好对接接口的请求参数，没有使用接口模式。

基于这种特性，分页使用了PageHelper插件，但是只使用获取指定范围记录这部分，查询的总条数是自己实现的（插件的总条数，需要创建查询对象实现），直接在查询后多发送一次数据库请求，使用自定义的拦截器，拦截查询的请求（默认注入一个参数，等于指定值），在正常的SQL前后 拼接上 "select count(1) coun from (" +sql+")"，直接在数据库中，查询结果集的条数。如下：

if ( mapParam.get("count") != null && mapParam.get("count").toString().equalsIgnoreCase("sum")) {
                // 从StatementHandler中取出查询的SQL
                String sql = (String) metaStatementHandler.getValue("delegate.boundSql.sql");

                sql = "select count(1) coun from ( " + sql + " )";
                // 将修改的SQL放回StatementHandler中
                metaStatementHandler.setValue("delegate.boundSql.sql", sql);
}

mybatis的 拦截器配置如下：

<plugins>
        <plugin interceptor="com.github.pagehelper.PageInterceptor">
            <!-- 支持通过Mapper接口参数来传递分页参数 -->
            <property name="supportMethodsArguments" value="true"/>
            <property name="rowBoundsWithCount" value="true"/>
            <property name="reasonable" value="true"/>
        </plugin>
        <!-- mybatis写出sql记录控件(拦截器) -->
        <!-- 自己写的那个拦截器 -->
        <plugin interceptor="com.xx.SqlInterceptorCount">
            <!-- 方言 -->
            <property name="dialect" value="oracle"/>
        </plugin>

    </plugins>

Java使用如下：

// 从数据库查询
result = QueryMapper.getObjFromDB(paras.get("qdi"),paras);
// 查询当前sql 的总条数
total = QueryMapper.getQdiSum(paras.get("qdi"),paras);

查 total 的时候，给paras，默认注入一个count参数。

paras.put("count", "sum");
hashMap = (HashMap)session.selectList(qdi, paras).get(0);

用起来还是很酸爽的。

--------------朴素的分割线------------------

进入主题

为迎合时代的发展，领导的召唤，项目架构改版，迁入springboot，方便统一管理（监控）

程序员就开干了啊。。。

。。。

。。。

。。。

。。。

。。。

// 程序员挥汗如雨中

几天后，springboot的项目开发完了，但是遇到个问题，mybatis的拦截器不生效，配置如下：

mybatis:
  mapper-locations: classpath:mapper/*.xml

事实上，并没有配置，不过可以加个mybatis的配置文件，像这样，如下：

mybatis:
  mapper-locations: classpath:mapper/*.xml
  configuration:
    interceptions: cn.utstarcom.statplatform7.interceptor.SqlInterceptorCount
  config-location: classpath:mybatis-config.xml

mbatis-config.xml 内容如下：

<plugins>
        <plugin interceptor="com.github.pagehelper.PageInterceptor">
            <!-- 支持通过Mapper接口参数来传递分页参数 -->
            <property name="supportMethodsArguments" value="true"/>
            <property name="rowBoundsWithCount" value="true"/>
            <property name="reasonable" value="true"/>
        </plugin>
        <!-- mybatis写出sql记录控件(拦截器) -->
        <!-- 自己写的那个拦截器 -->
        <plugin interceptor="com..SqlInterceptorCount">
            <!-- 方言 -->
            <property name="dialect" value="oracle"/>
        </plugin>

    </plugins>

拦截器内容如下：

@Intercepts({@Signature(type = StatementHandler.class, method = "prepare", args = {Connection.class, Integer.class})})
public class SqlInterceptorCount implements Interceptor {

    private Logger logger = LoggerFactory.getLogger(SqlInterceptorCount.class);
    private static final ObjectFactory DEFAULT_OBJECT_FACTORY = new DefaultObjectFactory();
    private static final ObjectWrapperFactory DEFAULT_OBJECT_WRAPPER_FACTORY = new DefaultObjectWrapperFactory();
    private static final ReflectorFactory DEFAULT_REFLECTOR_FACTORY = new DefaultReflectorFactory();

    @Override
    public Object intercept(Invocation invocation) throws Throwable {

        try {
            StatementHandler statementHandler = (StatementHandler) invocation.getTarget();
            MetaObject metaStatementHandler = MetaObject.forObject(statementHandler, DEFAULT_OBJECT_FACTORY,
                    DEFAULT_OBJECT_WRAPPER_FACTORY, DEFAULT_REFLECTOR_FACTORY);

//            HashMap mapParam = (HashMap) metaStatementHandler.getValue("delegate.boundSql.parameterObject");
            Object paramObject = metaStatementHandler.getValue("delegate.boundSql.parameterObject");
            if (paramObject instanceof Map) {
                HashMap paramMap = (HashMap)paramObject;
                if (paramMap.get("count") != null && paramMap.get("count").toString().equalsIgnoreCase("sum")) {
                    // get sql from StatementHandler
                    String sql = (String) metaStatementHandler.getValue("delegate.boundSql.sql");
                    sql = "select count(1) coun from ( " + sql + " )";
                    // set moidfy sql to StatementHandler
                    metaStatementHandler.setValue("delegate.boundSql.sql", sql);
                }
            }
        } catch (Exception e) {
            logger.error("throw exception and will not handle the mybatis sql");
        }
        // process the next query process
        return invocation.proceed();
    }

    /**
     *
     * @param o
     * @return
     */
    @Override
    public Object plugin(Object o) {

        return null;
    }

    @Override
    public void setProperties(Properties properties) {

    }
}

事实上，mybatis的配置文件和配置的 interceptions，不能使用（我当然知道是），启动没有问题，数据库也可以操作，但是拦截器没有生效。

1、在又是一大圈的乱找，发现个说法，在拦截器上加个  @Component 注解就可以了。

然后，还是不行，这个更惨，只要查询数据库，都会返回个null ，又没有报错信息，只是个null

2、又是网上一通乱找，看到个说法：说拦截器只有 plugin方法执行了，intercept方法没有执行

debug 一下，发现问题了，加上@Component 注解，plugin方法执行了，返回个null，知道为什么吗，看下plugin的代码：

@Override
    public Object plugin(Object o) {

        return null;
    }

我去

不返回null，返回什么

不返回null，返回什么

不返回null，返回什么

之后接简单了，plugin 方法是实现 Interceptor 接口的时候，IDEA生成的方法，有个默认实现，返回null

改改：

 @Override
    public Object plugin(Object o) {

//        logger.info("Interceptor : SqlInterceptorCount is running, but this is plugin method not intercept method");
        /*
            use Intercepts annotation，intercept StatementHandler method prepare
         */
        if (o instanceof StatementHandler) {
            return Plugin.wrap(o, this);
        }
        return o;
    }

在传入对象是 STATEMENTHandler的对象的时候，把方法转发到intercept 方法，拦截器可以正常执行了。

结论就是：

　　springboot下的mybatis拦截器，和单独用mybatis的拦截器一样，但是不需要配置，直接在拦截器前面加个@Component 注解，在plugin 方法中将请求转发到intercept方法拦截即可，不需要其他配置。

完整代码如下：

@Component
@Intercepts({@Signature(type = StatementHandler.class, method = "prepare", args = {Connection.class, Integer.class})})
public class SqlInterceptorCount implements Interceptor {

    private Logger logger = LoggerFactory.getLogger(SqlInterceptorCount.class);
    private static final ObjectFactory DEFAULT_OBJECT_FACTORY = new DefaultObjectFactory();
    private static final ObjectWrapperFactory DEFAULT_OBJECT_WRAPPER_FACTORY = new DefaultObjectWrapperFactory();
    private static final ReflectorFactory DEFAULT_REFLECTOR_FACTORY = new DefaultReflectorFactory();

    @Override
    public Object intercept(Invocation invocation) throws Throwable {

        try {
            StatementHandler statementHandler = (StatementHandler) invocation.getTarget();
            MetaObject metaStatementHandler = MetaObject.forObject(statementHandler, DEFAULT_OBJECT_FACTORY,
                    DEFAULT_OBJECT_WRAPPER_FACTORY, DEFAULT_REFLECTOR_FACTORY);

//            HashMap mapParam = (HashMap) metaStatementHandler.getValue("delegate.boundSql.parameterObject");
            Object paramObject = metaStatementHandler.getValue("delegate.boundSql.parameterObject");
            if (paramObject instanceof Map) {
                HashMap paramMap = (HashMap)paramObject;
                if (paramMap.get("count") != null && paramMap.get("count").toString().equalsIgnoreCase("sum")) {
                    // get sql from StatementHandler
                    String sql = (String) metaStatementHandler.getValue("delegate.boundSql.sql");
                    sql = "select count(1) coun from ( " + sql + " )";
                    // set moidfy sql to StatementHandler
                    metaStatementHandler.setValue("delegate.boundSql.sql", sql);
                }
            }
        } catch (Exception e) {
            logger.error("throw exception and will not handle the mybatis sql");
        }
        // process the next query process
        return invocation.proceed();
    }

    /**
     *
     * @param o
     * @return
     */
     @Override
    public Object plugin(Object o) {

//        logger.info("Interceptor : SqlInterceptorCount is running, but this is plugin method not intercept method");
        /*
            use Intercepts annotation，intercept StatementHandler method prepare
         */
        if (o instanceof StatementHandler) {
            return Plugin.wrap(o, this);
        }
        return o;
    }

    @Override
    public void setProperties(Properties properties) {

    }
}