mirror of
				https://github.com/TeaOSLab/EdgeAPI.git
				synced 2025-11-04 16:00:24 +08:00 
			
		
		
		
	
		
			
				
	
	
		
			575 lines
		
	
	
		
			17 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			575 lines
		
	
	
		
			17 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
package services
 | 
						|
 | 
						|
import (
 | 
						|
	"context"
 | 
						|
	"encoding/json"
 | 
						|
	"github.com/TeaOSLab/EdgeAPI/internal/db/models"
 | 
						|
	"github.com/TeaOSLab/EdgeAPI/internal/dnsclients"
 | 
						|
	"github.com/TeaOSLab/EdgeAPI/internal/errors"
 | 
						|
	rpcutils "github.com/TeaOSLab/EdgeAPI/internal/rpc/utils"
 | 
						|
	"github.com/TeaOSLab/EdgeAPI/internal/tasks"
 | 
						|
	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
 | 
						|
	"github.com/iwind/TeaGo/types"
 | 
						|
	"strconv"
 | 
						|
)
 | 
						|
 | 
						|
type NodeClusterService struct {
 | 
						|
	BaseService
 | 
						|
}
 | 
						|
 | 
						|
// 创建集群
 | 
						|
func (this *NodeClusterService) CreateNodeCluster(ctx context.Context, req *pb.CreateNodeClusterRequest) (*pb.CreateNodeClusterResponse, error) {
 | 
						|
	adminId, err := this.ValidateAdmin(ctx, 0)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	clusterId, err := models.SharedNodeClusterDAO.CreateCluster(adminId, req.Name, req.GrantId, req.InstallDir, req.DnsDomainId, req.DnsName)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	return &pb.CreateNodeClusterResponse{ClusterId: clusterId}, nil
 | 
						|
}
 | 
						|
 | 
						|
// 修改集群
 | 
						|
func (this *NodeClusterService) UpdateNodeCluster(ctx context.Context, req *pb.UpdateNodeClusterRequest) (*pb.RPCSuccess, error) {
 | 
						|
	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	err = models.SharedNodeClusterDAO.UpdateCluster(req.ClusterId, req.Name, req.GrantId, req.InstallDir)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	return this.Success()
 | 
						|
}
 | 
						|
 | 
						|
// 禁用集群
 | 
						|
func (this *NodeClusterService) DeleteNodeCluster(ctx context.Context, req *pb.DeleteNodeClusterRequest) (*pb.RPCSuccess, error) {
 | 
						|
	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	err = models.SharedNodeClusterDAO.DisableNodeCluster(req.ClusterId)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	return this.Success()
 | 
						|
}
 | 
						|
 | 
						|
// 查找单个集群
 | 
						|
func (this *NodeClusterService) FindEnabledNodeCluster(ctx context.Context, req *pb.FindEnabledNodeClusterRequest) (*pb.FindEnabledNodeClusterResponse, error) {
 | 
						|
	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	cluster, err := models.SharedNodeClusterDAO.FindEnabledNodeCluster(req.ClusterId)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	if cluster == nil {
 | 
						|
		return &pb.FindEnabledNodeClusterResponse{}, nil
 | 
						|
	}
 | 
						|
 | 
						|
	return &pb.FindEnabledNodeClusterResponse{Cluster: &pb.NodeCluster{
 | 
						|
		Id:         int64(cluster.Id),
 | 
						|
		Name:       cluster.Name,
 | 
						|
		CreatedAt:  int64(cluster.CreatedAt),
 | 
						|
		InstallDir: cluster.InstallDir,
 | 
						|
		GrantId:    int64(cluster.GrantId),
 | 
						|
		UniqueId:   cluster.UniqueId,
 | 
						|
		Secret:     cluster.Secret,
 | 
						|
	}}, nil
 | 
						|
}
 | 
						|
 | 
						|
// 查找集群的API节点信息
 | 
						|
