package gourmet import ( "context" "go_dreamfactory/comm" "go_dreamfactory/lego/core" "go_dreamfactory/pb" "go_dreamfactory/utils" "go.mongodb.org/mongo-driver/bson" "go.mongodb.org/mongo-driver/mongo/options" "google.golang.org/protobuf/proto" ) //参数校验 func (this *apiComp) GetRandUserCheck(session comm.IUserSession, req *pb.GourmetGetRandUserReq) (code pb.ErrorCode) { if req.People <= 0 { code = pb.ErrorCode_ReqParameterError return } return } /// 获取一些玩家数据 func (this *apiComp) GetRandUser(session comm.IUserSession, req *pb.GourmetGetRandUserReq) (code pb.ErrorCode, data proto.Message) { var ( szDbUser []*pb.DBUser randOnlineUsers []string ) code = this.GetRandUserCheck(session, req) if code != pb.ErrorCode_Success { return // 参数校验失败直接返回 } // 获取在线玩家信息 onlineList, err := this.module.ModuleUser.CrossUserOnlineList() if err != nil { code = pb.ErrorCode_DBError return } var szUid []string for _, v := range onlineList { if v.Uid == session.GetUserId() || v.Uid == "" { // 过滤自己信息 continue } szUid = append(szUid, v.Uid) } // 随机在线玩家信息 if len(onlineList) > int(req.People) { randArr := utils.Numbers(0, len(onlineList), int(req.People)) for _, v := range randArr { if onlineList[v].Uid != "" { randOnlineUsers = append(randOnlineUsers, onlineList[v].Uid) } } } else { // 数量不足 则有多少给多少 for _, v := range szUid { randOnlineUsers = append(randOnlineUsers, v) } left := int(req.People) - len(randOnlineUsers) if left > 0 { // 一个人也没有 那就从db 中随机取 //tag := session.GetServiecTag() //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()) { temp := &pb.DBUser{} if err = _data.Decode(temp); err == nil { randOnlineUsers = append(randOnlineUsers, temp.Uid) } } } //} } } for _, v := range randOnlineUsers { user, _ := this.module.ModuleUser.GetCrossUser(v) szDbUser = append(szDbUser, user) // 转成user对象 } session.SendMsg(string(this.module.GetType()), GourmetGetRandUserResp, &pb.GourmetGetRandUserResp{User: szDbUser}) return }