Merge a0ccbf18839d567358d2519ceec8d37a06be1932 into 8e824c7e8eb58265a6e4c32edb5ad2bed2f1be67

.golangci.yml

@@ -28,8 +28,6 @@ run:
   #   - util
   #   - .*~
   #   - api/swagger/docs
-
-  
   #   - server/docs
   #   - components/mnt/config/certs
   #   - logs

LICENSE

@@ -1,201 +1,35 @@
-                                 Apache License
-                           Version 2.0, January 2004
-                        http://www.apache.org/licenses/
+# Open Source License
 
-   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+OpenIM is licensed under the Apache License 2.0, with the following additional conditions:
 
-   1. Definitions.
+1. OpenIM may be utilized commercially, including as a backend service for other applications or as an application development platform for enterprises. 
+A commercial license must be obtained from the producer if:
+    - Under no circumstances may you operate a multi-tenant or multi-business environment using the OpenIM source code, whether or not you have modified the repository code. In other words, a single instance of OpenIM may not simultaneously serve multiple enterprises, nor may it serve multiple lines of business within the same enterprise.
+   - If you intend to operate in such a multi-tenant or multi-business manner, you must obtain a commercial license from the producer in advance.
 
-      "License" shall mean the terms and conditions for use, reproduction,
-      and distribution as defined by Sections 1 through 9 of this document.
+2. As a contributor, you should agree that:
 
-      "Licensor" shall mean the copyright owner or entity authorized by
-      the copyright owner that is granting the License.
+   a. The producer can adjust the open-source agreement to be more strict or more relaxed as deemed necessary.  
+   b. Your contributed code may be used for commercial purposes, including but not limited to its cloud business operations.
 
-      "Legal Entity" shall mean the union of the acting entity and all
-      other entities that control, are controlled by, or are under common
-      control with that entity. For the purposes of this definition,
-      "control" means (i) the power, direct or indirect, to cause the
-      direction or management of such entity, whether by contract or
-      otherwise, or (ii) ownership of fifty percent (50%) or more of the
-      outstanding shares, or (iii) beneficial ownership of such entity.
+Apart from the specific conditions mentioned above, all other rights and restrictions follow the Apache License 2.0. Detailed information about the Apache License 2.0 can be found at http://www.apache.org/licenses/LICENSE-2.0.
 
-      "You" (or "Your") shall mean an individual or Legal Entity
-      exercising permissions granted by this License.
 
-      "Source" form shall mean the preferred form for making modifications,
-      including but not limited to software source code, documentation
-      source, and configuration files.
+For any licensing-related questions or to obtain a commercial license, please contact contact@openim.io.
 
-      "Object" form shall mean any form resulting from mechanical
-      transformation or translation of a Source form, including but
-      not limited to compiled object code, generated documentation,
-      and conversions to other media types.
 
-      "Work" shall mean the work of authorship, whether in Source or
-      Object form, made available under the License, as indicated by a
-      copyright notice that is included in or attached to the work
-      (an example is provided in the Appendix below).
+© 2024 OpenIMSDK
 
-      "Derivative Works" shall mean any work, whether in Source or Object
-      form, that is based on (or derived from) the Work and for which the
-      editorial revisions, annotations, elaborations, or other modifications
-      represent, as a whole, an original work of authorship. For the purposes
-      of this License, Derivative Works shall not include works that remain
-      separable from, or merely link (or bind by name) to the interfaces of,
-      the Work and Derivative Works thereof.
+----------
 
-      "Contribution" shall mean any work of authorship, including
-      the original version of the Work and any modifications or additions
-      to that Work or Derivative Works thereof, that is intentionally
-      submitted to Licensor for inclusion in the Work by the copyright owner
-      or by an individual or Legal Entity authorized to submit on behalf of
-      the copyright owner. For the purposes of this definition, "submitted"
-      means any form of electronic, verbal, or written communication sent
-      to the Licensor or its representatives, including but not limited to
-      communication on electronic mailing lists, source code control systems,
-      and issue tracking systems that are managed by, or on behalf of, the
-      Licensor for the purpose of discussing and improving the Work, but
-      excluding communication that is conspicuously marked or otherwise
-      designated in writing by the copyright owner as "Not a Contribution."
+Licensed under the Apache License, Version 2.0 (the "License");  
+you may not use this file except in compliance with the License.  
+You may obtain a copy of the License at
 
-      "Contributor" shall mean Licensor and any individual or Legal Entity
-      on behalf of whom a Contribution has been received by Licensor and
-      subsequently incorporated within the Work.
+    http://www.apache.org/licenses/LICENSE-2.0
 
