52 lines
1.3 KiB
Go
52 lines
1.3 KiB
Go
|
package service
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"go-common/app/interface/main/push-archive/dao"
|
|||
|
|
"go-common/app/interface/main/push-archive/model"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
var (
|
|||
|
|
// 存放实验组mid尾号
|
|||
|
|
fansTestGroup = make(map[int]struct{})
|
|||
|
|
// 存放对照组mid尾号
|
|||
|
|
fansComparisonGroup = make(map[int]struct{})
|
|||
|
|
// 指定mid放进测试组
|
|||
|
|
fansTestMids = make(map[int64]struct{})
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func (s *Service) mappingAbtest() {
|
|||
|
|
for _, n := range s.c.Abtest.TestGroup {
|
|||
|
|
fansTestGroup[n] = struct{}{}
|
|||
|
|
}
|
|||
|
|
for _, n := range s.c.Abtest.ComparisonGroup {
|
|||
|
|
fansComparisonGroup[n] = struct{}{}
|
|||
|
|
}
|
|||
|
|
for _, n := range s.c.Abtest.TestMids {
|
|||
|
|
fansTestMids[n] = struct{}{}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 将所有粉丝通过abtest规则，拆分成 (实验流量||对照组流量) && 其余流量
|
|||
|
|
func (s *Service) fansByAbtest(group *dao.FanGroup, fans []int64) (result, others []int64) {
|
|||
|
|
for _, fan := range fans {
|
|||
|
|
n := int(fan % 10)
|
|||
|
|
if group.Hitby == model.GroupDataTypeAbtest {
|
|||
|
|
if _, ok := fansTestMids[fan]; ok {
|
|||
|
|
result = append(result, fan)
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
if _, ok := fansTestGroup[n]; ok {
|
|||
|
|
result = append(result, fan)
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
} else if group.Hitby == model.GroupDataTypeAbComparison {
|
|||
|
|
if _, ok := fansComparisonGroup[n]; ok {
|
|||
|
|
result = append(result, fan)
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
others = append(others, fan)
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|