用ThreadLocal来优化下代码吧

2020-09-30 572

最近接手了一个老项目，看到一个很有意思的现象。

这个项目中大量的方法入参都会带上user信息，比如这样

它的意图是希望在方法内使用user的信息，但是如此大范围的传递用户信息，第一感觉就是不优雅。那有什么办法可以优化一下呢？

我们第一反应是，可以存一个全局变量，在初始位置将用户信息存入全局变量，然后在需要的地方去get一下。

那在WEB应用中，每个请求都是一个独立线程，怎么去标记呢？

可以用线程的id去作为map的key，将该请求的用户信息作为map的value。

咦？很熟悉的感觉。

没错，Java已经帮我们封装好了这么一个对象，它就是我们今天要说的ThreadLocal。

什么是ThreadLocal
如何使用ThreadLocal优化userid层层传递的问题
ThreadLocal原理是啥
ThreadLocal的实战要点

1.什么是ThreadLocal

先来看下JDK的注释：

简单翻译过来，就是说：

ThreadLocal提供了线程隔离的局部变量，通过get( )和set( )方法操作当前线程对应的变量，而且不会和其他线程冲突，实现了基于线程的数据隔离。

2.如何使用ThreadLocal进行优化

话不多说，基于我们开头的例子，我迫不及待地用ThreadLocal来优化一下。

2.1 构建基于ThreadLocal的上下文

定义一个SessionUser类，存储用户信息，包括用户id、用户名。

然后定义一个基于ThreadLocal的上下文SessionUserContext，代码如下所示。

2.2 信息存入ThreadLocal中

在我们的优化案例中，就是存入用户信息。

解析请求中的用户信息有很多方法。本文以HandlerIntercept为例，说明下MVC中的一种方式。

实现HandlerIntercept接口
重写preHandler方法
解析HttpServletRequest，获取用户信息
用户信息存于SessionUserContext

源码如下所示。

2.3 在需要的地方获取信息

原本需要传入CurrentUser的参数都可以去掉了。

在需要用户信息的时候，直接从SessionUserContext中获取即可。

哈哈，是不是看起来一下子清爽了很多。

可以在任何地方获取user信息，不再需要层层传递用户信息了。

3.ThreadLocal实现原理

上面我们已经知道了怎么通过ThreadLocal进行优化。

下面，我们要知其然知其所以然，一起看看ThreadLocal实现原理吧。

3.1 set方法

Set方法应该是ThreadLocal的核心逻辑了。

主要三步：

获取当前线程
获取ThreadLocalMap对象
如果ThreadLocalMap对象存在，则将当前线程对象作为key，要存储的对象作为value存到map中如果ThreadLocalMap对象不存在，就调用creatMap( )进行创建

3.2 ThreadLocalMap是什么。

ThreadLocalMap是一个定义在ThreadLocal类内部的静态类，里面还定义了一个Entry类作为存储值的地方。

ThreadLocalMap的key是当前ThreadLocal对象，value是我们要存储的值（对象）。

调用creatMap的时候，就是新建一个ThreadLocalMap对象

同时，ThreadLocalMap在Thread类中作为一个属性存在。

每个线程Thread维护了ThreadLocalMap这么一个Map，这个map的key是LocalThread对象本身，value则是要存储的对象

3.3 get方法

Get方法就比较简单了，就是从map中取值的过程。

3.4 ThreadLocal小结

现在，让我们重新梳理一遍，看看ThreadLocal是如何实现变量的线程隔离的：

每个Thread维护着一个ThreadLocalMap的引用
ThreadLocalMap是ThreadLocal的内部类，用Entry来进行存储，key是ThreadLocal对象，值是传递进来的对象
调用ThreadLocal的get()/set()方法时，实际上就是以ThreadLocal对象为key，在ThreadLocalMap中读写value

4.实战要点

在一开始的优化设计中，不知道大家有没有注意到对ThreadLocal的remove调用。

这里就需要谈谈ThreadLocal使用时的，两个要点。尤其是在使用线程池的时候使用ThreadLocal。

4.1 避免内存泄露

在ThreadLocalMap介绍的时候，我们可以看到，ThreadLocalMap是Thread的一个属性。因此，ThreadLocalMap和Thread的生命周期是一样的。

如果没有手动删除对应的ThreadLocal的key，那么就会造成内存泄漏无法回收。尤其在线程池环境下，线程会被不断复用。

4.2 线程池避免重复线程变量影响

以上文优化案例为例。

在MVC中，每次请求进来会使用线程池复用线程。如果请求带了用户信息，那么就会重置ThreadLocal对应的用户信息，如果请求没有带用户信息，必须手动清除一下当前ThreadLocal对应的变量，否则后面使用过程中可能会造成混乱。