-   2. Grant of Copyright License. Subject to the terms and conditions of
-      this License, each Contributor hereby grants to You a perpetual,
-      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
-      copyright license to reproduce, prepare Derivative Works of,
-      publicly display, publicly perform, sublicense, and distribute the
-      Work and such Derivative Works in Source or Object form.
-
-   3. Grant of Patent License. Subject to the terms and conditions of
-      this License, each Contributor hereby grants to You a perpetual,
-      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
-      (except as stated in this section) patent license to make, have made,
-      use, offer to sell, sell, import, and otherwise transfer the Work,
-      where such license applies only to those patent claims licensable
-      by such Contributor that are necessarily infringed by their
-      Contribution(s) alone or by combination of their Contribution(s)
-      with the Work to which such Contribution(s) was submitted. If You
-      institute patent litigation against any entity (including a
-      cross-claim or counterclaim in a lawsuit) alleging that the Work
-      or a Contribution incorporated within the Work constitutes direct
-      or contributory patent infringement, then any patent licenses
-      granted to You under this License for that Work shall terminate
-      as of the date such litigation is filed.
-
-   4. Redistribution. You may reproduce and distribute copies of the
-      Work or Derivative Works thereof in any medium, with or without
-      modifications, and in Source or Object form, provided that You
-      meet the following conditions:
-
-      (a) You must give any other recipients of the Work or
-          Derivative Works a copy of this License; and
-
-      (b) You must cause any modified files to carry prominent notices
-          stating that You changed the files; and
-
-      (c) You must retain, in the Source form of any Derivative Works
-          that You distribute, all copyright, patent, trademark, and
-          attribution notices from the Source form of the Work,
-          excluding those notices that do not pertain to any part of
-          the Derivative Works; and
-
-      (d) If the Work includes a "NOTICE" text file as part of its
-          distribution, then any Derivative Works that You distribute must
-          include a readable copy of the attribution notices contained
-          within such NOTICE file, excluding those notices that do not
-          pertain to any part of the Derivative Works, in at least one
-          of the following places: within a NOTICE text file distributed
-          as part of the Derivative Works; within the Source form or
-          documentation, if provided along with the Derivative Works; or,
-          within a display generated by the Derivative Works, if and
-          wherever such third-party notices normally appear. The contents
-          of the NOTICE file are for informational purposes only and
-          do not modify the License. You may add Your own attribution
-          notices within Derivative Works that You distribute, alongside
-          or as an addendum to the NOTICE text from the Work, provided
-          that such additional attribution notices cannot be construed
-          as modifying the License.
-
-      You may add Your own copyright statement to Your modifications and
-      may provide additional or different license terms and conditions
-      for use, reproduction, or distribution of Your modifications, or
-      for any such Derivative Works as a whole, provided Your use,
-      reproduction, and distribution of the Work otherwise complies with
-      the conditions stated in this License.
-
-   5. Submission of Contributions. Unless You explicitly state otherwise,
-      any Contribution intentionally submitted for inclusion in the Work
-      by You to the Licensor shall be under the terms and conditions of
-      this License, without any additional terms or conditions.
-      Notwithstanding the above, nothing herein shall supersede or modify
-      the terms of any separate license agreement you may have executed
-      with Licensor regarding such Contributions.
-
-   6. Trademarks. This License does not grant permission to use the trade
-      names, trademarks, service marks, or product names of the Licensor,
-      except as required for reasonable and customary use in describing the
-      origin of the Work and reproducing the content of the NOTICE file.
-
-   7. Disclaimer of Warranty. Unless required by applicable law or
-      agreed to in writing, Licensor provides the Work (and each
-      Contributor provides its Contributions) on an "AS IS" BASIS,
-      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
-      implied, including, without limitation, any warranties or conditions
-      of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
-      PARTICULAR PURPOSE. You are solely responsible for determining the
-      appropriateness of using or redistributing the Work and assume any
-      risks associated with Your exercise of permissions under this License.
-
-   8. Limitation of Liability. In no event and under no legal theory,
-      whether in tort (including negligence), contract, or otherwise,
-      unless required by applicable law (such as deliberate and grossly
-      negligent acts) or agreed to in writing, shall any Contributor be
-      liable to You for damages, including any direct, indirect, special,
-      incidental, or consequential damages of any character arising as a
-      result of this License or out of the use or inability to use the
-      Work (including but not limited to damages for loss of goodwill,
-      work stoppage, computer failure or malfunction, or any and all
-      other commercial damages or losses), even if such Contributor
-      has been advised of the possibility of such damages.
-
-   9. Accepting Warranty or Additional Liability. While redistributing
-      the Work or Derivative Works thereof, You may choose to offer,
-      and charge a fee for, acceptance of support, warranty, indemnity,
-      or other liability obligations and/or rights consistent with this
-      License. However, in accepting such obligations, You may act only
-      on Your own behalf and on Your sole responsibility, not on behalf
-      of any other Contributor, and only if You agree to indemnify,
-      defend, and hold each Contributor harmless for any liability
-      incurred by, or claims asserted against, such Contributor by reason
-      of your accepting any such warranty or additional liability.
-
-   END OF TERMS AND CONDITIONS
-
-   APPENDIX: How to apply the Apache License to your work.
-
-      To apply the Apache License to your work, attach the following
-      boilerplate notice, with the fields enclosed by brackets "[]"
-      replaced with your own identifying information. (Don't include
-      the brackets!)  The text should be enclosed in the appropriate
-      comment syntax for the file format. We also recommend that a
-      file or class name and description of purpose be included on the
-      same "printed page" as the copyright notice for easier
-      identification within third-party archives.
-
-   Copyright [yyyy] [name of copyright owner]
-
-   Licensed under the Apache License, Version 2.0 (the "License");
-   you may not use this file except in compliance with the License.
-   You may obtain a copy of the License at
-
-       http://www.apache.org/licenses/LICENSE-2.0
-
-   Unless required by applicable law or agreed to in writing, software
-   distributed under the License is distributed on an "AS IS" BASIS,
-   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-   See the License for the specific language governing permissions and
-   limitations under the License.
+Unless required by applicable law or agreed to in writing, software  
+distributed under the License is distributed on an "AS IS" BASIS,  
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.  
+See the License for the specific language governing permissions and  
+limitations under the License.

README.md

@@ -132,7 +132,7 @@ Thank you for contributing to building a powerful instant messaging solution!
 
 ## :closed_book: License
 
-This software is licensed under the Apache License 2.0
+For more details, please refer to [here](./LICENSE).
 
 
 

README_zh_CN.md

@@ -131,10 +131,9 @@
 
 感谢您的贡献，一起来打造强大的即时通讯解决方案！
 
-## :closed_book:  开源许可证 License
-
-This software is licensed under the Apache License 2.0
+## :closed_book: 许可证
 
