diff --git a/internal/db/models/node_dao_limit.go b/internal/db/models/node_dao_limit.go index 7b640c99..f801bd75 100644 --- a/internal/db/models/node_dao_limit.go +++ b/internal/db/models/node_dao_limit.go @@ -7,6 +7,13 @@ import ( "github.com/iwind/TeaGo/dbs" ) +func (this *NodeDAO) CountAllAuthorityNodes(tx *dbs.Tx) (int64, error) { + return this.Query(tx). + State(NodeStateEnabled). + Where("clusterId IN (SELECT id FROM " + SharedNodeClusterDAO.Table + " WHERE state=1)"). + Count() +} + func (this *NodeDAO) CheckNodesLimit(tx *dbs.Tx) error { return nil }