package smithy import ( "go_dreamfactory/comm" "go_dreamfactory/pb" "go_dreamfactory/utils" "google.golang.org/protobuf/proto" ) //参数校验 func (this *apiComp) GetRandUserCheck(session comm.IUserSession, req *pb.SmithyGetRandUserReq) (code pb.ErrorCode) { if req.People <= 0 { code = pb.ErrorCode_ReqParameterError return } return } /// 获取一些玩家数据 func (this *apiComp) GetRandUser(session comm.IUserSession, req *pb.SmithyGetRandUserReq) (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.UserOnlineList() if err != nil { code = pb.ErrorCode_DBError return } var szUid []string for _, v := range onlineList { if v.Uid == session.GetUserId() { // 过滤自己信息 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 szUid[v] != "" { randOnlineUsers = append(randOnlineUsers, szUid[v]) } } } else { // 数量不足 则有多少给多少 for _, v := range szUid { randOnlineUsers = append(randOnlineUsers, v) } } for _, v := range randOnlineUsers { szDbUser = append(szDbUser, this.module.ModuleUser.GetUser(v)) // 转成user对象 } session.SendMsg(string(this.module.GetType()), SmithyGetRandUserResp, &pb.SmithyGetRandUserResp{User: szDbUser}) return }