104 lines
2.4 KiB
Go
104 lines
2.4 KiB
Go
package friend
|
|
|
|
import (
|
|
"go_dreamfactory/comm"
|
|
"go_dreamfactory/pb"
|
|
"go_dreamfactory/utils"
|
|
|
|
"google.golang.org/protobuf/proto"
|
|
)
|
|
|
|
func (this *apiComp) AgreeCheck(session comm.IUserSession, req *pb.FriendAgreeReq) (code pb.ErrorCode) {
|
|
if len(req.FriendIds) == 0 {
|
|
code = pb.ErrorCode_ReqParameterError
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
//单个/批量同意
|
|
func (this *apiComp) Agree(session comm.IUserSession, req *pb.FriendAgreeReq) (code pb.ErrorCode, data proto.Message) {
|
|
if code = this.AgreeCheck(session, req); code != pb.ErrorCode_Success {
|
|
return
|
|
}
|
|
var (
|
|
err error
|
|
self *pb.DBFriend
|
|
target *pb.DBFriend
|
|
Resp *pb.FriendAgreeResp
|
|
optNum int32
|
|
)
|
|
|
|
defer func() {
|
|
Resp = &pb.FriendAgreeResp{
|
|
Num: optNum,
|
|
}
|
|
err := session.SendMsg(string(this.module.GetType()), FriendSubTypeAgree, Resp)
|
|
if err != nil {
|
|
code = pb.ErrorCode_SystemError
|
|
return
|
|
}
|
|
}()
|
|
|
|
//获取玩家自己好友数据
|
|
self = this.module.modelFriend.GetFriend(session.GetUserId())
|
|
if self == nil {
|
|
code = pb.ErrorCode_FriendSelfNoData
|
|
return
|
|
}
|
|
|
|
//同意的好友
|
|
agreeIds := []string{}
|
|
for _, friendId := range req.FriendIds {
|
|
if _, ok := utils.Find(self.FriendIds, friendId); !ok {
|
|
//不在好友列表中就加入
|
|
agreeIds = append(agreeIds, friendId)
|
|
}
|
|
}
|
|
|
|
//将目标加入到自己的好友列表中
|
|
for _, userId := range agreeIds {
|
|
if _, ok := utils.Find(self.FriendIds, userId); !ok {
|
|
if self.FriendIds == nil {
|
|
self.FriendIds = []string{}
|
|
}
|
|
self.FriendIds = append(self.FriendIds, userId)
|
|
}
|
|
|
|
//双向添加：将自己加入到申请人的好友列表中
|
|
target = this.module.modelFriend.GetFriend(userId)
|
|
if target == nil {
|
|
code = pb.ErrorCode_FriendTargetNoData
|
|
return
|
|
}
|
|
if _, ok := utils.Find(target.FriendIds, self.UId); !ok {
|
|
if target.FriendIds == nil {
|
|
target.FriendIds = []string{}
|
|
}
|
|
target.FriendIds = append(target.FriendIds, self.UId)
|
|
}
|
|
err = this.module.modelFriend.Change(target.UId, map[string]interface{}{
|
|
"friendIds": target.FriendIds,
|
|
})
|
|
if err != nil {
|
|
code = pb.ErrorCode_DBError
|
|
}
|
|
|
|
//将目标从申请列表中删除
|
|
self.ApplyIds = utils.DeleteString(self.ApplyIds, userId)
|
|
optNum++
|
|
}
|
|
|
|
//更新
|
|
err = this.module.modelFriend.Change(self.UId, map[string]interface{}{
|
|
"applyIds": self.ApplyIds,
|
|
"friendIds": self.FriendIds,
|
|
})
|
|
if err != nil {
|
|
code = pb.ErrorCode_DBError
|
|
return
|
|
}
|
|
|
|
return
|
|
}
|