func (this *NodeClusterService) FindAPINodesWithNodeCluster(ctx context.Context, req *pb.FindAPINodesWithNodeClusterRequest) (*pb.FindAPINodesWithNodeClusterResponse, error) {
 | 
						|
	// 校验请求
 | 
						|
	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	cluster, err := models.SharedNodeClusterDAO.FindEnabledNodeCluster(req.ClusterId)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	if cluster == nil {
 | 
						|
		return nil, errors.New("can not find cluster with id '" + strconv.FormatInt(req.ClusterId, 10) + "'")
 | 
						|
	}
 | 
						|
 | 
						|
	result := &pb.FindAPINodesWithNodeClusterResponse{}
 | 
						|
	result.UseAllAPINodes = cluster.UseAllAPINodes == 1
 | 
						|
 | 
						|
	apiNodeIds := []int64{}
 | 
						|
	if len(cluster.ApiNodes) > 0 && cluster.ApiNodes != "null" {
 | 
						|
		err = json.Unmarshal([]byte(cluster.ApiNodes), &apiNodeIds)
 | 
						|
		if err != nil {
 | 
						|
			return nil, err
 | 
						|
		}
 | 
						|
		if len(apiNodeIds) > 0 {
 | 
						|
			apiNodes := []*pb.APINode{}
 | 
						|
			for _, apiNodeId := range apiNodeIds {
 | 
						|
				apiNode, err := models.SharedAPINodeDAO.FindEnabledAPINode(apiNodeId)
 | 
						|
				if err != nil {
 | 
						|
					return nil, err
 | 
						|
				}
 | 
						|
				apiNodeAddrs, err := apiNode.DecodeAccessAddrStrings()
 | 
						|
				if err != nil {
 | 
						|
					return nil, err
 | 
						|
				}
 | 
						|
				apiNodes = append(apiNodes, &pb.APINode{
 | 
						|
					Id:          int64(apiNode.Id),
 | 
						|
					IsOn:        apiNode.IsOn == 1,
 | 
						|
					ClusterId:   int64(apiNode.ClusterId),
 | 
						|
					Name:        apiNode.Name,
 | 
						|
					Description: apiNode.Description,
 | 
						|
					AccessAddrs: apiNodeAddrs,
 | 
						|
				})
 | 
						|
			}
 | 
						|
			result.ApiNodes = apiNodes
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	return result, nil
 | 
						|
}
 | 
						|
 | 
						|
// 查找所有可用的集群
 | 
						|
func (this *NodeClusterService) FindAllEnabledNodeClusters(ctx context.Context, req *pb.FindAllEnabledNodeClustersRequest) (*pb.FindAllEnabledNodeClustersResponse, error) {
 | 
						|
	// 校验请求
 | 
						|
	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	clusters, err := models.SharedNodeClusterDAO.FindAllEnableClusters()
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	result := []*pb.NodeCluster{}
 | 
						|
	for _, cluster := range clusters {
 | 
						|
		result = append(result, &pb.NodeCluster{
 | 
						|
			Id:        int64(cluster.Id),
 | 
						|
			Name:      cluster.Name,
 | 
						|
			CreatedAt: int64(cluster.CreatedAt),
 | 
						|
			UniqueId:  cluster.UniqueId,
 | 
						|
			Secret:    cluster.Secret,
 | 
						|
		})
 | 
						|
	}
 | 
						|
 | 
						|
	return &pb.FindAllEnabledNodeClustersResponse{
 | 
						|
		Clusters: result,
 | 
						|
	}, nil
 | 
						|
}
 | 
						|
 | 
						|
// 查找所有变更的集群
 | 
						|
func (this *NodeClusterService) FindAllChangedNodeClusters(ctx context.Context, req *pb.FindAllChangedNodeClustersRequest) (*pb.FindAllChangedNodeClustersResponse, error) {
 | 
						|
	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	clusterIds, err := models.SharedNodeDAO.FindChangedClusterIds()
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	if len(clusterIds) == 0 {
 | 
						|
		return &pb.FindAllChangedNodeClustersResponse{
 | 
						|
			Clusters: []*pb.NodeCluster{},
 | 
						|
		}, nil
 | 
						|
	}
 | 
						|
	result := []*pb.NodeCluster{}
 | 
						|
	for _, clusterId := range clusterIds {
 | 
						|
		cluster, err := models.SharedNodeClusterDAO.FindEnabledNodeCluster(clusterId)
 | 
						|
		if err != nil {
 | 
						|
			return nil, err
 | 
						|
		}
 | 
						|
		if cluster == nil {
 | 
						|
			continue
 | 
						|
		}
 | 
						|
		result = append(result, &pb.NodeCluster{
 | 
						|
			Id:        int64(cluster.Id),
 | 
						|
			Name:      cluster.Name,
 | 
						|
			CreatedAt: int64(cluster.CreatedAt),
 | 
						|
			UniqueId:  cluster.UniqueId,
 | 
						|
			Secret:    cluster.Secret,
 | 
						|
		})
 | 
						|
	}
 | 
						|
	return &pb.FindAllChangedNodeClustersResponse{Clusters: result}, nil
 | 
						|
}
 | 
						|
 | 
						|
// 计算所有集群数量
 | 
						|
func (this *NodeClusterService) CountAllEnabledNodeClusters(ctx context.Context, req *pb.CountAllEnabledNodeClustersRequest) (*pb.RPCCountResponse, error) {
 | 
						|
	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	count, err := models.SharedNodeClusterDAO.CountAllEnabledClusters()
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	return this.SuccessCount(count)
 | 
						|
}
 | 
						|
 | 
						|
// 列出单页集群
 | 
						|
func (this *NodeClusterService) ListEnabledNodeClusters(ctx context.Context, req *pb.ListEnabledNodeClustersRequest) (*pb.ListEnabledNodeClustersResponse, error) {
 | 
						|
	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	clusters, err := models.SharedNodeClusterDAO.ListEnabledClusters(req.Offset, req.Size)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	result := []*pb.NodeCluster{}
 | 
						|
	for _, cluster := range clusters {
 | 
						|
		result = append(result, &pb.NodeCluster{
 | 
						|
			Id:          int64(cluster.Id),
 | 
						|
			Name:        cluster.Name,
 | 
						|
			CreatedAt:   int64(cluster.CreatedAt),
 | 
						|
			GrantId:     int64(cluster.GrantId),
 | 
						|
			InstallDir:  cluster.InstallDir,
 | 
						|
			UniqueId:    cluster.UniqueId,
 | 
						|
			Secret:      cluster.Secret,
 | 
						|
			DnsName:     cluster.DnsName,
 | 
						|
			DnsDomainId: int64(cluster.DnsDomainId),
 | 
						|
		})
 | 
						|
	}
 | 
						|
 | 
						|
	return &pb.ListEnabledNodeClustersResponse{Clusters: result}, nil
 | 
						|
}
 | 
						|
 | 
						|
// 查找集群的健康检查配置
 | 
						|
func (this *NodeClusterService) FindNodeClusterHealthCheckConfig(ctx context.Context, req *pb.FindNodeClusterHealthCheckConfigRequest) (*pb.FindNodeClusterHealthCheckConfigResponse, error) {
 | 
						|
	// 校验请求
 | 
						|
	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	config, err := models.SharedNodeClusterDAO.FindClusterHealthCheckConfig(req.ClusterId)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	configJSON, err := json.Marshal(config)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	return &pb.FindNodeClusterHealthCheckConfigResponse{HealthCheckJSON: configJSON}, nil
 | 
						|
}
 | 
						|
 | 
						|
// 修改集群健康检查设置
 | 
						|
func (this *NodeClusterService) UpdateNodeClusterHealthCheck(ctx context.Context, req *pb.UpdateNodeClusterHealthCheckRequest) (*pb.RPCSuccess, error) {
 | 
						|
	// 校验请求
 | 
						|
	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	err = models.SharedNodeClusterDAO.UpdateClusterHealthCheck(req.ClusterId, req.HealthCheckJSON)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	return this.Success()
 | 
						|
}
 | 
						|
 | 
						|
// 执行健康检查
 | 
						|
func (this *NodeClusterService) ExecuteNodeClusterHealthCheck(ctx context.Context, req *pb.ExecuteNodeClusterHealthCheckRequest) (*pb.ExecuteNodeClusterHealthCheckResponse, error) {
 | 
						|
	// 校验请求
 | 
						|
	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	executor := tasks.NewHealthCheckExecutor(req.ClusterId)
 | 
						|
	results, err := executor.Run()
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	pbResults := []*pb.ExecuteNodeClusterHealthCheckResponse_Result{}
 | 
						|
	for _, result := range results {
 | 
						|
		pbResults = append(pbResults, &pb.ExecuteNodeClusterHealthCheckResponse_Result{
 | 
						|
			Node: &pb.Node{
 | 
						|
				Id:   int64(result.Node.Id),
 | 
						|
				Name: result.Node.Name,
 | 
						|
			},
 | 
						|
			NodeAddr: result.NodeAddr,
 | 
						|
			IsOk:     result.IsOk,
 | 
						|
			Error:    result.Error,
 | 
						|
			CostMs:   types.Float32(result.CostMs),
 | 
						|
		})
 | 
						|
	}
 | 
						|
	return &pb.ExecuteNodeClusterHealthCheckResponse{Results: pbResults}, nil
 | 
						|
}
 | 
						|
 | 
						|
// 计算使用某个认证的集群数量
 | 
						|
func (this *NodeClusterService) CountAllEnabledNodeClustersWithGrantId(ctx context.Context, req *pb.CountAllEnabledNodeClustersWithGrantIdRequest) (*pb.RPCCountResponse, error) {
 | 
						|
	// 校验请求
 | 
						|
	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	count, err := models.SharedNodeClusterDAO.CountAllEnabledClustersWithGrantId(req.GrantId)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	return this.SuccessCount(count)
 | 
						|
}
 | 
						|
 | 
						|
// 查找使用某个认证的所有集群
 | 
						|
func (this *NodeClusterService) FindAllEnabledNodeClustersWithGrantId(ctx context.Context, req *pb.FindAllEnabledNodeClustersWithGrantIdRequest) (*pb.FindAllEnabledNodeClustersWithGrantIdResponse, error) {
 | 
						|
	// 校验请求
 | 
						|
	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	clusters, err := models.SharedNodeClusterDAO.FindAllEnabledClustersWithGrantId(req.GrantId)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	result := []*pb.NodeCluster{}
 | 
						|
	for _, cluster := range clusters {
 | 
						|
		result = append(result, &pb.NodeCluster{
 | 
						|
			Id:        int64(cluster.Id),
 | 
						|
			Name:      cluster.Name,
 | 
						|
			CreatedAt: int64(cluster.CreatedAt),
 | 
						|
			UniqueId:  cluster.UniqueId,
 | 
						|
			Secret:    cluster.Secret,
 | 
						|
		})
 | 
						|
	}
 | 
						|
	return &pb.FindAllEnabledNodeClustersWithGrantIdResponse{Clusters: result}, nil
 | 
						|
}
 | 
						|
 | 
						|
// 查找集群的DNS配置
 | 
						|
func (this *NodeClusterService) FindEnabledNodeClusterDNS(ctx context.Context, req *pb.FindEnabledNodeClusterDNSRequest) (*pb.FindEnabledNodeClusterDNSResponse, error) {
 | 
						|
	// 校验请求
 | 
						|
	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	dnsInfo, err := models.SharedNodeClusterDAO.FindClusterDNSInfo(req.NodeClusterId)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	if dnsInfo == nil {
 | 
						|
		return &pb.FindEnabledNodeClusterDNSResponse{
 | 
						|
			Name:     "",
 | 
						|
			Domain:   nil,
 | 
						|
			Provider: nil,
 | 
						|
		}, nil
 | 
						|
	}
 | 
						|
 | 
						|
	dnsConfig, err := dnsInfo.DecodeDNSConfig()
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	if dnsInfo.DnsDomainId == 0 {
 | 
						|
		return &pb.FindEnabledNodeClusterDNSResponse{
 | 
						|
			Name:            dnsInfo.DnsName,
 | 
						|
			Domain:          nil,
 | 
						|
			Provider:        nil,
 | 
						|
			NodesAutoSync:   dnsConfig.NodesAutoSync,
 | 
						|
			ServersAutoSync: dnsConfig.ServersAutoSync,
 | 
						|
		}, nil
 | 
						|
	}
 | 
						|
 | 
						|
	domain, err := models.SharedDNSDomainDAO.FindEnabledDNSDomain(int64(dnsInfo.DnsDomainId))
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	if domain == nil {
 | 
						|
		return &pb.FindEnabledNodeClusterDNSResponse{
 | 
						|
			Name:     dnsInfo.DnsName,
 | 
						|
			Domain:   nil,
 | 
						|
			Provider: nil,
 | 
						|
		}, nil
 | 
						|
	}
 | 
						|
	pbDomain := &pb.DNSDomain{
 | 
						|
		Id:   int64(domain.Id),
 | 
						|
		Name: domain.Name,
 | 
						|
		IsOn: domain.IsOn == 1,
 | 
						|
	}
 | 
						|
 | 
						|
	provider, err := models.SharedDNSProviderDAO.FindEnabledDNSProvider(int64(domain.ProviderId))
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	var pbProvider *pb.DNSProvider = nil
 | 
						|
	if provider != nil {
 | 
						|
		pbProvider = &pb.DNSProvider{
 | 
						|
			Id:       int64(provider.Id),
 | 
						|
			Name:     provider.Name,
 | 
						|
			Type:     provider.Type,
 | 
						|
			TypeName: dnsclients.FindProviderTypeName(provider.Type),
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	return &pb.FindEnabledNodeClusterDNSResponse{
 | 
						|
		Name:            dnsInfo.DnsName,
 | 
						|
		Domain:          pbDomain,
 | 
						|
		Provider:        pbProvider,
 | 
						|
		NodesAutoSync:   dnsConfig.NodesAutoSync,
 | 
						|
		ServersAutoSync: dnsConfig.ServersAutoSync,
 | 
						|
	}, nil
 | 
						|
}
 | 
						|
 | 
						|
// 计算使用某个DNS服务商的集群数量
 | 
						|
func (this *NodeClusterService) CountAllEnabledNodeClustersWithDNSProviderId(ctx context.Context, req *pb.CountAllEnabledNodeClustersWithDNSProviderIdRequest) (*pb.RPCCountResponse, error) {
 | 
						|
	// 校验请求
 | 
						|
	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	count, err := models.SharedNodeClusterDAO.CountAllEnabledClustersWithDNSProviderId(req.DnsProviderId)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	return this.SuccessCount(count)
 | 
						|
}
 | 
						|
 | 
						|
// 计算使用某个DNS域名的集群数量
 | 
						|
func (this *NodeClusterService) CountAllEnabledNodeClustersWithDNSDomainId(ctx context.Context, req *pb.CountAllEnabledNodeClustersWithDNSDomainIdRequest) (*pb.RPCCountResponse, error) {
 | 
						|
	// 校验请求
 | 
						|
	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	count, err := models.SharedNodeClusterDAO.CountAllEnabledClustersWithDNSDomainId(req.DnsDomainId)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	return this.SuccessCount(count)
 | 
						|
}
 | 
						|
 | 
						|
// 检查集群域名是否已经被使用
 | 
						|
func (this *NodeClusterService) CheckNodeClusterDNSName(ctx context.Context, req *pb.CheckNodeClusterDNSNameRequest) (*pb.CheckNodeClusterDNSNameResponse, error) {
 | 
						|
	// 校验请求
 | 
						|
	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	exists, err := models.SharedNodeClusterDAO.ExistClusterDNSName(req.DnsName, req.NodeClusterId)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	return &pb.CheckNodeClusterDNSNameResponse{IsUsed: exists}, nil
 | 
						|
}
 | 
						|
 | 
						|
// 修改集群的域名设置
 | 
						|
func (this *NodeClusterService) UpdateNodeClusterDNS(ctx context.Context, req *pb.UpdateNodeClusterDNSRequest) (*pb.RPCSuccess, error) {
 | 
						|
	// 校验请求
 | 
						|
	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	err = models.SharedNodeClusterDAO.UpdateClusterDNS(req.NodeClusterId, req.DnsName, req.DnsDomainId, req.NodesAutoSync, req.ServersAutoSync)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	return this.Success()
 | 
						|
}
 | 
						|
 | 
						|
// 检查集群的DNS是否有变化
 | 
						|
func (this *NodeClusterService) CheckNodeClusterDNSChanges(ctx context.Context, req *pb.CheckNodeClusterDNSChangesRequest) (*pb.CheckNodeClusterDNSChangesResponse, error) {
 | 
						|
	// 校验请求
 | 
						|
	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	cluster, err := models.SharedNodeClusterDAO.FindClusterDNSInfo(req.NodeClusterId)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	if cluster == nil || len(cluster.DnsName) == 0 || cluster.DnsDomainId <= 0 {
 | 
						|
		return &pb.CheckNodeClusterDNSChangesResponse{IsChanged: false}, nil
 | 
						|
	}
 | 
						|
 | 
						|
	domainId := int64(cluster.DnsDomainId)
 | 
						|
	domain, err := models.SharedDNSDomainDAO.FindEnabledDNSDomain(domainId)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	if domain == nil {
 | 
						|
		return &pb.CheckNodeClusterDNSChangesResponse{IsChanged: false}, nil
 | 
						|
	}
 | 
						|
	records, err := domain.DecodeRecords()
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	service := &DNSDomainService{}
 | 
						|
	changes, _, _, _, _, err := service.findClusterDNSChanges(cluster, records, domain.Name)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	return &pb.CheckNodeClusterDNSChangesResponse{IsChanged: len(changes) > 0}, nil
 | 
						|
}
 | 
						|
 | 
						|
// 查找集群的TOA配置
 | 
						|
func (this *NodeClusterService) FindEnabledNodeClusterTOA(ctx context.Context, req *pb.FindEnabledNodeClusterTOARequest) (*pb.FindEnabledNodeClusterTOAResponse, error) {
 | 
						|
	_, err := this.ValidateAdmin(ctx, 0)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	// TODO 检查权限
 | 
						|
 | 
						|
	config, err := models.SharedNodeClusterDAO.FindClusterTOAConfig(req.NodeClusterId)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	configJSON, err := json.Marshal(config)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	return &pb.FindEnabledNodeClusterTOAResponse{ToaJSON: configJSON}, nil
 | 
						|
}
 | 
						|
 | 
						|
// 修改集群的TOA设置
 | 
						|
func (this *NodeClusterService) UpdateNodeClusterTOA(ctx context.Context, req *pb.UpdateNodeClusterTOARequest) (*pb.RPCSuccess, error) {
 | 
						|
	_, err := this.ValidateAdmin(ctx, 0)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	// TODO 检查权限
 | 
						|
 | 
						|
	err = models.SharedNodeClusterDAO.UpdateClusterTOA(req.NodeClusterId, req.ToaJSON)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	// 更新节点版本
 | 
						|
	err = models.SharedNodeDAO.UpdateAllNodesLatestVersionMatch(req.NodeClusterId)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	return this.Success()
 | 
						|
}
 |