101 lines
2.7 KiB
Go
101 lines
2.7 KiB
Go
package gourmet
|
|
|
|
import (
|
|
"context"
|
|
"go_dreamfactory/comm"
|
|
"go_dreamfactory/lego/core"
|
|
"go_dreamfactory/pb"
|
|
"go_dreamfactory/sys/db"
|
|
"go_dreamfactory/utils"
|
|
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
"go.mongodb.org/mongo-driver/mongo/options"
|
|
)
|
|
|
|
//参数校验
|
|
func (this *apiComp) GetRandUserCheck(session comm.IUserSession, req *pb.GourmetGetRandUserReq) (errdata *pb.ErrorData) {
|
|
if req.People <= 0 {
|
|
errdata = &pb.ErrorData{
|
|
Code: pb.ErrorCode_ReqParameterError,
|
|
Title: pb.ErrorCode_ReqParameterError.ToString(),
|
|
}
|
|
return
|
|
}
|
|
return
|
|
}
|
|
|
|
/// 获取一些玩家数据
|
|
func (this *apiComp) GetRandUser(session comm.IUserSession, req *pb.GourmetGetRandUserReq) (errdata *pb.ErrorData) {
|
|
var (
|
|
szDbUser []*pb.DBUser
|
|
mapUser map[string]struct{}
|
|
)
|
|
mapUser = make(map[string]struct{}, 0)
|
|
if errdata = this.GetRandUserCheck(session, req); errdata != nil {
|
|
return // 参数校验失败直接返回
|
|
}
|
|
|
|
// 获取在线玩家信息
|
|
onlineList, err := this.module.ModuleUser.UserOnlineList()
|
|
if err != nil {
|
|
errdata = &pb.ErrorData{
|
|
Code: pb.ErrorCode_DBError,
|
|
Title: pb.ErrorCode_DBError.ToString(),
|
|
Message: err.Error(),
|
|
}
|
|
return
|
|
}
|
|
|
|
var szUid []string
|
|
for _, v := range onlineList {
|
|
if v.Uid == session.GetUserId() || v.Uid == "" { // 过滤自己信息
|
|
continue
|
|
}
|
|
szUid = append(szUid, v.Uid)
|
|
}
|
|
// 随机在线玩家信息
|
|
if len(szUid) > int(req.People) {
|
|
randArr := utils.RandomNumbers(0, len(szUid), int(req.People))
|
|
for _, v := range randArr {
|
|
if szUid[v] != "" {
|
|
mapUser[szUid[v]] = struct{}{}
|
|
}
|
|
}
|
|
} else { // 数量不足 则有多少给多少
|
|
for _, v := range szUid {
|
|
mapUser[v] = struct{}{}
|
|
}
|
|
left := int(req.People) - len(mapUser)
|
|
if left > 0 { // 一个人也没有 那就从db 中随机取
|
|
tag, _, b := utils.UIdSplit(session.GetUserId())
|
|
if b {
|
|
if conn, err := db.ServerDBConn(tag); err == nil {
|
|
dbModel := db.NewDBModel(comm.TableHero, conn)
|
|
if _data, err1 := dbModel.DB.Find(core.SqlTable(comm.TableUser), bson.M{}, options.Find().SetSort(bson.M{"lv": -1}).SetLimit(int64(req.People))); err1 == nil {
|
|
for _data.Next(context.TODO()) {
|
|
temp := &pb.DBUser{}
|
|
if err = _data.Decode(temp); err == nil {
|
|
if len(mapUser)+len(szDbUser) >= int(req.People) {
|
|
break
|
|
}
|
|
if _, ok := mapUser[temp.Uid]; !ok {
|
|
szDbUser = append(szDbUser, temp)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
for k := range mapUser {
|
|
if user, err := this.module.ModuleUser.GetUser(k); err == nil {
|
|
szDbUser = append(szDbUser, user) // 转成user对象
|
|
} else {
|
|
this.module.Errorf("%v", err)
|
|
}
|
|
}
|
|
session.SendMsg(string(this.module.GetType()), GourmetGetRandUserResp, &pb.GourmetGetRandUserResp{User: szDbUser})
|
|
return
|
|
}
|