Examples with TagRouterRule org.apache.dubbo.rpc.cluster.router.tag.model.TagRouterRule used on opensource projects

Example 1 with TagRouterRule

use of org.apache.dubbo.rpc.cluster.router.tag.model.TagRouterRule in project dubbo by alibaba.

the class TagRouter method route.

@Override
public <T> List<Invoker<T>> route(List<Invoker<T>> invokers, URL url, Invocation invocation) throws RpcException {
    if (CollectionUtils.isEmpty(invokers)) {
        return invokers;
    }
    // since the rule can be changed by config center, we should copy one to use.
    final TagRouterRule tagRouterRuleCopy = tagRouterRule;
    if (tagRouterRuleCopy == null || !tagRouterRuleCopy.isValid() || !tagRouterRuleCopy.isEnabled()) {
        return filterUsingStaticTag(invokers, url, invocation);
    }
    List<Invoker<T>> result = invokers;
    String tag = StringUtils.isEmpty(invocation.getAttachment(TAG_KEY)) ? url.getParameter(TAG_KEY) : invocation.getAttachment(TAG_KEY);
    // if we are requesting for a Provider with a specific tag
    if (StringUtils.isNotEmpty(tag)) {
        List<String> addresses = tagRouterRuleCopy.getTagnameToAddresses().get(tag);
        // filter by dynamic tag group first
        if (CollectionUtils.isNotEmpty(addresses)) {
            result = filterInvoker(invokers, invoker -> addressMatches(invoker.getUrl(), addresses));
            // if result is not null OR it's null but force=true, return result directly
            if (CollectionUtils.isNotEmpty(result) || tagRouterRuleCopy.isForce()) {
                return result;
            }
        } else {
            // dynamic tag group doesn't have any item about the requested app OR it's null after filtered by
            // dynamic tag group but force=false. check static tag
            result = filterInvoker(invokers, invoker -> tag.equals(invoker.getUrl().getParameter(TAG_KEY)));
        }
        // to false, which means it will invoke any providers without a tag unless it's explicitly disallowed.
        if (CollectionUtils.isNotEmpty(result) || isForceUseTag(invocation)) {
            return result;
        } else // FAILOVER: return all Providers without any tags.
        {
            List<Invoker<T>> tmp = filterInvoker(invokers, invoker -> addressNotMatches(invoker.getUrl(), tagRouterRuleCopy.getAddresses()));
            return filterInvoker(tmp, invoker -> StringUtils.isEmpty(invoker.getUrl().getParameter(TAG_KEY)));
        }
    } else {
        // List<String> addresses = tagRouterRule.filter(providerApp);
        // return all addresses in dynamic tag group.
        List<String> addresses = tagRouterRuleCopy.getAddresses();
        if (CollectionUtils.isNotEmpty(addresses)) {
            result = filterInvoker(invokers, invoker -> addressNotMatches(invoker.getUrl(), addresses));
            // 1. all addresses are in dynamic tag group, return empty list.
            if (CollectionUtils.isEmpty(result)) {
                return result;
            }
        // 2. if there are some addresses that are not in any dynamic tag group, continue to filter using the
        // static tag group.
        }
        return filterInvoker(result, invoker -> {
            String localTag = invoker.getUrl().getParameter(TAG_KEY);
            return StringUtils.isEmpty(localTag) || !tagRouterRuleCopy.getTagNames().contains(localTag);
        });
    }
}

Example 2 with TagRouterRule

use of org.apache.dubbo.rpc.cluster.router.tag.model.TagRouterRule in project dubbo by alibaba.

the class TagRouterTest method tagRouterRuleParseTest.

/**
 * TagRouterRule parse test when the tags addresses is null
 *
 * <pre>
 *     ~ -> null
 *     null -> null
 * </pre>
 */
@Test
public void tagRouterRuleParseTest() throws Exception {
    String tagRouterRuleConfig = "---\n" + "force: false\n" + "runtime: true\n" + "enabled: false\n" + "priority: 1\n" + "key: demo-provider\n" + "tags:\n" + "  - name: tag1\n" + "    addresses: null\n" + "  - name: tag2\n" + "    addresses: [\"30.5.120.37:20880\"]\n" + "  - name: tag3\n" + "    addresses: []\n" + "  - name: tag4\n" + "    addresses: ~\n" + "...";
    TagRouterRule tagRouterRule = TagRuleParser.parse(tagRouterRuleConfig);
    // assert tags
    assert tagRouterRule.getTagNames().contains("tag1");
    assert tagRouterRule.getTagNames().contains("tag2");
    assert tagRouterRule.getTagNames().contains("tag3");
    assert tagRouterRule.getTagNames().contains("tag4");
    // assert addresses
    assert tagRouterRule.getAddresses().contains("30.5.120.37:20880");
    assert tagRouterRule.getTagnameToAddresses().get("tag1") == null;
    assert tagRouterRule.getTagnameToAddresses().get("tag2").size() == 1;
    assert tagRouterRule.getTagnameToAddresses().get("tag3") == null;
    assert tagRouterRule.getTagnameToAddresses().get("tag4") == null;
    assert tagRouterRule.getAddresses().size() == 1;
}