Merge pull request #5 from volcengine/feature/support_domain

Feature/support domain

Author: oMinGWo

client/client.go

@@ -16,6 +16,7 @@ import (
 	"github.com/volcengine/datatester-go-sdk/handler"
 	"github.com/volcengine/datatester-go-sdk/log"
 	"github.com/volcengine/datatester-go-sdk/meta/manager"
+	"github.com/volcengine/datatester-go-sdk/utils"
 )
 
 type AbClient struct {
@@ -180,7 +181,10 @@ func (t *AbClient) GetAllExperimentConfigs(decisionId string,
 			continue
 		}
 		for k, v := range confMap {
-			configs[k] = v
+			existConfig, conflict := configs[k]
+			if !conflict || utils.IsHigherPriorityConfig(existConfig, v) {
+				configs[k] = v
+			}
 		}
 	}
 	t.updateUserAbInfo(decisionId, experiment2variant)
@@ -212,7 +216,10 @@ func (t *AbClient) getAllExperimentConfigs4Activate(variantKey, decisionId strin
 			continue
 		}
 		for k, v := range confMap {
-			configs[k] = v
+			existConfig, conflict := configs[k]
+			if !conflict || utils.IsHigherPriorityConfig(existConfig, v) {
+				configs[k] = v
+			}
 		}
 		vid2ExperimentIdMap[variant.Id] = variant.ExperimentId
 	}
@@ -426,7 +433,10 @@ func (t *AbClient) GetAllFeatureConfigs(decisionId string,
 			continue
 		}
 		for k, v := range confMap {
-			configs[k] = v
+			existConfig, conflict := configs[k]
+			if !conflict || utils.IsHigherPriorityConfig(existConfig, v) {
+				configs[k] = v
+			}
 		}
 	}
 	return configs, nil

distributor/distributor.go

@@ -254,6 +254,11 @@ func (v *VariantsDistributor) tabExperimentVariant(c *config.ProductConfig, expe
 	if !ok {
 		return e.Variant{}, fmt.Errorf("no layer[%s] exist in config", experiment.LayerID)
 	}
+	// check 互斥域
+	hitDomain := layer.HitDomain(decisionId)
+	if !hitDomain {
+		return e.Variant{}, nil
+	}
 	eid, err := v.tabLayerExperimentId(layer, decisionId)
 	if err != nil || len(eid) == 0 || eid != experiment.Id {
 		return e.Variant{}, err

entities/domain.go

@@ -0,0 +1,28 @@
+package entities
+
+import (
+	"strings"
+
+	"github.com/volcengine/datatester-go-sdk/distributor/bucketer"
+)
+
+type Domain struct {
+	ID           string `json:"id"`
+	Name         string `json:"name"`
+	HashStrategy string `json:"hash_strategy"`
+	Begin        uint32 `json:"begin"`
+	Length       uint32 `json:"length"`
+}
+
+// Hit 方法检查给定的 decisionID 是否在当前 Domain 的索引范围内
+func (d Domain) Hit(decisionID string) bool {
+	// 通过连接 decisionID 和 d.Name 获取流量桶索引
+	index, err := bucketer.NewMmh3BucketService().GetTrafficBucketIndex(
+		strings.Join([]string{decisionID, d.Name}, ":"))
+	// 如果获取索引过程中出现错误，返回 false
+	if err != nil {
+		return false
+	}
+	// 如果索引值在 d.Begin 和 d.Begin + d.Length 之间，返回 true，否则返回 false
+	return index >= d.Begin && index < d.Begin+d.Length
+}

entities/layer.go

@@ -14,4 +14,25 @@ type Layer struct {
 	Name              string                      `json:"name"`
 	TrafficAllocation []release.TrafficAllocation `json:"traffic_allocation"`
 	ExperimentIds     []string                    `json:"experiment_ids"`
+	Domain            *Domain                     `json:"domain"`
+	ParentDomains     []Domain                    `json:"parent_domains"`
+}
+
+// HitDomain 方法用于检查给定的 decisionID 是否命中当前层及其所有父域
+func (l Layer) HitDomain(decisionID string) bool {
+	// 层不属于互斥域，直接返回
+	if l.Domain == nil || l.Domain.ID == "0" {
+		return true
+	}
+	hit := l.Domain.Hit(decisionID)
+	if !hit {
+		return false
+	}
+	for _, d := range l.ParentDomains {
+		hit = d.Hit(decisionID)
+		if !hit {
+			return false
+		}
+	}
+	return true
 }

utils/conflict.go

@@ -0,0 +1,18 @@
+package utils
+
+import (
+	"strconv"
+	"strings"
+)
+
+func IsHigherPriorityConfig(existConf, newConf map[string]interface{}) bool {
+	vid1 := existConf["vid"].(string)
+	vid2 := newConf["vid"].(string)
+	v1, err1 := strconv.ParseInt(vid1, 10, 64)
+	v2, err2 := strconv.ParseInt(vid2, 10, 64)
+	// 转换失败，则按照字符串对比，一般不会出现
+	if err1 != nil || err2 != nil {
+		return strings.Compare(vid2, vid1) < 0
+	}
+	return v2 < v1
+}