阿里开源的TransmittableThreadLocal的正确使用姿势

TransmittableThreadLocal是阿里巴巴开源的一个线程本地变量,它是ThreadLocal的一个增强版,可以在线程池等多线程环境下使用,解决了ThreadLocal在多线程环境下的一些问题。在多线程环境下,ThreadLocal可以避免线程安全问题,但是在使用线程池等多线程环境时,ThreadLocal可能会出现一些问题。例如,当使用线程池时,线程池中的线程可能会被多个任务共享,

 TransmittableThreadLocal是阿里巴巴开源的一个线程本地变量,它是ThreadLocal的一个增强版,可以在线程池等多线程环境下使用,解决了ThreadLocal在多线程环境下的一些问题。

在多线程环境下,ThreadLocal可以避免线程安全问题,但是在使用线程池等多线程环境时,ThreadLocal可能会出现一些问题。例如,当使用线程池时,线程池中的线程可能会被多个任务共享,如果使用ThreadLocal存储数据,可能会导致数据被错误地共享。

        TransmittableThreadLocal解决了这个问题,它可以在多线程环境下正确地传递数据。具体来说,当一个线程从线程池中获取到一个线程时,TransmittableThreadLocal会自动将当前线程的ThreadLocal副本传递给新的线程,从而保证数据的正确性。

        下面有几个简单的示例,演示了如何使用TransmittableThreadLocal

1. 直接使用 TransmittableThreadLocal

public class TransmittableThreadLocalDemo {
    private static TransmittableThreadLocal<String> threadLocal = new TransmittableThreadLocal<>();

    public static void main(String[] args) {
        // 在主线程中设置TransmittableThreadLocal的值
        threadLocal.set("Hello, World!");

        // 在新线程中获取TransmittableThreadLocal的值
        Thread thread = new Thread(() -> {
            String value = threadLocal.get();
            System.out.println("TransmittableThreadLocal value in new thread: " + value);
        });
        thread.start();
    }
}

 在这个示例中,我们首先在主线程中将一个字符串存储到TransmittableThreadLocal中。然后,我们创建一个新线程,在新线程中获取TransmittableThreadLocal的值并打印出来。由于TransmittableThreadLocal可以正确地传递数据,因此新线程中获取到的值与主线程中设置的值是相同的。

        需要注意的是,TransmittableThreadLocal是ThreadLocal的一个增强版,因此在使用时需要注意一些细节。例如,TransmittableThreadLocal的使用方式与ThreadLocal略有不同,需要使用TtlRunnable和TtlCallable等类来包装Runnable和Callable。此外,TransmittableThreadLocal也需要进行垃圾回收,否则可能会导致内存泄漏

2. 在使用线程池的情况下使用 TransmittableThreadLocal

        在使用线程池的情况下,如果需要使用TransmittableThreadLocal,可以使用阿里巴巴的TTL库来实现。TTL库提供了一些工具类,可以方便地在使用线程池的情况下使用TransmittableThreadLocal。

        具体来说, 可以使用TtlRunnable和TtlCallable等类来包装 Runnable 和 Callable,从而在执行任务时自动传递ThreadLocal的值。例如,下面是一个使用线程池和TransmittableThreadLocal的示例:

public class ThreadPoolDemo {
    private static TransmittableThreadLocal<String> threadLocal = new TransmittableThreadLocal<>();

    public static void main(String[] args) throws InterruptedException {
        ExecutorService executorService = Executors.newFixedThreadPool(2);

        // 在主线程中设置TransmittableThreadLocal的值
        threadLocal.set("Hello, World!");

        // 在线程池中执行任务
        executorService.execute(TtlRunnable.get(() -> {
            String value = threadLocal.get();
            System.out.println("TransmittableThreadLocal value in new thread: " + value);
        }));

        // 等待任务执行完成
        executorService.shutdown();
        executorService.awaitTermination(1, TimeUnit.SECONDS);
    }
}

  在这个示例中,我们首先创建了一个线程池,然后在主线程中设置了一个TransmittableThreadLocal的值。接着,我们使用TtlRunnable包装了一个Runnable任务,并将其提交到线程池中执行。在执行任务时,TtlRunnable会自动传递ThreadLocal的值,从而保证了数据的正确性


3. 使用 TtlExecutors 的使用案例 (推荐)

        TtlExecutors是阿里巴巴开源的一个工具类,用于在使用线程池的情况下使用TransmittableThreadLocal。具体来说,TtlExecutors提供了一些工具方法,可以方便地将线程池和TransmittableThreadLocal结合起来使用。

下面是一个使用TtlExecutors的示例代码:      

import com.alibaba.ttl.TransmittableThreadLocal;
import com.alibaba.ttl.threadpool.TtlExecutors;

import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

public class TtlExecutorsDemo {
    private static TransmittableThreadLocal<String> threadLocal = new TransmittableThreadLocal<>();

    public static void main(String[] args) {
        // 创建一个固定大小的线程池
        ExecutorService executorService = Executors.newFixedThreadPool(2);

        // 在主线程中设置TransmittableThreadLocal的值
        threadLocal.set("Hello, World!");

        // 使用TtlExecutors包装线程池
        ExecutorService ttlExecutorService = TtlExecutors.getTtlExecutorService(executorService);

        // 提交任务到线程池中
        ttlExecutorService.submit(() -> {
            String value = threadLocal.get();
            System.out.println("Task is running in thread: " + Thread.currentThread().getName() + ", TransmittableThreadLocal value: " + value);
        });

        // 关闭线程池
        ttlExecutorService.shutdown();
    }
}

在这个示例中,我们首先创建了一个固定大小的线程池,并在主线程中设置了一个TransmittableThreadLocal的值。接着,我们使用TtlExecutors.getTtlExecutorService()方法将线程池包装成一个支持TransmittableThreadLocal的线程池。最后,我们使用包装后的线程池提交了一个任务,并在任务中获取了TransmittableThreadLocal的值

4. 拓展

        除了在使用线程池的情况下使用TransmittableThreadLocal外,还有一些其他的使用案例。

例如,在使用Dubbo等分布式框架时,TransmittableThreadLocal可以用于在服务调用链路中传递数据。具体来说,可以在服务提供者和消费者中使用TransmittableThreadLocal来传递一些上下文信息,例如请求ID、用户信息等。这样可以方便地在服务调用链路中获取这些信息,从而实现一些功能,例如日志追踪、权限校验等。

另外,TransmittableThreadLocal还可以用于一些需要在多个线程之间共享数据的场景。例如,在使用Netty等网络编程框架时,可以使用TransmittableThreadLocal来在不同的事件处理器之间共享一些上下文信息,例如连接信息、请求信息等。

总之,TransmittableThreadLocal可以用于在多线程环境下传递数据,解决了ThreadLocal在多线程环境下的一些问题。在使用TransmittableThreadLocal时需要注意一些细节,例如需要进行垃圾回收等。同时,也需要根据具体的使用场景来选择合适的使用方式。

  • 20
    点赞
  • 23
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值