func (self *InterfaceForProject) LeaveGroup(interfaceCMD *business.Command) (resultPacket Packet, err error) {
//1.从interfaceCMD 中解析出所需要的信息
//2.处理，并且得到结果
//3.把结果写入CMD，并且做成Packet 返回。
//1.解析...................................
dialogid := interfaceCMD.GetGroupid()
if dialogid == 0 {
feedbackCMD.Result = proto.Uint32(2)
feedbackCMD.ResultDescription = proto.String("no dialogid")
resultPacket = *MakeInterfacePacket(feedbackCMD, 103)
return resultPacket, err
}
//var uids []uint64
// uids = make([]uint64, 0, len(interfaceCMD.GetUid()))
uids := interfaceCMD.GetUid()[:]
if uids == nil {
feedbackCMD.Result = proto.Uint32(2)
feedbackCMD.ResultDescription = proto.String("no uid")
resultPacket = *MakeInterfacePacket(feedbackCMD, 103)
return resultPacket, err
}
////检测uid,dialogid是否合法。。。未完成********************
//2.处理...................................
//查询数据库的dialog表，给对应的uid添加当前时间戳。修该chatMap的dialogId的成员。
//1.修改数据库dialog表,添加时间戳
dbUpdateDialog, err := self.InterfaceDB.Db.Prepare("update dialog set timestamp=? where dialog_id=? and uid=?")
if err != nil {
ImBusinessLog("Prepare to update dialog error:", err)
feedbackCMD.Result = proto.Uint32(1)
feedbackCMD.ResultDescription = proto.String("IM error.")
resultPacket = *MakeInterfacePacket(feedbackCMD, 103)
return resultPacket, err
}
defer dbUpdateDialog.Close()
for _, uid := range uids {
_, err = dbUpdateDialog.Exec(time.Now().Format("2006-01-02 15:04:05"), dialogid, uid)
if err != nil {
ImBusinessLog("Update dialog error : ", err)
feedbackCMD.Result = proto.Uint32(1)
feedbackCMD.ResultDescription = proto.String("IM error.")
resultPacket = *MakeInterfacePacket(feedbackCMD, 103)
return resultPacket, err
}
//2.修改chatMap
_, flag := SessionTable[uid]
if flag == true {
delete(SessionTable[uid].chatMap, dialogid)
}
}
//3.打包...................................
feedbackCMD.Result = proto.Uint32(0)
feedbackCMD.ResultDescription = proto.String("OK")
resultPacket = *MakeInterfacePacket(feedbackCMD, 103)
return resultPacket, nil
}
func (self *InterfaceForProject) PushMessage(interfaceCMD *business.Command) (resultPacket Packet, err error) {
//1.从interfaceCMD 中解析出所需要的信息
//2.处理，并且得到结果
//3.把结果写入CMD，并且做成Packet 返回。
//1.解析...................................
if interfaceCMD.GetUid() == nil {
feedbackCMD.Result = proto.Uint32(2)
feedbackCMD.ResultDescription = proto.String("no uid")
resultPacket = *MakeInterfacePacket(feedbackCMD, 102)
return resultPacket, err
}
senderUid := interfaceCMD.GetUid()[0]
dialogID := interfaceCMD.GetGroupid()
if dialogID == 0 {
feedbackCMD.Result = proto.Uint32(2)
feedbackCMD.ResultDescription = proto.String("no dialogid")
resultPacket = *MakeInterfacePacket(feedbackCMD, 102)
return resultPacket, err
}
if interfaceCMD.GetMessage() == nil {
feedbackCMD.Result = proto.Uint32(2)
feedbackCMD.ResultDescription = proto.String("no message")
resultPacket = *MakeInterfacePacket(feedbackCMD, 102)
return resultPacket, err
}
message := interfaceCMD.GetMessage()[0]
random := GenRandom()
var (
requestMess = &protocol.Message{
Data: proto.String(message),
Random: proto.Uint32(random),
}
requestCMD = &protocol.Command{
Sender: proto.Uint64(senderUid), //sender's uid
Receiver: proto.Uint64(dialogID), //0=server
Message: requestMess,
}
)
pushMessagePacket := *MakePacket(requestCMD, 2)
_, fl := SessionTable[senderUid]
if fl == false {
//fmt.Println("uid no in")
feedbackCMD.Result = proto.Uint32(1)
feedbackCMD.ResultDescription = proto.String("sender isn't online")
resultPacket = *MakeInterfacePacket(feedbackCMD, 102)
return resultPacket, nil
}
SessionTable[senderUid].incoming <- pushMessagePacket
feedbackCMD.Result = proto.Uint32(0)
feedbackCMD.ResultDescription = proto.String("success")
resultPacket = *MakeInterfacePacket(feedbackCMD, 102)
return resultPacket, nil
//2.处理...................................
//2.1 从dialogID （act_dialog表---dialog表） 中得到receiverUid,若不为senderUid，则转发，或者存离线
//2.2 查找receiverUid失败，则返回要求服务器先建群
//3.打包...................................
}
func (self *InterfaceForProject) LogOut(interfaceCMD *business.Command) (resultPacket Packet, err error) {
//1.从interfaceCMD 中解析出所需要的信息
//2.处理，并且得到结果
//3.把结果写入CMD，并且做成Packet 返回。
//1.解析...................................
if interfaceCMD.GetUid() == nil {
feedbackCMD.Result = proto.Uint32(2)
feedbackCMD.ResultDescription = proto.String("no uid.")
resultPacket = *MakeInterfacePacket(feedbackCMD, 106)
return resultPacket, err
}
uid := interfaceCMD.GetUid()[0]
//2.处理...................................
//删除IM的user表对应的uid，得到session表chatMap的所有dialog_ID，
//并删除session表
delete(UserTable, uid)
delete(SessionTable, uid)
//3.打包...................................
feedbackCMD.Result = proto.Uint32(0)
feedbackCMD.ResultDescription = proto.String("OK")
resultPacket = *MakeInterfacePacket(feedbackCMD, 106)
return resultPacket, nil
}
func (self *InterfaceForProject) LogIn(interfaceCMD *business.Command) (resultPacket Packet, err error) {
//1.从interfaceCMD 中解析出所需要的信息
//2.处理，并且得到结果
//3.把结果写入CMD，并且做成Packet 返回。
//1.解析...................................
if interfaceCMD.GetUid() == nil {
feedbackCMD.Result = proto.Uint32(2)
feedbackCMD.ResultDescription = proto.String("no uid.")
resultPacket = *MakeInterfacePacket(feedbackCMD, 105)
return resultPacket, err
}
uid := interfaceCMD.GetUid()[0]
//验证uid，未完成/********************
//2.处理...................................
//在IM的内存表user表添加用户ID和token，返回token
//1.1 内存表添加user
token := MakeToken()
UserTable[uid] = Token{token, 1}
//1.2 数据库添加user
err = self.InterfaceDB.AddUser(uid)
if err != nil {
ImBusinessLog("Interface.LogIn.AddUser err", err)
feedbackCMD.Result = proto.Uint32(1)
feedbackCMD.ResultDescription = proto.String("IM error.")
resultPacket = *MakeInterfacePacket(feedbackCMD, 105)
return resultPacket, err
}
//3.打包...................................
feedbackCMD.Result = proto.Uint32(0)
feedbackCMD.ResultDescription = proto.String(token)
resultPacket = *MakeInterfacePacket(feedbackCMD, 105)
return resultPacket, nil
}
func (self *InterfaceForProject) InqueryHistory(interfaceCMD *business.Command) (resultPacket Packet, err error) {
//1.从interfaceCMD 中解析出所需要的信息
//2.处理，并且得到结果
//3.把结果写入CMD，并且做成Packet 返回。
//1.解析...................................
if interfaceCMD.GetUid() == nil {
feedbackCMD.Result = proto.Uint32(2)
feedbackCMD.ResultDescription = proto.String("no uid.")
resultPacket = *MakeInterfacePacket(feedbackCMD, 107)
return resultPacket, err
}
uid := interfaceCMD.GetUid()[0]
dialogid := interfaceCMD.GetGroupid()
if dialogid == 0 {
feedbackCMD.Result = proto.Uint32(2)
feedbackCMD.ResultDescription = proto.String("no dialogid.")
resultPacket = *MakeInterfacePacket(feedbackCMD, 107)
return resultPacket, err
}
//2.处理...................................
//如果dialogid为空，查询dialog表获取有uid的dialog_ID。
//根据dialog_ID查询数据库的message表，对于dialog表的uid带有时间戳的，则返回时间戳之前dialog的message
//其他的返回所有的message。
var t string
var info string
messageTimestamp := make([]string, 0, maxMessage) //
messageText := make([]string, 0, maxMessage)
if dialogid != allUserMessage { ////1.查询特定用户的聊天信息//dialogid不为空，
//1.1获取uid时间戳
dbUserTimestamp, err := self.InterfaceDB.Db.Query("select timestamp from dialog where uid=? and dialog_id=?", uid, dialogid)
if err != nil {
ImBusinessLog("Get user timestamp fail 1 ：", err)
feedbackCMD.Result = proto.Uint32(1)
feedbackCMD.ResultDescription = proto.String("IM error.")
resultPacket = *MakeInterfacePacket(feedbackCMD, 107)
return resultPacket, err
}
defer dbUserTimestamp.Close()
var uidTimes string
if dbUserTimestamp.Next() {
err = dbUserTimestamp.Scan(&uidTimes)
if err != nil {
ImBusinessLog("Get user timestamp fail 2:", err)
feedbackCMD.Result = proto.Uint32(1)
feedbackCMD.ResultDescription = proto.String("IM error.")
resultPacket = *MakeInterfacePacket(feedbackCMD, 107)
return resultPacket, err
}
}
//1.2查询message表
if uidTimes != "0000-00-00 00:00:00" { //用户带时间戳，查找时间戳之前的message
dbSelectMessages, err := self.InterfaceDB.Db.Query("select timestamp, information from message where dialog_id=? and timestamp<? order by timestamp desc", dialogid, uidTimes)
if err != nil {
ImBusinessLog("get message fail 1:", err)
feedbackCMD.Result = proto.Uint32(1)
feedbackCMD.ResultDescription = proto.String("IM error.")
resultPacket = *MakeInterfacePacket(feedbackCMD, 107)
return resultPacket, err
}
defer dbSelectMessages.Close()
for dbSelectMessages.Next() {
err := dbSelectMessages.Scan(&t, &info)
if err != nil {
ImBusinessLog("get message fail 2", err)
feedbackCMD.Result = proto.Uint32(1)
feedbackCMD.ResultDescription = proto.String("IM error.")
resultPacket = *MakeInterfacePacket(feedbackCMD, 107)
return resultPacket, err
}
messageTimestamp = append(messageTimestamp, t)
messageText = append(messageText, info)
}
//fmt.Println("用户带时间戳", messageTimestamp)
} else { //1.3.用户不带时间戳，查找所有dialogid的message,不考虑时间戳
dbSelectMessages, err := self.InterfaceDB.Db.Query("select timestamp, information from message where dialog_id=? order by timestamp desc", dialogid)
if err != nil {
ImBusinessLog("find all dialogid message query err", err)
feedbackCMD.Result = proto.Uint32(1)
feedbackCMD.ResultDescription = proto.String("IM error.")
resultPacket = *MakeInterfacePacket(feedbackCMD, 107)
return resultPacket, err
}
defer dbSelectMessages.Close()
for dbSelectMessages.Next() {
err := dbSelectMessages.Scan(&t, &info)
if err != nil {
ImBusinessLog("find all dialogid message err", err)
feedbackCMD.Result = proto.Uint32(1)
feedbackCMD.ResultDescription = proto.String("IM error.")
resultPacket = *MakeInterfacePacket(feedbackCMD, 107)
return resultPacket, err
}
messageTimestamp = append(messageTimestamp, t)
messageText = append(messageText, info)
}
}
} else { ////2.查询所有的聊天信息 。dialogid为空,查询dialog表获取有uid1的dialog_ID
//2.1获取的所有uid时间戳为0的dialogid的消息
partMsgTimestamp := make([]string, 0, maxMessage)
partMsgText := make([]string, 0, maxMessage)
//fmt.Println("查询所有的聊天信息************")
dbSelectAllMsgs, err := self.InterfaceDB.Db.Query("select timestamp, information from message where dialog_id in (select dialog_id from dialog where uid=? and timestamp in (0) ) order by timestamp desc ", uid)
if err != nil {
imdatabase.DatabaseLog(" Select from messageTable error:", err)
ImBusinessLog(" Select from messageTable error:", err)
feedbackCMD.Result = proto.Uint32(1)
feedbackCMD.ResultDescription = proto.String("IM error.")
resultPacket = *MakeInterfacePacket(feedbackCMD, 107)
return resultPacket, err
}
defer dbSelectAllMsgs.Close()
for dbSelectAllMsgs.Next() {
err := dbSelectAllMsgs.Scan(&t, &info)
if err != nil {
ImBusinessLog(" Scan from messageTable error:", err)
imdatabase.DatabaseLog(" Scan from messageTable error:", err)
feedbackCMD.Result = proto.Uint32(1)
feedbackCMD.ResultDescription = proto.String("IM error.")
resultPacket = *MakeInterfacePacket(feedbackCMD, 107)
return resultPacket, err
}
partMsgTimestamp = append(partMsgTimestamp, t)
partMsgText = append(partMsgText, info)
}
part1Length := len(partMsgTimestamp)
//fmt.Println("所有记录11", partMsgTimestamp, "partlenght:", part1Length)
//第一部分的长度，即第二部分的起始位置
//2.2获取uid带有时间戳的dialogid，以及时间戳
dbSelectUidTimestamp, err := self.InterfaceDB.Db.Query("select dialog_id,timestamp from dialog where uid=? and timestamp not in (0) order by timestamp", uid)
if err != nil {
imdatabase.DatabaseLog(" Select from dialogTable error:", err)
ImBusinessLog(" Select from dialogTable error:", err)
feedbackCMD.Result = proto.Uint32(1)
feedbackCMD.ResultDescription = proto.String("IM error.")
resultPacket = *MakeInterfacePacket(feedbackCMD, 107)
return resultPacket, err
}
defer dbSelectUidTimestamp.Close()
var userDial uint64
var userTimes string
for dbSelectUidTimestamp.Next() {
err := dbSelectUidTimestamp.Scan(&userDial, &userTimes)
if err != nil {
ImBusinessLog(" Scan from dialogTable error:", err)
feedbackCMD.Result = proto.Uint32(1)
feedbackCMD.ResultDescription = proto.String("IM error.")
resultPacket = *MakeInterfacePacket(feedbackCMD, 107)
return resultPacket, err
}
dbSelectMsgs, err := self.InterfaceDB.Db.Query("select timestamp, information from message where dialog_id=? and timestamp<? order by timestamp ", userDial, userTimes)
if err != nil {
ImBusinessLog("Select from message error:", err)
feedbackCMD.Result = proto.Uint32(1)
feedbackCMD.ResultDescription = proto.String("IM error.")
resultPacket = *MakeInterfacePacket(feedbackCMD, 107)
return resultPacket, err
}
defer dbSelectMsgs.Close()
for dbSelectMsgs.Next() {
err = dbSelectMsgs.Scan(&t, &info)
if err != nil {
ImBusinessLog("Scan from message error:", err)
feedbackCMD.Result = proto.Uint32(1)
feedbackCMD.ResultDescription = proto.String("IM error.")
resultPacket = *MakeInterfacePacket(feedbackCMD, 107)
return resultPacket, err
}
partMsgTimestamp = append(partMsgTimestamp, t)
partMsgText = append(partMsgText, info)
}
//fmt.Println(...)
QuickSort(partMsgTimestamp, partMsgText, part1Length, len(partMsgTimestamp)-1)
MergeSort(partMsgTimestamp, partMsgText, &messageTimestamp, &messageText, part1Length, len(partMsgTimestamp)-1)
/// messageTimestamp = append(messageTimestamp, partMsgTimestamp...)
// messageText = append(messageText, partMsgText...)
}
}
//3.打包...................................
returnCMD := &business.Command{
Sender: proto.Uint32(0),
Result: proto.Uint32(0),
ResultDescription: proto.String("OK"),
}
returnCMD.Message = make([]string, 0, maxMessage)
returnCMD.Timestamp = make([]string, 0, maxMessage)
returnCMD.Message = append(returnCMD.Message, messageText...)
returnCMD.Timestamp = append(returnCMD.Timestamp, messageTimestamp...)
resultPacket = *MakeInterfacePacket(returnCMD, 107)
return resultPacket, nil
}
func (self *InterfaceForProject) CreateGroup(interfaceCMD *business.Command) (resultPacket Packet, err error) {
//1.从interfaceCMD 中解析出所需要的信息
//2.处理，并且得到结果
//3.把结果写入CMD，并且做成Packet 返回。
//1.解析...................................
uids := interfaceCMD.GetUid()
if uids == nil {
feedbackCMD.Result = proto.Uint32(2) //业务系统没有输入uid，请求无法实现
feedbackCMD.ResultDescription = proto.String("no uid")
resultPacket = *MakeInterfacePacket(feedbackCMD, 101)
return resultPacket, err
}
//检测uid是否合法。。。待优化。。。
var uidstring string
for k, v := range uids {
if k == len(uids)-1 {
uidstring += strconv.Itoa(int(v))
} else {
uidstring += strconv.Itoa(int(v)) + ","
}
}
checkUser, err := self.InterfaceDB.Db.Query("select count(uid) from user where uid in (" + uidstring + ")")
if err != nil {
ImBusinessLog("Interface check uids err", err)
feedbackCMD.Result = proto.Uint32(1)
feedbackCMD.ResultDescription = proto.String("IM error.")
resultPacket = *MakeInterfacePacket(feedbackCMD, 101)
return resultPacket, err
}
defer checkUser.Close()
var sumUids int //在user表中的输入的uid总数
if checkUser.Next() {
err = checkUser.Scan(&sumUids)
if err != nil {
ImBusinessLog("Interface check uids err", err)
feedbackCMD.Result = proto.Uint32(1)
feedbackCMD.ResultDescription = proto.String("IM error.")
resultPacket = *MakeInterfacePacket(feedbackCMD, 101)
return resultPacket, err
}
}
if sumUids != len(uids) {
feedbackCMD.Result = proto.Uint32(3) //业务系统输入的uid有误，请求无法实现
feedbackCMD.ResultDescription = proto.String("uid error.")
resultPacket = *MakeInterfacePacket(feedbackCMD, 101)
return resultPacket, err
}
//2.处理...................................
//IM在数据库的dialog表（扫表？）建立一条信息，包含dialog_ID，所有uid。
//将dialog_ID添加到act_dialog中。返回dialog_ID。
//1.新建dialogId，
dialogid := uniq.GetUniq() //不合理，应当唯一，且和uid区分开
err = self.InterfaceDB.AddDialog(dialogid, uids)
if err != nil {
ImBusinessLog("Interface.CreateGroup.AddDialog err", err)
feedbackCMD.Result = proto.Uint32(1)
feedbackCMD.ResultDescription = proto.String("IM error.")
resultPacket = *MakeInterfacePacket(feedbackCMD, 101)
return resultPacket, err
}
//2.将dialog_ID添加到act_dialog中
err = self.InterfaceDB.AddActDialog(dialogid)
if err != nil {
ImBusinessLog("Interface.CreateGroup.AddActDialog err", err)
feedbackCMD.Result = proto.Uint32(1)
feedbackCMD.ResultDescription = proto.String("IM error.")
resultPacket = *MakeInterfacePacket(feedbackCMD, 101)
return resultPacket, err
}
//3.打包...................................
var returnCMD = &business.Command{
Sender: proto.Uint32(0),
Groupid: proto.Uint64(dialogid),
Result: proto.Uint32(0),
ResultDescription: proto.String("OK"),
}
resultPacket = *MakeInterfacePacket(returnCMD, 101)
return resultPacket, nil
}