88 lines
2.2 KiB
Go
88 lines
2.2 KiB
Go
package chat
|
|
|
|
import (
|
|
"fmt"
|
|
"go_dreamfactory/comm"
|
|
"go_dreamfactory/pb"
|
|
"time"
|
|
|
|
"go.mongodb.org/mongo-driver/bson/primitive"
|
|
"google.golang.org/protobuf/proto"
|
|
)
|
|
|
|
//参数校验
|
|
func (this *apiComp) SendCheck(session comm.IUserSession, req *pb.ChatSendReq) (code pb.ErrorCode) {
|
|
if (req.Channel == pb.ChatChannel_Union && req.TargetId == "") || (req.Channel == pb.ChatChannel_Private && req.TargetId == "") {
|
|
code = pb.ErrorCode_ReqParameterError
|
|
}
|
|
return
|
|
}
|
|
|
|
///消息发送请求
|
|
func (this *apiComp) Send(session comm.IUserSession, req *pb.ChatSendReq) (code pb.ErrorCode, data proto.Message) {
|
|
var (
|
|
err error
|
|
max int32
|
|
msg *pb.DBChat
|
|
user *pb.DBUser
|
|
max_chat int32
|
|
)
|
|
if code = this.SendCheck(session, req); code != pb.ErrorCode_Success {
|
|
return
|
|
}
|
|
if user = this.module.ModuleUser.GetUser(session.GetUserId()); user == nil {
|
|
this.module.Errorf("GetUser is nill")
|
|
code = pb.ErrorCode_DBError
|
|
return
|
|
}
|
|
|
|
msg = &pb.DBChat{
|
|
Id: primitive.NewObjectID().Hex(),
|
|
Channel: req.Channel,
|
|
Suid: session.GetUserId(),
|
|
Avatar: user.Avatar,
|
|
Content: req.Content,
|
|
Ctime: time.Now().Unix(),
|
|
}
|
|
if max, err = this.module.configure.GetChannelRecordMax(); err != nil {
|
|
code = pb.ErrorCode_ConfigNoFound
|
|
return
|
|
}
|
|
if max_chat, err = this.module.configure.GetChannelRecordMax(); err != nil {
|
|
code = pb.ErrorCode_ConfigNoFound
|
|
return
|
|
}
|
|
switch msg.Channel {
|
|
case pb.ChatChannel_World:
|
|
if err = this.module.modelChat.addChatMsg(worldchatkey, int64(max), msg); err != nil {
|
|
code = pb.ErrorCode_DBError
|
|
return
|
|
}
|
|
if err = this.module.PushWorld(msg); err != nil {
|
|
code = pb.ErrorCode_DBError
|
|
return
|
|
}
|
|
break
|
|
case pb.ChatChannel_Union:
|
|
msg.UnionId = req.TargetId
|
|
if err = this.module.modelChat.addChatMsg(fmt.Sprintf("%s:%s", unionchatkey, req.TargetId), int64(max_chat), msg); err != nil {
|
|
code = pb.ErrorCode_DBError
|
|
return
|
|
}
|
|
if err = this.module.PushUnion(req.TargetId, msg); err != nil {
|
|
code = pb.ErrorCode_DBError
|
|
return
|
|
}
|
|
break
|
|
case pb.ChatChannel_Private:
|
|
msg.Ruid = req.TargetId
|
|
if err = this.module.PushUser(msg); err != nil {
|
|
code = pb.ErrorCode_DBError
|
|
return
|
|
}
|
|
break
|
|
}
|
|
session.SendMsg(string(this.module.GetType()), "send", &pb.ChatSendResp{})
|
|
return
|
|
}
|