真实项目中 ThreadLocal 的妙用

  • 时间:
  • 浏览:0

一、那此是 ThreadLocal

ThreadLocal 提供了程序的局部变量,每个程序时需通过 set() 和 get() 来对你什儿 局部变量进行操作,但不想和或多或少程序的局部变量冲突,实现了程序间的据隔离。

简单讲:二个 获取用户的请求程序 A,可能性向 ThreadLocal 填充变量 AValue(只有被程序 A 操作),该变量对或多或少获取用户的请求程序 B、C...是隔离的.

最简单的使用土措施:

类似于一次 HTTP 请求程序中,利用 ThreadLocal 存储 Cookie 对象,进行请况管理。set Cookie:

private ThreadLocal httpThreadLocal = new ThreadLocal();

httpThreadLocal.set(“Cookie: sid=13420771402233”)

后面 存储格式是 String ,实际场景存储的是具体的对象。在这次 HTTP 请求过程中,任何之后 时需获取 Cookie 。获取土措施很简单 get Cookie:

String cookieValue = (String) httpThreadLocal.get();

Thread 与 ThreadLocal 对象引用关系图

二、你熟悉的场景

2.1 数据库连接池

比如一次请求程序进来,业务 Dao 时需更新 user 表和 user-detail 表。可能性是 new 出二个 数据库 Connection ,分别不同的 Connection 操作 user 表和 user-detail 表,就无法保证事务。那末 数据库连接池是怎样才能保证的?

答案是:利用 ThreadLocal 存储唯一 Connection 对象。每次请求程序,pool.getConnection 获取连接的之后 不是原来操作:

  • 会从 ThreadLocal 获取 Connection 对象。可能性有,则保证了后面 多个数据库操作共用同二个 Connection ,从而保证了事务。
  • 可能性那末 ,往 ThreadLocal 新增Connection 对象,并返回到程序
错误的做法
public class XXXService {

    private Connection conn;
}

可能性 conn 是程序不安全的。原来会原因分析分析多个请求公用二个 连接。请求量很大的请况下,延迟各种。你懂。

或多或少,使用 ThreadLocal 保证每个请求程序的 Connection 是唯一的。即每个程序有被委托人的连接。

继续讲到 Spring 框架,在事务之后开使时,会给当前程序二个 Jdbc Connection,在整个事务过程,不是使用该程序绑定的connection来执行数据库操作,实现了事务的隔离性。Spring框架后面 假使 用的ThreadLocal来实现你什儿 隔离

比如你访问百度、我访问百度,会有不同 Cookie 。或多或少你只有访问我的 Cookie,我假使 能。顾名思义,使用 ThreadLocal 保证每个 HTTP 请求程序的 Cookie 是唯一的。

Cookie 原来能够做 Session 等请况管理。

三、实战场景

总结一下假使 :ThreadLocal 时需让同二个 程序中上下文之间数据共享

在后面 章节 二、你熟悉的场景 虽然介绍了好多好多 现有场景。那末 我这边具体的实战场景是那此?

简单的例子:

适用满足你什儿 二个 条件的场景:1.每个程序独有的或多或少信息,2.那此信息又会在多个土措施或类中用到。

  1. 二个 请求程序,后面 有二个 异步小程序,各有二个 土措施。分别解决 A 或 B 业务
  2. 你什儿 土措施是传递不可变的入参
  3. 另你什儿 假使 ThreadLocal,插进 ThreadLocal 的入参,会被各个土措施共享。或多或少多个请求程序互不影响
比较复杂的例子:

一次发货操作:会根据入参,进行组件化、流程编排话。那末 入参会被各个地方用到,或多或少或多或少流程组件是异步的(类似于 new thread 操作的)。这之后 时需定二个 XXContext 上下文:

public class XXContext {
    
    private static ThreadLocal<Map<Class<?>, Object>> context = new InheritableThreadLocal<>();
    
    /**
     * 把参数设置到上下文的Map中
     */
    public static void put(Object obj) {
        Map<Class<?>, Object> map = context.get();
        if (map == null) {
            map = new HashMap<>();
            context.set(map);
        }
        if (obj instanceof Enum) {
            map.put(obj.getClass().getSuperclass(), obj);
        } else {
            map.put(obj.getClass(), obj);
        }
    }
    
    /**
     * 从上下文中,根据类名取出参数
     */
    @SuppressWarnings("unchecked")
    public static <T> T get(Class<T> c) {
        Map<Class<?>, Object> map = context.get();
        if (map == null) {
            return null;
        }
        return (T) map.get(c);
    }
    
    /**
     * 清空ThreadLocal的数据
     */
    public static void clean() {
        context.remove();
    }
}

代码解析:

  • 不是 static 操作,类似于 DateUtil 玩法
  • 记得每次请求程序后清理。时需 AOP 去清理,加个注解就行。可能性同二个 请求程序可能性被业务方公用。

(完)