static Map<Long, RouteRule> route2RouteRule( List<Route> routes, Map<String, List<String>> clusters) { Map<Long, RouteRule> rules = new HashMap<Long, RouteRule>(); // route -> RouteRule if (routes != null && routes.size() > 0) { for (Route route : routes) { rules.put(route.getId(), RouteRule.parseQuitely(route)); } } // expand the cluster parameters into conditions of routerule if (clusters != null && clusters.size() > 0) { Map<Long, RouteRule> rrs = new HashMap<Long, RouteRule>(); for (Map.Entry<Long, RouteRule> entry : rules.entrySet()) { RouteRule rr = entry.getValue(); Map<String, RouteRule.MatchPair> when = RouteRuleUtils.expandCondition( rr.getWhenCondition(), "consumer.cluster", "consumer.host", clusters); Map<String, RouteRule.MatchPair> then = RouteRuleUtils.expandCondition( rr.getThenCondition(), "provider.cluster", "provider.host", clusters); rrs.put(entry.getKey(), RouteRule.createFromCondition(when, then)); } rules = rrs; } return rules; }
public static RouteRule copyWithRemove( RouteRule copy, Set<String> whenParams, Set<String> thenParams) { Map<String, MatchPair> when = new HashMap<String, RouteRule.MatchPair>(); for (Entry<String, MatchPair> entry : copy.getWhenCondition().entrySet()) { if (whenParams == null || !whenParams.contains(entry.getKey())) { when.put(entry.getKey(), entry.getValue()); } } Map<String, MatchPair> then = new HashMap<String, RouteRule.MatchPair>(); for (Entry<String, MatchPair> entry : copy.getThenCondition().entrySet()) { if (thenParams == null || !thenParams.contains(entry.getKey())) { then.put(entry.getKey(), entry.getValue()); } } return new RouteRule(when, then); }
/** * 使用新的条件值来替换。 * * @param copy 替换的Base * @param whenCondition 要替换的whenCondition,如果Base没有项目,则直接插入。 * @param thenCondition 要替换的thenCondition,如果Base没有项目,则直接插入。 * @return 替换后的RouteRule */ public static RouteRule copyWithReplace( RouteRule copy, Map<String, MatchPair> whenCondition, Map<String, MatchPair> thenCondition) { if (null == copy) { throw new NullPointerException("Argument copy is null!"); } Map<String, MatchPair> when = new HashMap<String, RouteRule.MatchPair>(); when.putAll(copy.getWhenCondition()); if (whenCondition != null) { when.putAll(whenCondition); } Map<String, MatchPair> then = new HashMap<String, RouteRule.MatchPair>(); then.putAll(copy.getThenCondition()); if (thenCondition != null) { then.putAll(thenCondition); } return new RouteRule(when, then); }
public static boolean matchRoute( String consumerAddress, String consumerQueryUrl, Route route, Map<String, List<String>> clusters) { RouteRule rule = RouteRule.parseQuitely(route); Map<String, RouteRule.MatchPair> when = RouteRuleUtils.expandCondition( rule.getWhenCondition(), "consumer.cluster", "consumer.host", clusters); Map<String, String> consumerSample = ParseUtils.parseQuery("consumer.", consumerQueryUrl); final int index = consumerAddress.lastIndexOf(":"); String consumerHost = null; if (index != -1) { consumerHost = consumerAddress.substring(0, index); } else { consumerHost = consumerAddress; } consumerSample.put("consumer.host", consumerHost); return RouteRuleUtils.isMatchCondition(when, consumerSample, consumerSample); }
static Route getFirstRouteMatchedWhenConditionOfRule( String serviceName, Map<String, String> consumerSample, List<Route> routes, Map<Long, RouteRule> routeRuleMap) { if (serviceName == null || serviceName.length() == 0) { return null; } if (routes != null && routes.size() > 0) { for (Route route : routes) { if (isSerivceNameMatched(route.getService(), serviceName)) { RouteRule rule = routeRuleMap.get(route.getId()); // 当满足when条件时 if (rule != null && RouteRuleUtils.isMatchCondition( rule.getWhenCondition(), consumerSample, consumerSample)) { return route; // 第一个满足即返回 } } } } return null; }
// FIXME clusters和routes的合并,可以在clusters或routes变化时预先做 // FIXME 从Util方法中分离出Cache的操作 public static Map<String, String> route( String serviceName, String consumerAddress, String consumerQueryUrl, Map<String, String> serviceUrls, List<Route> routes, Map<String, List<String>> clusters, List<Route> routed) { if (serviceUrls == null || serviceUrls.size() == 0) { return serviceUrls; } if (routes == null || routes.isEmpty()) { return serviceUrls; } Map<Long, RouteRule> rules = route2RouteRule(routes, clusters); final Map<String, String> consumerSample = ParseUtils.parseQuery("consumer.", consumerQueryUrl); final int index = consumerAddress.lastIndexOf(":"); final String consumerHost; if (consumerAddress != null && index != -1) { consumerHost = consumerAddress.substring(0, index); } else { consumerHost = consumerAddress; } consumerSample.put("consumer.host", consumerHost); Map<String, Map<String, String>> url2ProviderSample = new HashMap<String, Map<String, String>>(); for (Map.Entry<String, String> entry : serviceUrls.entrySet()) { URI uri; try { uri = new URI(entry.getKey()); } catch (URISyntaxException e) { throw new IllegalStateException( "fail to parse url(" + entry.getKey() + "):" + e.getMessage(), e); } Map<String, String> sample = new HashMap<String, String>(); sample.putAll(ParseUtils.parseQuery("provider.", entry.getValue())); sample.put("provider.protocol", uri.getScheme()); sample.put("provider.host", uri.getHost()); sample.put("provider.port", String.valueOf(uri.getPort())); url2ProviderSample.put(entry.getKey(), sample); } Map<String, Set<String>> url2Methods = new HashMap<String, Set<String>>(); // consumer可以通过consumer.methods Key指定需要的方法 String methodsString = consumerSample.get("consumer.methods"); String[] methods = methodsString == null || methodsString.length() == 0 ? new String[] {Route.ALL_METHOD} : methodsString.split(ParseUtils.METHOD_SPLIT); for (String method : methods) { consumerSample.put("method", method); // NOTE: // <*方法>只配置 <no method key> // method1方法匹配 <no method key> 和 <method = method1>, 此时要把<no method key>的Route的优先级降低即可 if (routes != null && routes.size() > 0) { for (Route route : routes) { if (isSerivceNameMatched(route.getService(), serviceName)) { RouteRule rule = rules.get(route.getId()); // 当满足when条件时 if (rule != null && RouteRuleUtils.isMatchCondition( rule.getWhenCondition(), consumerSample, consumerSample)) { if (routed != null && !routed.contains(route)) { routed.add(route); } Map<String, RouteRule.MatchPair> then = rule.getThenCondition(); if (then != null) { Map<String, Map<String, String>> tmp = getUrlsMatchedCondition(then, consumerSample, url2ProviderSample); // 如果规则的结果是空,则该规则无效,使用所有Provider if (route.isForce() || !tmp.isEmpty()) { url2ProviderSample = tmp; } } } } } } for (String url : url2ProviderSample.keySet()) { Set<String> mts = url2Methods.get(url); if (mts == null) { mts = new HashSet<String>(); url2Methods.put(url, mts); } mts.add(method); } } // end of for methods return appendMethodsToUrls(serviceUrls, url2Methods); }