springcloud gateway 根据请求参数自定义路由，实现灰度发布

最新推荐文章于 2024-09-20 15:35:29 发布

zhangzhen02

最新推荐文章于 2024-09-20 15:35:29 发布

阅读量3.2k

点赞数

分类专栏： springcloud gateway 文章标签： java

本文链接：https://blog.csdn.net/zhangzhen02/article/details/110135995

版权

springcloud 同时被 2 个专栏收录

7 篇文章

订阅专栏

gateway

7 篇文章

订阅专栏

gateway 实现灰度发布比单纯的ribbon 更简单一些。
一，实现一个自定义的路由


import com.netflix.client.config.IClientConfig;
import com.netflix.loadbalancer.AbstractLoadBalancerRule;
import com.netflix.loadbalancer.Server;
import org.springframework.stereotype.Component;

import java.util.List;

@Component
public class CustomizeBalanceRule extends AbstractLoadBalancerRule {
    @Override
    public void initWithNiwsConfig(IClientConfig iClientConfig) {

    }

    @Override
    public Server choose(Object key) {
        List<Server> allServers = getLoadBalancer().getAllServers();
        if (key.equals("v1")){
            return allServers.get(0);
        }
        return allServers.get(1);
    }
}

二，实现一个自定义的负载均衡filter

import org.springframework.cloud.client.ServiceInstance;
import org.springframework.cloud.client.loadbalancer.LoadBalancerClient;
import org.springframework.cloud.gateway.config.LoadBalancerProperties;
import org.springframework.cloud.gateway.filter.LoadBalancerClientFilter;
import org.springframework.cloud.netflix.ribbon.RibbonLoadBalancerClient;
import org.springframework.stereotype.Component;
import org.springframework.web.server.ServerWebExchange;

import java.net.URI;

import static org.springframework.cloud.gateway.support.ServerWebExchangeUtils.GATEWAY_REQUEST_URL_ATTR;

@Component
public class CustomizeLoadBalancerClientFilter extends LoadBalancerClientFilter {


    public CustomizeLoadBalancerClientFilter(LoadBalancerClient loadBalancer, LoadBalancerProperties properties) {
        super(loadBalancer, properties);
    }
    @Override
    protected ServiceInstance choose(ServerWebExchange exchange) {

        String version = exchange.getRequest().getHeaders().getFirst("version");
        if (version == null) {
            version="default1";
        }
        if (this.loadBalancer instanceof RibbonLoadBalancerClient) {
            RibbonLoadBalancerClient client = (RibbonLoadBalancerClient) this.loadBalancer;
            String serviceId = ((URI) exchange.getAttribute(GATEWAY_REQUEST_URL_ATTR)).getHost();
            return client.choose(serviceId, version);
        }
        return super.choose(exchange);
    }
}

以上。