player.go 18 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616
  1. package models
  2. import (
  3. "context"
  4. "fmt"
  5. jsoniter "github.com/json-iterator/go"
  6. mhayaString "github.com/mhaya/extend/string"
  7. mhayaTime "github.com/mhaya/extend/time"
  8. "github.com/mhaya/game/game_cluster/internal/constant"
  9. "github.com/mhaya/game/game_cluster/internal/data"
  10. "github.com/mhaya/game/game_cluster/internal/mdb"
  11. "github.com/mhaya/game/game_cluster/internal/param"
  12. "github.com/mhaya/game/game_cluster/internal/third"
  13. "github.com/mhaya/game/game_cluster/internal/token"
  14. "time"
  15. mhayaLogger "github.com/mhaya/logger"
  16. "math/rand"
  17. "strings"
  18. )
  19. type Player struct {
  20. UserName string `json:"userName" bson:"userName"`
  21. OpenId string `json:"openId" bson:"openId"`
  22. UserNameMaybe string `json:"userNameMaybe" bson:"userNameMaybe"`
  23. NickName string `json:"nickName" bson:"nickName"`
  24. Pid string `json:"pid" bson:"pid"`
  25. XID string `json:"xID" bson:"xID"`
  26. TonWall string `json:"tonWall" bson:"tonWall"`
  27. Email string `json:"email" bson:"email"`
  28. Mobile string `json:"mobile" bson:"mobile"`
  29. Avatar string `json:"avatar" bson:"avatar"`
  30. IsRobot int `json:"isRobot" bson:"isRobot"`
  31. IsLeader int `json:"isLeader" bson:"isLeader"`
  32. IsVip int `json:"isVip" bson:"isVip"`
  33. IsFirstBindingXID int `json:"isFirstBindingXID" bson:"isFirstBindingXID"`
  34. Level int `json:"level" bson:"level"`
  35. Exp int `json:"exp" bson:"exp"`
  36. Gender int `json:"gender" bson:"gender"`
  37. Birthday string `json:"birthday" bson:"birthday"`
  38. Successions int `json:"successions" bson:"successions"`
  39. MaxSuccessions int `json:"maxSuccessions" bson:"maxSuccessions"`
  40. PrevTime int64 `json:"prevTime" bson:"prevTime"`
  41. LoginTime int64 `json:"loginTime" bson:"loginTime"`
  42. LoginIP string `json:"loginIP" bson:"loginIP"`
  43. LoginFailure int `json:"loginFailure" bson:"loginFailure"`
  44. JoinIP string `json:"joinIP" bson:"joinIP"`
  45. Rank int `json:"rank" bson:"rank"`
  46. JoinTime int64 `json:"joinTime" bson:"joinTime"`
  47. CreateTime int64 `json:"createTime" bson:"createTime"`
  48. UpdateTime int64 `json:"updateTime" bson:"updateTime"`
  49. DailyRefreshTime int64 `json:"dailyRefreshTime" bson:"dailyRefreshTime"`
  50. WeeklyRefreshTime int64 `json:"weeklyRefreshTime" bson:"weeklyRefreshTime"`
  51. RollDay int64 `json:"rollDay" bson:"rollDay"`
  52. Status int `json:"status" bson:"status"`
  53. Roll *Roll `json:"roll" bson:"roll"`
  54. Guide int `json:"guide" bson:"guide"`
  55. Share *Share `json:"share" bson:"share"`
  56. FirstReward int `json:"firstReward" bson:"firstReward"`
  57. Index int `json:"-" bson:"index"`
  58. Item ItemBasePack `json:"item" bson:"item"`
  59. SinIn SignIn `json:"sinIn" bson:"sinIn"`
  60. DailyTask DailyTask `json:"dailyTask" bson:"dailyTask"`
  61. AchieveTask AchieveTask `json:"achieveTask" bson:"achieveTask"`
  62. RankReward RankReward `json:"rankReward" bson:"rankReward"`
  63. Invite Invite `json:"invite" bson:"invite"`
  64. InviteReward *InviteReward `json:"inviteReward" bson:"inviteReward"`
  65. PlayerReward map[int]*PlayerRewardBase `json:"-" bson:"-"`
  66. }
  67. func NewPlayer(token *token.Token) *Player {
  68. return &Player{
  69. UserName: token.PlayerID,
  70. OpenId: token.OpenID,
  71. NickName: token.Nickname,
  72. IsRobot: 0,
  73. Successions: 1,
  74. MaxSuccessions: 1,
  75. LoginTime: mhayaTime.Now().Unix(),
  76. CreateTime: mhayaTime.Now().Unix(),
  77. UpdateTime: mhayaTime.Now().Unix(),
  78. RollDay: mhayaTime.Now().Unix(),
  79. Exp: 0,
  80. Status: 1,
  81. Rank: 0,
  82. Item: NewItem(),
  83. Guide: 1,
  84. Level: 1,
  85. Index: 1,
  86. FirstReward: 1,
  87. Roll: &Roll{RollNums: 0, RollDays: 0},
  88. DailyTask: NewDailyTask(),
  89. Share: &Share{IsDrawShare: 0},
  90. SinIn: NewSignIn(),
  91. AchieveTask: NewAchieveTask(),
  92. RankReward: NewRankReward(),
  93. Invite: NewInvite(),
  94. }
  95. }
  96. func NewBotPlayer(token *token.Token) *Player {
  97. return &Player{
  98. UserName: token.PlayerID,
  99. OpenId: token.OpenID,
  100. NickName: token.Nickname,
  101. IsRobot: 1,
  102. Successions: 1,
  103. MaxSuccessions: 1,
  104. LoginTime: mhayaTime.Now().Unix(),
  105. CreateTime: mhayaTime.Now().Unix(),
  106. UpdateTime: mhayaTime.Now().Unix(),
  107. RollDay: mhayaTime.Now().Unix(),
  108. Exp: 0,
  109. Status: 1,
  110. Rank: 0,
  111. Item: NewItem(),
  112. Guide: 1,
  113. Level: 1,
  114. Index: 1,
  115. FirstReward: 1,
  116. Roll: &Roll{RollNums: 0, RollDays: 0},
  117. DailyTask: NewDailyTask(),
  118. Share: &Share{IsDrawShare: 0},
  119. SinIn: NewSignIn(),
  120. AchieveTask: NewAchieveTask(),
  121. RankReward: NewRankReward(),
  122. Invite: NewInvite(),
  123. }
  124. }
  125. func (p *Player) Init() {
  126. if p.InviteReward == nil {
  127. p.InviteReward = &InviteReward{IsClaim: 0}
  128. }
  129. }
  130. // InitDaily 每日刷新
  131. func (p *Player) InitDaily() {
  132. if !mhayaTime.CreateFromTimestamp(p.DailyRefreshTime).IsToday() {
  133. //刷新每日任务
  134. p.DailyTask.RefreshDailyTask()
  135. //刷新抽奖每日分享
  136. p.Share.RefreshShare()
  137. //刷新是否有排行榜奖励
  138. p.RankReward.InitRank(p.UserName, 1)
  139. //更新时间到当天
  140. p.DailyRefreshTime = mhayaTime.Now().Unix()
  141. }
  142. }
  143. func (p *Player) InitWeekly() {
  144. if !mhayaTime.CreateFromTimestamp(p.WeeklyRefreshTime).IsThisWeek() {
  145. p.SinIn.RefreshSignIn()
  146. //刷新是否有排行榜奖励
  147. p.RankReward.InitRank(p.UserName, 2)
  148. p.WeeklyRefreshTime = mhayaTime.Now().Unix()
  149. }
  150. }
  151. func (p *Player) InitAchieveTask() {
  152. p.AchieveTask.RefreshAchieveTask()
  153. }
  154. func (p *Player) LevelUp() bool {
  155. ret, ok := data.LevelConfig.Get(p.Level + 1)
  156. if !ok {
  157. return false
  158. }
  159. curLevel := p.Level
  160. if p.Item[ItemRoll].CurNum >= ret.DiceNum && p.Item[ItemInvite].CurNum >= ret.Invite {
  161. p.Level += 1
  162. if p.Level > curLevel {
  163. p.Item[ItemRoll].AddMaxNum(ret.DiceLimit)
  164. p.Item[ItemRoll].CurNum = 0
  165. p.Item[ItemInvite].CurNum = 0
  166. //升级邀请奖励
  167. p.SetPassiveInviteReward()
  168. p.SetPlayerRewardLog(SourceLevelUp, p.Level, []data.ItemReward{{ItemID: ItemAllRoll, ItemBaseType: 2, Amount: ret.DiceLimit}}, []data.ItemReward{}, p.Level)
  169. return true
  170. }
  171. }
  172. return false
  173. }
  174. func (p *Player) RefreshRoll() bool {
  175. ret, ok := data.DiscreteRuleConfig.Get(RefreshRoll)
  176. if !ok {
  177. return false
  178. }
  179. arr := strings.Split(ret.Condition, ",")
  180. if len(arr) != 2 {
  181. return false
  182. }
  183. h, _ := mhayaString.ToInt64(arr[0])
  184. m, _ := mhayaString.ToInt64(arr[1])
  185. s := mhayaTime.CreateFromTimestamp(p.RollDay).DiffInSeconds(mhayaTime.Now())
  186. if s >= h {
  187. p.RollDay = mhayaTime.Now().Unix()
  188. num := p.Item[ItemRoll].AddAll()
  189. if num > 0 {
  190. p.SetPlayerRewardLog(SourceRefreshRoll, 1, []data.ItemReward{{ItemID: ItemRoll, ItemBaseType: 1, Amount: num}}, []data.ItemReward{}, 0)
  191. }
  192. mhayaLogger.Infof("RefreshRoll by condition 1 rollDay:%d,num:%v", s, num)
  193. return true
  194. } else if s >= m {
  195. p.RollDay = mhayaTime.Now().Unix()
  196. num := p.Item[ItemRoll].AddSame()
  197. if num > 0 {
  198. p.SetPlayerRewardLog(SourceRefreshRoll, 2, []data.ItemReward{{ItemID: ItemRoll, ItemBaseType: 1, Amount: num}}, []data.ItemReward{}, 0)
  199. }
  200. mhayaLogger.Infof("RefreshRoll by condition 2 rollDay:%d,num:%v", s, num)
  201. return true
  202. }
  203. return false
  204. }
  205. func (p *Player) CheckTaskByType1(userID string, id int) bool {
  206. switch id {
  207. case 1: //关注关注推特
  208. return true
  209. case 2: //加入TG频道
  210. if third.GetIfChannel(userID) == 1 {
  211. return true
  212. } else {
  213. return false
  214. }
  215. case 3: //关注Yutube账号
  216. return true
  217. case 4: //加入DC频道
  218. return true
  219. }
  220. return false
  221. }
  222. func (p *Player) FinishAchieveTaskByType1(id int) bool {
  223. if a, ok := p.AchieveTask[id]; ok {
  224. if a.Num == 0 && a.Validate == 1 { //待验证
  225. if p.CheckTaskByType1(p.UserName, id) {
  226. p.AchieveTask[id].Validate = 2 //验证成功
  227. p.AchieveTask[id].Num = 1
  228. p.AchieveTask[id].Status = 1
  229. } else {
  230. mhayaLogger.Error("FinishAchieveTaskByType1 验证失败 username :%v, id : v%", p.UserName, id)
  231. return false
  232. }
  233. } else {
  234. if a.Validate == 0 {
  235. p.AchieveTask[id].Validate = 1
  236. }
  237. }
  238. return true
  239. }
  240. return false
  241. }
  242. // FinishAchieveTaskByType2 完成成就类型2
  243. func (p *Player) FinishAchieveTaskByType2() bool {
  244. var cur int
  245. for _, v := range p.AchieveTask {
  246. if v.Type == 2 {
  247. ret, ok := data.AchieveTaskConfig.Get(v.ID)
  248. if !ok {
  249. continue
  250. }
  251. if v.Num < ret.Condition && v.Num >= cur {
  252. p.AchieveTask[v.ID].Num += 1
  253. p.AchieveTask[v.ID].Status = 1
  254. p.AchieveTask[v.ID].RewardNum += 1
  255. return true
  256. }
  257. cur = ret.Condition
  258. }
  259. }
  260. return false
  261. }
  262. func (p *Player) FinishDailyTask(tp, num int) bool {
  263. var f bool
  264. for i := 0; i < num; i++ {
  265. for _, v := range p.DailyTask {
  266. if v.Type == tp && v.Status == 0 {
  267. p.DailyTask[v.ID].Num += 1
  268. if p.DailyTask[v.ID].Num == v.Condition {
  269. p.DailyTask[v.ID].Status = 1
  270. f = true
  271. continue
  272. }
  273. }
  274. }
  275. }
  276. return f
  277. }
  278. func (p *Player) RollOnce(num int) *param.RollOnce {
  279. var roll param.RollOnce
  280. levelConfig, ok := data.LevelConfig.Get(p.Level)
  281. if !ok {
  282. return nil
  283. }
  284. if levelConfig == nil {
  285. return nil
  286. }
  287. r1 := rand.Int31n(5) + 1
  288. r2 := rand.Int31n(5) + 1
  289. r3 := rand.Int31n(5) + 1
  290. //rule 4
  291. if (r1 == 1 && r2 == 1 && r3 == 1) || (r1 == 6 && r2 == 6 && r3 == 6) {
  292. roll.Score = levelConfig.Ratio * levelConfig.Rule4
  293. roll.RollType = constant.RollSpecialTripleBet
  294. } else if r1 == r2 && r2 == r3 {
  295. roll.Score = levelConfig.Ratio * levelConfig.Rule3
  296. roll.RollType = constant.RollTripleBet
  297. } else if r1 == r2 || r1 == r3 || r2 == r3 {
  298. roll.Score = levelConfig.Ratio * levelConfig.Rule2
  299. roll.RollType = constant.RollDoubleBet
  300. } else {
  301. roll.Score = levelConfig.Ratio * levelConfig.Rule1
  302. if r1+r2+r3 > 9 {
  303. roll.RollType = constant.RollBigBet
  304. } else {
  305. roll.RollType = constant.RollSmallBet
  306. }
  307. }
  308. roll.Rand = make(map[int]int32)
  309. roll.Rand[1] = r1
  310. roll.Rand[2] = r2
  311. roll.Rand[3] = r3
  312. p.Item[ItemScore].Add(roll.Score * num)
  313. p.Item[ItemRoll].sub(1 * num)
  314. p.SetPlayerRewardLog(SourceRoll, 0, []data.ItemReward{{ItemID: ItemScore, ItemBaseType: 1, Amount: roll.Score * num}}, []data.ItemReward{{ItemID: ItemRoll, ItemBaseType: 1, Amount: 1 * num}}, roll)
  315. p.Roll.RollNums += 1 * num
  316. roll.IsDailyTask = p.FinishDailyTask(1, num)
  317. roll.IsLevelUp = p.LevelUp()
  318. return &roll
  319. }
  320. // Draw 最多尝试重复执行50次
  321. func (p *Player) Draw() *param.Draw {
  322. for i := 0; i < 50; i++ {
  323. draw, ok := p.DrawOnce()
  324. if ok {
  325. return draw
  326. }
  327. }
  328. return nil
  329. }
  330. func (p *Player) DrawOnce() (*param.Draw, bool) {
  331. var ret map[int]*data.DrawConfigRow
  332. var draw param.Draw
  333. if p.Item[ItemDrawsNumber].Num <= 10 {
  334. ret = data.DrawConfig.GetByType(1)
  335. } else {
  336. ret = data.DrawConfig.GetByType(2)
  337. }
  338. weight := int(rand.Int31n(9999))
  339. var cur int
  340. var curID int
  341. var drawId int
  342. for _, v := range ret {
  343. if v.Weight == 0 {
  344. continue
  345. }
  346. if weight >= cur && weight < cur+v.Weight {
  347. curID = v.ID
  348. drawId = v.Order
  349. break
  350. }
  351. cur += v.Weight
  352. }
  353. if d, ok := ret[curID]; ok {
  354. if d.HourTotalCondition > 0 {
  355. i, err := GetDrawHour(curID)
  356. if err != nil {
  357. return nil, false
  358. }
  359. if int(i) > d.HourTotalCondition {
  360. return nil, false
  361. }
  362. }
  363. if d.DailyTotalCondition > 0 {
  364. i, err := GetDrawDailyKey(curID)
  365. if err != nil {
  366. return nil, false
  367. }
  368. if int(i) > d.DailyTotalCondition {
  369. return nil, false
  370. }
  371. }
  372. if d.WeeklyTotalCondition > 0 {
  373. i, err := GetDrawWeeklyKey(curID)
  374. if err != nil {
  375. return nil, false
  376. }
  377. if int(i) > d.WeeklyTotalCondition {
  378. return nil, false
  379. }
  380. }
  381. p.Item.AddItem(d.Reward)
  382. p.Item[ItemDrawsNumber].sub(1)
  383. draw.Reward = d.Reward
  384. draw.ID = drawId
  385. draw.CurID = curID
  386. if len(p.Roll.Draw) == 0 {
  387. p.Roll.Draw = make(map[int]int)
  388. }
  389. if len(p.Roll.DrawRatio) == 0 {
  390. p.Roll.DrawRatio = make(map[int]float64)
  391. }
  392. p.Roll.Draw[curID] += 1
  393. p.Roll.DrawTotal += 1
  394. p.Roll.DrawRatio[curID] = float64(p.Roll.Draw[curID]) / float64(p.Roll.DrawTotal)
  395. p.SetPlayerRewardLog(SourceDraw, 0, draw.Reward, []data.ItemReward{{1, ItemDrawsNumber, 1}}, draw)
  396. } else {
  397. return nil, false
  398. }
  399. return &draw, true
  400. }
  401. func GetDrawHour(id int) (int64, error) {
  402. s := mhayaTime.CreateFromTime(mhayaTime.Now().Hour(), 0, 0).ToDateTimeFormat()
  403. key := fmt.Sprintf("%v:%v:%v", constant.DrawHourKey, id, s)
  404. return mdb.RDB.IncrBy(context.Background(), key, 1).Result()
  405. }
  406. func GetDrawDailyKey(id int) (int64, error) {
  407. key := fmt.Sprintf("%v:%v:%v", constant.DrawDailyKey, id, mhayaTime.Now().Format(mhayaTime.DateFormat))
  408. return mdb.RDB.IncrBy(context.Background(), key, 1).Result()
  409. }
  410. func GetDrawWeeklyKey(id int) (int64, error) {
  411. key := fmt.Sprintf("%v:%v:%v", constant.DrawWeeklyKey, id, mhayaTime.Now().Week())
  412. return mdb.RDB.IncrBy(context.Background(), key, 1).Result()
  413. }
  414. func (p *Player) SignIn() *param.ChangeData {
  415. var sign param.ChangeData
  416. p.SinIn.SignDailyTime = mhayaTime.Now().Unix()
  417. p.SinIn.Sign[p.SinIn.Num].IsSign = 1
  418. p.SinIn.Sign[p.SinIn.Num].SignTime = mhayaTime.Now().Unix()
  419. var addItem AddItem
  420. at := addItem.Add(p.SinIn.Sign[p.SinIn.Num].Reward)
  421. sign.AddItem = at
  422. p.Item.AddItem(p.SinIn.Sign[p.SinIn.Num].Reward)
  423. p.SinIn.Num += 1
  424. p.SetPlayerRewardLog(SourceSignIn, p.SinIn.Num, p.SinIn.Sign[p.SinIn.Num].Reward, []data.ItemReward{}, nil)
  425. sign.IsLevelUp = p.LevelUp()
  426. return &sign
  427. }
  428. func (p *Player) SetPlayerRewardLog(tp, tid int, addReward, subReward []data.ItemReward, desc interface{}) {
  429. p.Index += 1
  430. if len(p.PlayerReward) == 0 {
  431. p.PlayerReward = make(map[int]*PlayerRewardBase)
  432. }
  433. p.PlayerReward[p.Index] = &PlayerRewardBase{
  434. UserName: p.UserName,
  435. AddReward: addReward,
  436. SubReward: subReward,
  437. Desc: desc,
  438. Source: tp,
  439. SourceId: tid,
  440. CreateTime: mhayaTime.Now().Unix(),
  441. }
  442. }
  443. func (p *Player) SetPassiveInviteReward() {
  444. ret2, ok := data.DiscreteRuleConfig.Get(InvitePlayer)
  445. c, tok := mhayaString.ToInt(ret2.Condition)
  446. if !ok || !tok || c != p.Level {
  447. return
  448. }
  449. ret3, ok := data.DiscreteRuleConfig.Get(InviteVipPlayer)
  450. c, tok = mhayaString.ToInt(ret3.Condition)
  451. if !ok || !tok || c != p.Level {
  452. return
  453. }
  454. //被动邀请
  455. key := fmt.Sprintf("%v:1:%v", constant.InviteKey, p.UserName)
  456. pUid := mdb.RDB.Get(context.Background(), key).Val()
  457. pbase, err := mdb.RDB.Get(context.Background(), GetPlayBaseKey(pUid)).Bytes()
  458. if err != nil {
  459. return
  460. }
  461. mdb.RDB.Del(context.Background(), key)
  462. var base param.PlayerBase
  463. err = jsoniter.Unmarshal(pbase, &base)
  464. if err != nil {
  465. return
  466. }
  467. var addItem []data.ItemReward
  468. if p.IsVip == 0 {
  469. //主动邀请记录
  470. key = fmt.Sprintf("%v:%v:%v", constant.InviteKey, InvitePlayer, pUid)
  471. mdb.RDB.IncrBy(context.Background(), key, 1)
  472. addItem = append(addItem, ret2.Reward...)
  473. } else {
  474. key = fmt.Sprintf("%v:%v:%v", constant.InviteKey, InviteVipPlayer, pUid)
  475. mdb.RDB.IncrBy(context.Background(), key, 1)
  476. addItem = append(addItem, ret3.Reward...)
  477. }
  478. if base.IsLeader == 1 {
  479. ret, ok := data.DiscreteRuleConfig.Get(InvitePlayerByLeader)
  480. if !ok {
  481. return
  482. }
  483. addItem = append(addItem, ret.Reward...)
  484. }
  485. if p.InviteReward.IsClaim == 2 {
  486. p.InviteReward.Reward = []data.ItemReward{}
  487. p.InviteReward.IsClaim = 0
  488. p.InviteReward.RewardTime = 0
  489. }
  490. p.InviteReward.Reward = append(p.InviteReward.Reward, addItem...)
  491. }
  492. func (p *Player) SetInviteReward() {
  493. ret, ok := data.DiscreteRuleConfig.Get(InvitePlayerByLeader)
  494. if !ok {
  495. return
  496. }
  497. ret2, ok := data.DiscreteRuleConfig.Get(InvitePlayer)
  498. c, tok := mhayaString.ToInt(ret2.Condition)
  499. if !ok || !tok || c != p.Level {
  500. return
  501. }
  502. ret3, ok := data.DiscreteRuleConfig.Get(InviteVipPlayer)
  503. c, tok = mhayaString.ToInt(ret3.Condition)
  504. if !ok || !tok || c != p.Level {
  505. return
  506. }
  507. var addItem []data.ItemReward
  508. key1 := fmt.Sprintf("%v:%v:%v", constant.InviteKey, InvitePlayer, p.UserName)
  509. key2 := fmt.Sprintf("%v:%v:%v", constant.InviteKey, InviteVipPlayer, p.UserName)
  510. freeNum, _ := mdb.RDB.Get(context.Background(), key1).Int()
  511. VipNum, _ := mdb.RDB.Get(context.Background(), key2).Int()
  512. for i := 0; i < freeNum; i++ {
  513. addItem = append(addItem, ret2.Reward...)
  514. p.SetPlayerRewardLog(SourceInvite, InvitePlayer, []data.ItemReward{}, []data.ItemReward{}, 2)
  515. if p.IsLeader == 1 {
  516. addItem = append(addItem, ret.Reward...)
  517. p.SetPlayerRewardLog(SourceInvite, InvitePlayerByLeader, []data.ItemReward{}, []data.ItemReward{}, 2)
  518. }
  519. mdb.RDB.DecrBy(context.Background(), key1, 1)
  520. }
  521. for i := 0; i < VipNum; i++ {
  522. addItem = append(addItem, ret3.Reward...)
  523. p.SetPlayerRewardLog(SourceInvite, InviteVipPlayer, []data.ItemReward{}, []data.ItemReward{}, 2)
  524. if p.IsLeader == 1 {
  525. addItem = append(addItem, ret.Reward...)
  526. p.SetPlayerRewardLog(SourceInvite, InvitePlayerByLeader, []data.ItemReward{}, []data.ItemReward{}, 2)
  527. }
  528. mdb.RDB.DecrBy(context.Background(), key2, 1)
  529. }
  530. if (freeNum + VipNum) > 0 {
  531. SetRank(constant.RankSourceInvite, p.IsRobot, p.UserName, freeNum+VipNum)
  532. }
  533. if p.InviteReward.IsClaim == 2 {
  534. p.InviteReward.Reward = []data.ItemReward{}
  535. p.InviteReward.IsClaim = 0
  536. p.InviteReward.RewardTime = 0
  537. }
  538. p.InviteReward.Reward = append(p.InviteReward.Reward, addItem...)
  539. }
  540. func (p *Player) SetInvite(uid string) bool {
  541. key := fmt.Sprintf("%v:1:%v", constant.InviteKey, p.UserName)
  542. ret, ok := data.DiscreteRuleConfig.Get(InviteMaxExpireTime)
  543. if !ok {
  544. return false
  545. }
  546. t, _ := mhayaString.ToInt64(ret.Condition)
  547. err := mdb.RDB.Set(context.Background(), key, uid, time.Duration(t)*24*time.Hour).Err()
  548. if err != nil {
  549. return false
  550. }
  551. return true
  552. }