原创：阿丸笔记（微信公众号：aone_note），欢迎分享，转载请保留出处。

扫描下方二维码可以关注我哦～

觉得不错，就点个再看吧👇

本文分享自微信公众号 - 阿丸笔记（aone_note）。
如有侵权，请联系 support@oschina.cn 删除。
本文参与“OSC源创计划”，欢迎正在阅读的你也加入，一起分享。

微信关注我们

原文链接：https://my.oschina.net/u/4061399/blog/4657232

转载内容版权归作者及来源网站所有！

低调大师中文资讯倾力打造互联网数据资讯、行业资源、电子商务、移动互联网、网络营销平台。持续更新报道IT业界、互联网、市场资讯、驱动更新,是最及时权威的产业资讯及硬件资讯报道平台。

基于Elasticsearch的日志系统设计实现

之家经销商技术部日志系统设计实现 1. 背景之家经销商业务迅猛发展，技术团队日益壮大，系统的数量和复杂度快速上升，从3年前的20人到现在170人的技术团队。在对业务提供支持的过程中，我们自己面临了不少的问题，主要痛点如下：痛点1：发现问题的时间点严重滞后于端用户以前问题反馈的路径是：端用户反馈给顾问，顾问找产品确认，产品找开发和测试。这条反馈路径相当长并且低效。一般来说，当顾问收到用户反馈的时候，造成的影响面已经不小了，这中间还有信息流转，沟通不畅等问题，根本无法做到先于用户发现问题。痛点2：开发排查问题效率低下大量的文本和不合理的日志记录，不仅没有带来效率提升，反而是一个阻碍，从几百兆的文本日志里查找关键错误信息对任何人都是一个很大的挑战，更别说效率了。所以基本是靠回忆上线改动，然后猜测验证，中间还有反复的过程，对开发来说，也是相当痛苦。痛点3: 开发对自己系统运行状况毫无把握在以前，你如果问开发同事，你们接口的性能咋样，平均耗时，tp99是多少，一天有多少个500错误，耗时大于500ms 的请求一天有多少个，开发同事只能用绝望的眼神告诉你：不知道。看到这里，...

2020-09-30

733

简介 java中为了保证共享数据的安全性，我们引入了锁的机制。有了锁就有可能产生死锁。死锁的原因就是多个线程锁住了对方所需要的资源，然后现有的资源又没有释放，从而导致循环等待的情况。通常来说如果不同的线程对加锁和释放锁的顺序不一致的话，就很有可能产生死锁。不同的加锁顺序我们来看一个不同加锁顺序的例子： public class DiffLockOrder { private int amount; public DiffLockOrder(int amount){ this.amount=amount; } public void transfer(DiffLockOrder target,int transferAmount){ synchronized (this){ synchronized (target){ if(amount< transferAmount){ System.out.println("余额不足！"); }else{ amount=amount-transferAmount; target.amount=target.amount+transfe...

2020-10-01

590

资源下载

更多资源

腾讯云软件源

为解决软件依赖安装时官方源访问速度慢的问题，腾讯云为一些软件搭建了缓存服务。您可以通过使用腾讯云软件源站来提升依赖包的安装速度。为了方便用户自由搭建服务架构，目前腾讯云软件源站支持公网访问和内网访问。

Nacos

Nacos /nɑ:kəʊs/ 是 Dynamic Naming and Configuration Service 的首字母简称，一个易于构建 AI Agent 应用的动态服务发现、配置管理和AI智能体管理平台。Nacos 致力于帮助您发现、配置和管理微服务及AI智能体应用。Nacos 提供了一组简单易用的特性集，帮助您快速实现动态服务发现、服务配置、服务元数据、流量管理。Nacos 帮助您更敏捷和容易地构建、交付和管理微服务平台。

Spring

Spring框架（Spring Framework）是由Rod Johnson于2002年提出的开源Java企业级应用框架，旨在通过使用JavaBean替代传统EJB实现方式降低企业级编程开发的复杂性。该框架基于简单性、可测试性和松耦合性设计理念，提供核心容器、应用上下文、数据访问集成等模块，支持整合Hibernate、Struts等第三方框架，其适用范围不仅限于服务器端开发，绝大多数Java应用均可从中受益。

WebStorm

WebStorm 是jetbrains公司旗下一款JavaScript 开发工具。目前已经被广大中国JS开发者誉为“Web前端开发神器”、“最强大的HTML5编辑器”、“最智能的JavaScript IDE”等。与IntelliJ IDEA同源，继承了IntelliJ IDEA强大的JS部分的功能。