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.
73 lines
2.2 KiB
73 lines
2.2 KiB
package apiChat
|
|
|
|
import (
|
|
"Open_IM/pkg/common/config"
|
|
"Open_IM/pkg/common/log"
|
|
"Open_IM/pkg/grpc-etcdv3/getcdv3"
|
|
"Open_IM/pkg/proto/chat"
|
|
"Open_IM/pkg/utils"
|
|
"context"
|
|
"github.com/gin-gonic/gin"
|
|
"net/http"
|
|
"strings"
|
|
)
|
|
|
|
type paramsUserPullMsg struct {
|
|
ReqIdentifier *int `json:"reqIdentifier" binding:"required"`
|
|
SendID string `json:"sendID" binding:"required"`
|
|
OperationID string `json:"operationID" binding:"required"`
|
|
Data struct {
|
|
SeqBegin *int64 `json:"seqBegin" binding:"required"`
|
|
SeqEnd *int64 `json:"seqEnd" binding:"required"`
|
|
}
|
|
}
|
|
|
|
func UserPullMsg(c *gin.Context) {
|
|
params := paramsUserPullMsg{}
|
|
if err := c.BindJSON(¶ms); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"errCode": 400, "errMsg": err.Error()})
|
|
return
|
|
}
|
|
|
|
token := c.Request.Header.Get("token")
|
|
if !utils.VerifyToken(token, params.SendID) {
|
|
c.JSON(http.StatusBadRequest, gin.H{"errCode": 400, "errMsg": "token validate err"})
|
|
return
|
|
}
|
|
pbData := pbChat.PullMessageReq{}
|
|
pbData.UserID = params.SendID
|
|
pbData.OperationID = params.OperationID
|
|
pbData.SeqBegin = *params.Data.SeqBegin
|
|
pbData.SeqEnd = *params.Data.SeqEnd
|
|
grpcConn := getcdv3.GetConn(config.Config.Etcd.EtcdSchema, strings.Join(config.Config.Etcd.EtcdAddr, ","), config.Config.RpcRegisterName.OpenImOfflineMessageName)
|
|
msgClient := pbChat.NewChatClient(grpcConn)
|
|
reply, err := msgClient.PullMessage(context.Background(), &pbData)
|
|
if err != nil {
|
|
log.ErrorByKv("PullMessage error", pbData.OperationID, "err", err.Error())
|
|
return
|
|
}
|
|
log.InfoByKv("rpc call success to pullMsgRep", pbData.OperationID, "ReplyArgs", reply.String(), "maxSeq", reply.GetMaxSeq(),
|
|
"MinSeq", reply.GetMinSeq(), "singLen", len(reply.GetSingleUserMsg()), "groupLen", len(reply.GetGroupUserMsg()))
|
|
|
|
msg := make(map[string]interface{})
|
|
if v := reply.GetSingleUserMsg(); v != nil {
|
|
msg["single"] = v
|
|
} else {
|
|
msg["single"] = []pbChat.GatherFormat{}
|
|
}
|
|
if v := reply.GetGroupUserMsg(); v != nil {
|
|
msg["group"] = v
|
|
} else {
|
|
msg["group"] = []pbChat.GatherFormat{}
|
|
}
|
|
msg["maxSeq"] = reply.GetMaxSeq()
|
|
msg["minSeq"] = reply.GetMinSeq()
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"errCode": reply.ErrCode,
|
|
"errMsg": reply.ErrMsg,
|
|
"reqIdentifier": *params.ReqIdentifier,
|
|
"data": msg,
|
|
})
|
|
|
|
}
|