server/game/game_cluster/internal/mdb/models/rank.go

package models

import (
	"context"
	"fmt"
	jsoniter "github.com/json-iterator/go"
	mhayaTime "github.com/mhaya/extend/time"
	"github.com/mhaya/game/game_cluster/internal/constant"
	"github.com/mhaya/game/game_cluster/internal/data"
	"github.com/mhaya/game/game_cluster/internal/mdb"
	"github.com/mhaya/game/game_cluster/internal/param"
	"time"
)

type RankBase struct {
	ID        int               `json:"id" bson:"id"`
	Rank      int               `json:"rank" bson:"rank"`
	score     int               `json:"score" bson:"score"`
	IsClaim   int               `json:"isClaim" bson:"isClaim"`
	ClaimTime int64             `json:"claimTime" bson:"claimTime"`
	Reward    []data.ItemReward `json:"reward" bson:"reward"`
}

type RankReward map[int]*RankBase

func NewRankReward() map[int]*RankBase {
	return make(map[int]*RankBase)
}

func (r RankReward) InitRank(uid string, tp int) {
	ret := data.RankConfig.GetAll()
	for _, v := range ret {
		if v.Type == tp {
			key := GetRankKey(v.ID, 1)
			rank := GetPlayerRank(key, uid)
			score := GetPlayerRankScore(key, uid)
			r[v.ID] = &RankBase{
				ID:        v.ID,
				Rank:      rank,
				score:     score,
				IsClaim:   0,
				ClaimTime: 0,
				Reward:    data.RankRewardConfig.GetRewardByRank(v.ID, rank),
			}
		}
	}
}

func GetPlayBaseKey(uid string) string {
	return fmt.Sprintf("%v:%v", constant.PlayerBaseKey, uid)
}

func GetRankKey(id, tp int) string {
	ret, ok := data.RankConfig.Get(id)
	if !ok {
		return ""
	}
	key := fmt.Sprintf("%s", constant.RankKey)
	if ret.StartTime != "" && ret.EndTime != "" {
		stime, err := time.Parse(ret.StartTime, mhayaTime.DateTimeFormat)
		if err != nil {
			return ""
		}
		sTime := mhayaTime.CreateFromTimestamp(stime.Unix())
		if sTime.After(time.Now()) {
			return ""
		}

		etime, err := time.Parse(ret.EndTime, mhayaTime.DateTimeFormat)
		if err != nil {
			return ""
		}

		eTime := mhayaTime.CreateFromTimestamp(etime.Unix())
		if eTime.Before(time.Now()) {
			return ""
		}
	}
	switch ret.Type {
	case 1:
		if tp == 0 {
			key = fmt.Sprintf("%s:%v:%v", key, id, mhayaTime.Now().Format(mhayaTime.DateFormat))
		} else {
			key = fmt.Sprintf("%s:%v:%v", key, id, mhayaTime.Now().Add(-24*time.Hour).Format(mhayaTime.DateFormat))
		}
	case 2:
		if tp == 0 {
			key = fmt.Sprintf("%s:%v:%v", key, id, mhayaTime.Now().StartOfWeek().Format(mhayaTime.DateFormat))
		} else {
			key = fmt.Sprintf("%s:%v:%v", key, id, mhayaTime.Now().StartOfWeek().Add(-7*24*time.Hour).Format(mhayaTime.DateFormat))
		}
	}

	return key
}

func SetRank(source, isRobot int, uid string, score int) error {
	ret := data.RankConfig.GetAll()
	ctx := context.Background()
	i := float64(2*1e12-time.Now().UnixMilli()) / 1e13
	for _, v := range ret {
		if v.Source == source {
			if v.IsBot == 0 && isRobot == 1 {
				continue
			}
			key := GetRankKey(v.ID, 0)
			err := mdb.RDB.ZIncrBy(ctx, key, float64(score)+i, uid).Err()
			if err != nil {
				return err
			}
		}
	}
	return nil
}

func GetPlayerRank(key, uid string) int {
	ctx := context.Background()
	rank, err := mdb.RDB.ZRank(ctx, key, uid).Result()
	if err != nil {
		return -1
	}
	return int(rank) + 1
}

func GetPlayerRankScore(key, uid string) int {
	ctx := context.Background()
	score, err := mdb.RDB.ZScore(ctx, key, uid).Result()
	if err != nil {
		return 0
	}
	return int(score)
}

func GetRankList(id, tp int) map[int]*param.PlayerBase {
	ret, ok := data.RankConfig.Get(id)
	if !ok {
		return nil
	}
	ctx := context.Background()
	key := GetRankKey(id, tp)
	ranks, err := mdb.RDB.ZRevRangeWithScores(ctx, key, 0, int64(ret.ShowTotal)-1).Result()
	if err != nil {
		return nil
	}

	rankList := make(map[int]*param.PlayerBase)
	for k, v := range ranks {
		uid := v.Member.(string)
		playerBaseKey := GetPlayBaseKey(uid)
		info, err := mdb.RDB.Get(ctx, playerBaseKey).Result()
		if err != nil {
			continue
		}
		var playerBase param.PlayerBase
		jsoniter.Unmarshal([]byte(info), &playerBase)

		rankList[k+1] = &param.PlayerBase{
			UserName:      playerBase.UserName,
			UserNameMaybe: playerBase.UserNameMaybe,
			Avatar:        playerBase.Avatar,
			OpenId:        playerBase.OpenId,
			NickName:      playerBase.NickName,
			Rank:          k + 1,
			Score:         int(v.Score),
			Reward:        data.RankRewardConfig.GetRewardByRank(id, k+1),
		}
	}
	return rankList
}