64 lines
1.6 KiB
Go
64 lines
1.6 KiB
Go
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
|
|
}
|