使用Hystrix的插件机制,解决在使用线程隔离时,threadlocal的传递问题

final int keepAliveTime = threadPoolProperties.keepAliveTimeMinutes().get();
final int maxQueueSize = threadPoolProperties.maxQueueSize().get();


// 1
return new ThreadPoolExecutor(dynamicCoreSize, dynamicCoreSize, keepAliveTime, TimeUnit.MINUTES, workQueue, threadFactory);
}
}

上面的1处,会去创建线程池。但是,这里直接就是要了 jdk 的默认线程池类来创建,这还怎么搞?类型都定死了。没法扩展了。。。

发现hystrix的插件机制

但是,回过头来,又仔细看了看,这个getThreadPool 是 HystrixConcurrencyStrategy类的一个方法,这个方法也是个实例方法。

方法不能改,那,实例能换吗?再看看前面的代码:

ok,那接着分析:

public HystrixConcurrencyStrategy getConcurrencyStrategy() {
if (concurrencyStrategy.get() == null) {
//1 check for an implementation from Archaius first
Object impl = getPluginImplementation(HystrixConcurrencyStrategy.class);
concurrencyStrategy.compareAndSet(null, (HystrixConcurrencyStrategy) impl);
}
return concurrencyStrategy.get();
}

1处,根据这个类,获取实现,感觉有点戏。

// 1
T p = getPluginImplementationViaProperties(pluginClass, dynamicProperties);
if (p != null) return p;
// 2
return findService(pluginClass, classLoader);
}

  • 1处,从一个动态属性中获取,后来经查,发现是如果集成了Netflix Archaius就可以动态获取属性,类似于一个配置中心

  • 2处,如果前面没找到,就是要 JDK 的SPI机制。

private static T findService(
Class spi,
ClassLoader classLoader) throws ServiceConfigurationError {

ServiceLoader sl = ServiceLoader.load(spi,
classLoader);
for (T s : sl) {
if (s != null)
return s;
}
return null;
}

那就好说了。SPI ,我们自定义一个实现,就可以替换掉默认的了,hystrix做的还是不错,扩展性可以。

现在知道可以自定义HystrixConcurrencyStrategy了,那要怎么自定义呢?

这个类,是个抽象类,大体有如下几个方法:

getThreadPool

getBlockingQueue(int maxQueueSize)

Callable wrapCallable(Callable callable)

getRequestVariable(final HystrixRequestVariableLifecycle rv)

说是抽象类,但其实并没有需要我们实现的方法,所有方法都有默认实现,我们只需要重写需要覆盖的方法即可。

我这里,看重了第三个方法:

/**

  • Provides an opportunity to wrap/decorate a {@code Callable} before execution.
  • This can be used to inject additional behavior such as copying of thread state (such as {@link ThreadLocal}).
  • Default Implementation
  • Pass-thru that does no wrapping.
  • @param callable
  •        {@code Callable<T>} to be executed via a {@link ThreadPoolExecutor}
    
  • @return {@code Callable} either as a pass-thru or wrapping the one given
    */
    public Callable wrapCallable(Callable callable) {
    return callable;
    }

方法注释如上,我简单说下,在执行前,提供一个机会,让你去wrap这个callable,即最终要丢到线程池执行的那个callable。

我们可以wrap一下原有的callable,在执行前,把当前线程的threadlocal变量存下来,即为A,然后设置到callable里面去;在callable执行的时候,就可以使用我们的A中的threadlocal来替换掉worker线程中的。

多说无益,这里直接看代码:

// 0
public class MyHystrixConcurrencyStrategy extends HystrixConcurrencyStrategy {

@Override
public Callable wrapCallable(Callable callable) {
/**

  • 1 获取当前线程的threadlocalmap
    */
    Object currentThreadlocalMap = getCurrentThreadlocalMap();

Callable finalCallable = new Callable() {
// 2
private Object callerThreadlocalMap = currentThreadlocalMap;
// 3
private Callable targetCallable = callable;

@Override
public T call() throws Exception {
/**

  • 4 将工作线程的原有线程变量保存起来
    /
    Object oldThreadlocalMapOfWorkThread = getCurrentThreadlocalMap();
    /
    *
    *5 将本线程的线程变量,设置为caller的线程变量
    */
    setCurrentThreadlocalMap(callerThreadlocalMap);

try {
// 6
return targetCallable.call();
}finally {
// 7
setCurrentThreadlocalMap(oldThreadlocalMapOfWorkThread);
log.info(“restore work thread’s threadlocal”);
}

}
};

return finalCallable;
}

