ribbon按參數路由,實作灰階釋出功能

遇到一個問題，需要根據請求參數進行自定義路由。

看了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);
    }
}
           

這樣就實作了根據參數路由了。