String id 为“应用名.服务名.端口名”；
• id这样设置是考虑到与其他接口命令对其；
• 该接口要与Tars Registry新增的接口名对上；

1.3 通过协议文件自动生成代码

Tars有个强大的功能，它能根据.tars里的配置文件自动生成相应Bean代码；

而在TarsGo里，对应代码如下：

执行上述命令后，对应代码会发生改变，如下：

这告诉我们老师发布的TarsGo代码里，有些代码不需要我们手动去更改，而是通过命令自动生成的。在Java中命令为在项目根路径执行

mvn tars:tars2java

。具体过程将在下半部分文章里详解，这里仅记录TarsGo代码在哪发生变化。

2. 【核心】增添Subset核心功能

Go语言在tars/subset.go内

2.1 Go语言修改部分

package tarsimport ("encoding/json""math/rand""regexp""strconv""sync""time""github.com/TarsCloud/TarsGo/tars/protocol/res/endpointf""github.com/TarsCloud/TarsGo/tars/protocol/res/queryf""github.com/TarsCloud/TarsGo/tars/util/consistenthash""github.com/TarsCloud/TarsGo/tars/util/endpoint""github.com/serialx/hashring")var (enableSubset = truesubsetMg     = &subsetManager{})type hashString stringfunc (h hashString) String() string {return string(h)}type subsetConf struct {enable    boolruleType  string // ratio/keyratioConf *ratioConfigkeyConf   *keyConfiglastUpdate time.Time}type ratioConfig struct {ring *hashring.HashRing}type keyRoute struct {action stringvalue  stringroute  string}type keyConfig struct {rules        []keyRoutedefaultRoute string}type subsetManager struct {lock  *sync.RWMutexcache map[string]*subsetConfregistry *queryf.QueryF}//根据服务名获取它的Subset方法，返回subsetConf配置项func (s *subsetManager) getSubsetConfig(servantName string) *subsetConf {s.lock.RLock()defer s.lock.RUnlock()var ret *subsetConf//如果缓存在最近时间之内，就直接返回if v, ok := s.cache[servantName]; ok {ret = vif v.lastUpdate.Add(time.Second * 10).After(time.Now()) {return ret}}//如果上次获取时间超时，则调用registry接口获取对应配置// get config from registryconf := &endpointf.SubsetConf{}retVal, err := s.registry.FindSubsetConfigById(servantName, conf)if err != nil || retVal != 0 {// log errorreturn ret}ret = &subsetConf{ruleType:   conf.RuleType,lastUpdate: time.Now(),}s.cache[servantName] = ret// 解析从registry那获取的配置信息// parse subset confif !conf.Enable {ret.enable = falsereturn ret}//按比例路由if conf.RuleType == "ratio" {kv := make(map[string]int)json.Unmarshal([]byte(conf.RuteData), &kv)ret.ratioConf = &ratioConfig{ring: hashring.NewWithWeights(kv)}} else {keyConf := &keyConfig{}kvlist := make([]map[string]string, 0)json.Unmarshal([]byte(conf.RuteData), &kvlist)for _, kv := range kvlist {//默认路由if vv, ok := kv["default"]; ok {keyConf.defaultRoute = vv}if vv, ok := kv["match"]; ok {//精确匹配keyConf.rules = append(keyConf.rules, keyRoute{action: "match",value:  vv,route:  kv["route"],})} else if vv, ok := kv["equal"]; ok {//正则匹配keyConf.rules = append(keyConf.rules, keyRoute{action: "equal",value:  vv,route:  kv["route"],})}}ret.keyConf = keyConf}return ret}func (s *subsetManager) getSubset(servantName, routeKey string) string {// check subset config existssubsetConf := subsetMg.getSubsetConfig(servantName)if subsetConf == nil {return ""}// route key to subsetif subsetConf.ruleType == "ratio" {return subsetConf.ratioConf.findSubet(routeKey)}return subsetConf.keyConf.findSubet(routeKey)}//根据subset规则过滤节点func subsetEndpointFilter(servantName, routeKey string, eps []endpoint.Endpoint) []endpoint.Endpoint {if !enableSubset {return eps}subset := subsetMg.getSubset(servantName, routeKey)if subset == "" {return eps}ret := make([]endpoint.Endpoint, 0)for i := range eps {if eps[i].Subset == subset {ret = append(ret, eps[i])}}return ret}func subsetHashEpFilter(servantName, routeKey string, m *consistenthash.ChMap) *consistenthash.ChMap {if !enableSubset {return m}subset := subsetMg.getSubset(servantName, routeKey)if subset == "" {return m}ret := consistenthash.NewChMap(32)for _, v := range m.GetNodes() {vv, ok := v.(endpoint.Endpoint)if ok && vv.Subset == subset {ret.Add(vv)}}return ret}func (k *ratioConfig) findSubet(key string) string {// 为空时使用随机方式if key == "" {key = strconv.Itoa(rand.Int())}v, _ := k.ring.GetNode(key)return v}func (k *keyConfig) findSubet(key string) string {for _, v := range k.rules {if v.action == "equal" && key == v.value {return v.route} else if v.action == "match" {if matched, _ := regexp.Match(v.value, []byte(key)); matched {return v.route}}}return k.defaultRoute}

2.2 新增地方的逻辑

subsetConf

ratioConfig

keyRoute

keyConfig

subsetManager

getSubsetConfig

getSubset

subsetEndpointFilter

subsetHashEpFilter

findSubet

findSubet

3. 添加常量与获取染色key的方法

在tars/util/current/tarscurrent相关包里，处理上下文信息相关；

3.1 Go语言修改部分

3.2 修改地方的逻辑

STATUS_ROUTE_KEY

4. 【核心】修改获取服务IP规则

在节点管理的相关文件里；方法是实现在第8点；

4.1 Go语言修改部分

4.2 修改地方的逻辑

current.STATUS_ROUTE_KEY

5. 实现透传染色Key功能（客户端）

在tars/tarsprotocol相关文件里；

5.1 Go语言修改部分

5.2 修改地方的逻辑

注意：

• 这里的染色Key为新建的，与源代码里的染色Key不同；
• 可以参考染色一致的实现方式，区别是Key的名称不同，实现思路类似；
• 在TarsJava客户端中，这里的最终执行的方法指

  TarsInvoker

  类里的三个执行方法；即：同步调用方法

  invokeWithSync()

  、异步调用方法

  invokeWithAsync()

  和协程调用方法

  invokeWithPromiseFuture()

6. 实现透传染色Key功能（服务端）

在tars/tarsprotocol.go相关文件里；

6.1 Go语言修改部分

6.2 修改地方的逻辑

• 在TarsJava服务端中，这里的最终执行的方法指

  TarsServantProcessor.process()

7. 给节点信息增添Subset字段

在节点信息相关文件里；

7.1 Go语言修改部分

7.2 修改地方的逻辑

注意：

• 不一定是String类型，只要在Endpoint对象结构体里添加一个Subset相关属性，有地方用即可；
• 这部分在Java中仅为Endpoint.java类，故放在一起；

* 8. 新增工具类

在工具类包里；

8.1 Go语言修改部分

最后

新人制作，如有错误，欢迎指出，感激不尽！欢迎关注公众号，会分享一些更日常的东西！如需转载，请标注出处！