99 lines
2.5 KiB
Go
99 lines
2.5 KiB
Go
package arena
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"go_dreamfactory/comm"
|
|
"go_dreamfactory/lego/core"
|
|
"go_dreamfactory/lego/sys/redis/pipe"
|
|
"go_dreamfactory/modules"
|
|
"go_dreamfactory/pb"
|
|
"go_dreamfactory/sys/db"
|
|
|
|
"github.com/go-redis/redis/v8"
|
|
)
|
|
|
|
// /竞技场 数据组件
|
|
type modelRank struct {
|
|
modules.MCompModel
|
|
module *Arena
|
|
}
|
|
|
|
// 组件初始化接口
|
|
func (this *modelRank) Init(service core.IService, module core.IModule, comp core.IModuleComp, opt core.IModuleOptions) (err error) {
|
|
this.TableName = comm.TableArenaRank
|
|
this.MCompModel.Init(service, module, comp, opt)
|
|
this.module = module.(*Arena)
|
|
|
|
return
|
|
}
|
|
|
|
// 更新排名
|
|
func (this *modelRank) updateArenaRank(users ...*pb.ArenaPlayer) (err error) {
|
|
var (
|
|
pipe *pipe.RedisPipe = this.DBModel.Redis.RedisPipe(context.TODO())
|
|
menbers []*redis.Z
|
|
cmd *redis.IntCmd
|
|
menbersCmd []*redis.IntCmd = make([]*redis.IntCmd, len(users))
|
|
rank int64
|
|
)
|
|
menbers = make([]*redis.Z, len(users))
|
|
for i, v := range users {
|
|
menbers[i] = &redis.Z{Score: float64(v.Integral), Member: v.Uinfo.Uid}
|
|
}
|
|
if cmd = pipe.ZAdd(this.TableName, menbers...); err != nil {
|
|
this.module.Errorln(err)
|
|
}
|
|
for i, v := range users {
|
|
menbersCmd[i] = pipe.ZRevRank(this.TableName, v.Uinfo.Uid)
|
|
}
|
|
if _, err = pipe.Exec(); err != nil {
|
|
this.module.Errorln(err)
|
|
return
|
|
}
|
|
if _, err = cmd.Result(); err != nil {
|
|
this.module.Errorln(err)
|
|
return
|
|
}
|
|
for i, v := range menbersCmd {
|
|
if rank, err = v.Result(); err != nil {
|
|
this.module.Errorln(err)
|
|
return
|
|
}
|
|
users[i].Rank = int32(rank + 1)
|
|
}
|
|
return
|
|
}
|
|
|
|
//获取指定区间的玩家人数
|
|
func (this *modelRank) queryRankByScoreArea(min, max int32) (count int64, err error) {
|
|
if count, err = this.DBModel.Redis.ZCount(fmt.Sprintf("%s-%s", this.DBModel.ServiceId, this.TableName), fmt.Sprintln(min), fmt.Sprintln(max)); err != nil {
|
|
this.module.Errorln(err)
|
|
return
|
|
}
|
|
return
|
|
}
|
|
|
|
// 获取排行榜前50的用户名单
|
|
func (this *modelRank) queryRankUser() (ranks []string, err error) {
|
|
var (
|
|
result []string
|
|
)
|
|
if result, err = this.DBModel.Redis.ZRevRange(fmt.Sprintf("%s-%s", this.DBModel.ServiceId, this.TableName), 0, 50).Result(); err != nil {
|
|
this.module.Errorln(err)
|
|
return
|
|
}
|
|
ranks = make([]string, 0)
|
|
for i := 0; i < len(result); i += 1 {
|
|
ranks = append(ranks, result[i])
|
|
}
|
|
return
|
|
}
|
|
|
|
// 比赛结算
|
|
func (this *modelRank) raceSettlement() {
|
|
if isopen := this.module.modelArena.isInMaintenance(); isopen {
|
|
this.DBModel.Del(fmt.Sprintf("%s-%s", this.DBModel.ServiceId, this.TableName), db.SetDBMgoLog(false))
|
|
}
|
|
}
|