ribbon按参数路由,实现灰度发布功能

最新推荐文章于 2024-03-29 16:36:04 发布

zhangzhen02

最新推荐文章于 2024-03-29 16:36:04 发布

阅读量776

点赞数 1

分类专栏： springcloud 文章标签： ribbon

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

版权

springcloud 专栏收录该内容

7 篇文章 1 订阅

订阅专栏

本文介绍了如何在Ribbon负载均衡器的基础上实现自定义参数路由。通过创建一个`MyRequestInterceptor`来拦截请求并保存参数到ThreadLocal中，然后定义`MyRule`扩展`AbstractLoadBalancerRule`，从ThreadLocal获取参数并据此选择服务器。这种方式使得可以根据请求头中的特定参数（如'version'）进行路由决策。

摘要由CSDN通过智能技术生成

遇到一个问题，需要根据请求参数进行自定义路由。
看了ribbon的实现源码，发现ribbon 在路由时，根本没有地方可以传参，以实现根据参数路由的功能。唯一的参数loadBalancerKey，是永远为null的坑爹设计。所以只好自己实现。
我的实现方式如下：

一，首先定义自定义RequestInterceptor,将参数传入到ThreadLocal里

@Component
public class MyRequestInterceptor implements RequestInterceptor {
    String VERSION="version";
    @Override
    public void apply(RequestTemplate requestTemplate) {
        Map<String,String> headers = getHeaders(getHttpServletRequest());
        String version=headers.getOrDefault(VERSION,"");
        if (!version.isEmpty()){
            VersionContext.set(version);
        }else {
            VersionContext.set("");
        }
    }
    private HttpServletRequest getHttpServletRequest() {
        try {
            return ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        }
    }

    private Map<String, String> getHeaders(HttpServletRequest request) {
        Map<String, String> map = new LinkedHashMap<>();
        Enumeration<String> enumeration = request.getHeaderNames();
        while (enumeration.hasMoreElements()) {
            String key = enumeration.nextElement();
            String value = request.getHeader(key);
            map.put(key, value);
        }
        return map;
    }
}

用到的VersionContext.java

public class VersionContext {
    private static ThreadLocal<String> version=new ThreadLocal<>();
    public static String get(){
        return version.get();
    }
    public static void set(String s){
        version.set(s);
    }
}

二，定义自定义路由MyRule

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

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

这样就实现了根据参数路由了。