文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

Spring中使用自定义ThreadLocal存储导致的坑怎么解决

2024-04-02 19:55

关注

这篇文章主要介绍了Spring中使用自定义ThreadLocal存储导致的坑怎么解决的相关知识,内容详细易懂,操作简单快捷,具有一定借鉴价值,相信大家阅读完这篇Spring中使用自定义ThreadLocal存储导致的坑怎么解决文章都会有所收获,下面我们一起来看看吧。

  Spring 中有时候我们需要存储一些和 Request 相关联的变量,例如用户的登陆有关信息等,它的生命周期和 Request 相同。一个容易想到的实现办法是使用 ThreadLocal:

  public class SecurityContextHolder {

  private static final ThreadLocal<SecurityContext> securityContext = new ThreadLocal<SecurityContext>();

  public static void set(SecurityContext context) {

  securityContext.set(context);

  }

  public static SecurityContext get() {

  return securityContext.get();

  }

  public static void clear() {

  securityContext.remove();

  }

  }

  使用一个自定义的 HandlerInterceptor 将有关信息注入进去:

  @Slf4j

  @Component

  public class RequestInterceptor implements HandlerInterceptor {

  @Override

  public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws

  Exception {

  try {

  SecurityContextHolder.set(retrieveRequestContext(request));

  } catch (Exception ex) {

  log.warn("读取请求信息失败", ex);

  }

  return true;

  }

  @Override

  public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, @Nullable

  ModelAndView modelAndView) throws Exception {

  SecurityContextHolder.clear();

  }

  通过这样,我们就可以在 Controller 中直接使用这个 context,很方便的获取到有关用户的信息:

  @Slf4j

  @RestController

  class Controller {

  public Result get() {

  long userId = SecurityContextHolder.get()。getUserId();

  // &hellip;

  }

  }

  这个方法也是很多博客中使用的。然而这个方法却存在着一个很隐蔽的坑: HandlerInterceptor 的 postHandle 并不总是会调用。

  当 Controller 中出现 Exception:

  @Slf4j

  @RestController

  class Controller {

  public Result get() {

  long userId = SecurityContextHolder.get()。getUserId();

  // &hellip;

  throw new RuntimeException();

  }

  }

  或者在 HandlerInterceptor 的 preHandle 中出现 Exception:

  @Slf4j

  @Component

  public class RequestInterceptor implements HandlerInterceptor {

  @Override

  public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws

  Exception {

  try {

  SecurityContextHolder.set(retrieveRequestContext(request));

  } catch (Exception ex) {

  log.warn("读取请求信息失败", ex);

  }

  // &hellip;

  throw new RuntimeException();

  //&hellip;

  return true;

  }

  }

  这些情况下, postHandle 并不会调用。这就导致了 ThreadLocal 变量不能被清理。

  在平常的 Java 环境中,ThreadLocal 变量随着 Thread 本身的销毁,是可以被销毁掉的。但 Spring 由于采用了线程池的设计,响应请求的线程可能会一直常驻,这就导致了变量一直不能被 GC 回收。更糟糕的是,这个没有被正确回收的变量,由于线程池对线程的复用,可能会串到别的 Request 当中,进而直接导致代码逻辑的错误。

  为了解决这个问题,我们可以使用 Spring 自带的 RequestContextHolder ,它背后的原理也是 ThreadLocal,不过它总会被更底层的 Servlet 的 Filter 清理掉,因此不存在泄露的问题。

  下面是一个使用 RequestContextHolder 重写的例子:

  public class SecurityContextHolder {

  private static final String SECURITY_CONTEXT_ATTRIBUTES = "SECURITY_CONTEXT";

  public static void setContext(SecurityContext context) {

  RequestContextHolder.currentRequestAttributes()。setAttribute(

  SECURITY_CONTEXT_ATTRIBUTES,

  context,

  RequestAttributes.SCOPE_REQUEST);

  }

  public static SecurityContext get() {

  return (SecurityContext)RequestContextHolder.currentRequestAttributes()

  。getAttribute(SECURITY_CONTEXT_ATTRIBUTES, RequestAttributes.SCOPE_REQUEST);

  }

  }

  除了使用 RequestContextHolder 还可以使用 Request Scope 的 Bean,或者使用 ThreadLocalTargetSource ,原理上是类似的。

  需要时刻注意 ThreadLocal 相当于线程内部的 static 变量,是一个非常容易产生泄露的点,因此使用 ThreadLocal 应该额外小心。

关于“Spring中使用自定义ThreadLocal存储导致的坑怎么解决”这篇文章的内容就介绍到这里,感谢各位的阅读!相信大家对“Spring中使用自定义ThreadLocal存储导致的坑怎么解决”知识都有一定的了解,大家如果还想学习更多知识,欢迎关注编程网行业资讯频道。

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     807人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     351人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     314人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     433人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     221人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