new feature: add batch send msg

internal/api/msg.go

@@ -252,14 +252,24 @@ func (m *MessageApi) BatchSendMsg(c *gin.Context) {
 	var recvIDs []string
 	var err error
 	if req.IsSendAll {
-		recvIDs, err = m.userRpcClient.GetAllUserIDs(c)
+		pageNumber := 1
-		if err != nil {
+		showNumber := 100
-			log.ZError(c, "GetAllUserIDs failed", err)
+		for {
-			apiresp.GinError(c, err)
+			recvIDsPart, err := m.userRpcClient.GetAllUserIDs(c, int32(pageNumber), int32(showNumber))
+			if err != nil {
+				log.ZError(c, "GetAllUserIDs failed", err)
+				apiresp.GinError(c, err)
+			}
+			if len(recvIDsPart) < showNumber {
+				recvIDs = append(recvIDs, recvIDsPart...)
+				break
+			}
+			pageNumber++
 		}
 	} else {
 		recvIDs = req.RecvIDs
 	}
+	log.ZDebug(c, "BatchSendMsg nums", "nums ", len(recvIDs))
 	sendMsgReq, err := m.getSendMsgReq(c, req.SendMsg)
 	if err != nil {
 		log.ZError(c, "decodeData failed", err)

internal/rpc/user/user.go

@@ -17,10 +17,11 @@ package user
 import (
 	"context"
 	"errors"
-	"github.com/OpenIMSDK/Open-IM-Server/pkg/common/log"
 	"strings"
 	"time"
 
+	"github.com/OpenIMSDK/Open-IM-Server/pkg/common/log"
+
 	"github.com/OpenIMSDK/Open-IM-Server/pkg/common/config"
 	"github.com/OpenIMSDK/Open-IM-Server/pkg/common/constant"
 	"github.com/OpenIMSDK/Open-IM-Server/pkg/common/convert"
@@ -233,7 +234,7 @@ func (s *userServer) GetGlobalRecvMessageOpt(ctx context.Context, req *pbuser.Ge
 }
 
 func (s *userServer) GetAllUserID(ctx context.Context, req *pbuser.GetAllUserIDReq) (resp *pbuser.GetAllUserIDResp, err error) {
-	userIDs, err := s.UserDatabase.GetAllUserID(ctx)
+	userIDs, err := s.UserDatabase.GetAllUserID(ctx, req.Pagination.PageNumber, req.Pagination.ShowNumber)
 	if err != nil {
 		return nil, err
 	}

pkg/common/db/controller/user.go

@@ -41,7 +41,7 @@ type UserDatabase interface {
 	//只要有一个存在就为true
 	IsExist(ctx context.Context, userIDs []string) (exist bool, err error)
 	//获取所有用户ID
-	GetAllUserID(ctx context.Context) ([]string, error)
+	GetAllUserID(ctx context.Context, pageNumber, showNumber int32) ([]string, error)
 	//函数内部先查询db中是否存在，存在则什么都不做；不存在则插入
 	InitOnce(ctx context.Context, users []*relation.UserModel) (err error)
 	// 获取用户总数
@@ -147,8 +147,8 @@ func (u *userDatabase) IsExist(ctx context.Context, userIDs []string) (exist boo
 	return false, nil
 }
 
-func (u *userDatabase) GetAllUserID(ctx context.Context) (userIDs []string, err error) {
+func (u *userDatabase) GetAllUserID(ctx context.Context, pageNumber, showNumber int32) (userIDs []string, err error) {
-	return u.userDB.GetAllUserID(ctx)
+	return u.userDB.GetAllUserID(ctx, pageNumber, showNumber)
 }
 
 func (u *userDatabase) CountTotal(ctx context.Context, before *time.Time) (count int64, err error) {

pkg/common/db/relation/user_model.go

@@ -84,8 +84,8 @@ func (u *UserGorm) Page(
 }
 
 // 获取所有用户ID
-func (u *UserGorm) GetAllUserID(ctx context.Context) (userIDs []string, err error) {
+func (u *UserGorm) GetAllUserID(ctx context.Context, pageNumber, showNumber int32) (userIDs []string, err error) {
-	return userIDs, errs.Wrap(u.db(ctx).Pluck("user_id", &userIDs).Error)
+	return userIDs, errs.Wrap(u.db(ctx).Limit(int(showNumber)).Offset(int((pageNumber-1)*showNumber)).Pluck("user_id", &userIDs).Error)
 }
 
 func (u *UserGorm) GetUserGlobalRecvMsgOpt(ctx context.Context, userID string) (opt int, err error) {

pkg/common/db/table/relation/user.go

@@ -63,7 +63,7 @@ type UserModelInterface interface {
 	Take(ctx context.Context, userID string) (user *UserModel, err error)
 	// 获取用户信息 不存在，不返回错误
 	Page(ctx context.Context, pageNumber, showNumber int32) (users []*UserModel, count int64, err error)
-	GetAllUserID(ctx context.Context) (userIDs []string, err error)
+	GetAllUserID(ctx context.Context, pageNumber, showNumber int32) (userIDs []string, err error)
 	GetUserGlobalRecvMsgOpt(ctx context.Context, userID string) (opt int, err error)
 	// 获取用户总数
 	CountTotal(ctx context.Context, before *time.Time) (count int64, err error)

pkg/rpcclient/user.go

@@ -147,8 +147,8 @@ func (u *UserRpcClient) Access(ctx context.Context, ownerUserID string) error {
 	return tokenverify.CheckAccessV3(ctx, ownerUserID)
 }
 
-func (u *UserRpcClient) GetAllUserIDs(ctx context.Context) ([]string, error) {
+func (u *UserRpcClient) GetAllUserIDs(ctx context.Context, pageNumber, showNumber int32) ([]string, error) {
-	resp, err := u.Client.GetAllUserID(ctx, &user.GetAllUserIDReq{})
+	resp, err := u.Client.GetAllUserID(ctx, &user.GetAllUserIDReq{Pagination: &sdkws.RequestPagination{PageNumber: pageNumber, ShowNumber: showNumber}})
 	if err != nil {
 		return nil, err
 	}