PageHelper 使用 ThreadLocal 的线程复用问题

欢迎关注方志朋的博客，回复”666“获面试宝典

来源：https://blog.csdn.net/qq_38245668/article/details/105984171/

前言

PageHelper 是较为常用的分页插件，通过实现 Mybatis 的 Interceptor 接口完成对 query sql 的动态分页，其中分页参数由 ThreadLocal 进行保存。

简单的分页执行过程：

1.设置 page 参数

2.执行 query 方法

3.Interceptor 接口中校验 ThreadLocal 中是否存在有设置的 page 参数

4.存在 page 参数，重新生成 count sql 和 page sql，并执行查询。不存在 page 参数，直接返回查询结果

5.执行 LOCAL_PAGE.remove() 清除 page 参数

问题场景

观察上述的执行过程，可以发现，如果在第 1 步和第 2 步之间发生异常，那么 LOCAL_PAGE 中当前线程对应的 page 参数并不会 remove。

在不使用线程池的情况下，当前线程在执行完毕后会被销毁，这时当前线程中的 threadLocals 参数将会被情况，也就清空了 LOCAL_PAGE 中当前线程的 page 参数。

但是如果使用了线程池，当前线程执行完毕，并不会被销毁，而是会将当前线程再次存放到池中，标记为空闲状态，以便后续使用。在后续使用这个线程的时候，由于线程的 threadLocals 依旧存在有值，尽管我们在第 1 步时未设置 page 参数，第 3 步的也能获取到page参数，从而生成 count sql 和 page sql，从而影响我们的正常查询。

SpringBoot 项目中会使用内置的 Tomcat 作为服务器，而Tomcat会默认使用线程来处理请求，从而便引发了上述问题。

解决方案

因为Tomcat的线程是用来处理request请求，那么在请求完成时，清空当前线程的threadLocals 属性值，也就是执行 LOCAL_PAGE.remove() 即可。

实现方式：

1.使用 aop，对所有 controller 进行处理

2.实现 HandlerInterceptor 或者 WebRequestInterceptor 对 request 请求的拦截器接口，通过 afterCompletion 方法执行 LOCAL_PAGE.remove() 。— 推荐，实现方式简单

这里使用第二种方式，实现 HandlerInterceptor 接口：

public class PageLocalWebInterceptor implements HandlerInterceptor {@Overridepublic void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {// PageHelper.clearPage() 内部调用 LOCAL_PAGE.remove()PageHelper.clearPage();}
}

定义配置类，配置类需实现 WebMvcConfigurer 接口完成对于WebMvc的相关配置，注册 PageLocalWebInterceptor ：

@Configuration
public class FrameworkAutoConfig implements WebMvcConfigurer {@Override
public void addInterceptors(InterceptorRegistry registry) {registry.addInterceptor(new PageLocalWebInterceptor());
}

热门内容：

干掉visio，这个画图神器真的绝了！！！
面试官问：生成订单30分钟未支付，则自动取消，该怎么实现？
不是我吹，这个idea插件你真没用过！

比 Xshell 还好用的 SSH 客户端神器，MobaXterm 太爱了！

RedisJson 横空出世，性能碾压ES和Mongo！

最近面试BAT，整理一份面试资料《Java面试BAT通关手册》，覆盖了Java核心技术、JVM、Java并发、SSM、微服务、数据库、数据结构等等。
获取方式：点“在看”，关注公众号并回复 666 领取，更多内容陆续奉上。

明天见(｡･ω･｡)ﾉ♡