You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
351 lines
13 KiB
351 lines
13 KiB
package service |
|
|
|
import ( |
|
"context" |
|
"math" |
|
"time" |
|
|
|
"go-common/app/job/main/figure-timer/conf" |
|
"go-common/app/job/main/figure-timer/model" |
|
"go-common/library/log" |
|
) |
|
|
|
const ( |
|
bizTypePosLawful = iota |
|
bizTypeNegLawful |
|
// bizTypePosWide |
|
// bizTypeNegWide |
|
bizTypePosFriendly |
|
bizTypeNegFriendly |
|
bizTypePosCreativity |
|
// bizTypeNegCreativity |
|
bizTypePosBounty |
|
// bizTypeNegBounty |
|
) |
|
|
|
// HandleFigure handle all figure score for a mid |
|
func (s *Service) HandleFigure(c context.Context, mid int64, weekVer int64) (err error) { |
|
var ( |
|
figure = &model.Figure{} |
|
userInfo *model.UserInfo |
|
actionCounter *model.ActionCounter |
|
records []*model.FigureRecord |
|
newRecord *model.FigureRecord |
|
weekVerRecordsFrom = time.Unix(weekVer, 0).AddDate(0, 0, -7*52).Unix() |
|
weekVerRecordsTo = time.Unix(weekVer, 0).AddDate(0, 0, -7).Unix() + 1 |
|
) |
|
//1. get user_info from hbase |
|
if userInfo, err = s.dao.UserInfo(c, mid, weekVer); err != nil { |
|
return |
|
} |
|
//2. get action_counter from hbase |
|
if actionCounter, err = s.dao.ActionCounter(c, mid, weekVer); err != nil { |
|
return |
|
} |
|
//3. get figure_records from hbase |
|
if records, err = s.dao.CalcRecords(c, mid, weekVerRecordsFrom, weekVerRecordsTo); err != nil { |
|
return |
|
} |
|
//4. calc figure |
|
figure, newRecord = s.CalcFigure(c, userInfo, []*model.ActionCounter{actionCounter}, records, weekVer) |
|
log.Info("User figure [%+v]", figure) |
|
log.Info("User newRecord [%+v]", newRecord) |
|
//5. save to db |
|
time.Sleep(time.Millisecond) |
|
if figure.ID, err = s.dao.UpsertFigure(c, figure); err != nil { |
|
log.Error("%+v", err) |
|
} |
|
//6. save new record |
|
if err = s.dao.PutCalcRecord(c, newRecord, weekVer); err != nil { |
|
log.Error("%+v", err) |
|
} |
|
//7. remove existed redis |
|
if err = s.dao.RemoveCache(c, mid); err != nil { |
|
log.Error("%+v", err) |
|
} |
|
rank.AddScore(figure.Score) |
|
return |
|
} |
|
|
|
// CalcFigure calc figure. |
|
func (s *Service) CalcFigure(c context.Context, userInfo *model.UserInfo, actionCounters []*model.ActionCounter, records []*model.FigureRecord, weekVer int64) (figure *model.Figure, newRecord *model.FigureRecord) { |
|
figure = &model.Figure{Mid: userInfo.Mid, Ver: weekVer} |
|
newRecord = &model.FigureRecord{Mid: userInfo.Mid, Version: time.Unix(weekVer, 0)} |
|
var ( |
|
posx float64 |
|
negx float64 |
|
newPosx, newNegx float64 |
|
k1, k2, k3, k4, k5 float64 = s.c.Property.Calc.K1, s.c.Property.Calc.K2, s.c.Property.Calc.K3, s.c.Property.Calc.K4, s.c.Property.Calc.K5 |
|
posOffset, negOffset int64 |
|
lawfulBase = s.c.Property.Calc.InitLawfulScore |
|
lawfulPosMax = s.c.Property.Calc.LawfulPosMax |
|
lawfulNegMax = s.c.Property.Calc.LawfulNegMax |
|
lawfulPosK = s.c.Property.Calc.LawfulPosK |
|
lawfulNegK1 = s.c.Property.Calc.LawfulNegK1 |
|
lawfulNegK2 = s.c.Property.Calc.LawfulNegK2 |
|
lawfulPosL = s.c.Property.Calc.LawfulPosL |
|
lawfulNegL = s.c.Property.Calc.LawfulNegL |
|
lawfulPosC3 = s.c.Property.Calc.LawfulPosC3 |
|
lawfulNegC1 = s.c.Property.Calc.LawfulNegC1 |
|
lawfulPosQ3 = s.c.Property.Calc.LawfulPosQ3 |
|
lawfulNegQ1 = s.c.Property.Calc.LawfulNegQ1 |
|
wideBase = s.c.Property.Calc.InitWideScore |
|
widePosMax = s.c.Property.Calc.WidePosMax |
|
widePosK = s.c.Property.Calc.WidePosK |
|
wideC1 = s.c.Property.Calc.WideC1 //有播放的活跃天数 |
|
wideQ1 = s.c.Property.Calc.WideQ1 |
|
wideC2 = s.c.Property.Calc.WideC2 // 账号累计经验值 |
|
wideQ2 = s.c.Property.Calc.WideQ2 |
|
friendlyBase = s.c.Property.Calc.InitFriendlyScore |
|
friendlyPosMax = s.c.Property.Calc.FriendlyPosMax |
|
friendlyNegMax = s.c.Property.Calc.FriendlyNegMax |
|
friendlyPosK = s.c.Property.Calc.FriendlyPosK |
|
friendlyNegK = s.c.Property.Calc.FriendlyNegK |
|
friendlyPosL = s.c.Property.Calc.FriendlyPosL |
|
friendlyNegL = s.c.Property.Calc.FriendlyNegL |
|
bountyBase = s.c.Property.Calc.InitBountyScore |
|
bountyMax = s.c.Property.Calc.BountyMax |
|
bountyPosL = s.c.Property.Calc.BountyPosL |
|
bountyK = s.c.Property.Calc.BountyK |
|
bountyQ1 = s.c.Property.Calc.BountyQ1 |
|
bountyC1 = s.c.Property.Calc.BountyC1 |
|
creativityBase = s.c.Property.Calc.InitCreativityScore |
|
creativityPosMax = s.c.Property.Calc.CreativityPosMax |
|
creativityPosK = s.c.Property.Calc.CreativityPosK |
|
creativityPosL1 = s.c.Property.Calc.CreativityPosL1 |
|
) |
|
//1. lawful |
|
newPosx, posx = s.calcActionX(lawfulPosL, bizTypePosLawful, actionCounters, records, weekVer) |
|
posx += lawfulPosQ3 * lawfulPosC3 * float64(userInfo.DisciplineCommittee) |
|
posOffset = calcOffset(lawfulPosMax, lawfulPosK, posx) |
|
|
|
spyScore := userInfo.SpyScore |
|
negx = lawfulNegQ1 * lawfulNegC1 * float64(80-float64(spyScore)) |
|
if negx < 0 { |
|
negx = 0.0 |
|
} |
|
negOffset = calcOffset(lawfulNegMax, lawfulNegK1, negx) |
|
newNegx, negx = s.calcActionX(lawfulNegL, bizTypeNegLawful, actionCounters, records, weekVer) |
|
negOffset += calcOffset(lawfulNegMax, lawfulNegK2, negx) |
|
figure.LawfulScore = int32(lawfulBase + posOffset - negOffset) |
|
if figure.LawfulScore < 0 { |
|
figure.LawfulScore = 0 |
|
} |
|
newRecord.XPosLawful, newRecord.XNegLawful = int64(newPosx), int64(newNegx) |
|
|
|
//2. wide |
|
newPosx, newNegx = 0, 0 |
|
posx = wideQ1*wideC1*float64(userInfo.ArchiveViews) + wideQ2*wideC2*float64(userInfo.Exp) |
|
posOffset = calcOffset(widePosMax, widePosK, posx) |
|
negx = 0 |
|
negOffset = 0 |
|
figure.WideScore = int32(wideBase + posOffset - negOffset) |
|
if figure.WideScore < 0 { |
|
figure.WideScore = 0 |
|
} |
|
newRecord.XPosWide, newRecord.XNegWide = int64(newPosx), int64(newNegx) |
|
|
|
//3. friendly |
|
newPosx, newNegx = 0, 0 |
|
newPosx, posx = s.calcActionX(friendlyPosL, bizTypePosFriendly, actionCounters, records, weekVer) |
|
posOffset = calcOffset(friendlyPosMax, friendlyPosK, posx) |
|
newNegx, negx = s.calcActionX(friendlyNegL, bizTypeNegFriendly, actionCounters, records, weekVer) |
|
negOffset = calcOffset(friendlyNegMax, friendlyNegK, negx) |
|
figure.FriendlyScore = int32(friendlyBase + posOffset - negOffset) |
|
if figure.FriendlyScore < 0 { |
|
figure.FriendlyScore = 0 |
|
} |
|
newRecord.XPosFriendly, newRecord.XNegFriendly = int64(newPosx), int64(newNegx) |
|
|
|
//4. bounty |
|
newPosx, newNegx = 0, 0 |
|
var bountyVIP float64 |
|
if userInfo.VIPStatus > 0 { |
|
bountyVIP = 1 |
|
} |
|
newPosx, posx = s.calcActionX(bountyPosL, bizTypePosBounty, actionCounters, records, weekVer) |
|
posx += bountyQ1 * bountyC1 * bountyVIP |
|
posOffset = calcOffset(bountyMax, bountyK, posx) |
|
negx = 0 |
|
negOffset = 0 |
|
figure.BountyScore = int32(bountyBase + posOffset - negOffset) |
|
if figure.BountyScore < 0 { |
|
figure.BountyScore = 0 |
|
} |
|
newRecord.XPosBounty, newRecord.XNegBounty = int64(newPosx), int64(newNegx) |
|
|
|
//5. creativity |
|
newPosx, newNegx = 0, 0 |
|
newPosx, posx = s.calcActionX(creativityPosL1, bizTypePosCreativity, actionCounters, records, weekVer) |
|
posOffset = calcOffset(creativityPosMax, creativityPosK, posx) |
|
negx = 0 |
|
negOffset = 0 |
|
figure.CreativityScore = int32(creativityBase + posOffset - negOffset) |
|
if figure.CreativityScore < 0 { |
|
figure.CreativityScore = 0 |
|
} |
|
newRecord.XPosCreativity, newRecord.XNegCreativity = int64(newPosx), int64(newNegx) |
|
|
|
//6. calc score |
|
figure.Score = int32(math.Floor(k1*float64(figure.LawfulScore) + k2*float64(figure.WideScore) + k3*float64(figure.FriendlyScore) + k4*float64(figure.CreativityScore) + k5*float64(figure.BountyScore))) |
|
return |
|
} |
|
|
|
// x must >= 0 |
|
func calcOffset(max, k, x float64) (score int64) { |
|
return int64(math.Floor(max * (1 - math.Pow(math.E, -(k*x))))) |
|
} |
|
|
|
func (s *Service) calcActionX(L float64, bizType int8, actionCounters []*model.ActionCounter, records []*model.FigureRecord, weekVer int64) (newx, totalx float64) { |
|
var ( |
|
day int64 = 24 * 3600 |
|
t float64 |
|
|
|
lawfulPosC1 = s.c.Property.Calc.LawfulPosC1 |
|
lawfulPosC2 = s.c.Property.Calc.LawfulPosC2 |
|
lawfulPosQ1 = s.c.Property.Calc.LawfulPosQ1 |
|
lawfulPosQ2 = s.c.Property.Calc.LawfulPosQ2 |
|
lawfulNegC2 = s.c.Property.Calc.LawfulNegC2 |
|
lawfulNegC3 = s.c.Property.Calc.LawfulNegC3 |
|
lawfulNegQ2 = s.c.Property.Calc.LawfulNegQ2 |
|
lawfulNegQ3 = s.c.Property.Calc.LawfulNegQ3 |
|
|
|
friendlyPosQ1 = s.c.Property.Calc.FriendlyPosQ1 |
|
friendlyPosC1 = s.c.Property.Calc.FriendlyPosC1 |
|
friendlyPosQ2 = s.c.Property.Calc.FriendlyPosQ2 |
|
friendlyPosC2 = s.c.Property.Calc.FriendlyPosC2 |
|
friendlyPosQ3 = s.c.Property.Calc.FriendlyPosQ3 |
|
friendlyPosC3 = s.c.Property.Calc.FriendlyPosC3 |
|
friendlyNegQ1 = s.c.Property.Calc.FriendlyNegQ1 |
|
friendlyNegC1 = s.c.Property.Calc.FriendlyNegC1 |
|
friendlyNegQ2 = s.c.Property.Calc.FriendlyNegQ2 |
|
friendlyNegC2 = s.c.Property.Calc.FriendlyNegC2 |
|
friendlyNegQ3 = s.c.Property.Calc.FriendlyNegQ3 |
|
friendlyNegC3 = s.c.Property.Calc.FriendlyNegC3 |
|
friendlyNegQ4 = s.c.Property.Calc.FriendlyNegQ4 |
|
friendlyNegC4 = s.c.Property.Calc.FriendlyNegC4 |
|
|
|
creativityQ1 = s.c.Property.Calc.CreativityQ1 |
|
creativityC1 = s.c.Property.Calc.CreativityC1 |
|
|
|
bountyQ2 = s.c.Property.Calc.BountyQ2 |
|
bountyC2 = s.c.Property.Calc.BountyC2 |
|
bountyQ3 = s.c.Property.Calc.BountyQ3 |
|
bountyC3 = s.c.Property.Calc.BountyC3 |
|
) |
|
if L == 0.0 { |
|
return 0, 0 |
|
} |
|
for _, ac := range actionCounters { |
|
t = float64(7 - (ac.Version.Unix()-weekVer)/day) |
|
if t <= 0 { |
|
continue |
|
} |
|
switch bizType { |
|
case bizTypePosLawful: |
|
if ac.ReportReplyPassed < 0 { |
|
ac.ReportReplyPassed = 0 |
|
} |
|
newx += actionX(lawfulPosQ1, lawfulPosC1, float64(ac.ReportReplyPassed), t, L) |
|
if ac.ReportDanmakuPassed < 0 { |
|
ac.ReportDanmakuPassed = 0 |
|
} |
|
newx += actionX(lawfulPosQ2, lawfulPosC2, float64(ac.ReportDanmakuPassed), t, L) |
|
case bizTypeNegLawful: |
|
if ac.PublishReplyDeleted < 0 { |
|
ac.PublishReplyDeleted = 0 |
|
} |
|
newx += actionX(lawfulNegQ2, lawfulNegC2, float64(ac.PublishReplyDeleted), t, L) |
|
if ac.PublishDanmakuDeleted < 0 { |
|
ac.PublishDanmakuDeleted = 0 |
|
} |
|
newx += actionX(lawfulNegQ3, lawfulNegC3, float64(ac.PublishDanmakuDeleted), t, L) |
|
case bizTypePosFriendly: |
|
newx += actionX(friendlyPosQ1, friendlyPosC1, float64(ac.CoinCount), t, L) |
|
newx += actionX(friendlyPosQ2, friendlyPosC2, float64(ac.ReplyCount), t, L) |
|
newx += actionX(friendlyPosQ3, friendlyPosC3, float64(ac.DanmakuCount), t, L) |
|
case bizTypeNegFriendly: |
|
newx += actionX(friendlyNegQ1, friendlyNegC1, float64(ac.CoinHighRisk), t, L) |
|
newx += actionX(friendlyNegQ2, friendlyNegC2, float64(ac.CoinLowRisk), t, L) |
|
if ac.PublishReplyDeleted < 0 { |
|
ac.PublishReplyDeleted = 0 |
|
} |
|
newx += actionX(friendlyNegQ3, friendlyNegC3, float64(ac.PublishReplyDeleted), t, L) |
|
if ac.PublishDanmakuDeleted < 0 { |
|
ac.PublishDanmakuDeleted = 0 |
|
} |
|
newx += actionX(friendlyNegQ4, friendlyNegC4, float64(ac.PublishDanmakuDeleted), t, L) |
|
case bizTypePosCreativity: |
|
replyLikeCount := float64(ac.ReplyLiked) - float64(ac.ReplyUnliked) |
|
if replyLikeCount < 0 { |
|
replyLikeCount = 0.0 |
|
} |
|
newx += actionX(creativityQ1, creativityC1, replyLikeCount, t, L) |
|
case bizTypePosBounty: |
|
newx += actionX(bountyQ2, bountyC2, float64(ac.PayMoney), t, L) |
|
newx += actionX(bountyQ3, bountyC3, float64(ac.PayLiveMoney), t, L) |
|
} |
|
} |
|
totalx = newx |
|
for _, r := range records { |
|
t = float64((weekVer - r.Version.Unix()) / day) |
|
if t <= 0 { |
|
continue |
|
} |
|
switch bizType { |
|
case bizTypePosLawful: |
|
totalx += actionX(1, 1, float64(r.XPosLawful), t, L) |
|
case bizTypeNegLawful: |
|
totalx += actionX(1, 1, float64(r.XNegLawful), t, L) |
|
case bizTypePosFriendly: |
|
totalx += actionX(1, 1, float64(r.XPosFriendly), t, L) |
|
case bizTypeNegFriendly: |
|
totalx += actionX(1, 1, float64(r.XNegFriendly), t, L) |
|
case bizTypePosCreativity: |
|
totalx += actionX(1, 1, float64(r.XPosCreativity), t, L) |
|
case bizTypePosBounty: |
|
totalx += actionX(1, 1, float64(r.XPosBounty), t, L) |
|
} |
|
} |
|
return |
|
} |
|
|
|
func actionX(q, c, x, t, L float64) float64 { |
|
return q * c * x * math.Pow(math.E, -math.Pow((t/L), 2)) |
|
} |
|
|
|
// InitFigure initialize user figure |
|
func (s *Service) InitFigure(c context.Context, mid int64, ver string) (figure *model.Figure, err error) { |
|
figure = &model.Figure{} |
|
figure.Mid = mid |
|
figure.LawfulScore = int32(conf.Conf.Property.Calc.InitLawfulScore) |
|
figure.WideScore = int32(conf.Conf.Property.Calc.InitWideScore) |
|
figure.FriendlyScore = int32(conf.Conf.Property.Calc.InitFriendlyScore) |
|
figure.BountyScore = int32(conf.Conf.Property.Calc.InitBountyScore) |
|
figure.CreativityScore = int32(conf.Conf.Property.Calc.InitCreativityScore) |
|
figure.Ver = s.curVer |
|
|
|
if figure.ID, err = s.dao.UpsertFigure(c, figure); err != nil { |
|
return |
|
} |
|
if err = s.dao.SetFigureCache(c, figure); err != nil { |
|
log.Error("%+v", err) |
|
} |
|
return |
|
} |
|
|
|
// get ever monday start time ts. |
|
func weekVersion(now time.Time) (ts int64) { |
|
var ( |
|
wd int |
|
w time.Weekday |
|
) |
|
w = now.Weekday() |
|
switch w { |
|
case time.Sunday: |
|
wd = 6 |
|
default: |
|
wd = int(w) - 1 |
|
} |
|
return time.Date(now.Year(), now.Month(), now.Day(), 0, 0, 0, 0, time.Local).AddDate(0, 0, -wd).Unix() |
|
}
|
|
|