organization

internal/rpc/office/office.go

@@ -20,6 +20,7 @@ import (
 	"strings"
 	"sync"
 	"time"
+	"unsafe"
 )
 
 type officeServer struct {
@@ -213,6 +214,12 @@ func (s *officeServer) SendMsg2Tag(_ context.Context, req *pbOffice.SendMsg2TagR
 			userIDList = append(userIDList[:i], userIDList[i+1:]...)
 		}
 	}
+	if unsafe.Sizeof(userIDList) > 1024*1024 {
+		log.NewDebug(req.OperationID, utils.GetSelfFuncName(), "size", unsafe.Sizeof(userIDList))
+		resp.CommonResp.ErrMsg = constant.ErrSendLimit.ErrMsg
+		resp.CommonResp.ErrCode = constant.ErrSendLimit.ErrCode
+		return
+	}
 	log.NewDebug(req.OperationID, utils.GetSelfFuncName(), "total userIDList result: ", userIDList)
 	user, err := imdb.GetUserByUserID(req.SendID)
 	if err != nil {

pkg/common/constant/error.go

@@ -54,6 +54,7 @@ var (
 	ErrArgs      = ErrInfo{ErrCode: 803, ErrMsg: ArgsMsg.Error()}
 	ErrStatus    = ErrInfo{ErrCode: 804, ErrMsg: StatusMsg.Error()}
 	ErrCallback  = ErrInfo{ErrCode: 809, ErrMsg: CallBackMsg.Error()}
+	ErrSendLimit = ErrInfo{ErrCode: 810, ErrMsg: "send msg limit, to many users"}
 )
 
 var (