我们知道,在我们的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值就行,和上面差不多我就不写了。
下面是一些配置,注册我们的拦截器,就是我刚刚写的那个类
数据库中的数据:
结果:只获取到两个数据证明我们成功了