+  OpenIMSDK 在 Apache License 2.0 许可下可用。查看[LICENSE 文件](https://github.com/openimsdk/open-im-server/blob/main/LICENSE)了解更多信息。
 
 ## 🔮 Thanks to our contributors!
 

config/share.yml

@@ -7,7 +7,3 @@ multiLogin:
   policy: 1
   # max num of tokens in one end
   maxNumOneEnd: 30
-
-rpcMaxBodySize:
-  requestMaxBodySize:  8388608
-  responseMaxBodySize: 8388608

internal/rpc/group/group.go

@@ -448,25 +448,12 @@ func (g *groupServer) InviteUserToGroup(ctx context.Context, req *pbgroup.Invite
 		return nil, err
 	}
 
-	const singleQuantity = 50
-	for start := 0; start < len(groupMembers); start += singleQuantity {
-		end := start + singleQuantity
-		if end > len(groupMembers) {
-			end = len(groupMembers)
-		}
-		currentMembers := groupMembers[start:end]
+	if err := g.db.CreateGroup(ctx, nil, groupMembers); err != nil {
+		return nil, err
+	}
 
-		if err := g.db.CreateGroup(ctx, nil, currentMembers); err != nil {
-			return nil, err
-		}
-
-		userIDs := datautil.Slice(currentMembers, func(e *model.GroupMember) string {
-			return e.UserID
-		})
-
-		if err = g.notification.GroupApplicationAgreeMemberEnterNotification(ctx, req.GroupID, req.SendMessage, opUserID, userIDs...); err != nil {
-			return nil, err
-		}
+	if err = g.notification.GroupApplicationAgreeMemberEnterNotification(ctx, req.GroupID, req.SendMessage, opUserID, req.InvitedUserIDs...); err != nil {
+		return nil, err
 	}
 	return &pbgroup.InviteUserToGroupResp{}, nil
 }

internal/rpc/group/notification.go

@@ -522,10 +522,6 @@ func (g *NotificationSender) MemberKickedNotification(ctx context.Context, tips
 }
 
 func (g *NotificationSender) GroupApplicationAgreeMemberEnterNotification(ctx context.Context, groupID string, SendMessage *bool, invitedOpUserID string, entrantUserID ...string) error {
-	return g.groupApplicationAgreeMemberEnterNotification(ctx, groupID, SendMessage, invitedOpUserID, entrantUserID...)
-}
-
-func (g *NotificationSender) groupApplicationAgreeMemberEnterNotification(ctx context.Context, groupID string, SendMessage *bool, invitedOpUserID string, entrantUserID ...string) error {
 	var err error
 	defer func() {
 		if err != nil {

pkg/common/config/config.go

@@ -348,15 +348,9 @@ type AfterConfig struct {
 }
 
 type Share struct {
-	Secret         string         `yaml:"secret"`
-	IMAdminUserID  []string       `yaml:"imAdminUserID"`
-	MultiLogin     MultiLogin     `yaml:"multiLogin"`
-	RPCMaxBodySize MaxRequestBody `yaml:"rpcMaxBodySize"`
-}
-
-type MaxRequestBody struct {
-	RequestMaxBodySize  int `yaml:"requestMaxBodySize"`
-	ResponseMaxBodySize int `yaml:"responseMaxBodySize"`
+	Secret        string     `yaml:"secret"`
+	IMAdminUserID []string   `yaml:"imAdminUserID"`
+	MultiLogin    MultiLogin `yaml:"multiLogin"`
 }
 
 type MultiLogin struct {

pkg/common/startrpc/start.go

@@ -21,7 +21,6 @@ import (
 	"net"
 	"os"
 	"os/signal"
-	"reflect"
 	"strconv"
 	"syscall"
 	"time"
@@ -46,36 +45,6 @@ func init() {
 	prommetrics.RegistryAll()
 }
 
-func getConfigRpcMaxRequestBody(value reflect.Value) *conf.MaxRequestBody {
-	for value.Kind() == reflect.Pointer {
-		value = value.Elem()
-	}
-	if value.Kind() == reflect.Struct {
-		num := value.NumField()
-		for i := 0; i < num; i++ {
-			field := value.Field(i)
-			if !field.CanInterface() {
-				continue
-			}
-			for field.Kind() == reflect.Pointer {
-				field = field.Elem()
-			}
-			switch elem := field.Interface().(type) {
-			case conf.Share:
-				return &elem.RPCMaxBodySize
-			case conf.MaxRequestBody:
-				return &elem
-			}
-			if field.Kind() == reflect.Struct {
-				if elem := getConfigRpcMaxRequestBody(field); elem != nil {
-					return elem
-				}
-			}
-		}
-	}
-	return nil
-}
-
 func Start[T any](ctx context.Context, disc *conf.Discovery, prometheusConfig *conf.Prometheus, listenIP,
 	registerIP string, autoSetPorts bool, rpcPorts []int, index int, rpcRegisterName string, notification *conf.Notification, config T,
 	watchConfigNames []string, watchServiceNames []string,
@@ -86,24 +55,7 @@ func Start[T any](ctx context.Context, disc *conf.Discovery, prometheusConfig *c
 		conf.InitNotification(notification)
 	}
 
-	maxRequestBody := getConfigRpcMaxRequestBody(reflect.ValueOf(config))
-
-	log.ZDebug(ctx, "rpc start", "rpcMaxRequestBody", maxRequestBody, "rpcRegisterName", rpcRegisterName, "registerIP", registerIP, "listenIP", listenIP)
-
-	options = append(options,
-		mw.GrpcServer(),
-	)
-	var clientOptions []grpc.DialOption
-	if maxRequestBody != nil {
-		if maxRequestBody.RequestMaxBodySize > 0 {
-			options = append(options, grpc.MaxRecvMsgSize(maxRequestBody.RequestMaxBodySize))
-			clientOptions = append(clientOptions, grpc.WithDefaultCallOptions(grpc.MaxCallSendMsgSize(maxRequestBody.RequestMaxBodySize)))
-		}
-		if maxRequestBody.ResponseMaxBodySize > 0 {
-			options = append(options, grpc.MaxSendMsgSize(maxRequestBody.ResponseMaxBodySize))
-			clientOptions = append(clientOptions, grpc.WithDefaultCallOptions(grpc.MaxCallRecvMsgSize(maxRequestBody.ResponseMaxBodySize)))
-		}
-	}
+	options = append(options, mw.GrpcServer())
 
 	registerIP, err := network.GetRpcRegisterIP(registerIP)
 	if err != nil {
@@ -132,9 +84,6 @@ func Start[T any](ctx context.Context, disc *conf.Discovery, prometheusConfig *c
 		mw.GrpcClient(), grpc.WithTransportCredentials(insecure.NewCredentials()),
 		grpc.WithDefaultServiceConfig(fmt.Sprintf(`{"LoadBalancingPolicy": "%s"}`, "round_robin")),
 	)
-	if len(clientOptions) > 0 {
-		client.AddOption(clientOptions...)
-	}
 
 	ctx, cancel := context.WithCancelCause(ctx)
 

tools/stress-test-v2/main.go

@@ -1,736 +0,0 @@
-package main
-
-import (
-	"bytes"
-	"context"
-	"encoding/json"
-	"flag"
-	"fmt"
-	"io"
-	"net/http"
-	"os"
-	"os/signal"
-	"sync"
-	"syscall"
-	"time"
-
-	"github.com/openimsdk/open-im-server/v3/pkg/apistruct"
-	"github.com/openimsdk/open-im-server/v3/pkg/common/config"
-	"github.com/openimsdk/protocol/auth"
-	"github.com/openimsdk/protocol/constant"
-	"github.com/openimsdk/protocol/group"
-	"github.com/openimsdk/protocol/sdkws"
-	pbuser "github.com/openimsdk/protocol/user"
-	"github.com/openimsdk/tools/log"
-	"github.com/openimsdk/tools/system/program"
-)
-
-// 1. Create 100K New Users
-// 2. Create 100 100K Groups
-// 3. Create 1000 999 Groups
-// 4. Send message to 100K Groups every second
-// 5. Send message to 999 Groups every minute
-
-var (
-	//  Use default userIDs List for testing, need to be created.
-	TestTargetUserList = []string{
-		// "<need-update-it>",
-	}
-	// DefaultGroupID = "<need-update-it>" // Use default group ID for testing, need to be created.
-)
-
-var (
-	ApiAddress string
-
-	// API method
-	GetAdminToken      = "/auth/get_admin_token"
-	UserCheck          = "/user/account_check"
-	CreateUser         = "/user/user_register"
-	ImportFriend       = "/friend/import_friend"
-	InviteToGroup      = "/group/invite_user_to_group"
-	GetGroupMemberInfo = "/group/get_group_members_info"
-	SendMsg            = "/msg/send_msg"
-	CreateGroup        = "/group/create_group"
-	GetUserToken       = "/auth/user_token"
-)
-
-const (
-	MaxUser            = 100000
-	Max100KGroup       = 100
-	Max999Group        = 1000
-	MaxInviteUserLimit = 999
-
-	CreateUserTicker         = 1 * time.Second
-	CreateGroupTicker        = 1 * time.Second
-	Create100KGroupTicker    = 1 * time.Second
-	Create999GroupTicker     = 1 * time.Second
-	SendMsgTo100KGroupTicker = 1 * time.Second
-	SendMsgTo999GroupTicker  = 1 * time.Minute
-)
-
-type BaseResp struct {
-	ErrCode int             `json:"errCode"`
-	ErrMsg  string          `json:"errMsg"`
-	Data    json.RawMessage `json:"data"`
-}
-
-type StressTest struct {
-	Conf                   *conf
-	AdminUserID            string
-	AdminToken             string
-	DefaultGroupID         string
-	DefaultUserID          string
-	UserCounter            int
-	CreateUserCounter      int
-	Create100kGroupCounter int
-	Create999GroupCounter  int
-	MsgCounter             int
-	CreatedUsers           []string
-	CreatedGroups          []string
-	Mutex                  sync.Mutex
-	Ctx                    context.Context
-	Cancel                 context.CancelFunc
-	HttpClient             *http.Client
-	Wg                     sync.WaitGroup
-	Once                   sync.Once
-}
-
-type conf struct {
-	Share config.Share
-	Api   config.API
-}
-
-func initConfig(configDir string) (*config.Share, *config.API, error) {
-	var (
-		share     = &config.Share{}
-		apiConfig = &config.API{}
-	)
-
-	err := config.Load(configDir, config.ShareFileName, config.EnvPrefixMap[config.ShareFileName], share)
-	if err != nil {
-		return nil, nil, err
-	}
-
-	err = config.Load(configDir, config.OpenIMAPICfgFileName, config.EnvPrefixMap[config.OpenIMAPICfgFileName], apiConfig)
-	if err != nil {
-		return nil, nil, err
-	}
-
-	return share, apiConfig, nil
-}
-
-// Post Request
-func (st *StressTest) PostRequest(ctx context.Context, url string, reqbody any) ([]byte, error) {
-	// Marshal body
-	jsonBody, err := json.Marshal(reqbody)
-	if err != nil {
-		log.ZError(ctx, "Failed to marshal request body", err, "url", url, "reqbody", reqbody)
-		return nil, err
-	}
-
-	req, err := http.NewRequest(http.MethodPost, url, bytes.NewReader(jsonBody))
-	if err != nil {
-		return nil, err
-	}
-	req.Header.Set("Content-Type", "application/json")
-	req.Header.Set("operationID", st.AdminUserID)
-	if st.AdminToken != "" {
-		req.Header.Set("token", st.AdminToken)
-	}
-
-	// log.ZInfo(ctx, "Header info is ", "Content-Type", "application/json", "operationID", st.AdminUserID, "token", st.AdminToken)
-
-	resp, err := st.HttpClient.Do(req)
-	if err != nil {
-		log.ZError(ctx, "Failed to send request", err, "url", url, "reqbody", reqbody)
-		return nil, err
-	}
-	defer resp.Body.Close()
-
-	respBody, err := io.ReadAll(resp.Body)
-	if err != nil {
-		log.ZError(ctx, "Failed to read response body", err, "url", url)
-		return nil, err
-	}
-
-	var baseResp BaseResp
-	if err := json.Unmarshal(respBody, &baseResp); err != nil {
-		log.ZError(ctx, "Failed to unmarshal response body", err, "url", url, "respBody", string(respBody))
-		return nil, err
-	}
-
-	if baseResp.ErrCode != 0 {
-		err = fmt.Errorf(baseResp.ErrMsg)
-		log.ZError(ctx, "Failed to send request", err, "url", url, "reqbody", reqbody, "resp", baseResp)
-		return nil, err
-	}
-
-	return baseResp.Data, nil
-}
-
-func (st *StressTest) GetAdminToken(ctx context.Context) (string, error) {
-	req := auth.GetAdminTokenReq{
-		Secret: st.Conf.Share.Secret,
-		UserID: st.AdminUserID,
-	}
-
-	resp, err := st.PostRequest(ctx, ApiAddress+GetAdminToken, &req)
-	if err != nil {
-		return "", err
-	}
-
-	data := &auth.GetAdminTokenResp{}
-	if err := json.Unmarshal(resp, &data); err != nil {
-		return "", err
-	}
-
-	return data.Token, nil
-}
-
-func (st *StressTest) CheckUser(ctx context.Context, userIDs []string) ([]string, error) {
-	req := pbuser.AccountCheckReq{
-		CheckUserIDs: userIDs,
-	}
-
-	resp, err := st.PostRequest(ctx, ApiAddress+UserCheck, &req)
-	if err != nil {
-		return nil, err
-	}
-
-	data := &pbuser.AccountCheckResp{}
-	if err := json.Unmarshal(resp, &data); err != nil {
-		return nil, err
-	}
-
-	unRegisteredUserIDs := make([]string, 0)
-
-	for _, res := range data.Results {
-		if res.AccountStatus == constant.UnRegistered {
-			unRegisteredUserIDs = append(unRegisteredUserIDs, res.UserID)
-		}
-	}
-
-	return unRegisteredUserIDs, nil
-}
-
-func (st *StressTest) CreateUser(ctx context.Context, userID string) (string, error) {
-	user := &sdkws.UserInfo{
-		UserID:   userID,
-		Nickname: userID,
-	}
-
-	req := pbuser.UserRegisterReq{
-		Users: []*sdkws.UserInfo{user},
-	}
-
-	_, err := st.PostRequest(ctx, ApiAddress+CreateUser, &req)
-	if err != nil {
-		return "", err
-	}
-
-	st.UserCounter++
-	return userID, nil
-}
-
-func (st *StressTest) CreateUserBatch(ctx context.Context, userIDs []string) error {
-	// The method can import a large number of users at once.
-	var userList []*sdkws.UserInfo
-
-	defer st.Once.Do(
-		func() {
-			st.DefaultUserID = userIDs[0]
-			fmt.Println("Default Send User Created ID:", st.DefaultUserID)
-		})
-
-	needUserIDs, err := st.CheckUser(ctx, userIDs)
-	if err != nil {
-		return err
-	}
-
-	for _, userID := range needUserIDs {
-		user := &sdkws.UserInfo{
-			UserID:   userID,
-			Nickname: userID,
-		}
-		userList = append(userList, user)
-	}
-
-	req := pbuser.UserRegisterReq{
-		Users: userList,
-	}
-
-	_, err = st.PostRequest(ctx, ApiAddress+CreateUser, &req)
-	if err != nil {
-		return err
-	}
-
-	st.UserCounter += len(userList)
-	return nil
-}
-
-func (st *StressTest) GetGroupMembersInfo(ctx context.Context, groupID string, userIDs []string) ([]string, error) {
-	needInviteUserIDs := make([]string, 0)
-
-	const maxBatchSize = 500
-	if len(userIDs) > maxBatchSize {
-		for i := 0; i < len(userIDs); i += maxBatchSize {
-			end := min(i+maxBatchSize, len(userIDs))
-			batchUserIDs := userIDs[i:end]
-
-			// log.ZInfo(ctx, "Processing group members batch", "groupID", groupID, "batch", i/maxBatchSize+1,
-			// 	"batchUserCount", len(batchUserIDs))
-
-			// Process a single batch
-			batchReq := group.GetGroupMembersInfoReq{
-				GroupID: groupID,
-				UserIDs: batchUserIDs,
-			}
-
-			resp, err := st.PostRequest(ctx, ApiAddress+GetGroupMemberInfo, &batchReq)
-			if err != nil {
-				log.ZError(ctx, "Batch query failed", err, "batch", i/maxBatchSize+1)
-				continue
-			}
-
-			data := &group.GetGroupMembersInfoResp{}
-			if err := json.Unmarshal(resp, &data); err != nil {
-				log.ZError(ctx, "Failed to parse batch response", err, "batch", i/maxBatchSize+1)
-				continue
-			}
-
-			// Process the batch results
-			existingMembers := make(map[string]bool)
-			for _, member := range data.Members {
-				existingMembers[member.UserID] = true
-			}
-
-			for _, userID := range batchUserIDs {
-				if !existingMembers[userID] {
-					needInviteUserIDs = append(needInviteUserIDs, userID)
-				}
-			}
-		}
-
-		return needInviteUserIDs, nil
-	}
-
-	req := group.GetGroupMembersInfoReq{
-		GroupID: groupID,
-		UserIDs: userIDs,
-	}
-
-	resp, err := st.PostRequest(ctx, ApiAddress+GetGroupMemberInfo, &req)
-	if err != nil {
-		return nil, err
-	}
-
-	data := &group.GetGroupMembersInfoResp{}
-	if err := json.Unmarshal(resp, &data); err != nil {
-		return nil, err
-	}
-
-	existingMembers := make(map[string]bool)
-	for _, member := range data.Members {
-		existingMembers[member.UserID] = true
-	}
-
-	for _, userID := range userIDs {
-		if !existingMembers[userID] {
-			needInviteUserIDs = append(needInviteUserIDs, userID)
-		}
-	}
-
-	return needInviteUserIDs, nil
-}
-
-func (st *StressTest) InviteToGroup(ctx context.Context, groupID string, userIDs []string) error {
-	req := group.InviteUserToGroupReq{
-		GroupID:        groupID,
-		InvitedUserIDs: userIDs,
-	}
-	_, err := st.PostRequest(ctx, ApiAddress+InviteToGroup, &req)
-	if err != nil {
-		return err
-	}
-
-	return nil
-}
-
-func (st *StressTest) SendMsg(ctx context.Context, userID string, groupID string) error {
-	contentObj := map[string]any{
-		// "content": fmt.Sprintf("index %d. The current time is %s", st.MsgCounter, time.Now().Format("2006-01-02 15:04:05.000")),
-		"content": fmt.Sprintf("The current time is %s", time.Now().Format("2006-01-02 15:04:05.000")),
-	}
-
-	req := &apistruct.SendMsgReq{
-		SendMsg: apistruct.SendMsg{
-			SendID:         userID,
-			SenderNickname: userID,
-			GroupID:        groupID,
-			ContentType:    constant.Text,
-			SessionType:    constant.ReadGroupChatType,
-			Content:        contentObj,
-		},
-	}
-
-	_, err := st.PostRequest(ctx, ApiAddress+SendMsg, &req)
-	if err != nil {
-		log.ZError(ctx, "Failed to send message", err, "userID", userID, "req", &req)
-		return err
-	}
-
-	st.MsgCounter++
-
-	return nil
-}
-
-// Max userIDs number is 1000
-func (st *StressTest) CreateGroup(ctx context.Context, groupID string, userID string, userIDsList []string) (string, error) {
-	groupInfo := &sdkws.GroupInfo{
-		GroupID:   groupID,
-		GroupName: groupID,
-		GroupType: constant.WorkingGroup,
-	}
-
-	req := group.CreateGroupReq{
-		OwnerUserID:   userID,
-		MemberUserIDs: userIDsList,
-		GroupInfo:     groupInfo,
-	}
-
-	resp := group.CreateGroupResp{}
-
-	response, err := st.PostRequest(ctx, ApiAddress+CreateGroup, &req)
-	if err != nil {
-		return "", err
-	}
-
-	if err := json.Unmarshal(response, &resp); err != nil {
-		return "", err
-	}
-
-	// st.GroupCounter++
-
-	return resp.GroupInfo.GroupID, nil
-}
-
-func main() {
-	var configPath string
-	// defaultConfigDir := filepath.Join("..", "..", "..", "..", "..", "config")
-	// flag.StringVar(&configPath, "c", defaultConfigDir, "config path")
-	flag.StringVar(&configPath, "c", "", "config path")
-	flag.Parse()
-
-	if configPath == "" {
-		_, _ = fmt.Fprintln(os.Stderr, "config path is empty")
-		os.Exit(1)
-		return
-	}
-
-	fmt.Printf(" Config Path: %s\n", configPath)
-
-	share, apiConfig, err := initConfig(configPath)
-	if err != nil {
-		program.ExitWithError(err)
-		return
-	}
-
-	ApiAddress = fmt.Sprintf("http://%s:%s", "127.0.0.1", fmt.Sprint(apiConfig.Api.Ports[0]))
-
-	ctx, cancel := context.WithCancel(context.Background())
-	// ch := make(chan struct{})
-
-	st := &StressTest{
-		Conf: &conf{
-			Share: *share,
-			Api:   *apiConfig,
-		},
-		AdminUserID: share.IMAdminUserID[0],
-		Ctx:         ctx,
-		Cancel:      cancel,
-		HttpClient: &http.Client{
-			Timeout: 50 * time.Second,
-		},
-	}
-
-	c := make(chan os.Signal, 1)
-	signal.Notify(c, os.Interrupt, syscall.SIGTERM)
-	go func() {
-		<-c
-		fmt.Println("\nReceived stop signal, stopping...")
-
-		go func() {
-			// time.Sleep(5 * time.Second)
-			fmt.Println("Force exit")
-			os.Exit(0)
-		}()
-
-		st.Cancel()
-	}()
-
-	token, err := st.GetAdminToken(st.Ctx)
-	if err != nil {
-		log.ZError(ctx, "Get Admin Token failed.", err, "AdminUserID", st.AdminUserID)
-	}
-
-	st.AdminToken = token
-	fmt.Println("Admin Token:", st.AdminToken)
-	fmt.Println("ApiAddress:", ApiAddress)
-
-	for i := range MaxUser {
-		userID := fmt.Sprintf("v2_StressTest_User_%d", i)
-		st.CreatedUsers = append(st.CreatedUsers, userID)
-		st.CreateUserCounter++
-	}
-
-	// err = st.CreateUserBatch(st.Ctx, st.CreatedUsers)
-	// if err != nil {
-	// 	log.ZError(ctx, "Create user failed.", err)
-	// }
-
-	const batchSize = 1000
-	totalUsers := len(st.CreatedUsers)
-	successCount := 0
-
-	if st.DefaultUserID == "" && len(st.CreatedUsers) > 0 {
-		st.DefaultUserID = st.CreatedUsers[0]
-	}
-
-	for i := 0; i < totalUsers; i += batchSize {
-		end := min(i+batchSize, totalUsers)
-
-		userBatch := st.CreatedUsers[i:end]
-		log.ZInfo(st.Ctx, "Creating user batch", "batch", i/batchSize+1, "count", len(userBatch))
-
-		err = st.CreateUserBatch(st.Ctx, userBatch)
-		if err != nil {
-			log.ZError(st.Ctx, "Batch user creation failed", err, "batch", i/batchSize+1)
-		} else {
-			successCount += len(userBatch)
-			log.ZInfo(st.Ctx, "Batch user creation succeeded", "batch", i/batchSize+1,
-				"progress", fmt.Sprintf("%d/%d", successCount, totalUsers))
-		}
-	}
-
-	// Execute create 100k group
-	st.Wg.Add(1)
-	go func() {
-		defer st.Wg.Done()
-
-		create100kGroupTicker := time.NewTicker(Create100KGroupTicker)
-		defer create100kGroupTicker.Stop()
-
-		for i := range Max100KGroup {
-			select {
-			case <-st.Ctx.Done():
-				log.ZInfo(st.Ctx, "Stop Create 100K Group")
-				return
-
-			case <-create100kGroupTicker.C:
-				// Create 100K groups
-				st.Wg.Add(1)
-				go func(idx int) {
-					defer st.Wg.Done()
-					defer func() {
-						st.Create100kGroupCounter++
-					}()
-
-					groupID := fmt.Sprintf("v2_StressTest_Group_100K_%d", idx)
-
-					if _, err = st.CreateGroup(st.Ctx, groupID, st.DefaultUserID, TestTargetUserList); err != nil {
-						log.ZError(st.Ctx, "Create group failed.", err)
-						// continue
-					}
-
-					for i := 0; i < MaxUser/MaxInviteUserLimit; i++ {
-						InviteUserIDs := make([]string, 0)
-						// ensure TargetUserList is in group
-						InviteUserIDs = append(InviteUserIDs, TestTargetUserList...)
-
-						startIdx := max(i*MaxInviteUserLimit, 1)
-						endIdx := min((i+1)*MaxInviteUserLimit, MaxUser)
-
-						for j := startIdx; j < endIdx; j++ {
-							userCreatedID := fmt.Sprintf("v2_StressTest_User_%d", j)
-							InviteUserIDs = append(InviteUserIDs, userCreatedID)
-						}
-
-						if len(InviteUserIDs) == 0 {
-							log.ZWarn(st.Ctx, "InviteUserIDs is empty", nil, "groupID", groupID)
-							continue
-						}
-
-						InviteUserIDs, err := st.GetGroupMembersInfo(ctx, groupID, InviteUserIDs)
-						if err != nil {
-							log.ZError(st.Ctx, "GetGroupMembersInfo failed.", err, "groupID", groupID)
-							continue
-						}
-
-						if len(InviteUserIDs) == 0 {
-							log.ZWarn(st.Ctx, "InviteUserIDs is empty", nil, "groupID", groupID)
-							continue
-						}
-
-						// Invite To Group
-						if err = st.InviteToGroup(st.Ctx, groupID, InviteUserIDs); err != nil {
-							log.ZError(st.Ctx, "Invite To Group failed.", err, "UserID", InviteUserIDs)
-							continue
-							// os.Exit(1)
-							// return
-						}
-					}
-				}(i)
-			}
-		}
-	}()
-
-	// create 999 groups
-	st.Wg.Add(1)
-	go func() {
-		defer st.Wg.Done()
-
-		create999GroupTicker := time.NewTicker(Create999GroupTicker)
-		defer create999GroupTicker.Stop()
-
-		for i := range Max999Group {
-			select {
-			case <-st.Ctx.Done():
-				log.ZInfo(st.Ctx, "Stop Create 999 Group")
-				return
-
-			case <-create999GroupTicker.C:
-				// Create 999 groups
-				st.Wg.Add(1)
-				go func(idx int) {
-					defer st.Wg.Done()
-					defer func() {
-						st.Create999GroupCounter++
-					}()
-
-					groupID := fmt.Sprintf("v2_StressTest_Group_1K_%d", idx)
-
-					if _, err = st.CreateGroup(st.Ctx, groupID, st.DefaultUserID, TestTargetUserList); err != nil {
-						log.ZError(st.Ctx, "Create group failed.", err)
-						// continue
-					}
-					for i := 0; i < MaxUser/MaxInviteUserLimit; i++ {
-						InviteUserIDs := make([]string, 0)
-						// ensure TargetUserList is in group
-						InviteUserIDs = append(InviteUserIDs, TestTargetUserList...)
-
-						startIdx := max(i*MaxInviteUserLimit, 1)
-						endIdx := min((i+1)*MaxInviteUserLimit, MaxUser)
-
-						for j := startIdx; j < endIdx; j++ {
-							userCreatedID := fmt.Sprintf("v2_StressTest_User_%d", j)
-							InviteUserIDs = append(InviteUserIDs, userCreatedID)
-						}
-
-						if len(InviteUserIDs) == 0 {
-							log.ZWarn(st.Ctx, "InviteUserIDs is empty", nil, "groupID", groupID)
-							continue
-						}
-
-						InviteUserIDs, err := st.GetGroupMembersInfo(ctx, groupID, InviteUserIDs)
-						if err != nil {
-							log.ZError(st.Ctx, "GetGroupMembersInfo failed.", err, "groupID", groupID)
-							continue
-						}
-
-						if len(InviteUserIDs) == 0 {
-							log.ZWarn(st.Ctx, "InviteUserIDs is empty", nil, "groupID", groupID)
-							continue
-						}
-
-						// Invite To Group
-						if err = st.InviteToGroup(st.Ctx, groupID, InviteUserIDs); err != nil {
-							log.ZError(st.Ctx, "Invite To Group failed.", err, "UserID", InviteUserIDs)
-							continue
-							// os.Exit(1)
-							// return
-						}
-					}
-				}(i)
-			}
-		}
-	}()
-
-	// Send message to 100K groups
-	st.Wg.Wait()
-	fmt.Println("All groups created successfully, starting to send messages...")
-	log.ZInfo(ctx, "All groups created successfully, starting to send messages...")
-
-	var groups100K []string
-	var groups999 []string
-
-	for i := range Max100KGroup {
-		groupID := fmt.Sprintf("v2_StressTest_Group_100K_%d", i)
-		groups100K = append(groups100K, groupID)
-	}
-
-	for i := range Max999Group {
-		groupID := fmt.Sprintf("v2_StressTest_Group_1K_%d", i)
-		groups999 = append(groups999, groupID)
-	}
-
-	send100kGroupLimiter := make(chan struct{}, 20)
-	send999GroupLimiter := make(chan struct{}, 100)
-
-	// execute Send message to 100K groups
-	go func() {
-		ticker := time.NewTicker(SendMsgTo100KGroupTicker)
-		defer ticker.Stop()
-
-		for {
-			select {
-			case <-st.Ctx.Done():
-				log.ZInfo(st.Ctx, "Stop Send Message to 100K Group")
-				return
-
-			case <-ticker.C:
-				// Send message to 100K groups
-				for _, groupID := range groups100K {
-					send100kGroupLimiter <- struct{}{}
-					go func(groupID string) {
-						defer func() { <-send100kGroupLimiter }()
-						if err := st.SendMsg(st.Ctx, st.DefaultUserID, groupID); err != nil {
-							log.ZError(st.Ctx, "Send message to 100K group failed.", err)
-						}
-					}(groupID)
-				}
-				// log.ZInfo(st.Ctx, "Send message to 100K groups successfully.")
-			}
-		}
-	}()
-
-	// execute Send message to 999 groups
-	go func() {
-		ticker := time.NewTicker(SendMsgTo999GroupTicker)
-		defer ticker.Stop()
-
-		for {
-			select {
-			case <-st.Ctx.Done():
-				log.ZInfo(st.Ctx, "Stop Send Message to 999 Group")
-				return
-
-			case <-ticker.C:
-				// Send message to 999 groups
-				for _, groupID := range groups999 {
-					send999GroupLimiter <- struct{}{}
-					go func(groupID string) {
-						defer func() { <-send999GroupLimiter }()
-
-						if err := st.SendMsg(st.Ctx, st.DefaultUserID, groupID); err != nil {
-							log.ZError(st.Ctx, "Send message to 999 group failed.", err)
-						}
-					}(groupID)
-				}
-				// log.ZInfo(st.Ctx, "Send message to 999 groups successfully.")
-			}
-		}
-	}()
-
-	<-st.Ctx.Done()
-	fmt.Println("Received signal to exit, shutting down...")
-}

tools/stress-test/main.go

@@ -72,22 +72,22 @@ type BaseResp struct {
 }
 
 type StressTest struct {
-	Conf           *conf
-	AdminUserID    string
-	AdminToken     string
-	DefaultGroupID string
-	DefaultUserID  string
-	UserCounter    int
-	GroupCounter   int
-	MsgCounter     int
-	CreatedUsers   []string
-	CreatedGroups  []string
-	Mutex          sync.Mutex
-	Ctx            context.Context
-	Cancel         context.CancelFunc
-	HttpClient     *http.Client
-	Wg             sync.WaitGroup
-	Once           sync.Once
+	Conf              *conf
+	AdminUserID       string
+	AdminToken        string
+	DefaultGroupID    string
+	DefaultSendUserID string
+	UserCounter       int
+	GroupCounter      int
+	MsgCounter        int
+	CreatedUsers      []string
+	CreatedGroups     []string
+	Mutex             sync.Mutex
+	Ctx               context.Context
+	Cancel            context.CancelFunc
+	HttpClient        *http.Client
+	Wg                sync.WaitGroup
+	Once              sync.Once
 }
 
 type conf struct {
@@ -384,6 +384,7 @@ func main() {
 					os.Exit(1)
 					return
 				}
+
 				// Invite To Group
 				if err = st.InviteToGroup(st.Ctx, userCreatedID); err != nil {
 					log.ZError(st.Ctx, "Invite To Group failed.", err, "UserID", userCreatedID)
@@ -392,7 +393,7 @@ func main() {
 				}
 
 				st.Once.Do(func() {
-					st.DefaultUserID = userCreatedID
+					st.DefaultSendUserID = userCreatedID
 					fmt.Println("Default Send User Created ID:", userCreatedID)
 					close(ch)
 				})
@@ -416,8 +417,8 @@ func main() {
 
 			case <-ticker.C:
 				// Send Message
-				if err = st.SendMsg(st.Ctx, st.DefaultUserID); err != nil {
-					log.ZError(st.Ctx, "Send Message failed.", err, "UserID", st.DefaultUserID)
+				if err = st.SendMsg(st.Ctx, st.DefaultSendUserID); err != nil {
+					log.ZError(st.Ctx, "Send Message failed.", err, "UserID", st.DefaultSendUserID)
 					continue
 				}
 			}
@@ -442,9 +443,9 @@ func main() {
 			case <-ticker.C:
 
 				// Create Group
-				_, err := st.CreateGroup(st.Ctx, st.DefaultUserID)
+				_, err := st.CreateGroup(st.Ctx, st.DefaultSendUserID)
 				if err != nil {
-					log.ZError(st.Ctx, "Create Group failed.", err, "UserID", st.DefaultUserID)
+					log.ZError(st.Ctx, "Create Group failed.", err, "UserID", st.DefaultSendUserID)
 					os.Exit(1)
 					return
 				}