83 lines
2.4 KiB
Go
83 lines
2.4 KiB
Go
package dao
|
||
|
||
import (
|
||
"strconv"
|
||
"strings"
|
||
|
||
"go-common/app/interface/main/push-archive/conf"
|
||
"go-common/app/interface/main/push-archive/model"
|
||
"go-common/library/log"
|
||
)
|
||
|
||
// Proportion 普通关注粉丝的灰度比例
|
||
type Proportion struct {
|
||
// 粉丝的后2位的最值
|
||
MinValue int
|
||
MaxValue int
|
||
}
|
||
|
||
// NewProportion new
|
||
func NewProportion(config []conf.Proportion) (ps []Proportion) {
|
||
var ppt float64
|
||
for _, g := range config {
|
||
valueStartFloat, err := strconv.ParseFloat(strings.TrimSpace(g.ProportionStartFrom), 64)
|
||
if err != nil {
|
||
log.Error("NewProportions config ArcPush.FanGroup.ProportionStartFrom strconv.ParseFloat(%s) error(%v)", g.ProportionStartFrom, err)
|
||
return
|
||
}
|
||
valueStartFrom := int(valueStartFloat)
|
||
// 比例验证
|
||
prop, err := strconv.ParseFloat(strings.TrimSpace(g.Proportion), 64)
|
||
if err != nil {
|
||
log.Error("NewProportions config ArcPush.FanGroup.Proportion(%s) strconv.ParseFloat err(%v)", g.Proportion, err)
|
||
return nil
|
||
}
|
||
if prop*100-float64(int(prop*100)) != 0 {
|
||
// 比例最多保留2位小数
|
||
log.Error("NewProportions config ArcPush.FanGroup.Proportion(%s) must keep at most 2 bits", g.Proportion)
|
||
return nil
|
||
}
|
||
ppt += prop
|
||
if prop <= 0 || prop > 1 || ppt > 1 || ppt <= 0 {
|
||
// 单个数在(0,1]区间，总和在(0, 1]区间
|
||
log.Error("NewProportions config ArcPush.FanGroup.Proportion(%s) must in (0, 1] and sum(%f) in (0, 1]", g.Proportion, ppt)
|
||
return nil
|
||
}
|
||
// 起始值和比例之和必须在00～99之间
|
||
maxValue := int(100*prop-1) + valueStartFrom
|
||
if maxValue >= 100 {
|
||
log.Error("NewProportions config ArcPush.FanGroup.Proportion(%s)+ProportionStartFrom must in [0, 99]", g.Proportion)
|
||
return
|
||
}
|
||
p := Proportion{
|
||
MinValue: valueStartFrom,
|
||
MaxValue: maxValue,
|
||
}
|
||
ps = append(ps, p)
|
||
}
|
||
return
|
||
}
|
||
|
||
// FansByProportion 根据比例分配该关注类型的粉丝, 以全站所有用户作为分母
|
||
func (d *Dao) FansByProportion(upper int64, fans map[int64]int) (attentions []int64, specials []int64) {
|
||
for mid, relationType := range fans {
|
||
if relationType == model.RelationSpecial {
|
||
specials = append(specials, mid)
|
||
continue
|
||
}
|
||
if len(d.Proportions) == 0 {
|
||
attentions = append(attentions, mid)
|
||
continue
|
||
}
|
||
// mid最后2位是否在抽样区间内
|
||
last2Digits := int(mid % 100)
|
||
for _, g := range d.Proportions {
|
||
if last2Digits >= g.MinValue && last2Digits <= g.MaxValue {
|
||
attentions = append(attentions, mid)
|
||
break
|
||
}
|
||
}
|
||
}
|
||
return
|
||
}
|