From 2454c503807978c70362a32f5c229849239002f3 Mon Sep 17 00:00:00 2001
From: wangchuxiao <wangchuxiao97@outlook.com>
Date: Tue, 17 May 2022 20:42:43 +0800
Subject: [PATCH] organization

---
 internal/rpc/office/office.go |  7 +++++++
 pkg/common/constant/error.go  | 11 ++++++-----
 2 files changed, 13 insertions(+), 5 deletions(-)

diff --git a/internal/rpc/office/office.go b/internal/rpc/office/office.go
index 10d09c938..1cade65eb 100644
--- a/internal/rpc/office/office.go
+++ b/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 {
diff --git a/pkg/common/constant/error.go b/pkg/common/constant/error.go
index c8aca5fe6..2537c3706 100644
--- a/pkg/common/constant/error.go
+++ b/pkg/common/constant/error.go
@@ -49,11 +49,12 @@ var (
 	ErrTokenUnknown     = ErrInfo{705, TokenUnknownMsg.Error()}
 	ErrTokenKicked      = ErrInfo{706, TokenUserKickedMsg.Error()}
 
-	ErrAccess   = ErrInfo{ErrCode: 801, ErrMsg: AccessMsg.Error()}
-	ErrDB       = ErrInfo{ErrCode: 802, ErrMsg: DBMsg.Error()}
-	ErrArgs     = ErrInfo{ErrCode: 803, ErrMsg: ArgsMsg.Error()}
-	ErrStatus   = ErrInfo{ErrCode: 804, ErrMsg: StatusMsg.Error()}
-	ErrCallback = ErrInfo{ErrCode: 809, ErrMsg: CallBackMsg.Error()}
+	ErrAccess    = ErrInfo{ErrCode: 801, ErrMsg: AccessMsg.Error()}
+	ErrDB        = ErrInfo{ErrCode: 802, ErrMsg: DBMsg.Error()}
+	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 (