Tars | 第8篇 TarsJava Subset最终代码的执行流程与原理分析


前言

中期汇报会后,对Tars Subset功能更加熟悉,并根据TarsGo的实现方式,对Java JDK实现代码进行翻新改造。于是有了以下两篇分析文章:

第5篇 基于TarsGo Subset路由规则的Java JDK实现方式(上篇)
https://www.cnblogs.com/dlhjw/p/15245113.html

第6篇 基于TarsGo Subset路由规则的Java JDK实现方式(下篇)
https://www.cnblogs.com/dlhjw/p/15245116.html

其中,《上篇》注重TarsGo分析,《下篇》注重TarsJava实现方式。不出意外的话,最终提交的考核成果就在下面的GitHub代码仓库中(以下简称“最终代码”),后续可能会有些许地方需要更改:

TarsJava 实现Subset路由规则JDK GitHub开源地址
https://github.com/dlhjw/TarsJava/commit/cc2fe884ecbe8455a8e1f141e21341f4f3dd98a3

最终代码与中期代码在整体思想逻辑上都是一致的都是:先判断Subset路由规则,再根据规则路由到定义的节点。不同点在于:中期在处理整个过程时,用一个方法filterEndpointsBySubset()实现;而最终的实现方式则是以subsetEndpointFilter()方法作为整个Subset流量路由的入口,通过subsetManager管理器调用getSubset()方法获取到路由规则的String类型的subset字段,与节点自身的subset字段一一比较过滤节点;其中Subset路由规则的判断封装在getSubset()方法里;

总的来说就是最终代码是在处理subset规则逻辑中增加了很多细节,比如:通过新增的registry接口获取subsetConf配置项;将获取到是配置项存入缓存中;以及将“判断Subset路由规则”进行层层封装,最终返回一个简单的String类型的subset字段与节点自身的subset字段比较等;

因此,在执行流程上相比较中期有些许区别,其中比较复杂的地方涉及对各种封装方法的调用与传参。本篇将结合最终代码的实现逻辑,以debug的方式,重点介绍其执行流程,即:

  • 首先找到过滤节点的方法入口;
  • 接着通过管理者尝试获取String类型的规则subset字段;
  • 通过getSubset()方法匹配路由规则;
  • 调用具体路由规则的findSubet()方法获取最终的规则subset字段;
  • 将规则subset字段与节点的subset字段对比,实现筛选功能;

通过下面分析可以得出一个Subset的业务执行流程结构图,如下:

  • subset.subsetEndpointFilter():整个业务流程的方法入口;
    • subsetManager.getSubset():通过XxxConf配置获取String类型的subset字段;
      • getSubsetConfig():从缓存或registry接口获取SubsetConf;
      • subsetConf.getRatioConf():获取RatioConf配置;
        • ratioConf.findSubet():通过RatioConf配置获取String类型的subset字段;
      • subsetConf.getKeyConf():获取KeyConf配置;
        • keyConf.findSubet():通过KeyConf配置获取String类型的subset字段;
    • for循环匹配subset字段;

当然,涉及到通过新增的registry接口获取subsetConf配置项等逻辑,这些细节就放在下面正文里讲吧;

最终代码的Subset测试方案设计请参考下面这篇文章:

第7篇 TarsJava Subset最终代码的测试方案设计
https://www.cnblogs.com/dlhjw/p/15245121.html

《测试方案设计》与《执行流程分析》两篇文章相辅相成,相互观阅能更快更好地理解整个Subset的业务流程与输出示例;


1. SubsetConf配置项的结构

在中期,笔者使用一个map来模拟subset的流量规则;而在最终代码里,是用多个对象来模拟Subset的配置,这些对象是理解整个Subset流量过滤规则的基础的,因此很有必要在这里做个介绍;

1.1 SubsetConf

public class SubsetConf {
    private boolean enanle;
    private String ruleType;
    private RatioConfig ratioConf;
    private KeyConfig keyConf;
    private Instant lastUpdate;
    ……
}

