如何解决异步任务上下文丢失问题
更新时间:2024年09月18日 10:48:18 作者:勿语&
在多线程编程中,异步任务可能会导致上下文信息丢失,为了解决这个问题,可以在执行异步任务前,通过自定义TaskDecorator拷贝主线程的上下文至子线程,这样可以确保上下文在异步执行过程中得以保留,将定制的TaskDecorator设置至线程池,可以有效地解决上下文丢失问题
解决异步任务上下文丢失问题
- 上下文丢失主要是因为主线程和子线程的上下文不能共享。
- 可以通过执行异步任务之前,将主线程上的上下文信息拷贝到子线程上。
自定义TaskDecorator 来拷贝主线程上的上下文信息到子线程,然后将自定义的 TaskDecorator实现类 设置到线程池上。
@Configuration public class ThreadPoolConfig { @Bean(name = "customizeTaskExecutor") public ThreadPoolTaskExecutor threadPoolTaskExecutor(){ ThreadPoolTaskExecutor poolExecutor = new ThreadPoolTaskExecutor(); // 核心线程数=cpu核心数+1 poolExecutor.setCorePoolSize(5); // 最大线程数=cpu核心数*2 poolExecutor.setMaxPoolSize(8); // 设置任务装饰器 poolExecutor.setTaskDecorator(taskDecorator()); // 任务被拒绝后,交给调用线程执行 poolExecutor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy()); return poolExecutor; } @Bean public TaskDecorator taskDecorator(){ return new TaskDecorator() { @Override public Runnable decorate(Runnable runnable) { RequestAttributes requestAttributes = RequestContextHolder.getRequestAttributes(); return ()->{ try { RequestContextHolder.setRequestAttributes(requestAttributes); runnable.run(); }finally { RequestContextHolder.resetRequestAttributes(); } }; } }; } }
Fegin异步情况丢失上下文问题
在微服务的开发中,我们经常需要服务之间的调用,并且为了提高效率使用异步的方式进行服务之间的调用,在这种异步的调用情况下会有一个严重的问题,丢失上文下
通过以上图片可以看出异步丢失上下文的原因是不在同一个线程,所有数据不能共享,Wie了解决这个问题,我们就需要把之前线程的请求头上下文,在次存放到其他线程的请求头上下文就行,具体实现如下:
案例:feign异步获取订单明细的案例代码
/** * 获取订单明细的vo * @return */ @Override public OrderConfirmVo orderConfirm() { MemberResponseVo member = OrderInterceptor.threadLocal.get(); OrderConfirmVo orderConfirmVo = new OrderConfirmVo(); System.out.println("主线程:"+ Thread.currentThread().getId()); //获取主线程的请求头信息 RequestAttributes requestAttributes = RequestContextHolder.getRequestAttributes(); //考虑到效率问题 使用异步编排 CompletableFuture<Void> getAddress = CompletableFuture.runAsync(() -> { //子线程中设置添加主线程的请求头信息 信息共享 否则远程调用异步处理丢失请求头信息 RequestContextHolder.setRequestAttributes(requestAttributes); System.out.println("address:"+ Thread.currentThread().getId()); //远程获取地址信息 List<MemberAddressVo> address = memberFeignService.getAddress(member.getId()); orderConfirmVo.setAddress(address); }, executor); CompletableFuture<Void> getItem = CompletableFuture.runAsync(() -> { //子线程中设置添加主线程的请求头信息 信息共享 否则远程调用异步处理丢失请求头信息 RequestContextHolder.setRequestAttributes(requestAttributes); System.out.println("item:"+ Thread.currentThread().getId()); //远程获取购物项 List<OrderItemVo> currentUserCartItems = cartFeignService.getCurrentUserCartItems(); orderConfirmVo.setItems(currentUserCartItems); }, executor).thenRunAsync(()->{ List<OrderItemVo> items = orderConfirmVo.getItems(); //获取所有商品的id List<String> collect = items.stream().map(item -> item.getSkuId()).collect(Collectors.toList()); List<Long> skuIds = collect.stream().map(item -> { return Long.parseLong(item); }).collect(Collectors.toList()); R<List<SkuHasStockVo>> skusHasStock = wmsFeignService.getSkusHasStock(skuIds); List<SkuHasStockVo> data = skusHasStock.getData(new TypeReference<List<SkuHasStockVo>>() { }); if(data!= null){ Map<Long, Boolean> collect1 = data.stream().collect(Collectors.toMap(SkuHasStockVo::getSkuId, SkuHasStockVo::getHasStock)); orderConfirmVo.setStocks(collect1); } },executor); //异步编排完成之后执行后续操作 try { CompletableFuture.allOf(getAddress,getItem).get(); } catch (InterruptedException e) { e.printStackTrace(); } catch (ExecutionException e) { e.printStackTrace(); } orderConfirmVo.setIntegration(member.getIntegration()); orderConfirmVo.setPayPrice(orderConfirmVo.getPayPrice()); orderConfirmVo.setTotal(orderConfirmVo.getTotal()); //TODO 放重处理 生成token令牌储存在redis String token = UUID.randomUUID().toString().replace("-", ""); orderConfirmVo.setOrderToken(token); redisTemplate.opsForValue().set(OrderConstant.ORDER_TOKEN+member.getId(),token); return orderConfirmVo; }
总结
以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。
相关文章
SpringBoot中使用websocket出现404的解决方法
在Springboot中使用websocket时,本地开发环境可以正常运行,但部署到服务器环境出现404问题,所以本文小编讲给大家详细介绍一下SpringBoot中使用websocket出现404的解决方法,需要的朋友可以参考下2023-09-09springboot vue接口测试前后端树节点编辑删除功能
这篇文章主要为大家介绍了springboot vue接口测试前后端树节点编辑删除功能,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪2022-05-05Spring JdbcTemplate实现添加与查询方法详解
JdbcTemplate是Spring框架自带的对JDBC操作的封装,目的是提供统一的模板方法使对数据库的操作更加方便、友好,效率也不错,这篇文章主要介绍了Spring JdbcTemplate执行数据库操作,需要的朋友可以参考下2022-11-11
最新评论