使用ThreadLocal来存取单线程内的数据

一.什么是ThreadLocal？

ThreadLocal，即线程本地变量。如果你创建了一个 ThreadLocal变量，那么访问这个变量的每个线程都会有这个变量的一个本地拷贝，多个线程操作这个变量的时候，实际是在操作自己本地内存里面的变量，从而起到线程隔离的作用，也就是两个线程间互不影响，避免了并发场景下的线程安全问题。

 这样就有利于我们使用ThreadLocal来存取数据。

上图灰色以及绿色是两个不同的线程，我们可以看出两个线程间的数据是独立互不影响，而利用这个ThreadLocal我们可以实现两个事情：

1. 通过单线程保存数据来减少参数传递
2. 在同一线程间共享数据

二.ThreadLocal的使用：

这里我们以将JWT令牌放入线程内并在Service层读取JWT令牌并获取内部用户 id 值为例 =>

1.导入ThreadLocal的工具类：

这里工具类是new一个ThreadLocal来维护全局唯一的ThreadLocal对象，然后使用其内部的三个方法就可以使用了 => 

public class ThreadLocalUtil {//提供ThreadLocal对象,private static final ThreadLocal THREAD_LOCAL = new ThreadLocal();//根据键获取值public static <T> T get(){return (T) THREAD_LOCAL.get();}//存储键值对public static void set(Object value){THREAD_LOCAL.set(value);}//清除ThreadLocal 防止内存泄漏public static void remove(){THREAD_LOCAL.remove();}
}

这里需要注意，因为ThreadLocal生命周期非常长，所以我们在使用完后需要调用内部方法 remove() 删除 ThreadLocal 以此来避免内存泄露。 

2.在拦截器内把JWT令牌存储到ThreadLocal中：

①当拦截器拦截请求，我们就会再preHandle()方法内开辟线程空间并保存JWT令牌。

②这里注意当响应完数据，也就是请求结束，我们就不会使用数据，而这个时候我们就需要移除数据，这个时候我们重写拦截器内部的afterCompletion()方法，在方法内调用remove()方法删除ThreadLocal。

@Component
public class LoginInterceptor implements HandlerInterceptor {@Overridepublic boolean preHandle(HttpServletRequest request , HttpServletResponse response , Object handle) throws Exception{String token = request.getHeader("Authorization"); //Authorization为请求头的名字try {//令牌验证Map<String,Object> claims = JwtUtil.parseToken(token);//线程开辟空间存储ThreadLocalUtil.set(claims);//放行return true;}catch (Exception e){//http响应状态码为401在·response.setStatus(401);//不放行return false;}}@Overridepublic void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {//清空线程数据ThreadLocalUtil.remove();}
}

3.使用ThreadLocal来获取存储数据使用：

@Service
public class CategoryServiceImpl implements CategoryService {@Autowiredprivate CategoryMapper categoryMapper;@Overridepublic void add(Category category) {category.setCreateTime(LocalDateTime.now());category.setUpdateTime(LocalDateTime.now());//在ThreadLocal中获取Map<String,Object> map = ThreadLocalUtil.get();Integer userId = (Integer) map.get("id");category.setCreateUser(userId);categoryMapper.add(category);}
}

好了，ThreadLocal的使用就到这里了，感谢收看！！！