增加DNS服务商账号管理

internal/rpc/services/service_dns_provider.go

@@ -0,0 +1,146 @@
+package services
+
+import (
+	"context"
+	"github.com/TeaOSLab/EdgeAPI/internal/db/models"
+	"github.com/TeaOSLab/EdgeAPI/internal/dnsproviders"
+	rpcutils "github.com/TeaOSLab/EdgeAPI/internal/rpc/utils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+)
+
+// DNS服务商相关服务
+type DNSProviderService struct {
+}
+
+// 创建服务商
+func (this *DNSProviderService) CreateDNSProvider(ctx context.Context, req *pb.CreateDNSProviderRequest) (*pb.CreateDNSProviderResponse, error) {
+	// 校验请求
+	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
+	if err != nil {
+		return nil, err
+	}
+
+	providerId, err := models.SharedDNSProviderDAO.CreateDNSProvider(req.Type, req.Name, req.ApiParamsJSON, req.RoutesJSON)
+	if err != nil {
+		return nil, err
+	}
+
+	return &pb.CreateDNSProviderResponse{DnsProviderId: providerId}, nil
+}
+
+// 修改服务商
+func (this *DNSProviderService) UpdateDNSProvider(ctx context.Context, req *pb.UpdateDNSProviderRequest) (*pb.RPCUpdateSuccess, error) {
+	// 校验请求
+	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
+	if err != nil {
+		return nil, err
+	}
+
+	err = models.SharedDNSProviderDAO.UpdateDNSProvider(req.DnsProviderId, req.Name, req.ApiParamsJSON, req.RoutesJSON)
+	if err != nil {
+		return nil, err
+	}
+	return rpcutils.RPCUpdateSuccess()
+}
+
+// 计算服务商数量
+func (this *DNSProviderService) CountAllEnabledDNSProviders(ctx context.Context, req *pb.CountAllEnabledDNSProvidersRequest) (*pb.CountAllEnabledDNSProvidersResponse, error) {
+	// 校验请求
+	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
+	if err != nil {
+		return nil, err
+	}
+
+	count, err := models.SharedDNSProviderDAO.CountAllEnabledDNSProviders()
+	if err != nil {
+		return nil, err
+	}
+	return &pb.CountAllEnabledDNSProvidersResponse{Count: count}, nil
+}
+
+// 列出单页服务商信息
+func (this *DNSProviderService) ListEnabledDNSProviders(ctx context.Context, req *pb.ListEnabledDNSProvidersRequest) (*pb.ListEnabledDNSProvidersResponse, error) {
+	// 校验请求
+	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
+	if err != nil {
+		return nil, err
+	}
+
+	providers, err := models.SharedDNSProviderDAO.ListEnabledDNSProviders(req.Offset, req.Size)
+	if err != nil {
+		return nil, err
+	}
+	result := []*pb.DNSProvider{}
+	for _, provider := range providers {
+		result = append(result, &pb.DNSProvider{
+			Id:            int64(provider.Id),
+			Name:          provider.Name,
+			Type:          provider.Type,
+			TypeName:      dnsproviders.FindProviderTypeName(provider.Type),
+			ApiParamsJSON: []byte(provider.ApiParams),
+			RoutesJSON:    []byte(provider.Routes),
+			DataUpdatedAt: int64(provider.DataUpdatedAt),
+		})
+	}
+	return &pb.ListEnabledDNSProvidersResponse{DnsProviders: result}, nil
+}
+
+// 删除服务商
+func (this *DNSProviderService) DeleteDNSProvider(ctx context.Context, req *pb.DeleteDNSProviderRequest) (*pb.RPCDeleteSuccess, error) {
+	// 校验请求
+	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
+	if err != nil {
+		return nil, err
+	}
+
+	err = models.SharedDNSProviderDAO.DisableDNSProvider(req.DnsProviderId)
+	if err != nil {
+		return nil, err
+	}
+	return rpcutils.RPCDeleteSuccess()
+}
+
+// 查找单个服务商
+func (this *DNSProviderService) FindEnabledDNSProvider(ctx context.Context, req *pb.FindEnabledDNSProviderRequest) (*pb.FindEnabledDNSProviderResponse, error) {
+	// 校验请求
+	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
+	if err != nil {
+		return nil, err
+	}
+
+	provider, err := models.SharedDNSProviderDAO.FindEnabledDNSProvider(req.DnsProviderId)
+	if err != nil {
+		return nil, err
+	}
+	if provider == nil {
+		return &pb.FindEnabledDNSProviderResponse{DnsProvider: nil}, nil
+	}
+
+	return &pb.FindEnabledDNSProviderResponse{DnsProvider: &pb.DNSProvider{
+		Id:            int64(provider.Id),
+		Name:          provider.Name,
+		Type:          provider.Type,
+		TypeName:      dnsproviders.FindProviderTypeName(provider.Type),
+		ApiParamsJSON: []byte(provider.ApiParams),
+		RoutesJSON:    []byte(provider.Routes),
+		DataUpdatedAt: int64(provider.DataUpdatedAt),
+	}}, nil
+}
+
+// 取得所有服务商类型
+func (this *DNSProviderService) FindAllDNSProviderTypes(ctx context.Context, req *pb.FindAllDNSProviderTypesRequest) (*pb.FindAllDNSProviderTypesResponse, error) {
+	// 校验请求
+	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
+	if err != nil {
+		return nil, err
+	}
+
+	result := []*pb.DNSProviderType{}
+	for _, t := range dnsproviders.AllProviderTypes {
+		result = append(result, &pb.DNSProviderType{
+			Name: t.GetString("name"),
+			Code: t.GetString("code"),
+		})
+	}
+	return &pb.FindAllDNSProviderTypesResponse{ProviderTypes: result}, nil
+}

internal/rpc/services/service_node_cluster.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"encoding/json"
 	"github.com/TeaOSLab/EdgeAPI/internal/db/models"
+	"github.com/TeaOSLab/EdgeAPI/internal/dnsproviders"
 	"github.com/TeaOSLab/EdgeAPI/internal/errors"
 	rpcutils "github.com/TeaOSLab/EdgeAPI/internal/rpc/utils"
 	"github.com/TeaOSLab/EdgeAPI/internal/tasks"
@@ -350,3 +351,59 @@ func (this *NodeClusterService) FindAllEnabledNodeClustersWithGrantId(ctx contex
 	}
 	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:   "",
+			Provider: nil,
+		}, nil
+	}
+
+	provider, err := models.SharedDNSProviderDAO.FindEnabledDNSProvider(int64(dnsInfo.DnsProviderId))
+	if err != nil {
+		return nil, err
+	}
+
+	var pbProvider *pb.DNSProvider = nil
+	if provider != nil {
+		pbProvider = &pb.DNSProvider{
+			Id:       int64(provider.Id),
+			Type:     provider.Type,
+			TypeName: dnsproviders.FindProviderTypeName(provider.Type),
+		}
+	}
+
+	return &pb.FindEnabledNodeClusterDNSResponse{
+		Name:     dnsInfo.DnsName,
+		Domain:   dnsInfo.DnsDomain,
+		Provider: pbProvider,
+	}, nil
+}
+
+// 计算使用某个DNS服务商的集群数量
+func (this *NodeClusterService) CountAllEnabledNodeClustersWithDNSProviderId(ctx context.Context, req *pb.CountAllEnabledNodeClustersWithDNSProviderIdRequest) (*pb.CountAllEnabledNodeClustersWithDNSProviderIdResponse, 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 &pb.CountAllEnabledNodeClustersWithDNSProviderIdResponse{Count: count}, nil
+}

internal/rpc/services/service_server.go

@@ -420,6 +420,7 @@ func (this *ServerService) ListEnabledServersMatch(ctx context.Context, req *pb.
 			ExcludeNodes:   []byte(server.ExcludeNodes),
 			ServerNamesJON: []byte(server.ServerNames),
 			CreatedAt:      int64(server.CreatedAt),
+			DnsName:        server.DnsName,
 			Cluster: &pb.NodeCluster{
 				Id:   int64(server.ClusterId),
 				Name: clusterName,
@@ -515,6 +516,7 @@ func (this *ServerService) FindEnabledServer(ctx context.Context, req *pb.FindEn
 		Type:             server.Type,
 		Name:             server.Name,
 		Description:      server.Description,
+		DnsName:          server.DnsName,
 		Config:           []byte(server.Config),
 		ServerNamesJON:   []byte(server.ServerNames),
 		HttpJSON:         []byte(server.Http),