// 客户端登陆
func HandleClientLogin(conn *net.TCPConn, recPacket *packet.Packet) {
// read
readMsg := &pb.PbClientLogin{}
packet.Unpack(recPacket, readMsg)
uuid := readMsg.GetUuid()
// 检测uuid合法性
if cid.UuidCheckExist(uuid) {
// 如果已经在线则关闭以前的conn
if cid.UuidCheckOnline(uuid) {
co := UuidMapConn.Get(uuid)
if co != nil {
CloseConn(co.(*net.TCPConn))
}
}
// 上线conn
InitConn(conn, uuid)
} else {
CloseConn(conn)
}
// fmt.Println("uuid:", readMsg.GetUuid())
// fmt.Println("version:", readMsg.GetVersion())
// fmt.Println("timestamp:", convert.TimestampToTimeString(readMsg.GetTimestamp()))
// write
writeMsg := &pb.PbServerAcceptLogin{
Login: proto.Bool(true),
TipsMsg: proto.String("登陆成功"),
Timestamp: proto.Int64(time.Now().Unix()),
}
SendPbData(conn, packet.PK_ServerAcceptLogin, writeMsg)
}
// 处理讨论组消息转发
func HandleGroupTextChat(conn *net.TCPConn, recPacket *packet.Packet) {
// read
readMsg := &pb.PbGroupTextChat{}
packet.Unpack(recPacket, readMsg)
from_uuid := ConnMapUuid.Get(conn).(string)
group_id := readMsg.GetGroupId()
txt_msg := readMsg.GetTextMsg()
// timestamp := readMsg.GetTimestamp()
// 验证发送者，这样会影响性能，视情况可以把这个验证去掉
if readMsg.GetFromUuid() != from_uuid || !groupinfo.ExistUuidFromGroup(group_id, from_uuid) {
CloseConn(conn)
return
}
// write
writeMsg := &pb.PbGroupTextChat{
FromUuid: proto.String(from_uuid),
GroupId: proto.String(group_id),
TextMsg: proto.String(txt_msg),
Timestamp: proto.Int64(time.Now().Unix()),
}
// 在线消息包
pac1, err := packet.Pack(packet.PK_GroupTextChat, writeMsg)
if err != nil {
log.Printf("%v\r\n", err)
return
}
// 离线消息包
pac2, err := packet.Pack(packet.PK_ServerResponseGroupOfflineMsg, writeMsg)
if err != nil {
log.Printf("%v\r\n", err)
return
}
// 将消息转发给所有组员(除了自己)，不在线则离线存储
group_members := groupinfo.GetAllUuid(group_id)
for i := 1; i < len(group_members); i++ {
if group_members[i] == from_uuid {
continue
}
if cid.UuidCheckOnline(group_members[i]) {
to_conn := UuidMapConn.Get(group_members[i]).(*net.TCPConn)
SendByteStream(to_conn, pac1.GetBytes())
} else {
groupmsg.AddMsg(group_members[i], string(pac2.GetBytes()))
}
}
}
// 处理C2C消息转发
func HandleC2CTextChat(conn *net.TCPConn, recPacket *packet.Packet) {
// read
readMsg := &pb.PbC2CTextChat{}
packet.Unpack(recPacket, readMsg)
from_uuid := ConnMapUuid.Get(conn).(string)
to_uuid := readMsg.GetToUuid()
txt_msg := readMsg.GetTextMsg()
// timestamp := readMsg.GetTimestamp()
// 验证发送者的真实性以及发送对象是否存，若消息伪造，则断开该连接
if readMsg.GetFromUuid() != from_uuid || !cid.UuidCheckExist(to_uuid) {
CloseConn(conn)
return
}
// write
writeMsg := &pb.PbC2CTextChat{
FromUuid: proto.String(from_uuid),
ToUuid: proto.String(to_uuid),
TextMsg: proto.String(txt_msg),
Timestamp: proto.Int64(time.Now().Unix()),
}
// 在线消息包
pac1, err := packet.Pack(packet.PK_C2CTextChat, writeMsg)
if err != nil {
log.Printf("%v\r\n", err)
return
}
// 离线消息包
pac2, err := packet.Pack(packet.PK_ServerResponseC2COfflineMsg, writeMsg)
if err != nil {
log.Printf("%v\r\n", err)
return
}
// 若 to_uuid 在线，则转发该消息，发送失败 或者 to_uuid不在线 则保存为离线消息
if cid.UuidCheckOnline(to_uuid) {
to_conn := UuidMapConn.Get(to_uuid).(*net.TCPConn)
if SendByteStream(to_conn, pac1.GetBytes()) != nil {
c2cmsg.AddMsg(to_uuid, string(pac2.GetBytes()))
report.AddCount(report.OfflineMsg, 1)
} else {
report.AddCount(report.OnlineMsg, 1)
}
} else {
c2cmsg.AddMsg(to_uuid, string(pac2.GetBytes()))
report.AddCount(report.OfflineMsg, 1)
}
}
// 处理登陆
func HandleClientLogin(conn *net.TCPConn, recPacket *packet.Packet) {
// read
readMsg := &pb.PbClientLogin{}
packet.Unpack(recPacket, readMsg)
uuid := readMsg.GetUuid()
// 检测uuid合法性
if cid.UuidCheckExist(uuid) {
// 如果已经在线则关闭以前的conn
if cid.UuidCheckOnline(uuid) {
co := UuidMapConn.Get(uuid)
if co != nil {
CloseConn(co.(*net.TCPConn))
}
}
// 上线conn
InitConn(conn, uuid)
} else {
CloseConn(conn)
}
// fmt.Println("uuid:", readMsg.GetUuid())
// fmt.Println("version:", readMsg.GetVersion())
// fmt.Println("timestamp:", convert.TimestampToTimeString(readMsg.GetTimestamp()))
// write
writeMsg := &pb.PbServerAcceptLogin{
Login: proto.Bool(true),
TipsMsg: proto.String("登陆成功"),
Timestamp: proto.Int64(time.Now().Unix()),
}
SendPbData(conn, packet.PK_ServerAcceptLogin, writeMsg)
// 检查是否有该uuid的离线消息存在，若有，则发送其离线消息
if offmsgNum := c2c.GetMsgNum(uuid); offmsgNum > 0 {
// 这里比较复杂，后续再优化(可以多个离线消息一起发送)
// 得到所有离线消息
// fmt.Println("转发离线消息")
msgs := c2c.GetMsgs(uuid, offmsgNum)
c2c.DeleteMsgs(uuid, offmsgNum, offmsgNum)
for i, _ := range msgs {
SendByteStream(conn, []byte(msgs[i]))
}
}
}