From b96fa2a995488dc1589417541f4bef56dc25de48 Mon Sep 17 00:00:00 2001
From: DacongDA <dacongda1@sina.com>
Date: Fri, 10 Jan 2025 22:28:25 +0800
Subject: [PATCH] feat: skip GetUserCount() if there is no quota limit (#3491)

---
 controllers/user.go |  8 +-------
 controllers/util.go | 10 ++++++++--
 2 files changed, 9 insertions(+), 9 deletions(-)

diff --git a/controllers/user.go b/controllers/user.go
index 128d6ca6..7e19db8b 100644
--- a/controllers/user.go
+++ b/controllers/user.go
@@ -353,13 +353,7 @@ func (c *ApiController) AddUser() {
 		return
 	}
 
-	count, err := object.GetUserCount("", "", "", "")
-	if err != nil {
-		c.ResponseError(err.Error())
-		return
-	}
-
-	if err := checkQuotaForUser(int(count)); err != nil {
+	if err := checkQuotaForUser(); err != nil {
 		c.ResponseError(err.Error())
 		return
 	}
diff --git a/controllers/util.go b/controllers/util.go
index 27a67786..d5751562 100644
--- a/controllers/util.go
+++ b/controllers/util.go
@@ -294,12 +294,18 @@ func checkQuotaForProvider(count int) error {
 	return nil
 }
 
-func checkQuotaForUser(count int) error {
+func checkQuotaForUser() error {
 	quota := conf.GetConfigQuota().User
 	if quota == -1 {
 		return nil
 	}
-	if count >= quota {
+
+	count, err := object.GetUserCount("", "", "", "")
+	if err != nil {
+		return err
+	}
+
+	if int(count) >= quota {
 		return fmt.Errorf("user quota is exceeded")
 	}
 	return nil