可以看出SubsetConf配置项里有以下属性:

  • enanle:表示是否开启Subset流量管理功能;
    • true:开启;false:关闭;
  • ruleType:表示流量管理的类型;
    • 目前有ratio按比例和key按参数两种模式;
  • RatioConfig:表示按比例路由配置项;
    • 里面定义了路由比例与路径等信息,详情请参考《1.2 RatioConfig》
  • KeyConfig:表示按参数路由配置项;
    • 里面定义了规则key与路由路径等信息,详情请参考《1.3 KeyConfig》
  • lastUpdate:表示该配置项上次更新时间,将在缓存那里起作用;

1.2 RatioConfig

public class RatioConfig {
    private Map<String, Integer> rules;
    ……
}

RatioConfig里只有一个map类型的rules路由规则,其中key为一个String类型的subset字段,用来跟节点的subset字段匹配,value为路由权重,如:{ {"v1" , 20} , {"v2" , 60} , {"v3" , 20} }表示路由到subset字段为v1的节点的概率为0.2;路由到subset字段为v2的节点的概率为0.6;路由到subset字段为v3的节点的概率为0.2;

1.3 KeyConfig

public class KeyConfig {
    private String defaultRoute;
    private List<KeyRoute> rules;
    ……
}    

KeyConfig里有两个属性,一个是defaultRoute默认路由路径;另一个是list类型的rules,里面是KeyRoute,其定义了按参数匹配的类型、规则key与路径,详情请见《1.4 KeyRoute》

1.4 KeyRoute

public class KeyRoute {
    private String action = null;
    private String value = null;
    private String route = null;
    public static final String TARS_ROUTE_KEY = "TARS_ROUTE_KEY";
    ……
}

KeyRoute里面有四个String类型的属性,如下:

  • action:用来定义参数匹配的类型;
    • 目前可设置的类型有:equals精确匹配、match正则匹配、default默认匹配;
  • value:这就是大名鼎鼎的规则key了。当action=equals时,还需满足规则key与请求key匹配,才能进行精确匹配;当action=match时,还需满足规则key与请求key正则匹配,才能进行正则匹配;action=default对规则key没要求;
  • route:用来规定路由路径,其值为一个String类型的subset字段,匹配到节点的subset字段;
  • TARS_ROUTE_KEY:一个常量字段,为Tars请求体里的status(map类型)的key;

1.5 SubsetConf的结构示意图

上述提到的配置类联系结构图如下:
SubsetConf结构图

从SubsetConf配置类可以看出,按比例路由和按参数在思路上有些许不同,因此下面dubug分析将分为两种情况;

2. 过滤节点的方法入口

新增的Sunset流量路由规则应该在查找服务端节点那里,其目的是对获取到的服务端节点根据subset规则进行过滤,因此整个Subset业务逻辑的入口函数在getServerNodes()方法的subsetEndpointFilter()里,如下图所示:
过滤节点的方法入口
我们给该方法打上断点,进行debug调试;


3. subsetEndpointFilter()方法解析

3.1 方法功能

方法名 subsetEndpointFilter()
方法所在类 Subset.java
方法逻辑 根据Subset路由规则过滤节点
传入参数 服务名、透传染色key、存活节点列表
传出参数 过滤后的存活节点列表

3.2 方法源码

public Holder<List<EndpointF>> subsetEndpointFilter(String servantName, String routeKey, Holder<List<EndpointF>> eps){
    if( subsetConf==null || !subsetConf.isEnanle() ){
        return eps;
    }
    if(eps.value == null || eps.value.isEmpty()){
        return eps;
    }
    //调用subsetManager,根据比例/匹配等规则获取到路由规则的subset
    String subset = subsetManager.getSubset(servantName, routeKey);
    if( "".equals(subset) || subset == null){
        return eps;
    }
    //和每一个eps的subset比较,淘汰不符合要求的
    Holder<List<EndpointF>> epsFilter = new Holder<>(new ArrayList<EndpointF>());
    for (EndpointF ep : eps.value) {
        if( subset.equals(ep.getSubset())){
            epsFilter.getValue().add(ep);
        }
    }
    return epsFilter;
}

