From adb10f8c1b5412b8d7cc34ca1e55d4942a9a6de4 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=88=98=E7=A5=A5=E8=B6=85?= Date: Sun, 5 Mar 2023 16:48:49 +0800 Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E8=AE=A1=E7=AE=97=E8=8A=82?= =?UTF-8?q?=E7=82=B9=E6=95=B0=E5=87=BD=E6=95=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- internal/db/models/node_dao_limit.go | 7 +++++++ 1 file changed, 7 insertions(+) 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 }