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，返回什么

之后接简单了，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) {

    }

}

巴特西

spring boot 实现mybatis拦截器

最新文章

热门文章