  • 0处,自定义了一个类,继承HystrixConcurrencyStrategy,准备覆盖其默认的wrap方法
  • 1处,获取外部线程的threadlocal
  • 2处,3处,这里已经是处于匿名内部类了,定义了2个field,分别存放1中的外部线程的threadlocal,以及要wrap的callable
  • 4处,此时已经处于run方法的执行逻辑了:保存worker线程的自身的线程局部变量
  • 5处,使用外部线程的threadlocal覆盖自身的
  • 6处,调用真正的业务逻辑
  • 7处,恢复为线程自身的threadlocal

获取线程的threadlocal的代码:

private Object getCurrentThreadlocalMap() {
Thread thread = Thread.currentThread();
try {
Field field = Thread.class.getDeclaredField(“threadLocals”);
field.setAccessible(true);
Object o = field.get(thread);
return o;
} catch (NoSuchFieldException | IllegalAccessException e) {
log.error(“{}”,e);
}
return null;
}

设置线程的threadlocal的代码:

private void setCurrentThreadlocalMap(Object newThreadLocalMap) {
Thread thread = Thread.currentThread();
try {
Field field = Thread.class.getDeclaredField(“threadLocals”);
field.setAccessible(true);
field.set(thread,newThreadLocalMap);

} catch (NoSuchFieldException | IllegalAccessException e) {
log.error(“{}”,e);
}
}

插件机制的相关资料

https://github.com/Netflix/Hystrix/wiki/Plugins

运行效果

controller代码

@RequestMapping(“/”)
public String hystrixOrder () {
// 1
SessionUtils.getSessionVOFromRedisAndPut2ThreadLocal();
// 2
SimpleHystrixCommand simpleHystrixCommand = new SimpleHystrixCommand(testService);
String res = simpleHystrixCommand.execute();
return res;
}

  • 1处,设置ThreadLocal变量

public static UserVO getSessionVOFromRedisAndPut2ThreadLocal() {
UserVO userVO = new UserVO();
userVO.setUserName(“test user”);

RequestContextHolder.set(userVO);
log.info(“set thread local:{} to context”,userVO);

return userVO;
}

  • 2处,new了一个HystrixCommand,然后execute执行

command中代码

public class SimpleHystrixCommand extends HystrixCommand {

private TestService testService;

public SimpleHystrixCommand(TestService testService) {
super(setter());
this.testService = testService;
}

@Override
protected String run() throws Exception {
// 1
String s = testService.getResult();
log.info(“get thread local:{}”,s);

/**

  • 如果睡眠时间,超过2s,会降级
  • {@link #getFallback()}
    */
    int millis = new Random().nextInt(3000);
    log.info(“will sleep {} millis”,millis);
    Thread.sleep(millis);

return s;
}

重点看1处代码:

public String getResult() {
UserVO userVO = RequestContextHolder.get();
log.info(“I am hystrix pool thread,try to get threadlocal:{}”,userVO);

return userVO.toString();
}

如上所示,会去获取ThreadLocal变量,并打印。

spi配置

在resources\META-INF\services目录下,创建文件:

com.netflix.hystrix.strategy.concurrency.HystrixConcurrencyStrategy

内容为下面一行:

com.learn.hystrix.utils.MyHystrixConcurrencyStrategy

执行效果

访问:http://localhost:8080/

2020-05-09 17:26:11.134 INFO 7452 — [nio-8080-exec-2] com.learn.hystrix.utils.SessionUtils : set thread local:UserVO(userName=test user) to context
2020-05-09 17:26:11.143 INFO 7452 — [x-member-pool-2] com.learn.hystrix.service.TestService : I am hystrix pool thread,try to get threadlocal:UserVO(userName=test user)
2020-05-09 17:26:11.143 INFO 7452 — [x-member-pool-2] c.l.h.command.SimpleHystrixCommand : get thread local:UserVO(userName=test user)
2020-05-09 17:26:11.144 INFO 7452 — [x-member-pool-2] c.l.h.command.SimpleHystrixCommand : will sleep 126 millis
2020-05-09 17:26:11.281 INFO 7452 — [x-member-pool-2] c.l.h.u.MyHystrixConcurrencyStrategy : restore work thread’s threadlocal

可以看到,已经发生了线程切换,在worker线程也取到了。

大家如果发现日志中出现了[ HystrixTimer-1] 线程的身影,不用担心,那只是因为我们的线程超时了,所以timer线程检测到了之后,去执行一个callable任务,那个runnable就是前面被我们包装过的那个callable。(这块超时的机制,todo吧,下次再讲)

总结

自我介绍一下,小编13年上海交大毕业,曾经在小公司待过,也去过华为、OPPO等大厂,18年进入阿里一直到现在。

深知大多数Java工程师,想要提升技能,往往是自己摸索成长或者是报班学习,但对于培训机构动则几千的学费,着实压力不小。自己不成体系的自学效果低效又漫长,而且极易碰到天花板技术停滞不前!

因此收集整理了一份《2024年Java开发全套学习资料》,初衷也很简单,就是希望能够帮助到想自学提升又不知道该从何学起的朋友,同时减轻大家的负担。img

既有适合小白学习的零基础资料,也有适合3年以上经验的小伙伴深入学习提升的进阶课程,基本涵盖了95%以上Java开发知识点,真正体系化!

由于文件比较大,这里只是将部分目录截图出来,每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频,并且会持续更新!

如果你觉得这些内容对你有帮助,可以扫码获取!!(备注Java获取)

img

最后

ActiveMQ消息中间件面试专题

