diff --git a/internal/db/models/acme/acme_user_dao.go b/internal/db/models/acme/acme_user_dao.go index dff285a8..2b9d5748 100644 --- a/internal/db/models/acme/acme_user_dao.go +++ b/internal/db/models/acme/acme_user_dao.go @@ -124,7 +124,7 @@ func (this *ACMEUserDAO) UpdateACMEUserRegistration(tx *dbs.Tx, acmeUserId int64 } // CountACMEUsersWithAdminId 计算用户数量 -func (this *ACMEUserDAO) CountACMEUsersWithAdminId(tx *dbs.Tx, adminId int64, userId int64) (int64, error) { +func (this *ACMEUserDAO) CountACMEUsersWithAdminId(tx *dbs.Tx, adminId int64, userId int64, accountId int64) (int64, error) { query := this.Query(tx) if adminId > 0 { query.Attr("adminId", adminId) @@ -132,6 +132,9 @@ func (this *ACMEUserDAO) CountACMEUsersWithAdminId(tx *dbs.Tx, adminId int64, us if userId > 0 { query.Attr("userId", userId) } + if accountId > 0 { + query.Attr("accountId", accountId) + } return query. State(ACMEUserStateEnabled). diff --git a/internal/rpc/services/service_acme_user.go b/internal/rpc/services/service_acme_user.go index 648c0a23..9bea2f58 100644 --- a/internal/rpc/services/service_acme_user.go +++ b/internal/rpc/services/service_acme_user.go @@ -91,7 +91,7 @@ func (this *ACMEUserService) CountACMEUsers(ctx context.Context, req *pb.CountAc tx := this.NullTx() - count, err := acmemodels.SharedACMEUserDAO.CountACMEUsersWithAdminId(tx, adminId, userId) + count, err := acmemodels.SharedACMEUserDAO.CountACMEUsersWithAdminId(tx, adminId, userId, req.AcmeProviderAccountId) if err != nil { return nil, err }