3.3 方法解析

进入subsetEndpointFilter()方法,首先会对subsetConf配置项进行非空校验,这里的配置项是从前端传过来的,正常情况下不为空;接着对传入的参数存活节点列表eps做非空校验。两步校验通过后,进行核心方法getSubset()的调用,返回String类型的规则的subset字段,该字段可以表示路由路径。然后将规则的subset字段与节点的subset字段一一比较,选出符合要求的节点;

测试结果详情请见《TarsJava Subset最终代码的测试方案设计》一文;

返回的String类型的规则的subset字段在下图中就是RotioConfig的rules中的value或者KeyConfig的rules中的route,表示最终路由路径。以下所有方法都是围绕如何找到这个规则的subset字段做文章;

SubsetConf结构图

我们进入getSubset()方法查看Tars是怎么获取到规则的subset字段的;


4. getSubset()方法解析

4.1 方法功能

方法名 getSubset()
方法所在类 SubsetManager.java
方法逻辑 根据路由规则先获取到比例 / 染色路由的配置,再通过配置获取String类型的subset字段
传入参数 服务名、透传染色key
传出参数 String类型的规则的subset字段

4.2 方法源码

public String getSubset(String servantName, String routeKey){
    //check subset config exists
    SubsetConf subsetConf = getSubsetConfig(servantName);
    if( subsetConf == null ){
        return null;
    }
    // route key to subset
    if("ratio".equals(subsetConf.getRuleType())){
        RatioConfig ratioConf = subsetConf.getRatioConf();
        if(ratioConf != null){
            return ratioConf.findSubet(routeKey);
        }
    }
    KeyConfig keyConf = subsetConf.getKeyConf();
    if ( keyConf != null ){
        return keyConf.findSubet(routeKey);
    }
    return null;
}

4.3 方法解析

该方法首先调用getSubsetConfig()方法获取到一个SubsetConf配置项,该配置项可以是从缓存中拿,也可以是通过registry接口获取(详情请见《5. getSubsetConfig()方法解析》);

确保SubsetConf配置项存在后,识别SubsetConf配置项的ruleType路由类型属性(按比例、按参数、默认),通过属性获取到对应的配置项(ratioConf、keyConf、keyConf),最后调用findSubet()方法即可获取到String类型的规则的subset字段,并返回(详情请见《6. 按比例路由的findSubet()方法解析》与《7. 按参数路由的findSubet()方法解析》);

我们先进入getSubsetConfig()方法探其源码;


5. getSubsetConfig()方法解析

5.1 方法功能

方法名
方法所在类 SubsetManager.java
方法逻辑 获取SubsetConf路由规则配置项,并存到subsetConf配置项
传入参数 服务名
传出参数 SubsetConf配置项

5.2 方法源码

