聊聊 Spring Cloud 全链路灰度发布 方案~（三）

因为不仅仅网关需要用到这个灰度发布策略，凡是涉及到OpenFeign调用的微服务如果需要配置灰度发布都需要用到，因此这里陈某定义了一个公用的gray-starter。

经过上述步骤网关的灰度发布则已经配置完成，此时只需要通过@RibbonClients指定对应哪个服务灰度发布。

openFeign透传灰度标记
上面在介绍网关的灰度发布配置时，是将灰度标记（grayTag=true）放在了请求头中，因此在下游服务中需要做的就只是从请求头中将灰度标记取出来，然后将其存入GrayRequestContextHolder上下文中。

这样一来下游服务中的GrayRule则能从GrayRequestContextHolder获取到灰度标记，从注册中心获取灰度服务进行调用了。

问题来了：如何从请求头中取出灰度标记？

在介绍OAuth2.0相关知识时，曾经出过一篇文章：实战！openFeign如何实现全链路JWT令牌信息不丢失？

其中介绍了令牌中继的解决方案，使用的是openFeign的请求拦截器去配置请求头信息。

如上图：openFeign在调用时并不是用的原先的Request，而是内部新建了一个Request，其中复制了请求的URL、请求参数一些信息，但是请求头并没有复制过去，因此openFeign调用会丢失请求头中的信息。

但是可以通过实现RequestInterceptor将原先的请求头给复制过去，代码如下：

@Component
@Slf4j
public class FeignRequestInterceptor implements RequestInterceptor {
    @Override
    public void apply(RequestTemplate template) {
        HttpServletRequest httpServletRequest = RequestContextUtils.getRequest();
        Map<String, String> headers = getHeaders(httpServletRequest);
        for (Map.Entry<String, String> entry : headers.entrySet()) {
            //② 设置请求头到新的Request中
            template.header(entry.getKey(), entry.getValue());
        }
    }

    /**
     * 获取原请求头
     */
    private Map<String, String> getHeaders(HttpServletRequest request) {
        Map<String, String> map = new LinkedHashMap<>();
        Enumeration<String> enumeration = request.getHeaderNames();
        if (enumeration != null) {
            while (enumeration.hasMoreElements()) {
                String key = enumeration.nextElement();
                String value = request.getHeader(key);
                //将灰度标记的请求头透传给下个服务
                if (StrUtil.equals(GrayConstant.GRAY_HEADER,key)&&Boolean.TRUE.toString().equals(value)){
                    //① 保存灰度发布的标记
                    GrayRequestContextHolder.setGrayTag(true);
                    map.put(key, value);
                }
            }
        }
        return map;
    }
}

①处的代码：从请求头中获取灰度发布的标记，设置到GrayRequestContextHolder上下文中

②处的代码：将这个请求头设置到新的Request中，继续向下游服务传递。

其实配置一下RequestInterceptor就已经完成了，关于灰度发布策略只需要复用网关的GrayRule

“注意：也需要使用@RibbonClients注解去标注文章服务调用的哪些服务需要灰度发布。”

代码如下：

@RibbonClients(value = {
        //指定对comments这个服务开启灰度部署
        @RibbonClient(value = "comments",configuration = GrayRuleConfig.class)
})
public class ArticleApplication {}

Nacos中服务如何做灰度标记
其实很简单，分为两种：

1、在配置文件中指定，如下：

spring:
  cloud:
    nacos:
      discovery:
        metadata:
          ## 灰度标记
          grayTag: true

2、在Nacos中动态的指定灰度标记

配置完成之后，在客户端请求的时候只需要携带grayTag=true这个请求头即可调用灰度服务。

总结
微服务中全链路灰度发布方案其实很简单，重要的就是灰度打标，整体流程如下：

1. 网关中通过全局过滤器实现灰度打标，将灰度标记放入请求头中传递给下游服务
2. 网关通过自定义的负载均衡策略，从注册中心获取灰度服务，进行转发
3. 在openFeign调用时需要从请求头中获取灰度标记，放入上下文中
4. openFeign调用同样是根据自定义的负载均衡策略从注册中心获取灰度服务，进行调用。

文章转自公众号：码猿技术专栏