优雅的使用 ThreadLocal

简介:

前言

在我们日常 Java Web 开发中难免遇到需要把一个参数层层的传递到最内层,然后中间层根本不需要使用这个参数,或者是仅仅在特定的工具类中使用,这样我们完全没有必要在每一个方法里面都传递这样一个通用的参数。如果有一个办法能够在任何一个类里面想用的时候直接拿来使用就太好了。JavaWeb项目大部分都是基于Tomcat,每次访问都是一个新的线程,这样让我们联想到了ThreadLocal,每一个线程都独享一个ThreadLocal,在接收请求的时候set特定内容,在需要的时候get这个值。下面我们就进入主题。

ThreadLocal

维持线程封闭性的一种更规范的方法就是使用ThreadLocal,这个类能使线程中的某个值与保存的值的对象关联起来。ThreadLocal提供getset等接口或方法,这些方法为每一个使用这个变量的线程都存有一份独立的副本,因此get总是返回由当前线程在调用set时设置的最新值。
ThreadLocal有如下方法:

1public T get({ }
2public void set(value{ }
3public void remove({ }
4protected T initialValue({ }

get()方法是用来获取ThreadLocal在当前线程中保存的变量副本  
set()用来设置当前线程中变量的副本  
remove()用来移除当前线程中变量的副本   
initialValue()是一个protected方法,一般是用来在使用时进行重写的,如果在没有set的时候就调用get,会调用initialValue方法初始化内容。
为了使用的更放心,我们简单的看一下具体的实现:

set方法

1public void set(value{
2        Thread t = Thread.currentThread();
3        ThreadLocalMap map = getMap(t);
4        if (map != null)
5            map.set(thisvalue);
6        else
7            createMap(t, value);
8    }

set方法会获取当前的线程,通过当前线程获取ThreadLocalMap对象。然后把需要存储的值放到这个map里面。如果没有就调用createMap创建对象。

getMap方法

1 ThreadLocalMap getMap(Thread t) {
2    return t.threadLocals;
3 }

getMap方法直接返回当前ThreadthreadLocals变量,这样说明了之所以说ThreadLocal线程局部变量就是因为它只是通过ThreadLocal变量存在了Thread本身而已。

createMap方法

1void createMap(Thread t, T firstValue) {
2    t.threadLocals = new ThreadLocalMap(this, firstValue);
3}

set的时候如果不存在threadLocals,直接创建对象。由上看出,放入mapkey是当前的ThreadLocalvalue是需要存放的内容,所以我们设置属性的时候需要注意存放和获取的是一个ThreadLocal

get方法

 1public T get() {
2        Thread t = Thread.currentThread();
3        ThreadLocalMap map = getMap(t);
4        if (map != null) {
5            ThreadLocalMap.Entry e = map.getEntry(this);
6            if (e != null)
7                return (T)e.value;
8        }
9        return setInitialValue();
10    }

get方法就比较简单,获取当前线程,尝试获取当前线程里面的threadLocals,如果没有获取到就调用setInitialValue方法,setInitialValue基本和set是一样的,就不累累述了。

场景

本文应用ThreadLocal的场景:在调用API接口的时候传递了一些公共参数,这些公共参数携带了一些设备信息,服务端接口根据不同的信息组装不同的格式数据返回给客户端。假定服务器端需要通过设备类型(device)来下发下载地址,当然接口也有同样的其他逻辑,我们只要在返回数据的时候判断好是什么类型的客户端就好了。如下:

场景一

请求

1GET api/users?device=android

返回

1 {
2  user : {  
3  },
4  link : "https://play.google.com/store/apps/details?id=***"
5 }

场景二

请求

1GET api/users?device=ios

返回

1 {
2  user : { 
3  },
4  link : "https://itunes.apple.com/us/app/**"
5 }

实现

首先准备一个BaseSigntureRequest类用来存放公共参数

 1public class BaseSignatureRequest {
2    private String device;
3
4    public String getDevice() {
5        return device;
6    }
7
8    public void setDevice(String device) {
9        this.device = device;
10    }
11}

然后准备一个staticThreadLocal类用来存放ThreadLocal,以便存储和获取时候的ThreadLocal一致。

1public class ThreadLocalCache {
2    public static ThreadLocal<BaseSignatureRequest> 
3     baseSignatureRequestThreadLocal = new ThreadLocal<>();
4}

然后编写一个Interceptor,在请求的时候获取device参数,存入当前线程的ThreadLocal中。这里需要注意的是,重写了afterCompletion方法,当请求结束的时候把ThreadLocal remove,移除不必须要键值对。

 1public class ParameterInterceptor implements HandlerInterceptor {
2    @Override
3    public boolean preHandle(HttpServletRequest request, HttpServletResponse response,
4                             Object handler)
 throws Exception 
{
5        String device = request.getParameter("device");
6        BaseSignatureRequest baseSignatureRequest = new BaseSignatureRequest();
7        baseSignatureRequest.setDevice(device);
8        ThreadLocalCache.baseSignatureRequestThreadLocal.set(baseSignatureRequest);
9        return true;
10    }
11
12    @Override
13    public void afterCompletion(HttpServletRequest request, HttpServletResponse response,
14                                Object handler, Exception ex)
 throws Exception 
{
15        ThreadLocalCache.baseSignatureRequestThreadLocal.remove();
16    }
17
18    @Override
19    public void postHandle(HttpServletRequest httpServletRequest,
20                           HttpServletResponse httpServletResponse, 
21                           Object o, ModelAndView modelAndView)
 throws Exception 
{
22
23    }
24}

当然需要在spring里面配置interceptor

1    <mvc:interceptors>
2        <mvc:interceptor>
3            <mvc:mapping path="/api/**"/>
4            <bean class="life.majiang.ParameterInterceptor"></bean>
5        </mvc:interceptor>
6    </mvc:interceptors>

最后在Converter里面转换实体的时候直接使用即可,这样就大功告成了。

 1public class UserConverter {
2    public static ResultDO toDO(User user{
3        ResultDO resultDO = new ResultDO();
4        resultDO.setUser(user);
5        BaseSignatureRequest baseSignatureRequest = ThreadLocalCache.baseSignatureRequestThreadLocal.get();
6        String device = baseSignatureRequest.getDevice();
7        if (StringUtils.equals(device, "ios")) {
8            resultDO.setLink("https://itunes.apple.com/us/app/**");
9        } else {
10            resultDO.setLink("https://play.google.com/store/apps/details?id=***");
11        }
12        return resultDO;
13    }

总结

这种机制很方便,因为他避免了在调用每一个方法时都要传递执行上下文信息,合理的使用ThreadLocal可以起到事倍功半的效果,但是需要避免滥用,例如将所有的全局变量作为ThreadLocal对象,ThreadLocal类似全局变量,他能降低代码的可重用性,并在类之间引入隐含的耦合性,所以再使用前需要格外小心。



原文发布时间为:2018-09-07
本文作者:码匠笔记
本文来自云栖社区合作伙伴“IT先森养成记”,了解相关信息可以关注“IT先森养成记”。
相关文章
|
6月前
|
存储 安全 Java
【ThreadLocal】
【ThreadLocal】
|
2天前
|
存储 Java 数据管理
ThreadLocal的使用
`ThreadLocal`是Java中的线程局部变量工具,确保每个线程都有自己的变量副本,互不干扰。适用于保持线程安全性数据和跨方法共享数据。基本用法包括创建实例、设置和获取值以及清除值。例如,创建ThreadLocal对象后,使用`.set()`设置值,`.get()`获取值,`.remove()`清除值。注意ThreadLocal可能引起内存泄漏,应适时清理,并谨慎使用以避免影响代码可读性和线程安全性。它是多线程编程中实现线程局部数据管理的有效手段。
25 10
|
3月前
|
存储 Java
ThreadLocal 有什么用
ThreadLocal 有什么用
21 0
|
7月前
|
存储 安全 Java
ThreadLocal介绍和应用
ThreadLocal介绍和应用
24 0
|
7月前
|
Java
ThreadLocal详解
ThreadLocal详解
39 0
|
8月前
|
存储 算法 安全
深入详解ThreadLocal
在我们日常的并发编程中,有一种神奇的机制在静悄悄地为我们解决着各种看似棘手的问题,它就是 ThreadLocal 。
21416 9
深入详解ThreadLocal
|
10月前
|
存储 SQL Java
ThreadLocal的其他应用
request对象跟PageHelper
84 0
|
存储 安全 Java
ThreadLocal 使用详解
ThreadLocal 是线程本地变量。当使用 ThreadLocal 维护变量时,ThreadLocal 为每个使用该变量的线程提供独立的变量副本,所以每一个线程都可以独立地改变自己的副本,而不会影响其它线程
456 0
|
存储 Java
ThreadLocal相关使用
ThreadLocal相关使用
162 0
ThreadLocal相关使用