public SubsetConf getSubsetConfig(String servantName){
    SubsetConf subsetConf = new SubsetConf();
    if( cache.containsKey(servantName) ){
        subsetConf = cache.get(servantName);
        //小于10秒从缓存中取
        if( Duration.between(subsetConf.getLastUpdate() , Instant.now()).toMillis() < 1000 ){
            return subsetConf;
        }
    }
    // get config from registry
    Holder<SubsetConf> subsetConfHolder = new Holder<SubsetConf>(subsetConf);
    int ret = queryProxy.findSubsetConfigById(servantName, subsetConfHolder);
    SubsetConf newSubsetConf = subsetConfHolder.getValue();
    if( ret == TarsHelper.SERVERSUCCESS ){
        return newSubsetConf;
    }
    //从registry中获取失败时,更新subsetConf添加进缓存
    subsetConf.setRuleType( newSubsetConf.getRuleType() );
    subsetConf.setLastUpdate( Instant.now() );
    cache.put(servantName, subsetConf);
    //解析subsetConf
    if( !newSubsetConf.isEnanle() ){
        subsetConf.setEnanle(false);
        return subsetConf;
    }
    if( "ratio".equals(newSubsetConf.getRuleType())){
        subsetConf.setRatioConf( newSubsetConf.getRatioConf() );
    } else {
        //按参数匹配
        KeyConfig newKeyConf = newSubsetConf.getKeyConf();
        List<KeyRoute> keyRoutes = newKeyConf.getRules();
        for ( KeyRoute kr: keyRoutes) {
            KeyConfig keyConf = new KeyConfig();
            //默认
            if("default".equals(kr.getAction())){
                keyConf.setDefaultRoute(newKeyConf.getDefaultRoute());
                subsetConf.setKeyConf(keyConf);
            }
            //精确匹配
            if("match".equals(kr.getAction())){
                List<KeyRoute> rule = new ArrayList<>();
                rule.add(new KeyRoute("match", kr.getValue() , kr.getRoute()));
                keyConf.setRules( rule );
            }
            //正则匹配
            if("equal".equals(kr.getAction())){
                List<KeyRoute> rule = new ArrayList<>();
                rule.add(new KeyRoute("equal", kr.getValue() , kr.getRoute()));
                keyConf.setRules( rule );
            }
        }
        subsetConf.setKeyConf(newKeyConf);
    }
    return subsetConf;
}

5.3 方法解析

getSubsetConfig()方法的获取主要分为两个逻辑:从缓存中获取与从registry接口获取,判断依据是:如果缓存中的subsetConf配置项的lastUpdate上次更新属性落后当前时间小于10秒,则直接从缓存中获取;否则重新调用registry接口获取subsetConf配置项;

测试结果详情请见《TarsJava Subset最终代码的测试方案设计》一文;

这是与另一个新增功能registry接口接触的方法,由于registry功能还未实现,这部分功能做不了测试,但其大概逻辑是通过registry接口查阅数据库获取到subsetConf配置项;

getSubsetConfig()方法执行完后,回到4. getSubset()方法里,我们再进入findSubet()方法窥其究竟;


6. 按比例路由的findSubet()方法解析

需要注意,按比例、按参数路由的findSubet()方法实现方式有些许不同,故在这里分开来讲;

6.1 方法功能

方法名
方法所在类 RatioConfig.java
方法逻辑 根据权重比例获取rules中最后的subset字段
传入参数 透传的染色key
传出参数 String类型的规则的subset字段

6.2 方法源码

public String findSubet(String routeKey){
     //routeKey为空时随机
     if( "".equals(routeKey) || routeKey == null){
         //赋值routeKey为获取的随机值
         Random random = new Random();
         int r = random.nextInt( rules.size() );
         routeKey = String.valueOf(r);
         int i = 0;
         for (String key : rules.keySet()) {
             if(i == r){
                 return key;
             }
             i++;
         }
     }
     //routeKey不为空时实现按比例算法
     int totalWeight = 0;
     int supWeight = 0;
     String subset = null;
     //获得总权重
     for (Integer value : rules.values()) {
         totalWeight+=value;
     }
     //获取随机数
     Random random = new Random();
     int r = random.nextInt(totalWeight);
     //根据随机数找到subset
     for (Map.Entry<String, Integer> entry : rules.entrySet()){
         supWeight+=entry.getValue();
         if( r < supWeight){
             subset = entry.getKey();
             return subset;
         }
     }
     return null;
 }

6.3 方法解析

首先判断透传的染色key是否为空,如果为空,则会进行等比例随机路由到任意一个value;如果不为空,则会计算权重,通过随机数匹配权重的方式按比例路由到对应的value,该value就是需要返回给入口函数的String类型的规则的subset字段;

该方法实现下图的蓝圈1号流程:

