在Hystrix系列之前的文章中提到过,如果使用线程池模式,那么存在一个ThreadLocal变量跨线程传递的问题,即在主线程的ThreadLocal变量,无法在线程池中使用,不过Hystrix内部提供了解决方案,但是个人觉得这个方案不是那么友好。
解决方案
在Hystrix中,如果想在跨线程时共享数据,必须通过HystrixRequestVariableDefault
申明变量
HystrixRequestVariableDefault name = new HystrixRequestVariableDefault();
name.set("占小狼");
其实在用法上,和ThreadLocal
是一样的,只是需要对现有代码的大量改造。
看下这个方案的实现原理,先从set
开始。
public void set(T value) {
HystrixRequestContext.getContextForCurrentThread().state.put(this, new LazyInitializer<T>(this, value));
}
Hystrix内部通过HystrixRequestContext
实现数据的跨线程传递,getContextForCurrentThread
得到的是当前线程的HystrixRequestContext
对象,每个HystrixRequestContext
对象都有一个对应ConcurrentHashMap
变量state
,负责保存通过HystrixRequestVariableDefault
初始化的数据。
通过set
方法,该对象和数据会被保存在一个当前线程所属的map中。为了实现数据的跨线程传递,只需要在初始化task的时候,把主线程的HystrixRequestContext
变量保存起来,在task执行的时候,重新赋值到子线程的上下文中,这样在子线程中就可以顺利拿到这些数据。
Hystrix中通过HystrixContextCallable
包装原始Callable
,并使用parentThreadState
保存了当前线程的HystrixRequestContext
变量。
任务执行时,先保存子线程现有的HystrixRequestContext
变量,再赋值主线程的HystrixRequestContext
变量,任务执行完成后,重新还原子线程。
如果不想使用Hystrix这种方式实现,也可以使用Hystrix提供的插件方式重新包装task,通过实现HystrixConcurrencyStrategy
类,重写wrapCallable
方法,和Hystrix的实现原理类似。
比如提供一个继承ThreadLocal
的XXXThreadLocal
类,那么业务方在使用时,就可以这样使用。
ThreadLocal name = new XXXThreadLocal();
name.set("占小狼");
这种方式看起来对已有逻辑只有一点小小的改动,对于新接入的也不那么陌生。