mirror of
https://github.com/TeaOSLab/EdgeAPI.git
synced 2025-11-02 05:40:27 +08:00
36 lines
1.0 KiB
Go
36 lines
1.0 KiB
Go
// Copyright 2022 GoEdge CDN goedge.cdn@gmail.com. All rights reserved. Official site: https://goedge.cn .
|
||
//go:build !plus
|
||
|
||
package models
|
||
|
||
import (
|
||
"errors"
|
||
teaconst "github.com/TeaOSLab/EdgeAPI/internal/const"
|
||
"github.com/iwind/TeaGo/dbs"
|
||
"github.com/iwind/TeaGo/types"
|
||
)
|
||
|
||
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 {
|
||
var maxNodes = teaconst.DefaultMaxNodes
|
||
|
||
// 检查节点数量
|
||
if maxNodes > 0 {
|
||
count, err := this.CountAllAuthorityNodes(tx)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if count >= int64(maxNodes) {
|
||
return errors.New("超出最大节点数限制：" + types.String(maxNodes) + "，当前已用：" + types.String(count) + "，请自行修改源码修改此限制（EdgeAPI/internal/const/const_community.go） 或者 购买商业版本授权。")
|
||
}
|
||
}
|
||
|
||
return nil
|
||
}
|