我是如何用 ThreadLocal 虐面试官的?
我是陈皮,一个在互联网 Coding 的 ITer,微信搜索「陈皮的JavaLib」第一时间阅读最新文章,回复【资料】,即可获得我精心整理的技术资料,电子书籍,一线大厂面试资料和优秀简历模板。
ThreadLocal 简介
Threadlocal 类提供了线程局部变量
功能。意思可以在指定线程内部存储数据,并且哪个线程存储的数据只能线程它自己有权限取得。
底层原理其实是在线程内部维护一个 Map 变量,然后 Threadlocal 对象作为 key,要存储的数据作为 value。而 Threadlocal 类作为一个设置和访问这个线程局部变量的入口。
Threadlocal 对象一般定义为私有静态的,而且通过它的 get 和 set 方法设置和获取线程局部变量。
private static final ThreadLocal<UserContext> THREAD_LOCAL = new ThreadLocal<>();
如何使用 ThreadLocal
ThreadLocal 使用方法很简单,它提供了三个公开的方法供外部调用。
- void set(T value):设置线程局部变量
- T get():获取线程局部变量
- void remove():删除线程局部变量
package com.chenpi;
/**
* @Description
* @Author 陈皮
* @Date 2021/6/27
* @Version 1.0
*/
public class ThreadLocalTest {
private static final ThreadLocal<String> THREAD_LOCAL = new ThreadLocal<>();
public static void main(String[] args) {
// 设置线程局部变量
THREAD_LOCAL.set("我是陈皮,个人公众号【陈皮的JavaLib】");
// 使用线程局部变量
peelChenpi();
// 删除线程局部变量
THREAD_LOCAL.remove();
// 使用线程局部变量
peelChenpi();
}
public static void peelChenpi() {
System.out.println(THREAD_LOCAL.get());
}
}
// 输出结果
我是陈皮,个人公众号【陈皮的JavaLib】
null
ThreadLocal 源码分析
ThreadLocal 底层原理是在线程内部维护一个 Map 变量,然后 Threadlocal 对象作为 key,要存储的数据作为 value。而 Threadlocal 类作为一个设置和访问这个线程局部变量的入口。
Thread 类中定义了一个 ThreadLocalMap
类型的变量 threadLocals,每个线程都有自己专属的 threadLocals 变量,ThreadLocalMap 类是由 ThreadLocal 维护的一个静态内部类。
ThreadLocal.ThreadLocalMap threadLocals = null;
Thread 的 threadLocals 变量是默认访问权限的,只能被同个包下的类访问,所以我们是不能直接使用 Thread 的 threadLocals 变量的,这也就是为什么能控制不同线程只能获取自己的数据,达到了线程隔离。Threadlocal 类是访问它的入口。
Thread t = Thread.currentThread();
ThreadLocalMap map = getMap(t);
ThreadLocalMap getMap(Thread t) {
return t.threadLocals;
}
ThreadLocal 类中的静态内部类 ThreadLocalMap 部分源码如下,底层是维护的了一个 Entry 类型数组 table。
static class ThreadLocalMap {
// Map中的Entry对象,弱引用类型,key是ThreadLocal对象,value是线程局部变量
static class Entry extends WeakReference<ThreadLocal<?>> {
Object value;
Entry(ThreadLocal<?> k, Object v) {
super(k);
value = v;
}
}
// 初始化容量16,必须是2的幂次方
private static final int INITIAL_CAPACITY = 16;
// 存储数据的数组,可扩容,长度必须是2的幂次方
private Entry[] table;
// table数组的大小
private int size = 0;
// table数组的阈值,达到则扩容
private int threshold; // Default to 0
}
为什么 ThreadLocalMap 内部存储机构是维护一个数组呢?因为一个线程是可以通过多个不同的 ThreadLocal 对象来设置多个线程局部变量的,这些局部变量都是存储在自己线程的同一个 ThreadLocalMap 对象中。通过不同的 ThreadLocal 对象可以取得当前线程的不同局部变量值。
package com.chenpi;
/**
* @Description
* @Author 陈皮
* @Date 2021/6/27
* @Version 1.0
*/
public class ThreadLocalTest {
private static final ThreadLocal<String> THREAD_LOCAL = new ThreadLocal<>();
private static final ThreadLocal<String> THREAD_LOCAL01 = new ThreadLocal<>();
public static void main(String[] args) {
THREAD_LOCAL.set("我是陈皮");
System.out.println(THREAD_LOCAL.get());
THREAD_LOCAL01.set("陈皮是我");
System.out.println(THREAD_LOCAL01.get());
}
}
那同一个线程的 ThreadLocalMap 对象的数组 table,当前线程的不同 ThreadLocal 是如何确定数组下标,如果数组下标冲突又是怎么解决的呢?其实它不同于 HashMap 底层数组+链表+红黑树的存储结构,它只有 Entry 数组。
ThreadLocal 有个静态的初始哈希值 nextHashCode
,然后每新建一个 ThreadLocal 对象都会在此哈希值的基础上自增一次,自增量为0x61c88647。
// 每 new 一个 ThreadLocal 对象都会自增一次哈希值
private final int threadLocalHashCode = nextHashCode();
// 初始哈希值,静态变量
private static AtomicInteger nextHashCode =
new AtomicInteger();
// 自增量
private static final int HASH_INCREMENT = 0x61c88647;
// 自增一次
private static int nextHashCode() {
return nextHashCode.getAndAdd(HASH_INCREMENT);
}
然后计算 table 数组下标是通过以下算法确定的,如果下标冲突,则下标会往后挪一位继续判断,直到不冲突为止。
// 首次创建 ThreadLocalMap 对象时,第一个元素的下标计算
int i = firstKey.threadLocalHashCode & (INITIAL_CAPACITY - 1);
// 后续元素的下标计算
int i = key.threadLocalHashCode & (len-1);
// 下标冲突时计算下一个下标的方法
private static int nextIndex(int i, int len) {
return ((i + 1 < len) ? i + 1 : 0);
}
我们看 ThreadLocal 类的 set 方法源码,它是设置线程局部变量的入口方法,实现原理也很简单。
- 首先获取当前线程的 ThreadLocalMap 变量
- 如果 ThreadLocalMap 变量存在,则将 ThreadLocal 对象和 T 数据以键值对的形式存储到 ThreadLocalMap 变量中
- 如果 ThreadLocalMap 变量不存在,则新建 ThreadLocalMap 变量并绑定到当前线程中,再将 ThreadLocal 对象和 T 数据以键值对的形式存储到 ThreadLocalMap 变量中
// 设置线程局部变量
public void set(T value) {
Thread t = Thread.currentThread();
ThreadLocalMap map = getMap(t);
if (map != null)
map.set(this, value);
else
createMap(t, value);
}
ThreadLocal 类的 get 方法,它是访问线程局部变量的入口方法,实现原理也很简单。
- 首先获取当前线程的 ThreadLocalMap 变量
- 如果 ThreadLocalMap 变量存在,则将 ThreadLocal 对象作为 key,在 ThreadLocalMap 变量中查找对应的线程局部变量
- 如果 ThreadLocalMap 变量不存在,则新建 ThreadLocalMap 变量并绑定到当前线程中,再将 ThreadLocal 对象和 null 以键值对的形式存储到 ThreadLocalMap 变量中
// 访问线程局部变量
public T get() {
Thread t = Thread.currentThread();
ThreadLocalMap map = getMap(t);
if (map != null) {
ThreadLocalMap.Entry e = map.getEntry(this);
if (e != null) {
@SuppressWarnings("unchecked")
T result = (T)e.value;
return result;
}
}
return setInitialValue();
}
private T setInitialValue() {
T value = initialValue();
Thread t = Thread.currentThread();
ThreadLocalMap map = getMap(t);
if (map != null)
map.set(this, value);
else
createMap(t, value);
return value;
}
protected T initialValue() {
return null;
}
ThreadLocal 类的 remove 方法,直接清除线程中 ThreadLocalMap 对象中以当前 ThreadLocal 对象为 key 的 Entry对象。
public void remove() {
ThreadLocalMap m = getMap(Thread.currentThread());
if (m != null)
m.remove(this);
}
你是否发现,ThreadLocal 类中的所有方法都是没有加锁的,因为 ThreadLocal 最终操作的都是对当前线程的 ThreadLocalMap 对象进行操作,既然线程处理自己的局部变量,就肯定不会有线程安全问题。
注意,同一个 ThreadLocal 变量在父线程中被设置值后,在子线程中是获取这个值的。即不具备继承性。具有继承性的是 InheritableThreadLocal 类,下期文章再讲解这个。
ThreadLocal 应用
ThreadLocal 具有线程隔离,线程安全的效果,如果数据是以线程为作用域并且不同线程具有不同的数据的时候,采用 ThreadLocal 是个不错的选择。
例如对于要用户登录的服务,对于每一个请求,我们可能需要校验用户是否登录,以及在登录后,后续的请求中会使用到用户信息,那我们就可以将登录校验过的用户信息放入线程局部变量中。
首先定义一个用户信息类,存放用户登录校验过的用户信息。
package com.chenpi;
import lombok.Data;
/**
* @Description
* @Author 陈皮
* @Date 2021/6/27
* @Version 1.0
*/
@Data
public class UserContext {
private String userId;
private String userName;
}
定义一个持有用户信息的管理工具类,主要用户管理当前线程的用户信息。
package com.chenpi;
/**
* @Description
* @Author 陈皮
* @Date 2021/6/27
* @Version 1.0
*/
public class UserContextHolder {
private static final ThreadLocal<UserContext> THREAD_LOCAL = new ThreadLocal<>();
private UserContextHolder() {}
public static void setUserContext(UserContext userContext) {
THREAD_LOCAL.set(userContext);
}
public static UserContext getUserContext() {
return THREAD_LOCAL.get();
}
public static void removeUserContext() {
THREAD_LOCAL.remove();
}
}
对需要用户权限的接口进行拦截,然后将用户信息存储到当前线程内部。注意,当请求完成后,需要将用户信息进行清除,避免内存泄露问题。
package com.chenpi;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.springframework.lang.Nullable;
import org.springframework.stereotype.Component;
import org.springframework.web.method.HandlerMethod;
import org.springframework.web.servlet.HandlerInterceptor;
/**
* @Description 用户权限验证拦截
* @Author 陈皮
* @Date 2021/6/27
* @Version 1.0
*/
@Component
public class UserPermissionInterceptor implements HandlerInterceptor {
@Override
public boolean preHandle(HttpServletRequest request, HttpServletResponse response,
Object handler) {
if (handler instanceof HandlerMethod) {
HandlerMethod handlerMethod = (HandlerMethod) handler;
// 获取用户权限校验注解
UserAuthenticate userAuthenticate =
handlerMethod.getMethod().getAnnotation(UserAuthenticate.class);
if (null == userAuthenticate) {
userAuthenticate = handlerMethod.getMethod().getDeclaringClass()
.getAnnotation(UserAuthenticate.class);
}
if (userAuthenticate != null && userAuthenticate.permission()) {
// 验证用户信息
UserContext userContext = userContextManager.getUserContext(request);
// 将用户信息存储到线程内部
UserContextHolder.setUserContext(userContext);
}
}
return true;
}
@Override
public void afterCompletion(HttpServletRequest request, HttpServletResponse response,
Object handler, @Nullable Exception ex) {
// 请求完后,清除当前线程的用户信息,避免内存泄露和用户信息混乱
UserContextHolder.removeUserContext();
}
}
至此,我们就能在当前请求的同一线程内,不用通过方法参数显示传递用户信息,可以通过工具类随时随地获取到当前用户信息了。
而且你会发现,如果方法调用链 A - B - C,AB 不需要用户信息,C 需要用户信息,那你需要层层通过方法参数传递用户信息。而使用 ThreadLocal 后,不用通过方法参数层层传递用户信息,避免了依赖污染,代码也更加简洁。
package com.chenpi;
import org.springframework.stereotype.Service;
/**
* @Description
* @Author 陈皮
* @Date 2021/6/27
* @Version 1.0
*/
@Service
public class UserService {
public void chenPiDeJavaLib() {
UserContext userContext = UserContextHolder.getUserContext();
}
}
我是如何用 ThreadLocal 虐面试官的?的更多相关文章
- 当面试官说 “你还有什么问题想问的” ,你该如何回答?
阅读本文大概需要 4 分钟. 作者:黄小斜 来源:程序员江湖 程序员面试时经常会听到面试官说一些套话,比如"今天的面试就到这里了,回去等通知吧","你还有什么问题想问我的 ...
- 阿里面试官用HashMap把我问倒了
本人是一名大三学生,最近在找暑期实习,其中也面试过两次阿里,一次菜鸟网络部门.一次网商银行部门,当然我都失败了,同时也让我印象很深刻,因此记录了其中一些面试心得,我觉得这个问题很值得分享,因此分享给大 ...
- 我是面试官--"自我介绍"
工作10余年,经历过很多次面试,也面试了N多人.这些年来,已经有好些位朋友(或同事)与我聊起相关话题,涉及面试,更关乎职业生涯规划.感触颇多,就借助自媒体的浪潮,与更多的程序员一起共谈面试经历,希望可 ...
- 面试官:小伙子,听说你看过ThreadLocal源码?(万字图文深度解析ThreadLocal)
前言 Ym8V9H.png (高清无损原图.pdf关注公众号后回复 ThreadLocal 获取,文末有公众号链接) 前几天写了一篇AQS相关的文章:我画了35张图就是为了让你深入 AQS,反响不错, ...
- 面经手册 · 第12篇《面试官,ThreadLocal 你要这么问,我就挂了!》
作者:小傅哥 博客:https://bugstack.cn 沉淀.分享.成长,让自己和他人都能有所收获! 一.前言 说到底,你真的会造火箭吗? 常说面试造火箭,入职拧螺丝.但你真的有造火箭的本事吗,大 ...
- 原创 | 我被面试官给虐懵了,竟然是因为我不懂Spring中的@Configuration
GitHub 3.7k Star 的Java工程师成神之路 ,不来了解一下吗? GitHub 3.7k Star 的Java工程师成神之路 ,真的不来了解一下吗? GitHub 3.7k Star 的 ...
- 我被面试官给虐懵了,竟然是因为我不懂Spring中的@Configuration
现在大部分的Spring项目都采用了基于注解的配置,采用了@Configuration 替换标签的做法.一行简单的注解就可以解决很多事情.但是,其实每一个注解背后都有很多值得学习和思考的内容.这些思考 ...
- 如何用json 与jsonp 的区别去回答你的面试官?
常常 有面试官这样问我们,虽然用过无数次,但是回答不上岂不是尴尬,那我们浅析一下它们的区别? 1. json JSON是一种基于文本的数据交换格式,用于描述复杂的数据,举个例子: var nax=[ ...
- 【MySQL】面试官问我:MySQL如何实现无数据插入,有数据更新?我是这样回答的!
写在前面 马上就是金九银十的跳槽黄金期了,很多读者都开始出去面试了.这不,又一名读者出去面试被面试官问了一个MySQL的问题:向MySQL中插入数据,如何实现MySQL中没有当前id标识的数据时插入数 ...
随机推荐
- 手把手教你部署验证freeswitch(避免踩坑)
前言:请各大网友尊重本人原创知识分享,谨记本人博客:南国以南i 介绍:freeswitch可集成ASR(语音识别)和TTS(文本转语音)创建智能电话机器人和用户通话,可用于问卷调查,自动催缴等业务,电 ...
- Java_接口
接口 接口中的方法全都是抽象方法,用来指定一些规则,让子类去重写.接口的作用主要体现在一下两点 1.指定规则2.利用这些规则给类做功能扩展 接口的定义和使用 接口使用interface关键字来定义,使 ...
- 微服务·API文档
阅文时长 | 3.92分钟 字数统计 | 2754.05字符 主要内容 | 1.什么是API文档 2.API文档的使用 3.声明与参考资料 『微服务·API文档』 编写人 | SCscHero 编写时 ...
- Django(33)Django操作cookie
前言 cookie:在网站中,http请求是无状态的.也就是说即使第一次和服务器连接后并且登录成功后,第二次请求服务器依然不能知道当前请求是哪个用户.cookie的出现就是为了解决这个问题,第一次登录 ...
- [Linux] Shell 脚本实例(超实用)
文件操作 为文件(test.sh)增加执行权限 chmod +x test.sh 列出当前文件夹下所有文件(每行输出一个) 1 #!/bin/bash 2 dir=`ls ./` 3 for i in ...
- centos7下cups + samba共打印服务
centos7下cups + samba共打印服务 2015年8月21日admin发表评论阅读评论 这个算是rhce课程的篇外篇,Samba除了提供文件共享外,也可以像windows下的共享一样提 ...
- HBase HA 集群环境搭建
安装准备 确定已安装并启动 HDFS(HA)集群 角色分配如下: node-01: namenode datanode regionserver hmaster zookeeper node-02: ...
- 解决Windows路径太长的目录以及文件名超长删除的问题
因Windows文件夹有长度限制,在路径太深,长度达到600多个字符时,删除文件时出现报错"源文件名长度大于文件系统支持的长度.请尝试将其移动到具有较短路径名称的位置,或者在执行此操作前尝试 ...
- Django(35)Django请求生命周期分析(超详细)
Django请求生命周期分析 1.客户端发送请求 在浏览器输入url地址,例如www.baidu.com,浏览器会自动补全协议(http),变为http://www.baidu.com,现在部分网站都 ...
- MySQL之数据查询语言(DQL)
数据查询语言DQL基本结构是由SELECT子句,FROM子句,WHERE子句组成的查询块: SELECT <字段> FROM <表名> WHERE <查询条件> - ...