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

office tag async send

Browse Source
This commit is contained in:
wangchuxiao 2022-05-18 15:48:42 +08:00
parent b0e349212d
commit 18a61061e1
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
internal/rpc/office/office.go
View File

@ -33,7 +33,7 @@ type officeServer struct {
func NewOfficeServer(port int) *officeServer { func NewOfficeServer(port int) *officeServer {
log.NewPrivateLog(constant.LogFileName) log.NewPrivateLog(constant.LogFileName)
ch := make(chan tagSendStruct, 10000) ch := make(chan tagSendStruct, 100000)
return &officeServer{ return &officeServer{
rpcPort: port, rpcPort: port,
rpcRegisterName: config.Config.RpcRegisterName.OpenImOfficeName, rpcRegisterName: config.Config.RpcRegisterName.OpenImOfficeName,
@ -106,7 +106,7 @@ func (s *officeServer) sendTagMsgRoutine() {
select { select {
case v := <-s.ch: case v := <-s.ch:
msg.TagSendMessage(v.operationID, v.user, v.userID, v.content, v.senderPlatformID) msg.TagSendMessage(v.operationID, v.user, v.userID, v.content, v.senderPlatformID)
time.Sleep(time.Millisecond * 500) time.Sleep(time.Millisecond * 100)
} }
} }
} }
@ -251,7 +251,7 @@ func (s *officeServer) SendMsg2Tag(_ context.Context, req *pbOffice.SendMsg2TagR
resp.CommonResp.ErrCode = constant.ErrDB.ErrCode resp.CommonResp.ErrCode = constant.ErrDB.ErrCode
return resp, nil return resp, nil
} }
var successUserIDList []string
for _, userID := range userIDList { for _, userID := range userIDList {
t := tagSendStruct{ t := tagSendStruct{
operationID: req.OperationID, operationID: req.OperationID,
@ -263,6 +263,7 @@ func (s *officeServer) SendMsg2Tag(_ context.Context, req *pbOffice.SendMsg2TagR
select { select {
case s.ch <- t: case s.ch <- t:
log.NewDebug(t.operationID, utils.GetSelfFuncName(), "msg: ", t, "send success") log.NewDebug(t.operationID, utils.GetSelfFuncName(), "msg: ", t, "send success")
successUserIDList = append(successUserIDList, userID)
// if channel is full, return grpc req // if channel is full, return grpc req
case <-time.After(1 * time.Second): case <-time.After(1 * time.Second):
log.NewError(t.operationID, utils.GetSelfFuncName(), s.ch, "channel is full") log.NewError(t.operationID, utils.GetSelfFuncName(), s.ch, "channel is full")
@ -274,9 +275,9 @@ func (s *officeServer) SendMsg2Tag(_ context.Context, req *pbOffice.SendMsg2TagR
var tagSendLogs db.TagSendLog var tagSendLogs db.TagSendLog
var wg sync.WaitGroup var wg sync.WaitGroup
wg.Add(len(userIDList)) wg.Add(len(successUserIDList))
var lock sync.Mutex var lock sync.Mutex
for _, userID := range userIDList { for _, userID := range successUserIDList {
go func(userID string) { go func(userID string) {
defer wg.Done() defer wg.Done()
userName, err := im_mysql_model.GetUserNameByUserID(userID) userName, err := im_mysql_model.GetUserNameByUserID(userID)