经验首页 前端设计 程序设计 Java相关 移动开发 数据库/运维 软件/图像 大数据/云计算 其他经验
当前位置:技术经验 » Java相关 » Java » 查看文章
Java中ThreadLocal的用法和原理
来源:cnblogs  作者:twilight0402  时间:2023/4/12 13:51:16  对本文有异议

用法

  • 隔离各个线程间的数据
  • 避免线程内每个方法都进行传参,线程内的所有方法都可以直接获取到ThreadLocal中管理的对象。
  1. package com.example.test1.service;
  2. import org.springframework.scheduling.annotation.Async;
  3. import org.springframework.stereotype.Component;
  4. import java.text.SimpleDateFormat;
  5. import java.util.Date;
  6. @Component
  7. public class AsyncTest {
  8. // 使用threadlocal管理
  9. private static final ThreadLocal<SimpleDateFormat> dateFormatLocal =
  10. ThreadLocal.withInitial(() -> new SimpleDateFormat("yyyy-MM-dd"));
  11. // 不用threadlocal进行管理,用于对比
  12. SimpleDateFormat dateFormat = new SimpleDateFormat();
  13. // 线程名称以task开头
  14. @Async("taskExecutor")
  15. public void formatDateSync(String format, Date date) throws InterruptedException {
  16. SimpleDateFormat simpleDateFormat = dateFormatLocal.get();
  17. simpleDateFormat.applyPattern(format);
  18. // 所有方法都可以直接使用这个变量,而不用根据形参传入
  19. doSomething();
  20. Thread.sleep(1000);
  21. System.out.println("sync " + Thread.currentThread().getName() + " | " + simpleDateFormat.format(date));
  22. // 线程执行完毕,清除数据
  23. dateFormatLocal.remove();
  24. }
  25. // 线程名称以task2开头
  26. @Async("taskExecutor2")
  27. public void formatDate(String format, Date date) throws InterruptedException {
  28. dateFormat.applyPattern(format);
  29. Thread.sleep(1000);
  30. System.out.println("normal " + Thread.currentThread().getName() + " | " + dateFormat.format(date));
  31. }
  32. }

使用junit进行测试:

  1. @Test
  2. void test2() throws InterruptedException {
  3. for(int index = 1; index <= 10; ++index){
  4. String format = index + "-yyyy-MM-dd";
  5. Date time = new Date();
  6. asyncTest.formatDate(format, time);
  7. }
  8. for(int index = 1; index <= 10; ++index){
  9. String format = index + "-yyyy-MM-dd";
  10. Date time = new Date();
  11. asyncTest.formatDateSync(format, time);
  12. }
  13. }

结果如下,可以看到没有被 ThreadLocal 管理的变量已经无法匹配正确的format。

  1. sync task--10 | 10-2023-04-11
  2. sync task--9 | 9-2023-04-11
  3. normal task2-3 | 2-2023-04-11
  4. normal task2-5 | 2-2023-04-11
  5. normal task2-10 | 2-2023-04-11
  6. normal task2-6 | 2-2023-04-11
  7. sync task--1 | 1-2023-04-11
  8. normal task2-7 | 2-2023-04-11
  9. normal task2-8 | 2-2023-04-11
  10. normal task2-9 | 2-2023-04-11
  11. sync task--6 | 6-2023-04-11
  12. sync task--3 | 3-2023-04-11
  13. sync task--2 | 2-2023-04-11
  14. sync task--7 | 7-2023-04-11
  15. sync task--4 | 4-2023-04-11
  16. sync task--8 | 8-2023-04-11
  17. normal task2-4 | 2-2023-04-11
  18. normal task2-1 | 2-2023-04-11
  19. sync task--5 | 5-2023-04-11
  20. normal task2-2 | 2-2023-04-11

实现原理

ThreadLocal中获取数据的过程:

  • 先获取对应的线程。
  • 通过 getMap(t)拿到线程中的 ThreadLocalMap
  • ThreadLocalMap 是一个重新实现的散列表,基于两个元素实现散列:
    • 用户定义的ThreadLocal对象,例如:dateFormatLocal
    • 封装了valueEntry对象。
  • 通过map.getEntry(this)方法,根据当前的 threadlocal对象在散列表中获得对应的Entry
  • 如果是第一次使用get(), 则使用 setInitialValue()调用用户重写的initialValue()方法创建map并使用用户指定的值初始化。

