springcloud-gray

public class CustomMetadataRule extends ZoneAvoidanceRule {

	// 检测灰度开关是否启动
	private HttpResult checkGraySwitch() {
		String url = "http://10.200.102.136:6015/eureka/apps/switch";
		HttpResult result = new HttpResult();
		result.statusCode = 500;
		try {
			result = HttpClient.get(url, null);
		} catch (Exception e1) {
			e1.printStackTrace();
		}

		return result;
	}

	@Override
	public Server choose(Object key) {

		// 获取是否存在存活的服务可调用
		List<Server> serverList = this.getPredicate().getEligibleServers(this.getLoadBalancer().getAllServers());
		// 获取不到服务
		if (CollectionUtils.isEmpty(serverList)) {
			return null;
		}

		// 获取灰度开关是否启动
		HttpResult result = checkGraySwitch();

		// 灰度开关被设置成关闭状态,默认走空metadata或者是特定标识是正常的服务,轮询访问
		Boolean isOpen = Boolean.parseBoolean(JSONObject.parseObject(result.content).getString("errorMsg"));
		if (result.statusCode == 200 && !isOpen) {
			isOpen = true;
			return RoundRobinRuleBySelf.getInstance().choose(this.getLoadBalancer(), key,isOpen);
		}

		// 灰度发布启动状态,未被设置成灰度对象,默认走空metadata或者是特定标识是正常的服务,轮询访问
		if (StringUtils.isEmpty(CoreHeaderInterceptor.label.get())) {
			isOpen = false;
			return RoundRobinRuleBySelf.getInstance().choose(this.getLoadBalancer(), key,isOpen);
		}
		
		// 灰度发布启动状态,被设置成灰度对象,走空特定标识的服务,轮询访问
		return RoundRobinRuleBySelf.getInstance().choose(this.getLoadBalancer(), key,!isOpen);
	}
}

public class CoreFeignRequestInterceptor implements RequestInterceptor {
	private static final Logger logger = LoggerFactory.getLogger(CoreHttpRequestInterceptor.class);

	@Override
	public void apply(RequestTemplate template) {
		String header = StringUtils.collectionToDelimitedString(CoreHeaderInterceptor.label.get(),CoreHeaderInterceptor.HEADER_LABEL_SPLIT);
		String tag = CoreHeaderInterceptor.tag.get();
		template.header(CoreHeaderInterceptor.HEADER_LABEL, header).header(CoreHeaderInterceptor.HEADER_TAG, tag);
		logger.info("label: " + header + " tag : " + tag);
	}

}

public class CoreHttpRequestInterceptor implements ClientHttpRequestInterceptor {
    private static final Logger logger = LoggerFactory.getLogger(CoreHttpRequestInterceptor.class);

    @Override
    public ClientHttpResponse intercept(HttpRequest request, byte[] body, ClientHttpRequestExecution execution) throws IOException {
        HttpRequestWrapper requestWrapper = new HttpRequestWrapper(request);

        String header = StringUtils.collectionToDelimitedString(CoreHeaderInterceptor.label.get(), CoreHeaderInterceptor.HEADER_LABEL_SPLIT);
        
        String tag = CoreHeaderInterceptor.tag.get();
        
        logger.info("label: "+header + " tag : " + tag);
        
        HttpHeaders headers = requestWrapper.getHeaders();
        headers.add(CoreHeaderInterceptor.HEADER_LABEL, header);
        headers.add(CoreHeaderInterceptor.HEADER_TAG, tag);
        
        return execution.execute(requestWrapper, body);
    }
}

@Configuration
@EnableWebMvc
public class CoreAutoConfiguration extends WebMvcConfigurerAdapter {

	@Bean
	public DefaultPropertiesFactory defaultPropertiesFactory() {
		return new DefaultPropertiesFactory();
	}

	@LoadBalanced
	@Bean
	public RestTemplate restTemplate() {
		RestTemplate restTemplate = new RestTemplate();
		restTemplate.getInterceptors().add(new CoreHttpRequestInterceptor());
		return restTemplate;
	}

  //用于配置feignClient透传生效
	@Bean
	public Feign.Builder feignBuilder() {
		return Feign.builder().requestInterceptor(new CoreFeignRequestInterceptor());
	}

	@Override
	public void addInterceptors(InterceptorRegistry registry) {
		registry.addInterceptor(new CoreHeaderInterceptor());
	}
}