|
- package rule
-
- import (
- "code.fnuoos.com/EggPlanet/egg_models.git/src/implement"
- "code.fnuoos.com/EggPlanet/egg_models.git/src/model"
- "code.fnuoos.com/EggPlanet/egg_system_rules.git/md"
- md2 "code.fnuoos.com/EggPlanet/egg_system_rules.git/rule/egg_energy/md"
- "code.fnuoos.com/EggPlanet/egg_system_rules.git/utils"
- "code.fnuoos.com/EggPlanet/egg_system_rules.git/utils/cache"
- "code.fnuoos.com/EggPlanet/egg_system_rules.git/utils/logx"
- "errors"
- "fmt"
- "math"
- "strconv"
- "strings"
- "time"
- "xorm.io/xorm"
- )
-
- // AddPublicPlatoonUserRelationCommission 新增公排用户关系记录
- func AddPublicPlatoonUserRelationCommission(engine *xorm.Engine, AddPublicPlatoonUserRelationCommissionReqList []*md2.AddPublicPlatoonUserRelationCommissionReq) (map[string]*model.PublicPlatoonUserRelation, error) {
- var resp = map[string]*model.PublicPlatoonUserRelation{}
- //查找 `public_platoon_basic_setting` 基础设置
- publicPlatoonBasicSettingDb := implement.NewPublicPlatoonBasicSettingDb(engine)
- publicPlatoonBasicSetting, err := publicPlatoonBasicSettingDb.PublicPlatoonBasicSettingGetOneByParams(map[string]interface{}{
- "key": "is_open",
- "value": 1,
- })
- if err != nil {
- return nil, err
- }
- if publicPlatoonBasicSetting == nil {
- return nil, errors.New("公排制度未开启")
- }
-
- publicPlatoonUserRelationDb := implement.NewPublicPlatoonUserRelationDb(engine)
- for _, param := range AddPublicPlatoonUserRelationCommissionReqList {
- isHasPublicPlatoonUserRelation, err11 := publicPlatoonUserRelationDb.PublicPlatoonUserRelationGetOneByParams(map[string]interface{}{
- "key": "uid",
- "value": param.Uid,
- })
- if err11 != nil {
- return nil, err11
- }
- if isHasPublicPlatoonUserRelation != nil {
- return nil, errors.New("当前用户已加入公排")
- }
-
- //1、判断是否有推荐人
- if param.RecommendUid != "" {
- //2、有推荐人
- //判断是否有uid为-1 (代表等待新用户填充) 的记录
- PublicPlatoonUserRelation, err1 := publicPlatoonUserRelationDb.PublicPlatoonUserRelationGetOneByPid(param.RecommendUid, map[string]interface{}{
- "key": "uid",
- "value": -1,
- })
- if err1 != nil {
- return nil, err1
- }
- if PublicPlatoonUserRelation != nil {
- now := time.Now()
- oldUid := PublicPlatoonUserRelation.Uid
- PublicPlatoonUserRelation.Uid = egg_system_rules.StrToInt64(param.Uid)
- PublicPlatoonUserRelation.RecommendUid = egg_system_rules.StrToInt64(param.RecommendUid)
- PublicPlatoonUserRelation.CreateAt = now.Format("2006-01-02 15:04:05")
- PublicPlatoonUserRelation.UpdateAt = now.Format("2006-01-02 15:04:05")
- PublicPlatoonUserRelation.JoinAt = now.Format("2006-01-02 15:04:05")
- updateAffected, err2 := publicPlatoonUserRelationDb.PublicPlatoonUserRelationUpdate(engine.NewSession(), PublicPlatoonUserRelation.Id, PublicPlatoonUserRelation)
- if err2 != nil {
- return nil, err2
- }
- if updateAffected == 0 {
- err = errors.New("更新 public_platoon_user_relation 记录失败")
- return nil, err
- } else {
- for n := 1; n <= 9; n++ {
- str := "father_uid" + strconv.Itoa(n)
- sql := fmt.Sprintf("UPDATE `public_platoon_user_relation` SET %s=%s where %s=%s", str, param.Uid, str, egg_system_rules.Int64ToStr(oldUid))
- fmt.Println(">>>>>>>sql>>>>>>>", sql)
- _, err = engine.Exec(sql)
- if err != nil {
- return nil, err
- }
- }
- resp[param.Uid] = PublicPlatoonUserRelation
- }
- continue
- }
- res, err := PublicPlatoon(engine, egg_system_rules.StrToInt64(param.Uid), egg_system_rules.StrToInt64(param.RecommendUid), *publicPlatoonBasicSetting)
- if err != nil {
- return nil, err
- }
- resp[param.Uid] = &res
- }
- }
- return resp, nil
- }
-
- /*
- 公排方法
-
- TODO 相关公式:
- 1: 每个等级的起始值(1+5^0+5^1+5^2+...+5^x-2), 每个等级的结束值(`5^0+5^1+5^2+...+5^x-1)
- 2: 根据position查找父级position { (position-1)/5 }
- 3: 根据position查找等级level {position-5^0-5^1-5^2-...-5^x 是否 <0 ? => x+1 }
- 4: 根据最新自增`id` 逆向推导 position {levelFirstPosition + (position-1)%5}
- */
- func PublicPlatoon(engine *xorm.Engine, uid, recommendUid int64, PublicPlatoonBasicSetting model.PublicPlatoonBasicSetting) (model.PublicPlatoonUserRelation, error) {
- var standardPublicPlatoonRelation md.StandardPublicPlatoonRelation
- var PublicPlatoonUserRelation model.PublicPlatoonUserRelation
- var fatherUid int64
- //1、 查找当前 public_platoon_user_relation 中 recommendUid 的记录
- publicPlatoonUserRelationDb := implement.NewPublicPlatoonUserRelationDb(engine)
- m, err := publicPlatoonUserRelationDb.PublicPlatoonUserRelationGetOneByParams(map[string]interface{}{
- "key": "uid",
- "value": recommendUid,
- })
- if err != nil {
- return model.PublicPlatoonUserRelation{}, err
- }
-
- //2、 查找当前 public_platoon_user_relation 中 recommendUid 所有的子记录
- PublicPlatoonUserRelations, err := publicPlatoonUserRelationDb.PublicPlatoonUserRelationFindByPid(m.Uid, "father_uid1", "position1")
- if err != nil {
- return model.PublicPlatoonUserRelation{}, err
- }
-
- if len(PublicPlatoonUserRelations) == 0 {
- fatherUid = m.Uid
-
- //证明是第一个
- standardPublicPlatoonRelation.FatherUid1 = m.Uid
- standardPublicPlatoonRelation.FatherUid2 = m.FatherUid1
- standardPublicPlatoonRelation.FatherUid3 = m.FatherUid2
- standardPublicPlatoonRelation.FatherUid4 = m.FatherUid3
- standardPublicPlatoonRelation.FatherUid5 = m.FatherUid4
- standardPublicPlatoonRelation.FatherUid6 = m.FatherUid5
- standardPublicPlatoonRelation.FatherUid7 = m.FatherUid6
- standardPublicPlatoonRelation.FatherUid8 = m.FatherUid7
- standardPublicPlatoonRelation.FatherUid9 = m.FatherUid8
-
- standardPublicPlatoonRelation.Pid1 = m.Position
- standardPublicPlatoonRelation.Pid2 = m.Position1
- standardPublicPlatoonRelation.Pid3 = m.Position2
- standardPublicPlatoonRelation.Pid4 = m.Position3
- standardPublicPlatoonRelation.Pid5 = m.Position4
- standardPublicPlatoonRelation.Pid6 = m.Position5
- standardPublicPlatoonRelation.Pid7 = m.Position6
- standardPublicPlatoonRelation.Pid8 = m.Position7
- standardPublicPlatoonRelation.Pid9 = m.Position8
-
- standardPublicPlatoonRelation.Position1 = SearchPositionFirstSonPosition(m.Position, int64(PublicPlatoonBasicSetting.SeveralTimes))
- standardPublicPlatoonRelation.Position2 = ReverseDeductionPositionByFather(m.Position1, int64(m.Level1), int64(PublicPlatoonBasicSetting.SeveralTimes), 0)
- standardPublicPlatoonRelation.Position3 = ReverseDeductionPositionByFather(m.Position2, int64(m.Level2), int64(PublicPlatoonBasicSetting.SeveralTimes), 0)
- standardPublicPlatoonRelation.Position4 = ReverseDeductionPositionByFather(m.Position3, int64(m.Level3), int64(PublicPlatoonBasicSetting.SeveralTimes), 0)
- standardPublicPlatoonRelation.Position5 = ReverseDeductionPositionByFather(m.Position4, int64(m.Level4), int64(PublicPlatoonBasicSetting.SeveralTimes), 0)
- standardPublicPlatoonRelation.Position6 = ReverseDeductionPositionByFather(m.Position5, int64(m.Level5), int64(PublicPlatoonBasicSetting.SeveralTimes), 0)
- standardPublicPlatoonRelation.Position7 = ReverseDeductionPositionByFather(m.Position6, int64(m.Level6), int64(PublicPlatoonBasicSetting.SeveralTimes), 0)
- standardPublicPlatoonRelation.Position8 = ReverseDeductionPositionByFather(m.Position7, int64(m.Level7), int64(PublicPlatoonBasicSetting.SeveralTimes), 0)
- standardPublicPlatoonRelation.Position9 = ReverseDeductionPositionByFather(m.Position8, int64(m.Level8), int64(PublicPlatoonBasicSetting.SeveralTimes), 0)
-
- if m.Level1 == 0 {
- m.Level1--
- }
- if m.Level2 == 0 {
- m.Level2--
- }
- if m.Level3 == 0 {
- m.Level3--
- }
- if m.Level4 == 0 {
- m.Level4--
- }
- if m.Level5 == 0 {
- m.Level5--
- }
- if m.Level6 == 0 {
- m.Level6--
- }
- if m.Level7 == 0 {
- m.Level7--
- }
- if m.Level8 == 0 {
- m.Level8--
- }
- if m.Level9 == 0 {
- m.Level9--
- }
-
- standardPublicPlatoonRelation.Level1 = m.Level + 1
- standardPublicPlatoonRelation.Level2 = m.Level1 + 1
- standardPublicPlatoonRelation.Level3 = m.Level2 + 1
- standardPublicPlatoonRelation.Level4 = m.Level3 + 1
- standardPublicPlatoonRelation.Level5 = m.Level4 + 1
- standardPublicPlatoonRelation.Level6 = m.Level5 + 1
- standardPublicPlatoonRelation.Level7 = m.Level6 + 1
- standardPublicPlatoonRelation.Level8 = m.Level7 + 1
- standardPublicPlatoonRelation.Level9 = m.Level8 + 1
- standardPublicPlatoonRelation.LevelTotal = m.LevelTotal + 1
- } else {
- //TODO::判断直属下级是否排满
- if len(PublicPlatoonUserRelations) >= PublicPlatoonBasicSetting.SeveralTimes {
- var fatherLevel float64
- makeSearchLevel(&m.Position, float64(PublicPlatoonBasicSetting.SeveralTimes), &fatherLevel)
- var times = 0
- data, father, err := FindSuitablePosition(engine, &m.Position, float64(PublicPlatoonBasicSetting.SeveralTimes), m.Uid, int64(fatherLevel), ×)
- if err != nil {
- return model.PublicPlatoonUserRelation{}, err
- }
- fatherUid = father
- standardPublicPlatoonRelation = data
- } else {
- fatherUid = m.Uid
-
- standardPublicPlatoonRelation.FatherUid1 = PublicPlatoonUserRelations[0].FatherUid1
- standardPublicPlatoonRelation.FatherUid2 = PublicPlatoonUserRelations[0].FatherUid2
- standardPublicPlatoonRelation.FatherUid3 = PublicPlatoonUserRelations[0].FatherUid3
- standardPublicPlatoonRelation.FatherUid4 = PublicPlatoonUserRelations[0].FatherUid4
- standardPublicPlatoonRelation.FatherUid5 = PublicPlatoonUserRelations[0].FatherUid5
- standardPublicPlatoonRelation.FatherUid6 = PublicPlatoonUserRelations[0].FatherUid6
- standardPublicPlatoonRelation.FatherUid7 = PublicPlatoonUserRelations[0].FatherUid7
- standardPublicPlatoonRelation.FatherUid8 = PublicPlatoonUserRelations[0].FatherUid8
- standardPublicPlatoonRelation.FatherUid9 = PublicPlatoonUserRelations[0].FatherUid9
-
- standardPublicPlatoonRelation.Pid1 = PublicPlatoonUserRelations[0].Pid1
- standardPublicPlatoonRelation.Pid2 = PublicPlatoonUserRelations[0].Pid2
- standardPublicPlatoonRelation.Pid3 = PublicPlatoonUserRelations[0].Pid3
- standardPublicPlatoonRelation.Pid4 = PublicPlatoonUserRelations[0].Pid4
- standardPublicPlatoonRelation.Pid5 = PublicPlatoonUserRelations[0].Pid5
- standardPublicPlatoonRelation.Pid6 = PublicPlatoonUserRelations[0].Pid6
- standardPublicPlatoonRelation.Pid7 = PublicPlatoonUserRelations[0].Pid7
- standardPublicPlatoonRelation.Pid8 = PublicPlatoonUserRelations[0].Pid8
- standardPublicPlatoonRelation.Pid9 = PublicPlatoonUserRelations[0].Pid9
-
- standardPublicPlatoonRelation.Position1 = PublicPlatoonUserRelations[0].Position1 + 1
- if PublicPlatoonUserRelations[0].Position2 != 0 {
- standardPublicPlatoonRelation.Position2 = PublicPlatoonUserRelations[0].Position2 + 1
- }
- if PublicPlatoonUserRelations[0].Position3 != 0 {
- standardPublicPlatoonRelation.Position3 = PublicPlatoonUserRelations[0].Position3 + 1
- }
- if PublicPlatoonUserRelations[0].Position4 != 0 {
- standardPublicPlatoonRelation.Position4 = PublicPlatoonUserRelations[0].Position4 + 1
- }
- if PublicPlatoonUserRelations[0].Position5 != 0 {
- standardPublicPlatoonRelation.Position5 = PublicPlatoonUserRelations[0].Position5 + 1
- }
- if PublicPlatoonUserRelations[0].Position6 != 0 {
- standardPublicPlatoonRelation.Position6 = PublicPlatoonUserRelations[0].Position6 + 1
- }
- if PublicPlatoonUserRelations[0].Position7 != 0 {
- standardPublicPlatoonRelation.Position7 = PublicPlatoonUserRelations[0].Position7 + 1
- }
- if PublicPlatoonUserRelations[0].Position8 != 0 {
- standardPublicPlatoonRelation.Position8 = PublicPlatoonUserRelations[0].Position8 + 1
- }
- if PublicPlatoonUserRelations[0].Position9 != 0 {
- standardPublicPlatoonRelation.Position9 = PublicPlatoonUserRelations[0].Position9 + 1
- }
-
- standardPublicPlatoonRelation.Level1 = PublicPlatoonUserRelations[0].Level1
- standardPublicPlatoonRelation.Level2 = PublicPlatoonUserRelations[0].Level2
- standardPublicPlatoonRelation.Level3 = PublicPlatoonUserRelations[0].Level3
- standardPublicPlatoonRelation.Level4 = PublicPlatoonUserRelations[0].Level4
- standardPublicPlatoonRelation.Level5 = PublicPlatoonUserRelations[0].Level5
- standardPublicPlatoonRelation.Level6 = PublicPlatoonUserRelations[0].Level6
- standardPublicPlatoonRelation.Level7 = PublicPlatoonUserRelations[0].Level7
- standardPublicPlatoonRelation.Level8 = PublicPlatoonUserRelations[0].Level8
- standardPublicPlatoonRelation.Level9 = PublicPlatoonUserRelations[0].Level9
- standardPublicPlatoonRelation.LevelTotal = PublicPlatoonUserRelations[0].LevelTotal
-
- }
-
- }
-
- //2、查找对应pid的 public_platoon_user_relation 记录
- var m1 model.PublicPlatoonUserRelation
- if has, err := engine.Where("uid=?", fatherUid).Get(&m1); err != nil || has == false {
- return PublicPlatoonUserRelation, err
- }
- var fatherUidStr string
- if m1.FatherUid == "" {
- //TODO::顶级
- fatherUidStr = egg_system_rules.Int64ToStr(m1.Id)
- } else {
- fatherUids := strings.Split(m1.FatherUid, "-")
- if len(fatherUids) >= PublicPlatoonBasicSetting.SeveralRows {
- fatherUidStr = egg_system_rules.Int64ToStr(m1.Id) + "-" + strings.Join(fatherUids[0:PublicPlatoonBasicSetting.SeveralRows:len(fatherUids)], "-")
- } else {
- fatherUidStr = egg_system_rules.Int64ToStr(m1.Id) + "-" + m1.FatherUid
- }
- }
-
- // 3、插入 public_platoon_user_relation 记录
- now := time.Now()
- PublicPlatoonUserRelation.Uid = uid
- PublicPlatoonUserRelation.FatherUid = fatherUidStr
-
- PublicPlatoonUserRelation.FatherUid1 = standardPublicPlatoonRelation.FatherUid1
- PublicPlatoonUserRelation.FatherUid2 = standardPublicPlatoonRelation.FatherUid2
- PublicPlatoonUserRelation.FatherUid3 = standardPublicPlatoonRelation.FatherUid3
- PublicPlatoonUserRelation.FatherUid4 = standardPublicPlatoonRelation.FatherUid4
- PublicPlatoonUserRelation.FatherUid5 = standardPublicPlatoonRelation.FatherUid5
- PublicPlatoonUserRelation.FatherUid6 = standardPublicPlatoonRelation.FatherUid6
- PublicPlatoonUserRelation.FatherUid7 = standardPublicPlatoonRelation.FatherUid7
- PublicPlatoonUserRelation.FatherUid8 = standardPublicPlatoonRelation.FatherUid8
- PublicPlatoonUserRelation.FatherUid9 = standardPublicPlatoonRelation.FatherUid9
- PublicPlatoonUserRelation.Pid1 = standardPublicPlatoonRelation.Pid1
- PublicPlatoonUserRelation.Pid2 = standardPublicPlatoonRelation.Pid2
- PublicPlatoonUserRelation.Pid3 = standardPublicPlatoonRelation.Pid3
- PublicPlatoonUserRelation.Pid4 = standardPublicPlatoonRelation.Pid4
- PublicPlatoonUserRelation.Pid5 = standardPublicPlatoonRelation.Pid5
- PublicPlatoonUserRelation.Pid6 = standardPublicPlatoonRelation.Pid6
- PublicPlatoonUserRelation.Pid7 = standardPublicPlatoonRelation.Pid7
- PublicPlatoonUserRelation.Pid8 = standardPublicPlatoonRelation.Pid8
- PublicPlatoonUserRelation.Pid9 = standardPublicPlatoonRelation.Pid9
- PublicPlatoonUserRelation.Position = 1
- PublicPlatoonUserRelation.Position1 = standardPublicPlatoonRelation.Position1
- PublicPlatoonUserRelation.Position2 = standardPublicPlatoonRelation.Position2
- PublicPlatoonUserRelation.Position3 = standardPublicPlatoonRelation.Position3
- PublicPlatoonUserRelation.Position4 = standardPublicPlatoonRelation.Position4
- PublicPlatoonUserRelation.Position5 = standardPublicPlatoonRelation.Position5
- PublicPlatoonUserRelation.Position6 = standardPublicPlatoonRelation.Position6
- PublicPlatoonUserRelation.Position7 = standardPublicPlatoonRelation.Position7
- PublicPlatoonUserRelation.Position8 = standardPublicPlatoonRelation.Position8
- PublicPlatoonUserRelation.Position9 = standardPublicPlatoonRelation.Position9
- PublicPlatoonUserRelation.Level = 1
- PublicPlatoonUserRelation.Level1 = standardPublicPlatoonRelation.Level1
- PublicPlatoonUserRelation.Level2 = standardPublicPlatoonRelation.Level2
- PublicPlatoonUserRelation.Level3 = standardPublicPlatoonRelation.Level3
- PublicPlatoonUserRelation.Level4 = standardPublicPlatoonRelation.Level4
- PublicPlatoonUserRelation.Level5 = standardPublicPlatoonRelation.Level5
- PublicPlatoonUserRelation.Level6 = standardPublicPlatoonRelation.Level6
- PublicPlatoonUserRelation.Level7 = standardPublicPlatoonRelation.Level7
- PublicPlatoonUserRelation.Level8 = standardPublicPlatoonRelation.Level8
- PublicPlatoonUserRelation.Level9 = standardPublicPlatoonRelation.Level9
- PublicPlatoonUserRelation.LevelTotal = standardPublicPlatoonRelation.LevelTotal
- PublicPlatoonUserRelation.RecommendUid = recommendUid
- PublicPlatoonUserRelation.JoinAt = now.Format("2006-01-02 15:04:05")
- PublicPlatoonUserRelation.CreateAt = now.Format("2006-01-02 15:04:05")
- PublicPlatoonUserRelation.UpdateAt = now.Format("2006-01-02 15:04:05")
- PublicPlatoonUserRelation.HasSonNum = 0
- _, err = publicPlatoonUserRelationDb.PublicPlatoonUserRelationInsert(&PublicPlatoonUserRelation)
- if err != nil {
- return model.PublicPlatoonUserRelation{}, err
- }
-
- // 4、改变直属父级的 has_son_num 数量
- update, err := engine.Where("uid = ?", fatherUid).SetExpr("has_son_num", "has_son_num + 1").Update(new(model.PublicPlatoonUserRelation))
- if err != nil {
- return PublicPlatoonUserRelation, err
- }
- if update == 0 {
- return PublicPlatoonUserRelation, errors.New("更新has_son_num数据出错")
- }
- return PublicPlatoonUserRelation, nil
- }
-
- // ReverseDeductionPosition 逆向推导 position
- func ReverseDeductionPosition(calcPosition, levelFirstPosition, severalTimes int) (position int) {
- remainder := (calcPosition - 1) % severalTimes
- position = levelFirstPosition + remainder
- return
- }
-
- func ReverseDeductionPositionByFather(fatherPosition, level, severalTimes, remainder int64) (position int64) {
- if fatherPosition == 0 {
- position = 0
- return
- }
- firstPosition := GetLevelForFirstPosition(level, severalTimes)
- endPosition := firstPosition + int64(math.Pow(float64(severalTimes), float64(level-1))) - 1
- position = fatherPosition + (fatherPosition-firstPosition)*severalTimes + (endPosition - fatherPosition) + 1
- return
- }
-
- func FindSuitablePosition(engine *xorm.Engine, position *int64, severalTimes float64, fuid, fatherLevel int64, times *int) (standardPublicPlatoonRelation md.StandardPublicPlatoonRelation, fatherUid int64, err error) {
- *times++
- positionName := "position" + egg_system_rules.IntToStr(*times)
- fatherName := "father_uid" + egg_system_rules.IntToStr(*times)
- firstSonPosition := SearchPositionFirstSonPosition(*position, int64(severalTimes))
- var tempLevel float64
- makeSearchLevel(&firstSonPosition, severalTimes, &tempLevel)
- firstSonLevel := int64(tempLevel)
- endSonPosition := firstSonPosition + int64(math.Pow(severalTimes, float64(firstSonLevel-fatherLevel))) - 1
- publicPlatoonUserRelationDb := implement.NewPublicPlatoonUserRelationDb(engine)
- list, _, err := publicPlatoonUserRelationDb.PublicPlatoonUserRelationFindCountByPosition(fuid, fatherName, positionName, int64(firstSonPosition), int64(endSonPosition))
- if err != nil {
- return standardPublicPlatoonRelation, fatherUid, err
- }
-
- if list[0].HasSonNum == int(severalTimes) {
- //TODO::则证明该层直属下级无空位
- position = &firstSonPosition
- return FindSuitablePosition(engine, position, severalTimes, fuid, fatherLevel, times)
- } else if list[0].HasSonNum == 0 {
- fatherUid = list[0].Uid
-
- //TODO::新的一层
- standardPublicPlatoonRelation.FatherUid1 = list[0].Uid
- standardPublicPlatoonRelation.FatherUid2 = list[0].FatherUid1
- standardPublicPlatoonRelation.FatherUid3 = list[0].FatherUid2
- standardPublicPlatoonRelation.FatherUid4 = list[0].FatherUid3
- standardPublicPlatoonRelation.FatherUid5 = list[0].FatherUid4
- standardPublicPlatoonRelation.FatherUid6 = list[0].FatherUid5
- standardPublicPlatoonRelation.FatherUid7 = list[0].FatherUid6
- standardPublicPlatoonRelation.FatherUid8 = list[0].FatherUid7
- standardPublicPlatoonRelation.FatherUid9 = list[0].FatherUid8
-
- standardPublicPlatoonRelation.Pid1 = list[0].Position
- standardPublicPlatoonRelation.Pid2 = list[0].Position1
- standardPublicPlatoonRelation.Pid3 = list[0].Position2
- standardPublicPlatoonRelation.Pid4 = list[0].Position3
- standardPublicPlatoonRelation.Pid5 = list[0].Position4
- standardPublicPlatoonRelation.Pid6 = list[0].Position5
- standardPublicPlatoonRelation.Pid7 = list[0].Position6
- standardPublicPlatoonRelation.Pid8 = list[0].Position7
- standardPublicPlatoonRelation.Pid9 = list[0].Position8
-
- standardPublicPlatoonRelation.Position1 = SearchPositionFirstSonPosition(list[0].Position, int64(severalTimes))
- standardPublicPlatoonRelation.Position2 = ReverseDeductionPositionByFather(list[0].Position1, int64(list[0].Level1), int64(severalTimes), 0)
- standardPublicPlatoonRelation.Position3 = ReverseDeductionPositionByFather(list[0].Position2, int64(list[0].Level2), int64(severalTimes), 0)
- standardPublicPlatoonRelation.Position4 = ReverseDeductionPositionByFather(list[0].Position3, int64(list[0].Level3), int64(severalTimes), 0)
- standardPublicPlatoonRelation.Position5 = ReverseDeductionPositionByFather(list[0].Position4, int64(list[0].Level4), int64(severalTimes), 0)
- standardPublicPlatoonRelation.Position6 = ReverseDeductionPositionByFather(list[0].Position5, int64(list[0].Level5), int64(severalTimes), 0)
- standardPublicPlatoonRelation.Position7 = ReverseDeductionPositionByFather(list[0].Position6, int64(list[0].Level6), int64(severalTimes), 0)
- standardPublicPlatoonRelation.Position8 = ReverseDeductionPositionByFather(list[0].Position7, int64(list[0].Level7), int64(severalTimes), 0)
- standardPublicPlatoonRelation.Position9 = ReverseDeductionPositionByFather(list[0].Position8, int64(list[0].Level8), int64(severalTimes), 0)
-
- if list[0].Level1 == 0 {
- list[0].Level1--
- }
- if list[0].Level2 == 0 {
- list[0].Level2--
- }
- if list[0].Level3 == 0 {
- list[0].Level3--
- }
- if list[0].Level4 == 0 {
- list[0].Level4--
- }
- if list[0].Level5 == 0 {
- list[0].Level5--
- }
- if list[0].Level6 == 0 {
- list[0].Level6--
- }
- if list[0].Level7 == 0 {
- list[0].Level7--
- }
- if list[0].Level8 == 0 {
- list[0].Level8--
- }
- if list[0].Level9 == 0 {
- list[0].Level9--
- }
- standardPublicPlatoonRelation.Level1 = list[0].Level + 1
- standardPublicPlatoonRelation.Level2 = list[0].Level1 + 1
- standardPublicPlatoonRelation.Level3 = list[0].Level2 + 1
- standardPublicPlatoonRelation.Level4 = list[0].Level3 + 1
- standardPublicPlatoonRelation.Level5 = list[0].Level4 + 1
- standardPublicPlatoonRelation.Level6 = list[0].Level5 + 1
- standardPublicPlatoonRelation.Level7 = list[0].Level6 + 1
- standardPublicPlatoonRelation.Level8 = list[0].Level7 + 1
- standardPublicPlatoonRelation.Level9 = list[0].Level8 + 1
- standardPublicPlatoonRelation.LevelTotal = list[0].LevelTotal + 1
- } else {
- fatherUid = list[0].Uid
-
- _, _, _, _, getPositionName, _, _, _ := GetPosition(list[0], *times+1)
- //PublicPlatoonUserRelations, err := db.PublicPlatoonUserRelationFindByPid(engine, m.Position, m.Uid, "pid1", "father_uid1", "position1")
-
- PublicPlatoonUserRelations, err := publicPlatoonUserRelationDb.PublicPlatoonUserRelationFindByPid(list[0].Uid, "father_uid1", getPositionName)
-
- if err != nil {
- return standardPublicPlatoonRelation, fatherUid, err
- }
-
- if PublicPlatoonUserRelations[0].Position2 == 0 {
- PublicPlatoonUserRelations[0].Position2 = -1
- }
- if PublicPlatoonUserRelations[0].Position3 == 0 {
- standardPublicPlatoonRelation.Position3 = -1
- }
- if PublicPlatoonUserRelations[0].Position4 == 0 {
- standardPublicPlatoonRelation.Position4 = -1
- }
- if PublicPlatoonUserRelations[0].Position5 == 0 {
- standardPublicPlatoonRelation.Position5 = -1
- }
- if PublicPlatoonUserRelations[0].Position6 == 0 {
- standardPublicPlatoonRelation.Position6 = -1
- }
- if PublicPlatoonUserRelations[0].Position7 == 0 {
- standardPublicPlatoonRelation.Position7 = -1
- }
- if PublicPlatoonUserRelations[0].Position8 == 0 {
- standardPublicPlatoonRelation.Position8 = -1
- }
- if PublicPlatoonUserRelations[0].Position9 == 0 {
- standardPublicPlatoonRelation.Position9 = -1
- }
-
- standardPublicPlatoonRelation = md.StandardPublicPlatoonRelation{
- FatherUid1: PublicPlatoonUserRelations[0].FatherUid1,
- FatherUid2: PublicPlatoonUserRelations[0].FatherUid2,
- FatherUid3: PublicPlatoonUserRelations[0].FatherUid3,
- FatherUid4: PublicPlatoonUserRelations[0].FatherUid4,
- FatherUid5: PublicPlatoonUserRelations[0].FatherUid5,
- FatherUid6: PublicPlatoonUserRelations[0].FatherUid6,
- FatherUid7: PublicPlatoonUserRelations[0].FatherUid7,
- FatherUid8: PublicPlatoonUserRelations[0].FatherUid8,
- FatherUid9: PublicPlatoonUserRelations[0].FatherUid9,
- Pid1: PublicPlatoonUserRelations[0].Pid1,
- Pid2: PublicPlatoonUserRelations[0].Pid2,
- Pid3: PublicPlatoonUserRelations[0].Pid3,
- Pid4: PublicPlatoonUserRelations[0].Pid4,
- Pid5: PublicPlatoonUserRelations[0].Pid5,
- Pid6: PublicPlatoonUserRelations[0].Pid6,
- Pid7: PublicPlatoonUserRelations[0].Pid7,
- Pid8: PublicPlatoonUserRelations[0].Pid8,
- Pid9: PublicPlatoonUserRelations[0].Pid9,
- Position1: PublicPlatoonUserRelations[0].Position1 + 1,
- Position2: PublicPlatoonUserRelations[0].Position2 + 1,
- Position3: PublicPlatoonUserRelations[0].Position3 + 1,
- Position4: PublicPlatoonUserRelations[0].Position4 + 1,
- Position5: PublicPlatoonUserRelations[0].Position5 + 1,
- Position6: PublicPlatoonUserRelations[0].Position6 + 1,
- Position7: PublicPlatoonUserRelations[0].Position7 + 1,
- Position8: PublicPlatoonUserRelations[0].Position8 + 1,
- Position9: PublicPlatoonUserRelations[0].Position9 + 1,
- Level1: PublicPlatoonUserRelations[0].Level1,
- Level2: PublicPlatoonUserRelations[0].Level2,
- Level3: PublicPlatoonUserRelations[0].Level3,
- Level4: PublicPlatoonUserRelations[0].Level4,
- Level5: PublicPlatoonUserRelations[0].Level5,
- Level6: PublicPlatoonUserRelations[0].Level6,
- Level7: PublicPlatoonUserRelations[0].Level7,
- Level8: PublicPlatoonUserRelations[0].Level8,
- Level9: PublicPlatoonUserRelations[0].Level9,
- LevelTotal: PublicPlatoonUserRelations[0].LevelTotal,
- }
- return standardPublicPlatoonRelation, fatherUid, nil
- }
- return
- }
-
- func GetPosition(m model.PublicPlatoonUserRelation, times int) (position, pid int64, level int, fatherUid int64, positionName, pidName, levelName, fatherUidName string) {
- if times == 1 {
- position = m.Position1
- positionName = "position1"
- pid = m.Pid1
- pidName = "pid1"
- level = m.Level1
- levelName = "level1"
- fatherUid = m.FatherUid1
- fatherUidName = "father_uid1"
- }
- if times == 2 {
- position = m.Position2
- positionName = "position2"
- pid = m.Pid2
- pidName = "pid2"
- level = m.Level2
- levelName = "level2"
- fatherUid = m.FatherUid2
- fatherUidName = "father_uid2"
- }
- if times == 3 {
- position = m.Position3
- positionName = "position3"
- pid = m.Pid3
- pidName = "pid3"
- level = m.Level3
- levelName = "level3"
- fatherUid = m.FatherUid3
- fatherUidName = "father_uid3"
- }
- if times == 4 {
- position = m.Position4
- positionName = "position4"
- pid = m.Pid4
- pidName = "pid4"
- level = m.Level4
- levelName = "level4"
- fatherUid = m.FatherUid4
- fatherUidName = "father_uid4"
- }
- if times == 5 {
- position = m.Position5
- positionName = "position5"
- pid = m.Pid5
- pidName = "pid5"
- level = m.Level5
- levelName = "level5"
- fatherUid = m.FatherUid5
- fatherUidName = "father_uid5"
- }
- if times == 6 {
- position = m.Position6
- positionName = "position6"
- pid = m.Pid6
- pidName = "pid6"
- level = m.Level6
- levelName = "level6"
- fatherUid = m.FatherUid6
- fatherUidName = "father_uid6"
- }
- if times == 7 {
- position = m.Position7
- positionName = "position7"
- pid = m.Pid7
- pidName = "pid7"
- level = m.Level7
- levelName = "level7"
- fatherUid = m.FatherUid7
- fatherUidName = "father_uid7"
- }
- if times == 8 {
- position = m.Position8
- positionName = "position8"
- pid = m.Pid8
- pidName = "pid8"
- level = m.Level8
- levelName = "level8"
- fatherUid = m.FatherUid8
- fatherUidName = "father_uid8"
- }
- if times == 9 {
- position = m.Position9
- positionName = "position9"
- pid = m.Pid9
- pidName = "pid9"
- level = m.Level9
- levelName = "level9"
- fatherUid = m.FatherUid9
- fatherUidName = "father_uid9"
- }
-
- return
- }
-
- // GetLevelForFirstPosition 返回当前等级的起始值
- func GetLevelForFirstPosition(level, severalTimes int64) (position int64) {
- position = position + 1
- for n := int64(0); n <= (level - 2); n++ {
- position += int64(math.Pow(float64(severalTimes), float64(n)))
- }
- return
- }
-
- // 查找归属父级id
- func MakeSearchPid(position int, row int) (pid int) {
- divisionValue := (position - 1) / row
- if divisionValue == 0 {
- pid = 1
- return
- } else {
- if (divisionValue*row + 1) == position {
- pid = divisionValue
- return
- } else {
- pid = divisionValue + 1
- return
- }
- }
- }
-
- // 查找当前位置下的第一个子集位置
- func SearchPositionFirstSonPosition(position int64, times int64) (sonPosition int64) {
- sonPosition = (position * times) - (times - 2)
- return
- }
-
- // 查找当前位位置相当于父级的等级
- func SearchPositionSonForLevel(fatherUid string, fatherPosition int64) (level int) {
- fatherUids := strings.Split(fatherUid, "-")
- for key, item := range fatherUids {
- if item == egg_system_rules.Int64ToStr(fatherPosition) {
- level = key + 1
- break
- }
- }
- return
- }
-
- // FindWaitForDealUsers 查询待处理的用户
- func FindWaitForDealUsers(engine *xorm.Engine, page, pageSize int) (err error, resp []int64) {
- now := time.Now().Format("2006-01-02")
- publicPlatoonUserRelationDb := implement.NewPublicPlatoonUserRelationDb(engine)
- lists, err := publicPlatoonUserRelationDb.PublicPlatoonUserRelationFindByParamsByPage(map[string]interface{}{
- "key": "wait_for_settlement_date",
- "value": now,
- }, page, pageSize)
- if err != nil {
- return
- }
- for _, list := range *lists {
- resp = append(resp, list.Uid)
- }
- return
- }
-
- // 递归查找等级
- func makeSearchLevel(position *int64, rows float64, times *float64) (level int) {
- difference := *position - int64(math.Pow(rows, *times))
- *times++
- if difference <= 0 {
- return int(*times)
- } else {
- position = &difference
- return makeSearchLevel(position, rows, times)
- }
- }
-
- // DealCommonWealthPunish 处理公排处罚
- func DealCommonWealthPunish(engine *xorm.Engine, uid int64, reason string) (err error, resp []map[string]string) {
- session := engine.NewSession()
- defer func() {
- session.Close()
- if err := recover(); err != nil {
- _ = zhios_order_relate_logx.Error(err)
- }
- }()
- session.Begin()
-
- //1、查找 `user_public_platoon_setting` 基础设置
- publicPlatoonBasicSettingDb := implement.NewPublicPlatoonBasicSettingDb(engine)
- PublicPlatoonBasicSetting, err := publicPlatoonBasicSettingDb.PublicPlatoonBasicSettingGetOneByParams(map[string]interface{}{
- "key": "is_open",
- "value": 1,
- })
- if err != nil {
- _ = session.Rollback()
- return
- }
-
- //2、查询出 `public_platoon_user_relation` 中相关记录 && 将该记录的uid置为 -1
- publicPlatoonUserRelationDb := implement.NewPublicPlatoonUserRelationDb(engine)
- params, err := publicPlatoonUserRelationDb.PublicPlatoonUserRelationGetOneByParams(map[string]interface{}{
- "key": "uid",
- "value": uid,
- })
- if err != nil {
- _ = session.Rollback()
- return
- }
- if params == nil {
- fmt.Println("未查询到公排关系记录!!!!!!!", uid)
- return
- }
-
- //TODO::判断是否为推荐用户
- if params.RecommendUid == 0 {
- params.Uid = params.Uid - int64(time.Now().Unix())
- } else {
- params.Uid = params.Uid - int64(time.Now().Unix())
- }
-
- for n := 1; n <= 9; n++ {
- str := "father_uid" + strconv.Itoa(n)
- sql := fmt.Sprintf("UPDATE `public_platoon_user_relation` SET %s=%s where %s=%s", str, egg_system_rules.Int64ToStr(params.Uid), str, egg_system_rules.Int64ToStr(uid))
- fmt.Println(">>>>>>>sql>>>>>>>", sql)
- _, err = session.Exec(sql)
- if err != nil {
- _ = session.Rollback()
- return
- }
- }
-
- updateAffected, err := publicPlatoonUserRelationDb.PublicPlatoonUserRelationUpdate(session, params.Id, params)
- if err != nil {
- _ = session.Rollback()
- return
- }
- if updateAffected == 0 {
- err = errors.New("更新 public_platoon_user_relation 记录失败")
- _ = session.Rollback()
- return
- }
-
- //3、新增一条 `public_platoon_user_relation` 记录
- params.Uid = uid
- res, err := PublicPlatoon(engine, params.Uid, params.RecommendUid, *PublicPlatoonBasicSetting)
- if err != nil {
- _ = session.Rollback()
- return
- }
-
- //4、新增一条 `public_platoon_user_system_punish_records` 记录
- now := time.Now()
- publicPlatoonUserSystemPunishRecordsDb := implement.NewPublicPlatoonUserSystemPunishRecordsDb(engine)
- insertAffected, err := publicPlatoonUserSystemPunishRecordsDb.PublicPlatoonUserSystemPunishRecordsInsert(session, &model.PublicPlatoonUserSystemPunishRecords{
- Uid: params.Uid,
- OldPosition: params.Position,
- NewPosition: res.Position,
- Date: now.AddDate(0, 0, 30).Format("2006-01-02"),
- Title: "系统处罚记录",
- Reason: reason,
- Type: 2,
- CreateAt: now.Format("2006-01-02 15:04:05"),
- UpdateAt: now.Format("2006-01-02 15:04:05"),
- })
- if err != nil {
- _ = session.Rollback()
- return
- }
- if insertAffected == 0 {
- err = errors.New("新增 public_platoon_user_system_punish_records 记录失败")
- _ = session.Rollback()
- return
- }
-
- err = session.Commit()
- return
- }
-
- const DealUserPublicPlatoonPunishLockKey = "deal_user_public_platoon_lock_key"
-
- // DealUserPublicPlatoonPunish 处理公排处罚
- func DealUserPublicPlatoonPunish(engine *xorm.Engine) (err error) {
- now := time.Now()
- fmt.Println(now.Hour())
- if !(now.Hour() > 22 && now.Hour() < 24) {
- //TODO::只在凌晨一点 ~ 凌晨 8 点运行
- return errors.New("非运行时间")
- }
- //TODO::增加“悲观锁”防止串行
- getString, _ := cache.GetString(DealUserPublicPlatoonPunishLockKey)
- if getString != "" {
- fmt.Println("!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!", "上一次结算未执行完")
- return errors.New("上一次结算未执行完")
- }
- cache.SetEx(DealUserPublicPlatoonPunishLockKey, "running", 3600*8) //8小时
-
- //查找 `public_platoon_basic_setting` 基础设置
- publicPlatoonBasicSettingDb := implement.NewPublicPlatoonBasicSettingDb(engine)
- PublicPlatoonBasicSetting, err := publicPlatoonBasicSettingDb.PublicPlatoonBasicSettingGetOneByParams(map[string]interface{}{
- "key": "is_open",
- "value": 1,
- })
- if err != nil {
- return err
- }
- if PublicPlatoonBasicSetting == nil {
- return errors.New("公排制度未开启")
- }
- publicPlatoonFreePunishWithUserDb := implement.NewPublicPlatoonFreePunishWithUserDb(engine)
- mapPublicPlatoonFreePunishWithUser, err := publicPlatoonFreePunishWithUserDb.FindAllPublicPlatoonFreePunishWithUser()
- if err != nil {
- return err
- }
- if PublicPlatoonBasicSetting.SystemPunishReplace == 1 {
- systemPunishReplaceValue := PublicPlatoonBasicSetting.SystemPunishReplaceValue
- startDate := now.AddDate(0, 0, -systemPunishReplaceValue).Format("2006-01-02") + " 00:00:00"
- var page = 1
- var pageSize = 100
- for {
- var users []model.User
- err = engine.Limit(pageSize, (page-1)*pageSize).Desc("uid").Find(&users)
- if err != nil {
- return err
- }
- if len(users) <= 0 {
- break
- }
- for _, user := range users {
- var list []model.EggSignIn
- err = engine.Where("start_time >= ?", startDate).And("uid = ?", user.Id).Find(&list)
- if len(list) <= 0 && mapPublicPlatoonFreePunishWithUser[user.Id] == nil {
- var PublicPlatoonRecordsPunishWithUser model.PublicPlatoonRecordsPunishWithUser
- has, err1 := engine.Where("uid = ?", user.Id).Get(&PublicPlatoonRecordsPunishWithUser)
- if err1 != nil {
- return err1
- }
- PublicPlatoonRecordsPunishWithUserDate, _ := time.ParseInLocation("2006-01-02", PublicPlatoonRecordsPunishWithUser.Date, time.Local)
- if has && now.AddDate(0, 0, -7).After(PublicPlatoonRecordsPunishWithUserDate) {
- //TODO::不进行重复处罚
- continue
- }
- //进行处罚
- err, _ = DealCommonWealthPunish(engine, user.Id, "公排处罚")
- if err != nil {
- return err
- }
-
- //添加 public_platoon_records_punish_with_user 记录
- publicPlatoonRecordsPunishWithUserDb := implement.NewPublicPlatoonRecordsPunishWithUserDb(engine)
- if has {
- PublicPlatoonRecordsPunishWithUser.Date = now.Format("2006-01-02")
- _, err2 := publicPlatoonRecordsPunishWithUserDb.PublicPlatoonRecordsPunishWithUserUpdate(PublicPlatoonRecordsPunishWithUser.Id, &PublicPlatoonRecordsPunishWithUser, "date")
- if err2 != nil {
- return err2
- }
- } else {
- _, err2 := publicPlatoonRecordsPunishWithUserDb.PublicPlatoonRecordsPunishWithUserInsert(&model.PublicPlatoonRecordsPunishWithUser{
- Uid: user.Id,
- Date: now.Format("2006-01-02"),
- })
- if err2 != nil {
- return err2
- }
- }
- }
- }
- page++
- }
- }
- return
- }
|