在这种设计方式下,线程死去的时候,线程共享变量ThreadLocalMap会被销毁。

  1. public 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. @SuppressWarnings("unchecked")
  8. T result = (T)e.value;
  9. return result;
  10. }
  11. }
  12. return setInitialValue();
  13. }

注意 Entry对象是弱引用:

  1. static class Entry extends WeakReference<ThreadLocal<?>> {
  2. /** The value associated with this ThreadLocal. */
  3. Object value;
  4. // k: ThreadLocal, v: value
  5. Entry(ThreadLocal<?> k, Object v) {
  6. super(k);
  7. value = v;
  8. }
  9. }

弱引用的常见用法是:

  1. WeakReference<RoleDTO> weakReference = new WeakReference<>(new RoleDTO());

因此,在Entry中,k 代表ThreadLocal对象,它是弱引用。v代表ThreadLocal管理的那个value,是强引用。

内存泄漏

内存泄漏是指无用对象(不再使用的对象)持续占有内存或无用对象的内存得不到及时释放,从而造成内存空间的浪费称为内存泄漏。随着垃圾回收器活动的增加以及内存占用的不断增加,程序性能会逐渐表现出来下降,极端情况下,会引发OutOfMemoryError导致程序崩溃。

内存泄漏问题主要在线程池中出现,因为线程池中的线程是不断执行的,从任务队列中不断获取新的任务执行。但是任务中可能有ThreadLocal对象,这些对象的ThreadLocal会保存在线程的ThreadLocalMap中,因此ThreadLocalMap会越来越大。

但是ThreadLocal是由任务(worker)传入的,一个任务执行结束后,对应的ThreadLocal对象会被销毁。线程中的关系是: Thread -> ThreadLoalMap -> Entry<ThreadLocal, Object>ThreadLocal由于是弱引用会,在GC的时候会被销毁,这会导致 ThreadLoalMap中存在Entry<null, Object>

使用remove()

由于线程池中的线程一直在运行,如果不对ThreadLoalMap进行清理,那Entry<null, Object>会一直占用内存。remove()方法会清除key==nullEntry

使用static修饰

ThreadLocal设置成static可以避免一个线程类多次传入线程池后重复创建Entry。例如,有一个用户定义的线程

  1. public class Test implements Runnable{
  2. private static ThreadLocal<Integer> local = new ThreadLocal<>();
  3. @Override
  4. public void run() {
  5. // do something
  6. }
  7. }

使用线程池处理10个任务。那么线程池中每个用来处理任务的线程的Thread.ThreadLocalMap中都会保存一个Entry<local, Integer>,由于添加了static关键字,所有每个线程中的Entry中的local变量引用的都是同一个变量。这时就算发生内存泄漏,所有的Test类也只有一个local对象,不会导致内存占用过多。

  1. @Test
  2. void contextLoads() {
  3. Runnable runnable = () -> {
  4. System.out.println(Thread.currentThread().getName());
  5. };
  6. for(int index = 1; index <= 10; ++index){
  7. taskExecutor2.submit(new com.example.test1.service.Test());
  8. }
  9. }

原文链接:https://www.cnblogs.com/twilight0402/p/17309557.html

 友情链接:直通硅谷  点职佳  北美留学生论坛

本站QQ群:前端 618073944 | Java 606181507 | Python 626812652 | C/C++ 612253063 | 微信 634508462 | 苹果 692586424 | C#/.net 182808419 | PHP 305140648 | 运维 608723728

W3xue 的所有内容仅供测试,对任何法律问题及风险不承担任何责任。通过使用本站内容随之而来的风险与本站无关。
关于我们  |  意见建议  |  捐助我们  |  报错有奖  |  广告合作、友情链接(目前9元/月)请联系QQ:27243702 沸活量
皖ICP备17017327号-2 皖公网安备34020702000426号