Java SpringCloud

什么是令牌中继?

令牌中继通俗的讲则是让令牌在微服务链路调用中传递下去,保证各个微服务能够获取令牌中的用户信息。
以下订单的例子来说,如下图:
下单流程
客户端携带令牌请求网关,网关鉴权成功后会将令牌中的用户信息解析出来放在请求头中下发给订单服务,同样的,订单服务需要将用户信息传递给账户服务获取该用户的账户信息。
那么问题来了?如何保证网关服务->订单服务->账户服务这条链路中的用户信息传递下去是个痛点

解决方案

令牌在openFeign调用过程中是不能自动中继的,因此必须手动的将令牌信息传递下去。
注意:openFeign在开启熔断降级后内部调用开启了子线程,因此传统的方案直接在RequestInterceptor中设置是不可行的。
那么如何保证子线程也能获取请求头中的用户信息呢?
答案是:RequestContextHolder这个神器。
RequestContextHolder内部通过InheritableThreadLocal实现子线程共享信息。
FeignCircuitBreakerInvocationHandler这个类中也是有如下一行代码:

  1. RequestContextHolder.setRequestAttributes(requestAttributes);

正是使用RequestContextHolder将request的信息保存在其中,因此实现令牌中继只需要读取RequestContextHolder的信息即可。
详细代码如下:

  1. @Component
  2. public class FeignRequestInterceptor implements RequestInterceptor {
  3. @Override
  4. public void apply(RequestTemplate template) {
  5. //从RequestContextHolder中获取HttpServletRequest
  6. HttpServletRequest httpServletRequest = RequestContextUtils.getRequest();
  7. //获取RequestContextHolder中的信息
  8. Map<String, String> headers = getHeaders(httpServletRequest);
  9. //放入feign的RequestTemplate中
  10. for (Map.Entry<String, String> entry : headers.entrySet()) {
  11. template.header(entry.getKey(), entry.getValue());
  12. }
  13. }
  14. /**
  15. * 获取原请求头
  16. */
  17. private Map<String, String> getHeaders(HttpServletRequest request) {
  18. Map<String, String> map = new LinkedHashMap<>();
  19. Enumeration<String> enumeration = request.getHeaderNames();
  20. if (enumeration != null) {
  21. while (enumeration.hasMoreElements()) {
  22. String key = enumeration.nextElement();
  23. String value = request.getHeader(key);
  24. map.put(key, value);
  25. }
  26. }
  27. return map;
  28. }
  29. }