Example #1
0
  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;
  }
Example #2
0
  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);
  }
Example #3
0
  /**
   * 使用新的条件值来替换。
   *
   * @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);
  }
Example #4
0
  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);
  }
Example #5
0
 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;
 }
Example #6
0
  // 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);
  }