You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
112 lines
3.9 KiB
112 lines
3.9 KiB
package messageCMS
|
|
|
|
import (
|
|
"Open_IM/pkg/cms_api_struct"
|
|
"Open_IM/pkg/common/config"
|
|
openIMHttp "Open_IM/pkg/common/http"
|
|
"Open_IM/pkg/common/log"
|
|
"Open_IM/pkg/grpc-etcdv3/getcdv3"
|
|
pbMessage "Open_IM/pkg/proto/message_cms"
|
|
pbCommon "Open_IM/pkg/proto/sdk_ws"
|
|
"Open_IM/pkg/utils"
|
|
"context"
|
|
"strings"
|
|
|
|
"Open_IM/pkg/common/constant"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func BroadcastMessage(c *gin.Context) {
|
|
var (
|
|
reqPb pbMessage.BoradcastMessageReq
|
|
)
|
|
etcdConn := getcdv3.GetConn(config.Config.Etcd.EtcdSchema, strings.Join(config.Config.Etcd.EtcdAddr, ","), config.Config.RpcRegisterName.OpenImMessageCMSName)
|
|
client := pbMessage.NewMessageCMSClient(etcdConn)
|
|
_, err := client.BoradcastMessage(context.Background(), &reqPb)
|
|
if err != nil {
|
|
log.NewError(reqPb.OperationID, utils.GetSelfFuncName(), "GetChatLogs rpc failed", err.Error())
|
|
openIMHttp.RespHttp200(c, err, nil)
|
|
return
|
|
}
|
|
openIMHttp.RespHttp200(c, constant.OK, nil)
|
|
}
|
|
|
|
func MassSendMassage(c *gin.Context) {
|
|
var (
|
|
reqPb pbMessage.MassSendMessageReq
|
|
)
|
|
etcdConn := getcdv3.GetConn(config.Config.Etcd.EtcdSchema, strings.Join(config.Config.Etcd.EtcdAddr, ","), config.Config.RpcRegisterName.OpenImMessageCMSName)
|
|
client := pbMessage.NewMessageCMSClient(etcdConn)
|
|
_, err := client.MassSendMessage(context.Background(), &reqPb)
|
|
if err != nil {
|
|
log.NewError(reqPb.OperationID, utils.GetSelfFuncName(), "GetChatLogs rpc failed", err.Error())
|
|
openIMHttp.RespHttp200(c, err, nil)
|
|
return
|
|
}
|
|
openIMHttp.RespHttp200(c, constant.OK, nil)
|
|
}
|
|
|
|
func WithdrawMessage(c *gin.Context) {
|
|
var (
|
|
reqPb pbMessage.WithdrawMessageReq
|
|
)
|
|
etcdConn := getcdv3.GetConn(config.Config.Etcd.EtcdSchema, strings.Join(config.Config.Etcd.EtcdAddr, ","), config.Config.RpcRegisterName.OpenImMessageCMSName)
|
|
client := pbMessage.NewMessageCMSClient(etcdConn)
|
|
_, err := client.WithdrawMessage(context.Background(), &reqPb)
|
|
if err != nil {
|
|
log.NewError(reqPb.OperationID, utils.GetSelfFuncName(), "GetChatLogs rpc failed", err.Error())
|
|
openIMHttp.RespHttp200(c, err, nil)
|
|
return
|
|
}
|
|
openIMHttp.RespHttp200(c, constant.OK, nil)
|
|
}
|
|
|
|
func GetChatLogs(c *gin.Context) {
|
|
var (
|
|
req cms_api_struct.GetChatLogsRequest
|
|
resp cms_api_struct.GetChatLogsResponse
|
|
reqPb pbMessage.GetChatLogsReq
|
|
)
|
|
if err := c.ShouldBindQuery(&req); err != nil {
|
|
log.NewError(reqPb.OperationID, utils.GetSelfFuncName(), "ShouldBindQuery failed ", err.Error())
|
|
openIMHttp.RespHttp200(c, constant.ErrArgs, resp)
|
|
return
|
|
}
|
|
reqPb.Pagination = &pbCommon.RequestPagination{
|
|
PageNumber: int32(req.PageNumber),
|
|
ShowNumber: int32(req.ShowNumber),
|
|
}
|
|
utils.CopyStructFields(&reqPb, &req)
|
|
log.NewInfo("", utils.GetSelfFuncName(), "req: ", req)
|
|
etcdConn := getcdv3.GetConn(config.Config.Etcd.EtcdSchema, strings.Join(config.Config.Etcd.EtcdAddr, ","), config.Config.RpcRegisterName.OpenImMessageCMSName)
|
|
client := pbMessage.NewMessageCMSClient(etcdConn)
|
|
respPb, err := client.GetChatLogs(context.Background(), &reqPb)
|
|
if err != nil {
|
|
log.NewError(reqPb.OperationID, utils.GetSelfFuncName(), "GetChatLogs rpc failed", err.Error())
|
|
openIMHttp.RespHttp200(c, err, resp)
|
|
return
|
|
}
|
|
//utils.CopyStructFields(&resp, &respPb)
|
|
for _, chatLog := range respPb.ChatLogs {
|
|
resp.ChatLogs = append(resp.ChatLogs, cms_api_struct.ChatLog{
|
|
SessionType: int(chatLog.SessionType),
|
|
ContentType: int(chatLog.ContentType),
|
|
SenderNickName: chatLog.SenderNickName,
|
|
SenderId: chatLog.SenderId,
|
|
SearchContent: chatLog.SearchContent,
|
|
WholeContent: chatLog.WholeContent,
|
|
ReceiverNickName: chatLog.ReciverNickName,
|
|
ReceiverID: chatLog.ReciverId,
|
|
GroupName: chatLog.GroupName,
|
|
GroupId: chatLog.GroupId,
|
|
Date: chatLog.Date,
|
|
})
|
|
}
|
|
resp.ShowNumber = int(respPb.Pagination.ShowNumber)
|
|
resp.CurrentPage = int(respPb.Pagination.CurrentPage)
|
|
resp.ChatLogsNum = int(respPb.ChatLogsNum)
|
|
log.NewInfo("", utils.GetSelfFuncName(), "resp", resp)
|
|
openIMHttp.RespHttp200(c, constant.OK, resp)
|
|
}
|