blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-04-26 03:26:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

user update

Browse Source
This commit is contained in:
Gordon 2023-03-20 19:34:50 +08:00
parent 99c486295b
commit 78d5fa30ba
4 changed files with 54 additions and 46 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

76
internal/api/msg.go
View File

@ -2,7 +2,6 @@ package api
import ( import (
"context" "context"
"errors"
"github.com/OpenIMSDK/Open-IM-Server/pkg/a2r" "github.com/OpenIMSDK/Open-IM-Server/pkg/a2r"
"github.com/OpenIMSDK/Open-IM-Server/pkg/apiresp" "github.com/OpenIMSDK/Open-IM-Server/pkg/apiresp"
"github.com/OpenIMSDK/Open-IM-Server/pkg/apistruct" "github.com/OpenIMSDK/Open-IM-Server/pkg/apistruct"
@ -23,23 +22,23 @@ import (
var _ context.Context // 解决goland编辑器bug var _ context.Context // 解决goland编辑器bug
func NewMsg(c discoveryregistry.SvcDiscoveryRegistry) *Msg { func NewMsg(c discoveryregistry.SvcDiscoveryRegistry) *Message {
return &Msg{c: c, validate: validator.New()} return &Message{c: c, validate: validator.New()}
} }
type Msg struct { type Message struct {
c discoveryregistry.SvcDiscoveryRegistry c discoveryregistry.SvcDiscoveryRegistry
validate *validator.Validate validate *validator.Validate
} }
func (Msg) SetOptions(options map[string]bool, value bool) { func (Message) SetOptions(options map[string]bool, value bool) {
utils.SetSwitchFromOptions(options, constant.IsHistory, value) utils.SetSwitchFromOptions(options, constant.IsHistory, value)
utils.SetSwitchFromOptions(options, constant.IsPersistent, value) utils.SetSwitchFromOptions(options, constant.IsPersistent, value)
utils.SetSwitchFromOptions(options, constant.IsSenderSync, value) utils.SetSwitchFromOptions(options, constant.IsSenderSync, value)
utils.SetSwitchFromOptions(options, constant.IsConversationUpdate, value) utils.SetSwitchFromOptions(options, constant.IsConversationUpdate, value)
} }
func (m Msg) newUserSendMsgReq(c *gin.Context, params *apistruct.ManagementSendMsgReq) *msg.SendMsgReq { func (m Message) newUserSendMsgReq(c *gin.Context, params *apistruct.ManagementSendMsgReq) *msg.SendMsgReq {
var newContent string var newContent string
var err error var err error
switch params.ContentType { switch params.ContentType {
@ -106,7 +105,7 @@ func (m Msg) newUserSendMsgReq(c *gin.Context, params *apistruct.ManagementSendM
return &pbData return &pbData
} }
func (m *Msg) client() (msg.MsgClient, error) { func (m *Message) client() (msg.MsgClient, error) {
conn, err := m.c.GetConn(config.Config.RpcRegisterName.OpenImMsgName) conn, err := m.c.GetConn(config.Config.RpcRegisterName.OpenImMsgName)
if err != nil { if err != nil {
return nil, err return nil, err
@ -114,46 +113,46 @@ func (m *Msg) client() (msg.MsgClient, error) {
return msg.NewMsgClient(conn), nil return msg.NewMsgClient(conn), nil
} }
func (m *Msg) GetSeq(c *gin.Context) { func (m *Message) GetSeq(c *gin.Context) {
a2r.Call(msg.MsgClient.GetMaxAndMinSeq, m.client, c) a2r.Call(msg.MsgClient.GetMaxAndMinSeq, m.client, c)
} }
func (m *Msg) PullMsgBySeqs(c *gin.Context) { func (m *Message) PullMsgBySeqs(c *gin.Context) {
a2r.Call(msg.MsgClient.PullMessageBySeqs, m.client, c) a2r.Call(msg.MsgClient.PullMessageBySeqs, m.client, c)
} }
func (m *Msg) DelMsg(c *gin.Context) { func (m *Message) DelMsg(c *gin.Context) {
a2r.Call(msg.MsgClient.DelMsgs, m.client, c) a2r.Call(msg.MsgClient.DelMsgs, m.client, c)
} }
func (m *Msg) DelSuperGroupMsg(c *gin.Context) { func (m *Message) DelSuperGroupMsg(c *gin.Context) {
a2r.Call(msg.MsgClient.DelSuperGroupMsg, m.client, c) a2r.Call(msg.MsgClient.DelSuperGroupMsg, m.client, c)
} }
func (m *Msg) ClearMsg(c *gin.Context) { func (m *Message) ClearMsg(c *gin.Context) {
a2r.Call(msg.MsgClient.ClearMsg, m.client, c) a2r.Call(msg.MsgClient.ClearMsg, m.client, c)
} }
func (m *Msg) SetMessageReactionExtensions(c *gin.Context) { func (m *Message) SetMessageReactionExtensions(c *gin.Context) {
a2r.Call(msg.MsgClient.SetMessageReactionExtensions, m.client, c) a2r.Call(msg.MsgClient.SetMessageReactionExtensions, m.client, c)
} }
func (m *Msg) GetMessageListReactionExtensions(c *gin.Context) { func (m *Message) GetMessageListReactionExtensions(c *gin.Context) {
a2r.Call(msg.MsgClient.GetMessagesReactionExtensions, m.client, c) a2r.Call(msg.MsgClient.GetMessagesReactionExtensions, m.client, c)
} }
func (m *Msg) AddMessageReactionExtensions(c *gin.Context) { func (m *Message) AddMessageReactionExtensions(c *gin.Context) {
a2r.Call(msg.MsgClient.AddMessageReactionExtensions, m.client, c) a2r.Call(msg.MsgClient.AddMessageReactionExtensions, m.client, c)
} }
func (m *Msg) DeleteMessageReactionExtensions(c *gin.Context) { func (m *Message) DeleteMessageReactionExtensions(c *gin.Context) {
a2r.Call(msg.MsgClient.DeleteMessageReactionExtensions, m.client, c) a2r.Call(msg.MsgClient.DeleteMessageReactionExtensions, m.client, c)
} }
func (m *Msg) SendMsg(c *gin.Context) { func (m *Message) SendMessage(c *gin.Context) {
params := apistruct.ManagementSendMsgReq{} params := apistruct.ManagementSendMsgReq{}
if err := c.BindJSON(&params); err != nil { if err := c.BindJSON(&params); err != nil {
apiresp.GinError(c, err) apiresp.GinError(c, errs.ErrArgs.WithDetail(err.Error()).Wrap())
return return
} }
var data interface{} var data interface{}
@ -181,32 +180,29 @@ func (m *Msg) SendMsg(c *gin.Context) {
data = apistruct.CustomElem{} data = apistruct.CustomElem{}
case constant.CustomOnlineOnly: case constant.CustomOnlineOnly:
data = apistruct.CustomElem{} data = apistruct.CustomElem{}
//case constant.HasReadReceipt:
//case constant.Typing:
//case constant.Quote:
default: default:
apiresp.GinError(c, errors.New("wrong contentType")) apiresp.GinError(c, errs.ErrArgs.WithDetail("not support err contentType").Wrap())
return return
} }
if err := mapstructure.WeakDecode(params.Content, &data); err != nil { if err := mapstructure.WeakDecode(params.Content, &data); err != nil {
apiresp.GinError(c, errs.ErrData) apiresp.GinError(c, errs.ErrArgs.Wrap(err.Error()))
return return
} else if err := m.validate.Struct(data); err != nil { } else if err := m.validate.Struct(data); err != nil {
apiresp.GinError(c, errs.ErrData) apiresp.GinError(c, errs.ErrArgs.Wrap(err.Error()))
return return
} }
switch params.SessionType { //switch params.SessionType {
case constant.SingleChatType: //case constant.SingleChatType:
if len(params.RecvID) == 0 { // if len(params.RecvID) == 0 {
apiresp.GinError(c, errs.ErrData) // apiresp.GinError(c, errs.ErrData)
return // return
} // }
case constant.GroupChatType, constant.SuperGroupChatType: //case constant.GroupChatType, constant.SuperGroupChatType:
if len(params.GroupID) == 0 { // if len(params.GroupID) == 0 {
apiresp.GinError(c, errs.ErrData) // apiresp.GinError(c, errs.ErrData)
return // return
} // }
} //}
pbReq := m.newUserSendMsgReq(c, &params) pbReq := m.newUserSendMsgReq(c, &params)
conn, err := m.c.GetConn(config.Config.RpcRegisterName.OpenImMsgName) conn, err := m.c.GetConn(config.Config.RpcRegisterName.OpenImMsgName)
if err != nil { if err != nil {
@ -232,18 +228,18 @@ func (m *Msg) SendMsg(c *gin.Context) {
apiresp.GinSuccess(c, resp) apiresp.GinSuccess(c, resp)
} }
func (m *Msg) ManagementBatchSendMsg(c *gin.Context) { func (m *Message) ManagementBatchSendMsg(c *gin.Context) {
a2r.Call(msg.MsgClient.SendMsg, m.client, c) a2r.Call(msg.MsgClient.SendMsg, m.client, c)
} }
func (m *Msg) CheckMsgIsSendSuccess(c *gin.Context) { func (m *Message) CheckMsgIsSendSuccess(c *gin.Context) {
a2r.Call(msg.MsgClient.GetSendMsgStatus, m.client, c) a2r.Call(msg.MsgClient.GetSendMsgStatus, m.client, c)
} }
func (m *Msg) GetUsersOnlineStatus(c *gin.Context) { func (m *Message) GetUsersOnlineStatus(c *gin.Context) {
a2r.Call(msg.MsgClient.GetSendMsgStatus, m.client, c) a2r.Call(msg.MsgClient.GetSendMsgStatus, m.client, c)
} }
func (m *Msg) AccountCheck(c *gin.Context) { func (m *Message) AccountCheck(c *gin.Context) {
a2r.Call(msg.MsgClient.GetSendMsgStatus, m.client, c) a2r.Call(msg.MsgClient.GetSendMsgStatus, m.client, c)
} }

2
internal/api/route.go
View File

@ -126,7 +126,7 @@ func NewGinRouter(zk discoveryregistry.SvcDiscoveryRegistry, rdb redis.Universal
m := NewMsg(zk) m := NewMsg(zk)
msgGroup.Use(mw.GinParseToken(rdb)) msgGroup.Use(mw.GinParseToken(rdb))
msgGroup.POST("/newest_seq", m.GetSeq) msgGroup.POST("/newest_seq", m.GetSeq)
msgGroup.POST("/send_msg", m.SendMsg) msgGroup.POST("/send_msg", m.SendMessage)
msgGroup.POST("/pull_msg_by_seq", m.PullMsgBySeqs) msgGroup.POST("/pull_msg_by_seq", m.PullMsgBySeqs)
msgGroup.POST("/del_msg", m.DelMsg) msgGroup.POST("/del_msg", m.DelMsg)
msgGroup.POST("/del_super_group_msg", m.DelSuperGroupMsg) msgGroup.POST("/del_super_group_msg", m.DelSuperGroupMsg)

6
internal/rpc/msg/send_pull.go
View File

@ -3,8 +3,8 @@ package msg
import ( import (
"context" "context"
"github.com/OpenIMSDK/Open-IM-Server/pkg/common/constant" "github.com/OpenIMSDK/Open-IM-Server/pkg/common/constant"
"github.com/OpenIMSDK/Open-IM-Server/pkg/common/log"
promePkg "github.com/OpenIMSDK/Open-IM-Server/pkg/common/prome" promePkg "github.com/OpenIMSDK/Open-IM-Server/pkg/common/prome"
"github.com/OpenIMSDK/Open-IM-Server/pkg/common/tokenverify"
"github.com/OpenIMSDK/Open-IM-Server/pkg/errs" "github.com/OpenIMSDK/Open-IM-Server/pkg/errs"
pbConversation "github.com/OpenIMSDK/Open-IM-Server/pkg/proto/conversation" pbConversation "github.com/OpenIMSDK/Open-IM-Server/pkg/proto/conversation"
"github.com/OpenIMSDK/Open-IM-Server/pkg/proto/msg" "github.com/OpenIMSDK/Open-IM-Server/pkg/proto/msg"
@ -259,6 +259,9 @@ func (m *msgServer) SendMsg(ctx context.Context, req *msg.SendMsgReq) (resp *msg
} }
func (m *msgServer) GetMaxAndMinSeq(ctx context.Context, req *sdkws.GetMaxAndMinSeqReq) (*sdkws.GetMaxAndMinSeqResp, error) { func (m *msgServer) GetMaxAndMinSeq(ctx context.Context, req *sdkws.GetMaxAndMinSeqReq) (*sdkws.GetMaxAndMinSeqResp, error) {
if err := tokenverify.CheckAccessV3(ctx, req.UserID); err != nil {
return nil, err
}
resp := new(sdkws.GetMaxAndMinSeqResp) resp := new(sdkws.GetMaxAndMinSeqResp)
m2 := make(map[string]*sdkws.MaxAndMinSeq) m2 := make(map[string]*sdkws.MaxAndMinSeq)
maxSeq, err := m.MsgDatabase.GetUserMaxSeq(ctx, req.UserID) maxSeq, err := m.MsgDatabase.GetUserMaxSeq(ctx, req.UserID)
@ -275,7 +278,6 @@ func (m *msgServer) GetMaxAndMinSeq(ctx context.Context, req *sdkws.GetMaxAndMin
for _, groupID := range req.GroupIDs { for _, groupID := range req.GroupIDs {
maxSeq, err := m.MsgDatabase.GetGroupMaxSeq(ctx, groupID) maxSeq, err := m.MsgDatabase.GetGroupMaxSeq(ctx, groupID)
if err != nil && errs.Unwrap(err) != redis.Nil { if err != nil && errs.Unwrap(err) != redis.Nil {
log.ZDebug(ctx, "redis err", "err", err.Error())
return nil, err return nil, err
} }
minSeq, err := m.MsgDatabase.GetGroupMinSeq(ctx, groupID) minSeq, err := m.MsgDatabase.GetGroupMinSeq(ctx, groupID)

16
pkg/apistruct/manage.go
View File

@ -34,7 +34,7 @@ type AccountCheckResp struct {
type ManagementSendMsg struct { type ManagementSendMsg struct {
SendID string `json:"sendID" binding:"required"` SendID string `json:"sendID" binding:"required"`
GroupID string `json:"groupID" ` GroupID string `json:"groupID" binding:"required_if=SessionType 2|required_if=SessionType 3"`
SenderNickname string `json:"senderNickname" ` SenderNickname string `json:"senderNickname" `
SenderFaceURL string `json:"senderFaceURL" ` SenderFaceURL string `json:"senderFaceURL" `
SenderPlatformID int32 `json:"senderPlatformID"` SenderPlatformID int32 `json:"senderPlatformID"`
@ -47,8 +47,18 @@ type ManagementSendMsg struct {
} }
type ManagementSendMsgReq struct { type ManagementSendMsgReq struct {
ManagementSendMsg SendID string `json:"sendID" binding:"required"`
RecvID string `json:"recvID" ` RecvID string `json:"recvID" binding:"required_if=SessionType 1|required_if=SessionType 4"`
GroupID string `json:"groupID" binding:"required_if=SessionType 2|required_if=SessionType 3"`
SenderNickname string `json:"senderNickname" `
SenderFaceURL string `json:"senderFaceURL" `
SenderPlatformID int32 `json:"senderPlatformID"`
Content map[string]interface{} `json:"content" binding:"required" swaggerignore:"true"`
ContentType int32 `json:"contentType" binding:"required"`
SessionType int32 `json:"sessionType" binding:"required"`
IsOnlineOnly bool `json:"isOnlineOnly"`
NotOfflinePush bool `json:"notOfflinePush"`
OfflinePushInfo *sdkws.OfflinePushInfo `json:"offlinePushInfo"`
} }
type ManagementSendMsgResp struct { type ManagementSendMsgResp struct {