  • 什么是ActiveMQ?
  • ActiveMQ服务器宕机怎么办?
  • 丢消息怎么办?
  • 持久化消息非常慢怎么办?
  • 消息的不均匀消费怎么办?
  • 死信队列怎么办?
  • ActiveMQ中的消息重发时间间隔和重发次数吗?

ActiveMQ消息中间件面试专题解析拓展:

BAT面试文档:ActiveMQ+redis+Spring+高并发多线程+JVM


redis面试专题及答案

  • 支持一致性哈希的客户端有哪些?
  • Redis与其他key-value存储有什么不同?
  • Redis的内存占用情况怎么样?
  • 都有哪些办法可以降低Redis的内存使用情况呢?
  • 查看Redis使用情况及状态信息用什么命令?
  • Redis的内存用完了会发生什么?
  • Redis是单线程的,如何提高多核CPU的利用率?

BAT面试文档:ActiveMQ+redis+Spring+高并发多线程+JVM


Spring面试专题及答案

  • 谈谈你对 Spring 的理解
  • Spring 有哪些优点?
  • Spring 中的设计模式
  • 怎样开启注解装配以及常用注解
  • 简单介绍下 Spring bean 的生命周期

Spring面试答案解析拓展

BAT面试文档:ActiveMQ+redis+Spring+高并发多线程+JVM


高并发多线程面试专题

  • 现在有线程 T1、T2 和 T3。你如何确保 T2 线程在 T1 之后执行,并且 T3 线程在 T2 之后执行?
  • Java 中新的 Lock 接口相对于同步代码块(synchronized block)有什么优势?如果让你实现一个高性能缓存,支持并发读取和单一写入,你如何保证数据完整性。
  • Java 中 wait 和 sleep 方法有什么区别?
  • 如何在 Java 中实现一个阻塞队列?
  • 如何在 Java 中编写代码解决生产者消费者问题?
  • 写一段死锁代码。你在 Java 中如何解决死锁?

高并发多线程面试解析与拓展

BAT面试文档:ActiveMQ+redis+Spring+高并发多线程+JVM


jvm面试专题与解析

  • JVM 由哪些部分组成?
  • JVM 内存划分?
  • Java 的内存模型?
  • 引用的分类?
  • GC什么时候开始?

JVM面试专题解析与拓展!

BAT面试文档:ActiveMQ+redis+Spring+高并发多线程+JVM

《一线大厂Java面试题解析+核心总结学习笔记+最新讲解视频+实战项目源码》点击传送门即可获取!

  • 写一段死锁代码。你在 Java 中如何解决死锁?

高并发多线程面试解析与拓展

[外链图片转存中…(img-pMFTHAq7-1711885713934)]


jvm面试专题与解析

  • JVM 由哪些部分组成?
  • JVM 内存划分?
  • Java 的内存模型?
  • 引用的分类?
  • GC什么时候开始?

JVM面试专题解析与拓展!

[外链图片转存中…(img-QQTubRsC-1711885713934)]

《一线大厂Java面试题解析+核心总结学习笔记+最新讲解视频+实战项目源码》点击传送门即可获取!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值