spring boot 并发请求，其他系统接口，丢失request的header信息【多线程、线程池、@Async 】
spring boot 并发请求，其他系统接口，丢失request的header信息【多线程、线程池、@Async 】
场景：一次迭代在灰度环境发版时，测试反馈说我开发的那个功能，查询接口有部分字段数据是空的，后续排查日志，发现日志如下：
feign.RetryableException: cannot retry due to redirection, in streaming mode executing POST
下面是业务、环境和分析过程下面是业务、环境和分析过程：
接口的业务场景 ：我这个接口类似是那种报表统计的接口，它会请求多个微服务，把请求到的数据，统一返回给前端，相当于设计模式中的门面模式了。
后续由于这个接口 是串行请求其他微服务的，速度有些慢，后面修改代码从串行请求，改成并行（多线程）获取数据
运维那边是通过判断http请求中cookie 或者 header中的某个数据，来区分请求是否要把流量打到灰度。
分析得出：应该是接口异步请求的时候cookie丢失，没走到灰度环境，找不到 这次迭代新开发的接口，导致的重定向到错误页面了。
验证：由于我代码是通过@Async异步注解，实现并行请求的，临时把五个接口的异步注解注释掉了，灰度在发版验证，数据能返回正常，说明流量打到灰度了
说明问题就是并发请求的时候，子线程获取不到 主线程的request 头信息，导致没有走到灰度
下图就是灰度环境的 流程图：
问题定位出来了，解决方案就是：让子线程能获取到主线程的 request 头信息，主线程把 数据透传到子线程。
我使用的是RequestContextHolder来透传数据
什么是 RequestContextHolder？
RequestContextHolder 是spring mvc的一个工具类，顾名思义,持有上下文的Request容器
如何使用：
//获取当前线程 request请求的属性
RequestAttributes requestAttributes = RequestContextHolder.currentRequestAttributes();
//设置当前线程 request请求的属性
RequestContextHolder.setRequestAttributes(attributes);
RequestContextHolder的 会用到的几个方法
currentRequestAttributes：获得当前线程请求的属性（头信息之类的）
setRequestAttributes(attributes)：设置当前线程 属性（设置头信息）
resetRequestAttributes：删除当前线程 绑定的属性
下面是他们的源码，可以简单看一下，原理是通过ThreadLocal来绑定数据的：
private static final ThreadLocal<RequestAttributes> requestAttributesHolder = new NamedThreadLocal<>("Request attributes"); private static final ThreadLocal<RequestAttributes> inheritableRequestAttributesHolder = new NamedInheritableThreadLocal<>("Request context");
//获得当前线程请求的属性（头信息之类的）
@Nullable public static RequestAttributes getRequestAttributes() {
RequestAttributes attributes = requestAttributesHolder.get(); if (attributes == null) {
attributes = inheritableRequestAttributesHolder.get();
} return attributes;
}
//设置当前线程 属性（设置头信息）
public static void setRequestAttributes(@Nullable RequestAttributes attributes) {
setRequestAttributes(attributes, false);
} //设置当前线程 属性（设置头信息）
public static void setRequestAttributes(@Nullable RequestAttributes attributes, boolean inheritable) { if (attributes == null) {
resetRequestAttributes();
} else { if (inheritable) {
inheritableRequestAttributesHolder.set(attributes);
requestAttributesHolder.remove();
} else {
requestAttributesHolder.set(attributes);
inheritableRequestAttributesHolder.remove();
}
}
} //删除当前线程 绑定的属性
public static void resetRequestAttributes() {
requestAttributesHolder.remove();
inheritableRequestAttributesHolder.remove();
}下面我编写了一套遇到问题的代码例子,以及解决的代码：
TestUserController
测试接口
@Slf4j@RestController@RequestMapping(value = "/v1/testUser")public class TestUserController { @Autowired
ITestRequestService testRequestService; @ApiOperation(value = "聚合数据接口(一)-串行获取数据")
@RequestMapping(value = "/listUser", method = RequestMethod.GET)
public Resp<List<User>> listUser(@RequestHeader(value = "token",required = false)String token){
TimeInterval timeInterval = DateUtil.timer();
DataResp dataResp = testRequestService.getDateResp();
log.info("聚合数据接口(一)-串行获取数据 总耗时：{}毫秒",timeInterval.interval()); return Resp.buildDataSuccess(dataResp).setTimeInterval(timeInterval.interval());
} @ApiOperation(value = "聚合数据接口(二)-并行获取数据@Async （子线程获取不到token）")
@RequestMapping(value = "/listUser2", method = RequestMethod.GET)
public Resp<List<User>> listUser2(@RequestHeader(value = "token",required = false)String token) throws ExecutionException, InterruptedException {
TimeInterval timeInterval = DateUtil.timer();
DataResp dataResp = testRequestService.getDateResp2();
log.info("聚合数据接口(二)-并行获取数据@Async （子线程获取不到token） 总耗时：{}毫秒",timeInterval.interval()); return Resp.buildDataSuccess(dataResp).setTimeInterval(timeInterval.interval());
} @ApiOperation(value = "聚合数据接口(三)-并行获取数据（子线程能获取到token）")
@RequestMapping(value = "/listUser3", method = RequestMethod.GET)
public Resp<List<User>> listUser3(@RequestHeader(value = "token",required = false)String token) throws ExecutionException, InterruptedException {
TimeInterval timeInterval = DateUtil.timer();
DataResp dataResp = testRequestService.getDateResp3();
log.info("聚合数据接口(三)-并行获取数据（子线程能获取到token） 总耗时：{}毫秒",timeInterval.interval()); return Resp.buildDataSuccess(dataResp).setTimeInterval(timeInterval.interval());
}
}TestRequestService
聚合数据的类
@Servicepublic class TestRequestService implements ITestRequestService {
@Autowired
IUserService userService;
@Autowired
IOrderService orderService; /**
* 自定义 - 线程池
*/
private static final ThreadPoolExecutor executorService = new ThreadPoolExecutor(50, 200, 180L, TimeUnit.SECONDS, new LinkedBlockingQueue<Runnable>(3000), new ThreadFactory() { final ThreadFactory defaultFactory = Executors.defaultThreadFactory();
@Override public Thread newThread(Runnable r) {
Thread thread = defaultFactory.newThread(r);
thread.setName("testRequest - " + thread.getName()); return thread;
}
}, new ThreadPoolExecutor.CallerRunsPolicy()); /**
* 聚合接口-串行获取数据
* @return
*/
@Override public DataResp getDateResp(){ //获取用户列表
List<User> userList = userService.listUser_1(); //获取订单列表
List<Order> orderList = orderService.listOrder_1(); return DataResp.builder().userList(userList).orderList(orderList).build();
}; /**
* 聚合接口-并行获取数据(@Async) 头信息传到子线程
* @return
*/
@Override public DataResp getDateResp2() throws ExecutionException, InterruptedException { //获取用户列表 start
Future<List<User>> userListFuture = userService.listUser_2(); List<User> userList = userListFuture.get(); //获取用户列表 end
//获取订单列表 start
Future<List<Order>> orderListFuture = orderService.listOrder_2(); List<Order> orderList = orderListFuture.get(); //获取订单列表 end
return DataResp.builder().userList(userListFuture.get()).orderList(orderList).build();
}; /**
* 聚合接口-并行获取数据
* @return
*/
@Override public DataResp getDateResp3() throws ExecutionException, InterruptedException {
RequestAttributes attributes = RequestContextHolder.getRequestAttributes(); //获取用户列表 start
Future<List<User>> userListFuture = CompletableFuture.supplyAsync(() -> {
RequestContextHolder.setRequestAttributes(attributes); try { List<User> resp = userService.listUser_3(); return resp;
}finally {
RequestContextHolder.resetRequestAttributes();
}
}, executorService); List<User> userList = userListFuture.get(); //获取用户列表 end
//获取订单列表 start
Future<List<Order>> orderListFuture = CompletableFuture.supplyAsync(() -> {
RequestContextHolder.setRequestAttributes(attributes); try { List<Order> resp = orderService.listOrder_3(); return resp;
}finally {
RequestContextHolder.resetRequestAttributes();
}
}, executorService); List<Order> orderList = orderListFuture.get(); //获取订单列表 end
return DataResp.builder().userList(userListFuture.get()).orderList(orderList).build();
};
}下面是两个请求 用户和订单请求类
OrderService 请求订单的服务的聚合方法
@Slf4j@Servicepublic class OrderService implements IOrderService { /**
* 获取订单code列表
* @return
*/
@Override
public List<String> listOrderCode(){ //使用httpUtil 模拟 feign请求服务接口 start
String reqUrl = Config.baseUrl.concat("/v1/order/list");
HttpRequest httpRequest = HttpUtil.createGet(reqUrl); //设置请求头信息
String token = WebUtil.getCurrentRequestHeaderToken();
httpRequest.header("token",token);
HttpResponse httpResponse = httpRequest.execute(); String body = httpResponse.body();
Resp<List<String>> respData = JSONUtil.toBean(body, Resp.class); //使用httpUtil 模拟 feign请求服务接口 end
if(respData.isSuccess()){ return respData.getData();
} return null;
}; /**
* 根据订单code获取 订单数据
* @param orderCode
* @return
*/
@Override
public Order getOrder(String orderCode){ //使用httpUtil 模拟 feign请求服务接口 start
String reqUrl = StrUtil.format(Config.baseUrl.concat("/v1/order/get?orderCode={}"),orderCode);
HttpRequest httpRequest = HttpUtil.createGet(reqUrl); //设置请求头信息
String token = WebUtil.getCurrentRequestHeaderToken();
httpRequest.header("token",token);
HttpResponse httpResponse = httpRequest.execute(); String body = httpResponse.body();
Gson gson = new Gson();
Resp<Order> respData = gson.fromJson(body , new TypeToken<Resp<Order>>(){}.getType()); //使用httpUtil 模拟 feign请求服务接口 end
if(respData.isSuccess()){ return respData.getData();
} return null;
}; /**
* 获取订单列表（串行获取）
* @return
*/
@Override
public List<Order> listOrder_1(){ //获取订单列表 start
List<Order> orderList = new ArrayList<>();
List<String> orderCodes = listOrderCode();
orderCodes.stream().forEach(orderCode->{
Order order = getOrder(orderCode);
orderList.add(order);
}); //获取订单列表 end
return orderList;
}; /**
* 获取订单列表（并行获取数据）
* stream也改成了parallelStream 并行for循环
* @return
*/
@Async
@Override
public Future<List<Order>> listOrder_2(){
log.info("listOrder_2 当前线程是：{}",Thread.currentThread().getName()); //获取订单列表 start
List<Order> orderList = new ArrayList<>();
List<String> orderCodes = listOrderCode(); if(CollUtil.isNotEmpty(orderCodes)){
orderCodes.parallelStream().forEach(orderCode->{
Order order = getOrder(orderCode); if(order!=null){
orderList.add(order);
}
});
} //获取订单列表 end
return new AsyncResult<List<Order>>(orderList);
}; /**
* 获取订单列表（并行获取数据）(把主线程的request的数据 透传给 子线程和子子线程)
* @return
*/
@Override
public List<Order> listOrder_3(){ //获取订单列表 start
List<Order> orderList = new ArrayList<>();
List<String> orderCodes = listOrderCode(); if(CollUtil.isNotEmpty(orderCodes)){
RequestAttributes attributes = RequestContextHolder.getRequestAttributes();
orderCodes.parallelStream().forEach(orderCode->{
RequestContextHolder.setRequestAttributes(attributes); try {
Order order = getOrder(orderCode); if(order!=null){
orderList.add(order);
}
}finally {
RequestContextHolder.resetRequestAttributes();
}
});
} //获取订单列表 end
return orderList;
};
}UserService 请求订单的服务的聚合方法
import java.util.ArrayList;import java.util.List;import java.util.concurrent.Future;@Slf4j@Servicepublic class UserService implements IUserService { @Override
public List<Integer> listUserId(){ //使用httpUtil 模拟 feign请求服务接口 start
String reqUrl = Config.baseUrl.concat("/v1/user/list");
HttpRequest httpRequest = HttpUtil.createGet(reqUrl); //设置请求头信息
String token = WebUtil.getCurrentRequestHeaderToken();
httpRequest.header("token",token);
HttpResponse httpResponse = httpRequest.execute(); String body = httpResponse.body();
Resp<List<Integer>> respData = JSONUtil.toBean(body, Resp.class); //使用httpUtil 模拟 feign请求服务接口 end
if(respData.isSuccess()){ return respData.getData();
} return null;
}; @Override
public User getUser(Integer userId){ //使用httpUtil 模拟 feign请求服务接口 start
String reqUrl = StrUtil.format(Config.baseUrl.concat("/v1/user/get?userId={}"),userId);
HttpRequest httpRequest = HttpUtil.createGet(reqUrl); //设置请求头信息
String token = WebUtil.getCurrentRequestHeaderToken();
httpRequest.header("token",token);
HttpResponse httpResponse = httpRequest.execute(); String body = httpResponse.body();
Gson gson = new Gson();
Resp<User> respData = gson.fromJson(body , new TypeToken<Resp<User>>(){}.getType()); //使用httpUtil 模拟 feign请求服务接口 end
if(respData.isSuccess()){ return respData.getData();
} return null;
}; @Override
public List<User> listUser_1(){ //获取用户列表 start
List<User> userList = new ArrayList<>();
List<Integer> userIds = listUserId();
userIds.stream().forEach(userId->{
User user = getUser(userId);
userList.add(user);
}); //获取用户列表 end
return userList;
}; @Async
@Override
public Future<List<User>> listUser_2(){
log.info("listUser_2 当前线程是：{}",Thread.currentThread().getName()); //获取用户列表 start
List<User> userList = new ArrayList<>();
List<Integer> userIds = listUserId(); if(CollUtil.isNotEmpty(userIds)){
userIds.parallelStream().forEach(userId->{
User user = getUser(userId); if(user!=null){
userList.add(user);
}
});
} //获取用户列表 end
return new AsyncResult<List<User>>(userList);
}; @Override
public List<User> listUser_3(){ //获取用户列表 start
List<User> userList = new ArrayList<>();
List<Integer> userIds = listUserId(); if(CollUtil.isNotEmpty(userIds)){
RequestAttributes attributes = RequestContextHolder.getRequestAttributes();
userIds.parallelStream().forEach(userId->{
RequestContextHolder.setRequestAttributes(attributes); try {
User user = getUser(userId); if(user!=null){
userList.add(user);
}
}finally {
RequestContextHolder.resetRequestAttributes();
}
});
} //获取用户列表 end
return userList;
};
}OrderController 你可以理解成其他其他微服务的接口(模拟写的一个接口，用来测试 请求接口的时候是否携带 请求头了)
@Slf4j@RestController@RequestMapping(value = "/v1/order")public class OrderController { @ApiOperation(value = "获取订单编号列表") @RequestMapping(value = "/list", method = RequestMethod.GET) public Resp<List<String>> list(HttpServletRequest request){ String token = request.getHeader("token"); if(StrUtil.isBlank(token)){ return Resp.buildFail("接口不存在 404");
}
List<String> userIds = new ArrayList<>();
userIds.add("11111");
userIds.add("22222");
userIds.add("33333");
userIds.add("44444");
userIds.add("55555");
userIds.add("6666");
userIds.add("7777");
handleBusinessTime(); return Resp.buildDataSuccess(userIds);
} @ApiOperation(value = "获取订单详情") @ApiImplicitParams({ @ApiImplicitParam(name = "orderCode", value = "订单CODE", paramType = "query"),
}) @RequestMapping(value = "/get", method = RequestMethod.GET) public Resp<Order> get(HttpServletRequest request,@RequestParam(value = "orderCode")String orderCode){ String token = request.getHeader("token"); if(StrUtil.isBlank(token)){ return Resp.buildFail("接口不存在 404");
}
handleBusinessTime(); String name = StrUtil.format("订单-{}-名",orderCode); return Resp.buildDataSuccess(Order.builder().code(orderCode).orderName(name).build());
} /**
* 这方法 模拟处理业务或者 去操作数据库 消耗的时间
*/
public static void handleBusinessTime(){ //去数据库查询数据耗时 start
int[] sleepTime = NumberUtil.generateRandomNumber(300,800,1); try { //Thread.sleep 休眠的时候 相当于 业务操作，或者请求数据库的需要消耗的时间
Thread.sleep(sleepTime[0]);
} catch (InterruptedException e) {
e.printStackTrace();
} //去数据库查询数据耗时 end
}
}@Slf4j@RestController@RequestMapping(value = "/v1/user")public class UserController { @ApiOperation(value = "获取用户列表-id") @ApiImplicitParams({ @ApiImplicitParam(name = "orderCode", value = "订单编号", paramType = "query"),
}) @RequestMapping(value = "/list", method = RequestMethod.GET) public Resp<List<Integer>> list(HttpServletRequest request){ String token = request.getHeader("token"); if(StrUtil.isBlank(token)){ return Resp.buildFail("接口不存在 404");
}
List<Integer> userIds = new ArrayList<>();
userIds.add(1);
userIds.add(2);
userIds.add(3);
userIds.add(4);
userIds.add(5);
handleBusinessTime(); return Resp.buildDataSuccess(userIds);
} @ApiOperation(value = "根据用户ID获取 用户信息") @ApiImplicitParams({ @ApiImplicitParam(name = "userId", value = "用户ID", paramType = "query"),
}) @RequestMapping(value = "/get", method = RequestMethod.GET) public Resp<User> get(HttpServletRequest request,@RequestParam(value = "userId")Integer userId){ String token = request.getHeader("token"); if(StrUtil.isBlank(token)){ return Resp.buildFail("接口不存在 404");
}
handleBusinessTime(); String name = StrUtil.format("用户{}号",userId); return Resp.buildDataSuccess(User.builder().id(userId).name(name).build());
} /**
* 这方法 模拟处理业务或者 去操作数据库 消耗的时间
*/
public static void handleBusinessTime(){ //去数据库查询数据耗时 start
int[] sleepTime = NumberUtil.generateRandomNumber(300,800,1); try { //Thread.sleep 休眠的时候 相当于 业务操作，或者请求数据库的需要消耗的时间
Thread.sleep(sleepTime[0]);
} catch (InterruptedException e) {
e.printStackTrace();
} //去数据库查询数据耗时 end
}
}下面三个接口的由来：
/v1/testUser/listUser 接口：就是串行调用其他服务接口 ，性能比较慢
/v1/testUser/listUser2 接口：是通过@Async 异步注解，并行调用其他 系统的接口，性能是提升上去了，但灰度环境 是需要根据请求头里面的数据判断是否把流量打到灰度环境
/v1/testUser/listUser3接口：对@Async注解没有找到透传 主线程request头信息的方案，就使用线程池+CompletableFuture.supplyAsync的方式 每次执行异步线程的时候，把主线程的 请求参数设置到子线程，然后通过try-finally 参数使用完之后RequestContextHolder.resetRequestAttributes() 删除参数。
注意：parallelStream它也是属于并行流操作，也要设置 请求头信息，虽说子线程（getDateResp3方法）能获取到主线程的请求头信息了，但是parallelStream 又相当于子线程的子线程了，它是获取不到的 主线程的attributes的，当时我就是没在parallelStream设置attributes，它没有走到灰度环境， 让我 耗费了两个多小时，代码加了四五次日志输出，才把这个问题定位出来，这是一个坑。。。
下面是代码：
基于这个问题，我还写了一篇 spring boot使用@Async的文章，大家感兴趣可以去看看 传送门～
我已经把上述代码例子放到gitee了，大家感兴趣可以clone 传送门～
人生很多烦恼，是因为想得太多，做得太少
来源https://www.cnblogs.com/x-kq/p/14911478.html