ThreadLocal 实战用法

ThreadLocal 实战用法

ThreadLocal 是什么

首先来看一下官方解释：

Creates a thread local variable. The initial value of the variable is determined by invoking the get method on the Supplier.

创建一个线程本地变量，变量的初始值是有调用get()方法的提供方所决定的。

• 它的作用是在一个线程内部，不同的方法直接共享变量。
• 对于多线程，每个线程都有一份自己的变量副本，它们各自拥有，互不干扰。

实战使用

场景 SpringBoot 后台开发

作用

在同一线程 不同的方法之间共享变量。

思路

1. 客户端访问接口，服务端开启一个线程进行处理
2. 拦截器验证，根据请求头中的token查询相关数据
3. 在拦截器代码中 将数据放入ThreadLocal
4. 在controller中 取出数据，调用service

实现

step 1. 定义ThreadLocal管理类，方便操作。这个类为可用单例模式，或者全部静态方法。

public class LoginInfoThreadLocal {
    // TrainerLoginVo 所要存放的业务数据
    private static final ThreadLocal<TrainerLoginVo> LOCAL = new ThreadLocal<TrainerLoginVo>();

    public static void set(TrainerLoginVo trainerLoginVo){
        LOCAL.set(trainerLoginVo);
    }

    public static TrainerLoginVo get(){
        return LOCAL.get();
    }

    public static void remove(){
        LOCAL.set(null);
    }

    public static String getUserCode() {
        return LOCAL.get().getUserCode();
    }
}

step2. 存入数据

// 在拦截器或其他业务方法中存入数据
TrainerLoginInfoThreadLocal.set(trainerLoginVo);

step3. 读取数据

@RequestMapping(value = "/test/getAll")
public Result<?> getAll() {
    // 在LoginInfoThreadLocal取数据
    String userCode = LoginInfoThreadLocal.getUserCode();
    service.getAll(userCode)
    return Result.success();
}