mybatis自定义拦截器分页查询

我们知道,在我们的sql语句中经常遇到的问题就是分页查询的问题,我们来看一看在原生的mybatis中怎么利用它的接口实现我们的分页查询,第一步我们应该知道在哪去插入分页的语句,我们知道我们的sql相关的执行处理都再一个叫做StatementHandler的类中,所以我们再该类创建以后我们对它里面的sql语句进行改写或者其他的处理即可:
在这里插入图片描述
在这里插入图片描述
所以我们只需要再StatementHandler的prepare方法中做一个拦截处理即可,
说干就干,我们知道我们的StatementHandler是做了增强处理的,直接通过拦截器做增强处理:
在这里插入图片描述

@Intercepts({@Signature(type = StatementHandler.class, method = "prepare",
        args = {Connection.class, Integer.class})})
public class PageHelper implements Interceptor {
    public Object intercept(Invocation invocation) throws Throwable {
        //获取我们的StatementHandler对象
        RoutingStatementHandler routingStatementHandler = (RoutingStatementHandler) invocation.getTarget();
        //2.获取sql语句
        BoundSql boundSql = routingStatementHandler.getBoundSql();
        String sql = boundSql.getSql();
        //3.判断是否为select语句  首先获取他的delegate  一般情况下 该属性都是默认创建prepareStatement
        Object delegate = getPrivateProperty(routingStatementHandler, "delegate");
        Object mappedStatement = getPrivateProperty(delegate, "mappedStatement");
        Object sqlCommandType=getPrivateProperty(mappedStatement,"sqlCommandType");
        if(sqlCommandType.equals(SqlCommandType.SELECT)){
            //获取它的私有属性
            RowBounds rowBounds=(RowBounds)getPrivateProperty(delegate,"rowBounds");
            //设置roundBounds的属性
            setPrivateProperty(rowBounds,"offset",0);
            setPrivateProperty(rowBounds,"limit",2);
        }
        return invocation.proceed();
    }
    public Object plugin(Object target) {
        return Plugin.wrap(target, this);
    }
    //获取对象的私有属性
    public Object getPrivateProperty(Object obj,String propertyName){
        Field field = null;
        System.out.println(obj.getClass());
        //遍历的原因,应为我们的每一次创建的StatementHandler都要经过这里,有些handler没有我们需要的字段所以要判断一下再处理
        for (Class<?> clazz=obj.getClass(); clazz != Object.class; clazz=clazz.getSuperclass()) {
            try {
                field = clazz.getDeclaredField(propertyName);
                if (field != null){
                    field.setAccessible(true);
                    break;
                }

            } catch (NoSuchFieldException e) {
            }
        }
        try {
            return field.get(obj);
        } catch (IllegalAccessException e) {
            e.printStackTrace();
        }
        return null;
    }
    //设置对象的私有属性
    public void setPrivateProperty(Object obj,String propertyName,Object value){
        Field field = null;
        for (Class<?> clazz=obj.getClass(); clazz != Object.class; clazz=clazz.getSuperclass()) {
            try {
                field = clazz.getDeclaredField(propertyName);
                if (field != null){
                    field.setAccessible(true);
                    try {
                        field.set(obj,value);
                    } catch (IllegalAccessException e) {
                        e.printStackTrace();
                    }
                    break;
                }
            } catch (NoSuchFieldException e) {
            }
        }
    }
}

我们大致的处理步骤:
1.获取我们的StatementHandler对象
2.获取boundSql对象:
3.获取rouBounds对象,但是该对象在我们的RoutingStatementHandler的私有属性delegate的属性:
4.获取他的delegate对象(一般情况下是prepareStatement)的mappedStatement属性 ,里面是我们解析出来的每一个sql语句对应的封装对象。
5.下一步,获取它的sqlCommandType,因为它里面包含了每个sql标签的属性,而这个属性就是标志它是什么类型的sql语句
6.判断是否是select类型即可
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
所以我定义了一个getProperty的方法来获取对应的私有参数,它没有get方法,所以这样获取该对象,
我们还要知道这个对象有啥用,为啥会获取它喃,我们看下图,在处理结果集的时候会调用一个skiprows方法该方法的作用就是省略我们不需要的数据,意思就是我们这样做处理不是吧我们的sql语句改写了,只是将我们不需要的sql省略了,改写sql的话直接修改boundSql的sql值就行,和上面差不多我就不写了。在这里插入图片描述
下面是一些配置,注册我们的拦截器,就是我刚刚写的那个类在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
数据库中的数据:
在这里插入图片描述

结果:只获取到两个数据证明我们成功了
在这里插入图片描述

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值