最后找到一个value作为规则的subset字段

而在按参数路由里,也有一个findSubet()方法,他们的传入传出参数相同,实现功能类似,但在逻辑上有些不同,下面来看一下吧;


7. 按参数路由的findSubet()方法解析

7.1 方法功能

方法名
方法所在类 KeyConfig.java
方法逻辑 根据参数匹配规则获取rules中最后的subset字段
传入参数 透传的染色key
传出参数 String类型的规则的subset字段

7.2 方法源码

public String findSubet(String routeKey){
    //非空校验
    if( routeKey == null || "".equals(routeKey) || rules == null){
        return null;
    }
    for ( KeyRoute rule: rules) {
        //根据根据分布式上下文信息获取 “请求的染色的key”
        String routeKeyReq;
        if( distributedContext != null){
            routeKeyReq = KeyRoute.getRouteKey(distributedContext);
        } else {
            logger.info("无分布式上下文信息distributedContext");
            return null;
        }
        //精确匹配
        if( "match".equals(rule.getAction())  ){
            if( routeKeyReq.equals(rule.getValue()) ){
                return rule.getRoute();
            } else {
                logger.info("染色key匹配不上,请求的染色key为:" + routeKeyReq + "; 规则的染色key为:" + rule.getValue());
            }
        }
        //正则匹配
        if( "equal".equals(rule.getAction()) ){
            if( StringUtils.matches(routeKeyReq, rule.getValue()) ){
                return rule.getRoute();
            } else {
                logger.info("正则匹配失败,请求的染色key为:" + routeKeyReq + "; 规则的染色key为:" + rule.getValue());
            }
        }
        //默认匹配
        if( "default".equals(rule.getAction()) ){
            //默认路由无需考虑染色key
            return rule.getRoute();
        }
    }
    return null;
}

7.3 方法解析

按参数匹配路由方式要求跟Taes请求体的染色key做匹配比较,因此这部分逻辑需要加上;

首先进行非空校验,校验通过后遍历KeyRoute,只需要找到符合条件的rules即可处理返回;在for循环里,首先根据分布式上下文信息获取 “请求的染色key”,根据KeyRoute的action分情况讨论:

  • action=match:精确匹配,需要判断请求的染色key规则的染色key是否相等,相等才能进行精确路由;不相等则不进行路由,并输出一句错误日志;
  • action=equal:正则匹配,其中正则式为请求的染色key,需要与规则的染色key进行匹配,同理匹配成功才能进行精确路由;否则则不进行路由,并输出一句错误日志;
  • action=default:默认路由,不需要使用染色key,路由到用户设定的默认路由路径;
  • 测试结果详情请见《TarsJava Subset最终代码的测试方案设计》一文;

该方法实现下图的蓝圈2号流程:

最后找到一个route作为规则的subset字段

通过上述步骤,最终获取到的规则的subset字段将返回到入口函数进行后续步骤:即将规则的subset字段与节点的subset字段一一比较,选出符合要求的节点;


8 总结:Subset业务执行流程结构图

通过上述分析,可以知道整个Subset的业务执行流程结构图如下:

  • subset.subsetEndpointFilter():整个业务流程的方法入口;
    • subsetManager.getSubset():通过XxxConf配置获取String类型的subset字段;
      • getSubsetConfig():从缓存或registry接口获取SubsetConf;
      • subsetConf.getRatioConf():获取RatioConf配置;
        • ratioConf.findSubet():通过RatioConf配置获取String类型的subset字段;
      • subsetConf.getKeyConf():获取KeyConf配置;
        • keyConf.findSubet():通过KeyConf配置获取String类型的subset字段;
    • for循环匹配subset字段;


最后

新人制作,如有错误,欢迎指出,感激不尽!
欢迎关注公众号,会分享一些更日常的东西!
如需转载,请标注出处!
posted @ 2021-09-16 10:16  多氯环己烷  阅读(88)  评论(0编辑  收藏  举报