liwei_3d/go_dreamfactory
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

获取远程用户

Browse Source
This commit is contained in:
meixiongfeng 2022-10-24 12:01:24 +08:00
parent 3ff349f397
commit 05a0e37ab9
2 changed files with 21 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
modules/gourmet/api_getranduser.go
View File

@ -24,9 +24,10 @@ func (this *apiComp) GetRandUserCheck(session comm.IUserSession, req *pb.Gourmet
/// 获取一些玩家数据 /// 获取一些玩家数据
func (this *apiComp) GetRandUser(session comm.IUserSession, req *pb.GourmetGetRandUserReq) (code pb.ErrorCode, data proto.Message) { func (this *apiComp) GetRandUser(session comm.IUserSession, req *pb.GourmetGetRandUserReq) (code pb.ErrorCode, data proto.Message) {
var ( var (
szDbUser []*pb.DBUser szDbUser []*pb.DBUser
randOnlineUsers []string mapUser map[string]struct{}
) )
mapUser = make(map[string]struct{}, req.People)
code = this.GetRandUserCheck(session, req) code = this.GetRandUserCheck(session, req)
if code != pb.ErrorCode_Success { if code != pb.ErrorCode_Success {
return // 参数校验失败直接返回 return // 参数校验失败直接返回
@ -43,41 +44,40 @@ func (this *apiComp) GetRandUser(session comm.IUserSession, req *pb.GourmetGetRa
if v.Uid == session.GetUserId() || v.Uid == "" { // 过滤自己信息 if v.Uid == session.GetUserId() || v.Uid == "" { // 过滤自己信息
continue continue
} }
mapUser[v.Uid] = struct{}{}
szUid = append(szUid, v.Uid) szUid = append(szUid, v.Uid)
} }
// 随机在线玩家信息 // 随机在线玩家信息
if len(onlineList) > int(req.People) { if len(szUid) > int(req.People) {
randArr := utils.Numbers(0, len(onlineList), int(req.People)) randArr := utils.Numbers(0, len(szUid), int(req.People))
for _, v := range randArr { for _, v := range randArr {
if onlineList[v].Uid != "" { if szUid[v] != "" {
randOnlineUsers = append(randOnlineUsers, onlineList[v].Uid) mapUser[szUid[v]] = struct{}{}
} }
} }
} else { // 数量不足 则有多少给多少 } else { // 数量不足 则有多少给多少
for _, v := range szUid { for _, v := range szUid {
randOnlineUsers = append(randOnlineUsers, v) mapUser[v] = struct{}{}
} }
left := int(req.People) - len(randOnlineUsers) left := int(req.People) - len(mapUser)
if left > 0 { // 一个人也没有 那就从db 中随机取 if left > 0 { // 一个人也没有 那就从db 中随机取
//tag := session.GetServiecTag() if _data, err1 := this.module.modelGourmet.DB.Find(core.SqlTable(comm.TableUser), bson.M{}, options.Find().SetSort(bson.M{"lv": -1}).SetLimit(int64(req.People))); err1 == nil {
//if conn, err := db.ServerDBConn(tag); err == nil {
//dbModel := db.NewDBModel(comm.TableUser, time.Hour, conn)
if _data, err1 := this.module.modelGourmet.DB.Find(core.SqlTable(comm.TableUser), bson.M{}, options.Find().SetSort(bson.M{"lv": -1}).SetLimit(int64(left))); err1 == nil {
for _data.Next(context.TODO()) { for _data.Next(context.TODO()) {
temp := &pb.DBUser{} temp := &pb.DBUser{}
if err = _data.Decode(temp); err == nil { if err = _data.Decode(temp); err == nil {
randOnlineUsers = append(randOnlineUsers, temp.Uid) if len(mapUser) >= int(req.People) {
break
}
if _, ok := mapUser[temp.Uid]; !ok {
szDbUser = append(szDbUser, temp)
}
} }
} }
} }
//}
} }
} }
for _, v := range randOnlineUsers { for k := range mapUser {
user, _ := this.module.ModuleUser.GetCrossUser(v) user, _ := this.module.ModuleUser.GetRmoteUser(k)
szDbUser = append(szDbUser, user) // 转成user对象 szDbUser = append(szDbUser, user) // 转成user对象
} }
session.SendMsg(string(this.module.GetType()), GourmetGetRandUserResp, &pb.GourmetGetRandUserResp{User: szDbUser}) session.SendMsg(string(this.module.GetType()), GourmetGetRandUserResp, &pb.GourmetGetRandUserResp{User: szDbUser})

4
modules/smithy/api_getranduser.go
View File

@ -69,7 +69,7 @@ func (this *apiComp) GetRandUser(session comm.IUserSession, req *pb.SmithyGetRan
break break
} }
if _, ok := mapUser[temp.Uid]; !ok { if _, ok := mapUser[temp.Uid]; !ok {
mapUser[temp.Uid] = struct{}{} szDbUser = append(szDbUser, temp)
} }
} }
} }
@ -77,7 +77,7 @@ func (this *apiComp) GetRandUser(session comm.IUserSession, req *pb.SmithyGetRan
} }
} }
for k := range mapUser { for k := range mapUser {
user, _ := this.module.ModuleUser.GetCrossUser(k) user, _ := this.module.ModuleUser.GetRmoteUser(k)
szDbUser = append(szDbUser, user) // 转成user对象 szDbUser = append(szDbUser, user) // 转成user对象
} }
session.SendMsg(string(this.module.GetType()), SmithyGetRandUserResp, &pb.SmithyGetRandUserResp{User: szDbUser}) session.SendMsg(string(this.module.GetType()), SmithyGetRandUserResp, &pb.SmithyGetRandUserResp{User: szDbUser})