TRKJ/EdgeAPI
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeAPI.git synced 2025-11-11 22:00:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

[区域]增加区域管理

Browse Source
This commit is contained in:
刘祥超
2020-12-10 15:03:03 +08:00
parent 82543e9fde
commit bddd0a3a4f
10 changed files with 317 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
internal/rpc/services/service_node.go
View File

@@ -975,14 +975,14 @@ func (this *NodeService) UpdateNodeLogin(ctx context.Context, req *pb.UpdateNode
}
// 计算某个节点分组内的节点数量
func (this *NodeService) CountAllEnabledNodesWithGroupId(ctx context.Context, req *pb.CountAllEnabledNodesWithGroupIdRequest) (*pb.RPCCountResponse, error) {
func (this *NodeService) CountAllEnabledNodesWithNodeGroupId(ctx context.Context, req *pb.CountAllEnabledNodesWithNodeGroupIdRequest) (*pb.RPCCountResponse, error) {
// 校验请求
_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
if err != nil {
return nil, err
}
count, err := models.SharedNodeDAO.CountAllEnabledNodesWithGroupId(req.GroupId)
count, err := models.SharedNodeDAO.CountAllEnabledNodesWithGroupId(req.NodeGroupId)
if err != nil {
return nil, err
}
@@ -1212,3 +1212,16 @@ func (this *NodeService) notifyNodeDNSChanged(nodeId int64) error {
}
return nil
}
// 计算某个区域下的节点数量
func (this *NodeService) CountAllEnabledNodesWithNodeRegionId(ctx context.Context, req *pb.CountAllEnabledNodesWithNodeRegionIdRequest) (*pb.RPCCountResponse, error) {
_, err := this.ValidateAdmin(ctx, 0)
if err != nil {
return nil, err
}
count, err := models.SharedNodeDAO.CountAllEnabledNodesWithRegionId(req.NodeRegionId)
if err != nil {
return nil, err
}
return this.SuccessCount(count)
}