Files
EdgeAPI/internal/rpc/services/service_node_cluster.go

1544 lines
44 KiB
Go
Raw Normal View History

2020-07-29 19:02:28 +08:00
package services
import (
"context"
"encoding/json"
2023-05-23 19:16:30 +08:00
teaconst "github.com/TeaOSLab/EdgeAPI/internal/const"
2020-07-29 19:02:28 +08:00
"github.com/TeaOSLab/EdgeAPI/internal/db/models"
"github.com/TeaOSLab/EdgeAPI/internal/db/models/dns"
"github.com/TeaOSLab/EdgeAPI/internal/db/models/dns/dnsutils"
"github.com/TeaOSLab/EdgeAPI/internal/dnsclients"
"github.com/TeaOSLab/EdgeAPI/internal/errors"
"github.com/TeaOSLab/EdgeAPI/internal/tasks"
2021-06-27 21:59:37 +08:00
"github.com/TeaOSLab/EdgeCommon/pkg/nodeconfigs"
2020-09-13 20:37:28 +08:00
"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
2021-06-27 21:59:37 +08:00
"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs"
2022-05-18 21:02:53 +08:00
"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/ddosconfigs"
2021-01-11 18:16:04 +08:00
"github.com/iwind/TeaGo/dbs"
"github.com/iwind/TeaGo/lists"
2021-01-11 18:16:04 +08:00
"github.com/iwind/TeaGo/maps"
"github.com/iwind/TeaGo/rands"
"github.com/iwind/TeaGo/types"
"strconv"
2020-07-29 19:02:28 +08:00
)
type NodeClusterService struct {
2020-11-24 15:02:44 +08:00
BaseService
2020-07-29 19:02:28 +08:00
}
2021-05-03 11:32:43 +08:00
// CreateNodeCluster 创建集群
2020-08-30 16:12:00 +08:00
func (this *NodeClusterService) CreateNodeCluster(ctx context.Context, req *pb.CreateNodeClusterRequest) (*pb.CreateNodeClusterResponse, error) {
2022-07-22 14:35:17 +08:00
adminId, err := this.ValidateAdmin(ctx)
2020-08-30 16:12:00 +08:00
if err != nil {
return nil, err
}
// 全局服务配置
var serverGlobalConfig = serverconfigs.NewGlobalServerConfig()
if len(req.GlobalServerConfigJSON) > 0 {
err = json.Unmarshal(req.GlobalServerConfigJSON, serverGlobalConfig)
if err != nil {
return nil, err
}
err = serverGlobalConfig.Init()
if err != nil {
return nil, err
}
}
// 系统服务
var systemServices = map[string]maps.Map{}
2021-01-11 18:16:04 +08:00
if len(req.SystemServicesJSON) > 0 {
err = json.Unmarshal(req.SystemServicesJSON, &systemServices)
if err != nil {
return nil, err
}
}
var clusterId int64
err = this.RunTx(func(tx *dbs.Tx) error {
// 缓存策略
if req.HttpCachePolicyId <= 0 {
policyId, err := models.SharedHTTPCachePolicyDAO.CreateDefaultCachePolicy(tx, req.Name)
if err != nil {
return err
}
req.HttpCachePolicyId = policyId
}
// WAF策略
if req.HttpFirewallPolicyId <= 0 {
policyId, err := models.SharedHTTPFirewallPolicyDAO.CreateDefaultFirewallPolicy(tx, req.Name)
if err != nil {
return err
}
req.HttpFirewallPolicyId = policyId
}
// DNS
if req.DnsTTL < 0 {
req.DnsTTL = 0
}
2022-09-17 21:05:18 +08:00
clusterId, err = models.SharedNodeClusterDAO.CreateCluster(tx, adminId, req.Name, req.NodeGrantId, req.InstallDir, req.DnsDomainId, req.DnsName, req.DnsTTL, req.HttpCachePolicyId, req.HttpFirewallPolicyId, systemServices, serverGlobalConfig, req.AutoInstallNftables)
2021-01-11 18:16:04 +08:00
if err != nil {
return err
}
return nil
})
2020-08-30 16:12:00 +08:00
if err != nil {
return nil, err
}
return &pb.CreateNodeClusterResponse{NodeClusterId: clusterId}, nil
2020-08-30 16:12:00 +08:00
}
2021-05-03 11:32:43 +08:00
// UpdateNodeCluster 修改集群
func (this *NodeClusterService) UpdateNodeCluster(ctx context.Context, req *pb.UpdateNodeClusterRequest) (*pb.RPCSuccess, error) {
2022-07-22 14:35:17 +08:00
_, err := this.ValidateAdmin(ctx)
2020-08-30 16:12:00 +08:00
if err != nil {
return nil, err
}
2022-07-22 15:05:30 +08:00
var tx = this.NullTx()
2022-09-15 15:56:50 +08:00
// validate clock
var clockConfig = nodeconfigs.DefaultClockConfig()
if len(req.ClockJSON) > 0 {
err = json.Unmarshal(req.ClockJSON, clockConfig)
if err != nil {
return nil, errors.New("decode clock failed: " + err.Error())
}
err = clockConfig.Init()
if err != nil {
return nil, errors.New("validate clock failed: " + err.Error())
}
}
// ssh params
var sshParams *nodeconfigs.SSHParams
if len(req.SshParamsJSON) > 0 {
sshParams = nodeconfigs.DefaultSSHParams()
err = json.Unmarshal(req.SshParamsJSON, sshParams)
if err != nil {
return nil, err
}
}
err = models.SharedNodeClusterDAO.UpdateCluster(tx, req.NodeClusterId, req.Name, req.NodeGrantId, req.InstallDir, req.TimeZone, req.NodeMaxThreads, req.AutoOpenPorts, clockConfig, req.AutoRemoteStart, req.AutoInstallNftables, sshParams)
2020-08-30 16:12:00 +08:00
if err != nil {
return nil, err
}
2020-11-24 17:36:47 +08:00
return this.Success()
2020-08-30 16:12:00 +08:00
}
2021-05-03 11:32:43 +08:00
// DeleteNodeCluster 禁用集群
func (this *NodeClusterService) DeleteNodeCluster(ctx context.Context, req *pb.DeleteNodeClusterRequest) (*pb.RPCSuccess, error) {
2022-07-22 14:35:17 +08:00
_, err := this.ValidateAdmin(ctx)
2020-08-30 16:12:00 +08:00
if err != nil {
return nil, err
}
2021-07-31 22:23:11 +08:00
if req.NodeClusterId <= 0 {
return this.Success()
}
2022-07-22 15:05:30 +08:00
var tx = this.NullTx()
2021-07-31 22:23:11 +08:00
// 转移节点
err = models.SharedNodeDAO.TransferPrimaryClusterNodes(tx, req.NodeClusterId)
if err != nil {
return nil, err
}
// 删除现有的DNS相关任务
err = dns.SharedDNSTaskDAO.DeleteDNSTasksWithClusterId(tx, req.NodeClusterId)
if err != nil {
return nil, err
}
// 生成新的删除DNS任务
dnsInfo, err := models.SharedNodeClusterDAO.FindClusterDNSInfo(tx, req.NodeClusterId, nil)
2020-08-30 16:12:00 +08:00
if err != nil {
return nil, err
}
if dnsInfo != nil {
var oldDNSDomainId = int64(dnsInfo.DnsDomainId)
err = dns.SharedDNSTaskDAO.CreateClusterRemoveTask(tx, req.NodeClusterId, oldDNSDomainId, dnsInfo.DnsName)
if err != nil {
return nil, err
}
}
2020-08-30 16:12:00 +08:00
// 删除相关任务
2021-08-08 15:47:48 +08:00
err = models.SharedNodeTaskDAO.DeleteAllClusterTasks(tx, nodeconfigs.NodeRoleNode, req.NodeClusterId)
if err != nil {
return nil, err
}
// 删除集群
err = models.SharedNodeClusterDAO.DisableNodeCluster(tx, req.NodeClusterId)
if err != nil {
return nil, err
}
2020-11-24 15:02:44 +08:00
return this.Success()
2020-08-30 16:12:00 +08:00
}
2021-05-03 11:32:43 +08:00
// FindEnabledNodeCluster 查找单个集群
2020-08-30 16:12:00 +08:00
func (this *NodeClusterService) FindEnabledNodeCluster(ctx context.Context, req *pb.FindEnabledNodeClusterRequest) (*pb.FindEnabledNodeClusterResponse, error) {
2022-09-17 16:07:37 +08:00
_, userId, err := this.ValidateAdminAndUser(ctx, false)
2020-08-30 16:12:00 +08:00
if err != nil {
return nil, err
}
if userId > 0 {
// TODO 检查用户是否有权限
2022-09-17 15:11:34 +08:00
// 禁止通过REST访问
// TODO
}
2022-07-22 15:05:30 +08:00
var tx = this.NullTx()
cluster, err := models.SharedNodeClusterDAO.FindEnabledNodeCluster(tx, req.NodeClusterId)
2020-08-30 16:12:00 +08:00
if err != nil {
return nil, err
}
if cluster == nil {
return &pb.FindEnabledNodeClusterResponse{}, nil
}
2020-12-17 17:36:20 +08:00
return &pb.FindEnabledNodeClusterResponse{NodeCluster: &pb.NodeCluster{
2022-05-18 21:02:53 +08:00
Id: int64(cluster.Id),
Name: cluster.Name,
CreatedAt: int64(cluster.CreatedAt),
InstallDir: cluster.InstallDir,
NodeGrantId: int64(cluster.GrantId),
SshParamsJSON: cluster.SshParams,
2022-05-18 21:02:53 +08:00
UniqueId: cluster.UniqueId,
Secret: cluster.Secret,
HttpCachePolicyId: int64(cluster.CachePolicyId),
HttpFirewallPolicyId: int64(cluster.HttpFirewallPolicyId),
DnsName: cluster.DnsName,
DnsDomainId: int64(cluster.DnsDomainId),
IsOn: cluster.IsOn,
TimeZone: cluster.TimeZone,
NodeMaxThreads: int32(cluster.NodeMaxThreads),
AutoOpenPorts: cluster.AutoOpenPorts == 1,
2022-09-15 15:56:50 +08:00
ClockJSON: cluster.Clock,
2022-09-17 15:11:34 +08:00
AutoRemoteStart: cluster.AutoRemoteStart,
2022-09-17 21:05:18 +08:00
AutoInstallNftables: cluster.AutoInstallNftables,
2020-08-30 16:12:00 +08:00
}}, nil
}
2021-05-03 11:32:43 +08:00
// FindAPINodesWithNodeCluster 查找集群的API节点信息
func (this *NodeClusterService) FindAPINodesWithNodeCluster(ctx context.Context, req *pb.FindAPINodesWithNodeClusterRequest) (*pb.FindAPINodesWithNodeClusterResponse, error) {
// 校验请求
2022-07-22 14:35:17 +08:00
_, err := this.ValidateAdmin(ctx)
if err != nil {
return nil, err
}
2022-07-22 15:05:30 +08:00
var tx = this.NullTx()
cluster, err := models.SharedNodeClusterDAO.FindEnabledNodeCluster(tx, req.NodeClusterId)
if err != nil {
return nil, err
}
if cluster == nil {
return nil, errors.New("can not find cluster with id '" + strconv.FormatInt(req.NodeClusterId, 10) + "'")
}
result := &pb.FindAPINodesWithNodeClusterResponse{}
result.UseAllAPINodes = cluster.UseAllAPINodes == 1
apiNodeIds := []int64{}
if models.IsNotNull(cluster.ApiNodes) {
err = json.Unmarshal(cluster.ApiNodes, &apiNodeIds)
if err != nil {
return nil, err
}
if len(apiNodeIds) > 0 {
apiNodes := []*pb.APINode{}
for _, apiNodeId := range apiNodeIds {
2022-01-19 16:53:52 +08:00
apiNode, err := models.SharedAPINodeDAO.FindEnabledAPINode(tx, apiNodeId, nil)
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),
2022-03-22 21:45:07 +08:00
IsOn: apiNode.IsOn,
NodeClusterId: int64(apiNode.ClusterId),
Name: apiNode.Name,
Description: apiNode.Description,
AccessAddrs: apiNodeAddrs,
})
}
result.ApiNodes = apiNodes
}
}
return result, nil
}
2021-05-03 11:32:43 +08:00
// FindAllEnabledNodeClusters 查找所有可用的集群
2020-08-30 16:12:00 +08:00
func (this *NodeClusterService) FindAllEnabledNodeClusters(ctx context.Context, req *pb.FindAllEnabledNodeClustersRequest) (*pb.FindAllEnabledNodeClustersResponse, error) {
// 校验请求
2022-07-22 14:35:17 +08:00
_, err := this.ValidateAdmin(ctx)
2020-07-29 19:02:28 +08:00
if err != nil {
return nil, err
}
2022-07-22 15:05:30 +08:00
var tx = this.NullTx()
clusters, err := models.SharedNodeClusterDAO.FindAllEnableClusters(tx)
2020-07-29 19:02:28 +08:00
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,
2022-03-22 21:45:07 +08:00
IsOn: cluster.IsOn,
2020-07-29 19:02:28 +08:00
})
}
return &pb.FindAllEnabledNodeClustersResponse{
2020-12-17 17:36:20 +08:00
NodeClusters: result,
2020-07-29 19:02:28 +08:00
}, nil
}
2020-08-21 12:32:33 +08:00
2021-05-03 11:32:43 +08:00
// CountAllEnabledNodeClusters 计算所有集群数量
2020-11-12 14:41:28 +08:00
func (this *NodeClusterService) CountAllEnabledNodeClusters(ctx context.Context, req *pb.CountAllEnabledNodeClustersRequest) (*pb.RPCCountResponse, error) {
2022-07-22 14:35:17 +08:00
_, err := this.ValidateAdmin(ctx)
2020-08-30 16:12:00 +08:00
if err != nil {
return nil, err
}
2022-07-22 15:05:30 +08:00
var tx = this.NullTx()
count, err := models.SharedNodeClusterDAO.CountAllEnabledClusters(tx, req.Keyword)
2020-08-30 16:12:00 +08:00
if err != nil {
return nil, err
}
2020-11-24 17:36:47 +08:00
return this.SuccessCount(count)
2020-08-30 16:12:00 +08:00
}
2021-05-03 11:32:43 +08:00
// ListEnabledNodeClusters 列出单页集群
2020-08-30 16:12:00 +08:00
func (this *NodeClusterService) ListEnabledNodeClusters(ctx context.Context, req *pb.ListEnabledNodeClustersRequest) (*pb.ListEnabledNodeClustersResponse, error) {
2022-07-22 14:35:17 +08:00
_, err := this.ValidateAdmin(ctx)
2020-08-30 16:12:00 +08:00
if err != nil {
return nil, err
}
2022-07-22 15:05:30 +08:00
var tx = this.NullTx()
clusters, err := models.SharedNodeClusterDAO.ListEnabledClusters(tx, req.Keyword, req.Offset, req.Size)
2020-08-30 16:12:00 +08:00
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),
2021-05-25 17:48:55 +08:00
NodeGrantId: int64(cluster.GrantId),
InstallDir: cluster.InstallDir,
UniqueId: cluster.UniqueId,
Secret: cluster.Secret,
DnsName: cluster.DnsName,
DnsDomainId: int64(cluster.DnsDomainId),
2022-03-22 21:45:07 +08:00
IsOn: cluster.IsOn,
2021-10-12 14:40:24 +08:00
TimeZone: cluster.TimeZone,
2022-03-22 22:11:32 +08:00
IsPinned: cluster.IsPinned,
2020-08-30 16:12:00 +08:00
})
}
2020-12-17 17:36:20 +08:00
return &pb.ListEnabledNodeClustersResponse{NodeClusters: result}, nil
2020-08-21 12:32:33 +08:00
}
2021-05-03 11:32:43 +08:00
// FindNodeClusterHealthCheckConfig 查找集群的健康检查配置
func (this *NodeClusterService) FindNodeClusterHealthCheckConfig(ctx context.Context, req *pb.FindNodeClusterHealthCheckConfigRequest) (*pb.FindNodeClusterHealthCheckConfigResponse, error) {
// 校验请求
2022-07-22 14:35:17 +08:00
_, err := this.ValidateAdmin(ctx)
if err != nil {
return nil, err
}
2022-04-23 13:26:15 +08:00
var tx = this.NullTx()
config, err := models.SharedNodeClusterDAO.FindClusterHealthCheckConfig(tx, req.NodeClusterId)
if err != nil {
return nil, err
}
2022-04-23 13:26:15 +08:00
if config == nil {
return &pb.FindNodeClusterHealthCheckConfigResponse{HealthCheckJSON: nil}, nil
}
configJSON, err := json.Marshal(config)
if err != nil {
return nil, err
}
2020-12-02 14:26:03 +08:00
return &pb.FindNodeClusterHealthCheckConfigResponse{HealthCheckJSON: configJSON}, nil
}
2021-05-03 11:32:43 +08:00
// UpdateNodeClusterHealthCheck 修改集群健康检查设置
func (this *NodeClusterService) UpdateNodeClusterHealthCheck(ctx context.Context, req *pb.UpdateNodeClusterHealthCheckRequest) (*pb.RPCSuccess, error) {
// 校验请求
2022-07-22 14:35:17 +08:00
_, err := this.ValidateAdmin(ctx)
if err != nil {
return nil, err
}
2022-07-22 15:05:30 +08:00
var tx = this.NullTx()
err = models.SharedNodeClusterDAO.UpdateClusterHealthCheck(tx, req.NodeClusterId, req.HealthCheckJSON)
if err != nil {
return nil, err
}
2020-11-24 15:02:44 +08:00
return this.Success()
}
2021-05-03 11:32:43 +08:00
// ExecuteNodeClusterHealthCheck 执行健康检查
func (this *NodeClusterService) ExecuteNodeClusterHealthCheck(ctx context.Context, req *pb.ExecuteNodeClusterHealthCheckRequest) (*pb.ExecuteNodeClusterHealthCheckResponse, error) {
// 校验请求
2022-07-22 14:35:17 +08:00
_, err := this.ValidateAdmin(ctx)
if err != nil {
return nil, err
}
executor := tasks.NewHealthCheckExecutor(req.NodeClusterId)
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
}
2020-10-25 21:27:46 +08:00
2021-05-25 17:08:58 +08:00
// CountAllEnabledNodeClustersWithNodeGrantId 计算使用某个认证的集群数量
func (this *NodeClusterService) CountAllEnabledNodeClustersWithNodeGrantId(ctx context.Context, req *pb.CountAllEnabledNodeClustersWithNodeGrantIdRequest) (*pb.RPCCountResponse, error) {
2020-10-25 21:27:46 +08:00
// 校验请求
2022-07-22 14:35:17 +08:00
_, err := this.ValidateAdmin(ctx)
2020-10-25 21:27:46 +08:00
if err != nil {
return nil, err
}
2022-07-22 15:05:30 +08:00
var tx = this.NullTx()
2021-05-25 17:48:55 +08:00
count, err := models.SharedNodeClusterDAO.CountAllEnabledClustersWithGrantId(tx, req.NodeGrantId)
2020-10-25 21:27:46 +08:00
if err != nil {
return nil, err
}
2020-11-24 17:36:47 +08:00
return this.SuccessCount(count)
2020-10-25 21:27:46 +08:00
}
2021-05-25 17:08:58 +08:00
// FindAllEnabledNodeClustersWithNodeGrantId 查找使用某个认证的所有集群
func (this *NodeClusterService) FindAllEnabledNodeClustersWithNodeGrantId(ctx context.Context, req *pb.FindAllEnabledNodeClustersWithNodeGrantIdRequest) (*pb.FindAllEnabledNodeClustersWithNodeGrantIdResponse, error) {
2020-10-25 21:27:46 +08:00
// 校验请求
2022-07-22 14:35:17 +08:00
_, err := this.ValidateAdmin(ctx)
2020-10-25 21:27:46 +08:00
if err != nil {
return nil, err
}
2022-07-22 15:05:30 +08:00
var tx = this.NullTx()
2021-05-25 17:48:55 +08:00
clusters, err := models.SharedNodeClusterDAO.FindAllEnabledClustersWithGrantId(tx, req.NodeGrantId)
2020-10-25 21:27:46 +08:00
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,
2022-03-22 21:45:07 +08:00
IsOn: cluster.IsOn,
2020-10-25 21:27:46 +08:00
})
}
2021-05-25 17:08:58 +08:00
return &pb.FindAllEnabledNodeClustersWithNodeGrantIdResponse{NodeClusters: result}, nil
2020-10-25 21:27:46 +08:00
}
2020-11-11 21:32:25 +08:00
2021-05-03 11:32:43 +08:00
// FindEnabledNodeClusterDNS 查找集群的DNS配置
2020-11-11 21:32:25 +08:00
func (this *NodeClusterService) FindEnabledNodeClusterDNS(ctx context.Context, req *pb.FindEnabledNodeClusterDNSRequest) (*pb.FindEnabledNodeClusterDNSResponse, error) {
// 校验请求
2022-09-17 16:07:37 +08:00
_, _, err := this.ValidateAdminAndUser(ctx, false)
2020-11-11 21:32:25 +08:00
if err != nil {
return nil, err
}
var tx = this.NullTx()
2021-08-22 11:35:33 +08:00
dnsInfo, err := models.SharedNodeClusterDAO.FindClusterDNSInfo(tx, req.NodeClusterId, nil)
2020-11-11 21:32:25 +08:00
if err != nil {
return nil, err
}
if dnsInfo == nil {
return &pb.FindEnabledNodeClusterDNSResponse{
Name: "",
2020-11-12 14:41:28 +08:00
Domain: nil,
2020-11-11 21:32:25 +08:00
Provider: nil,
}, nil
}
dnsConfig, err := dnsInfo.DecodeDNSConfig()
if err != nil {
return nil, err
}
2020-11-12 14:41:28 +08:00
if dnsInfo.DnsDomainId == 0 {
return &pb.FindEnabledNodeClusterDNSResponse{
Name: dnsInfo.DnsName,
Domain: nil,
Provider: nil,
NodesAutoSync: dnsConfig.NodesAutoSync,
ServersAutoSync: dnsConfig.ServersAutoSync,
2022-09-21 15:09:39 +08:00
CnameRecords: dnsConfig.CNAMERecords,
Ttl: dnsConfig.TTL,
2022-09-21 15:09:39 +08:00
CnameAsDomain: dnsConfig.CNAMEAsDomain,
IncludingLnNodes: dnsConfig.IncludingLnNodes,
2020-11-12 14:41:28 +08:00
}, nil
}
2021-08-22 11:35:33 +08:00
domain, err := dns.SharedDNSDomainDAO.FindEnabledDNSDomain(tx, int64(dnsInfo.DnsDomainId), nil)
2020-11-12 14:41:28 +08:00
if err != nil {
return nil, err
}
if domain == nil {
return &pb.FindEnabledNodeClusterDNSResponse{
Name: dnsInfo.DnsName,
Domain: nil,
Provider: nil,
}, nil
}
var pbDomain = &pb.DNSDomain{
2020-11-12 14:41:28 +08:00
Id: int64(domain.Id),
Name: domain.Name,
2022-03-22 21:45:07 +08:00
IsOn: domain.IsOn,
2020-11-12 14:41:28 +08:00
}
2020-11-11 21:32:25 +08:00
provider, err := dns.SharedDNSProviderDAO.FindEnabledDNSProvider(tx, int64(domain.ProviderId))
2020-11-11 21:32:25 +08:00
if err != nil {
return nil, err
}
var defaultRoute = ""
2020-11-11 21:32:25 +08:00
var pbProvider *pb.DNSProvider = nil
if provider != nil {
pbProvider = &pb.DNSProvider{
Id: int64(provider.Id),
2020-11-14 09:41:58 +08:00
Name: provider.Name,
2020-11-11 21:32:25 +08:00
Type: provider.Type,
TypeName: dnsclients.FindProviderTypeName(provider.Type),
2020-11-11 21:32:25 +08:00
}
var manager = dnsclients.FindProvider(provider.Type, int64(provider.Id))
if manager != nil {
apiParams, err := provider.DecodeAPIParams()
if err != nil {
return nil, err
}
err = manager.Auth(apiParams)
if err != nil {
return nil, err
}
defaultRoute = manager.DefaultRoute()
}
2020-11-11 21:32:25 +08:00
}
return &pb.FindEnabledNodeClusterDNSResponse{
Name: dnsInfo.DnsName,
Domain: pbDomain,
Provider: pbProvider,
NodesAutoSync: dnsConfig.NodesAutoSync,
ServersAutoSync: dnsConfig.ServersAutoSync,
2022-09-21 15:09:39 +08:00
CnameRecords: dnsConfig.CNAMERecords,
Ttl: dnsConfig.TTL,
2022-09-21 15:09:39 +08:00
CnameAsDomain: dnsConfig.CNAMEAsDomain,
IncludingLnNodes: dnsConfig.IncludingLnNodes,
DefaultRoute: defaultRoute,
2020-11-11 21:32:25 +08:00
}, nil
}
2021-05-03 11:32:43 +08:00
// CountAllEnabledNodeClustersWithDNSProviderId 计算使用某个DNS服务商的集群数量
2020-11-12 14:41:28 +08:00
func (this *NodeClusterService) CountAllEnabledNodeClustersWithDNSProviderId(ctx context.Context, req *pb.CountAllEnabledNodeClustersWithDNSProviderIdRequest) (*pb.RPCCountResponse, error) {
2020-11-11 21:32:25 +08:00
// 校验请求
2022-07-22 14:35:17 +08:00
_, err := this.ValidateAdmin(ctx)
2020-11-11 21:32:25 +08:00
if err != nil {
return nil, err
}
var tx = this.NullTx()
count, err := models.SharedNodeClusterDAO.CountAllEnabledClustersWithDNSProviderId(tx, req.DnsProviderId)
2020-11-11 21:32:25 +08:00
if err != nil {
return nil, err
}
2020-11-24 17:36:47 +08:00
return this.SuccessCount(count)
2020-11-12 14:41:28 +08:00
}
2021-05-03 11:32:43 +08:00
// CountAllEnabledNodeClustersWithDNSDomainId 计算使用某个DNS域名的集群数量
2020-11-12 14:41:28 +08:00
func (this *NodeClusterService) CountAllEnabledNodeClustersWithDNSDomainId(ctx context.Context, req *pb.CountAllEnabledNodeClustersWithDNSDomainIdRequest) (*pb.RPCCountResponse, error) {
// 校验请求
2022-07-22 14:35:17 +08:00
_, err := this.ValidateAdmin(ctx)
2020-11-12 14:41:28 +08:00
if err != nil {
return nil, err
}
var tx = this.NullTx()
count, err := models.SharedNodeClusterDAO.CountAllEnabledClustersWithDNSDomainId(tx, req.DnsDomainId)
2020-11-12 14:41:28 +08:00
if err != nil {
return nil, err
}
2020-11-24 17:36:47 +08:00
return this.SuccessCount(count)
2020-11-11 21:32:25 +08:00
}
2021-05-03 11:32:43 +08:00
// FindAllEnabledNodeClustersWithDNSDomainId 查找使用某个域名的所有集群
2020-12-23 16:49:47 +08:00
func (this *NodeClusterService) FindAllEnabledNodeClustersWithDNSDomainId(ctx context.Context, req *pb.FindAllEnabledNodeClustersWithDNSDomainIdRequest) (*pb.FindAllEnabledNodeClustersWithDNSDomainIdResponse, error) {
// 校验请求
2022-07-22 14:35:17 +08:00
_, err := this.ValidateAdmin(ctx)
2020-12-23 16:49:47 +08:00
if err != nil {
return nil, err
}
var tx = this.NullTx()
clusters, err := models.SharedNodeClusterDAO.FindAllEnabledClustersWithDNSDomainId(tx, req.DnsDomainId)
2020-12-23 16:49:47 +08:00
if err != nil {
return nil, err
}
result := []*pb.NodeCluster{}
for _, cluster := range clusters {
// 默认线路
domainId := int64(cluster.DnsDomainId)
domain, err := dns.SharedDNSDomainDAO.FindEnabledDNSDomain(tx, domainId, nil)
if err != nil {
return nil, err
}
if domain == nil {
continue
}
defaultRoute, err := dnsutils.FindDefaultDomainRoute(tx, domain)
if err != nil {
return nil, err
}
2020-12-23 16:49:47 +08:00
result = append(result, &pb.NodeCluster{
Id: int64(cluster.Id),
Name: cluster.Name,
DnsName: cluster.DnsName,
DnsDomainId: int64(cluster.DnsDomainId),
DnsDefaultRoute: defaultRoute,
IsOn: cluster.IsOn,
2020-12-23 16:49:47 +08:00
})
}
return &pb.FindAllEnabledNodeClustersWithDNSDomainIdResponse{NodeClusters: result}, nil
}
2021-05-03 11:32:43 +08:00
// CheckNodeClusterDNSName 检查集群域名是否已经被使用
func (this *NodeClusterService) CheckNodeClusterDNSName(ctx context.Context, req *pb.CheckNodeClusterDNSNameRequest) (*pb.CheckNodeClusterDNSNameResponse, error) {
// 校验请求
2022-07-22 14:35:17 +08:00
_, err := this.ValidateAdmin(ctx)
if err != nil {
return nil, err
}
var tx = this.NullTx()
exists, err := models.SharedNodeClusterDAO.ExistClusterDNSName(tx, req.DnsName, req.NodeClusterId)
if err != nil {
return nil, err
}
return &pb.CheckNodeClusterDNSNameResponse{IsUsed: exists}, nil
}
2021-05-03 11:32:43 +08:00
// UpdateNodeClusterDNS 修改集群的域名设置
func (this *NodeClusterService) UpdateNodeClusterDNS(ctx context.Context, req *pb.UpdateNodeClusterDNSRequest) (*pb.RPCSuccess, error) {
// 校验请求
2022-07-22 14:35:17 +08:00
_, err := this.ValidateAdmin(ctx)
if err != nil {
return nil, err
}
var tx = this.NullTx()
err = models.SharedNodeClusterDAO.UpdateClusterDNS(tx, req.NodeClusterId, req.DnsName, req.DnsDomainId, req.NodesAutoSync, req.ServersAutoSync, req.CnameRecords, req.Ttl, req.CnameAsDomain, req.IncludingLnNodes)
if err != nil {
return nil, err
}
2020-11-24 15:02:44 +08:00
return this.Success()
}
2020-11-14 21:28:07 +08:00
2021-05-03 11:32:43 +08:00
// CheckNodeClusterDNSChanges 检查集群的DNS是否有变化
2020-11-14 21:28:07 +08:00
func (this *NodeClusterService) CheckNodeClusterDNSChanges(ctx context.Context, req *pb.CheckNodeClusterDNSChangesRequest) (*pb.CheckNodeClusterDNSChangesResponse, error) {
// 校验请求
2022-07-22 14:35:17 +08:00
_, err := this.ValidateAdmin(ctx)
2020-11-14 21:28:07 +08:00
if err != nil {
return nil, err
}
2022-07-22 15:05:30 +08:00
var tx = this.NullTx()
2021-08-22 11:35:33 +08:00
cluster, err := models.SharedNodeClusterDAO.FindClusterDNSInfo(tx, req.NodeClusterId, nil)
2020-11-14 21:28:07 +08:00
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)
2021-08-22 11:35:33 +08:00
domain, err := dns.SharedDNSDomainDAO.FindEnabledDNSDomain(tx, domainId, nil)
2020-11-14 21:28:07 +08:00
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
}
defaultRoute, err := dnsutils.FindDefaultDomainRoute(tx, domain)
if err != nil {
return nil, err
}
2020-11-14 21:28:07 +08:00
service := &DNSDomainService{}
changes, _, _, _, _, _, _, err := service.findClusterDNSChanges(cluster, records, domain.Name, defaultRoute)
2020-11-14 21:28:07 +08:00
if err != nil {
return nil, err
}
2020-11-15 11:57:49 +08:00
2020-11-14 21:28:07 +08:00
return &pb.CheckNodeClusterDNSChangesResponse{IsChanged: len(changes) > 0}, nil
}
2020-12-02 14:26:03 +08:00
2021-05-03 11:32:43 +08:00
// FindEnabledNodeClusterTOA 查找集群的TOA配置
2020-12-02 14:26:03 +08:00
func (this *NodeClusterService) FindEnabledNodeClusterTOA(ctx context.Context, req *pb.FindEnabledNodeClusterTOARequest) (*pb.FindEnabledNodeClusterTOAResponse, error) {
2022-07-22 14:35:17 +08:00
_, err := this.ValidateAdmin(ctx)
2020-12-02 14:26:03 +08:00
if err != nil {
return nil, err
}
// TODO 检查权限
2022-07-22 15:05:30 +08:00
var tx = this.NullTx()
2021-11-11 14:16:42 +08:00
config, err := models.SharedNodeClusterDAO.FindClusterTOAConfig(tx, req.NodeClusterId, nil)
2020-12-02 14:26:03 +08:00
if err != nil {
return nil, err
}
configJSON, err := json.Marshal(config)
if err != nil {
return nil, err
}
return &pb.FindEnabledNodeClusterTOAResponse{ToaJSON: configJSON}, nil
}
2021-05-03 11:32:43 +08:00
// UpdateNodeClusterTOA 修改集群的TOA设置
2020-12-02 14:26:03 +08:00
func (this *NodeClusterService) UpdateNodeClusterTOA(ctx context.Context, req *pb.UpdateNodeClusterTOARequest) (*pb.RPCSuccess, error) {
2022-07-22 14:35:17 +08:00
_, err := this.ValidateAdmin(ctx)
2020-12-02 14:26:03 +08:00
if err != nil {
return nil, err
}
// TODO 检查权限
2022-07-22 15:05:30 +08:00
var tx = this.NullTx()
err = models.SharedNodeClusterDAO.UpdateClusterTOA(tx, req.NodeClusterId, req.ToaJSON)
2020-12-02 14:26:03 +08:00
if err != nil {
return nil, err
}
return this.Success()
}
2021-05-03 11:32:43 +08:00
// CountAllEnabledNodeClustersWithHTTPCachePolicyId 计算使用某个缓存策略的集群数量
func (this *NodeClusterService) CountAllEnabledNodeClustersWithHTTPCachePolicyId(ctx context.Context, req *pb.CountAllEnabledNodeClustersWithHTTPCachePolicyIdRequest) (*pb.RPCCountResponse, error) {
2022-07-22 14:35:17 +08:00
_, err := this.ValidateAdmin(ctx)
if err != nil {
return nil, err
}
2022-07-22 15:05:30 +08:00
var tx = this.NullTx()
count, err := models.SharedNodeClusterDAO.CountAllEnabledNodeClustersWithHTTPCachePolicyId(tx, req.HttpCachePolicyId)
if err != nil {
return nil, err
}
return this.SuccessCount(count)
}
2021-05-03 11:32:43 +08:00
// FindAllEnabledNodeClustersWithHTTPCachePolicyId 查找使用缓存策略的所有集群
func (this *NodeClusterService) FindAllEnabledNodeClustersWithHTTPCachePolicyId(ctx context.Context, req *pb.FindAllEnabledNodeClustersWithHTTPCachePolicyIdRequest) (*pb.FindAllEnabledNodeClustersWithHTTPCachePolicyIdResponse, error) {
2022-07-22 14:35:17 +08:00
_, err := this.ValidateAdmin(ctx)
if err != nil {
return nil, err
}
2022-07-22 15:05:30 +08:00
var tx = this.NullTx()
result := []*pb.NodeCluster{}
clusters, err := models.SharedNodeClusterDAO.FindAllEnabledNodeClustersWithHTTPCachePolicyId(tx, req.HttpCachePolicyId)
if err != nil {
return nil, err
}
for _, cluster := range clusters {
result = append(result, &pb.NodeCluster{
Id: int64(cluster.Id),
Name: cluster.Name,
2022-03-22 21:45:07 +08:00
IsOn: cluster.IsOn,
})
}
return &pb.FindAllEnabledNodeClustersWithHTTPCachePolicyIdResponse{
NodeClusters: result,
}, nil
}
2021-05-03 11:32:43 +08:00
// CountAllEnabledNodeClustersWithHTTPFirewallPolicyId 计算使用某个WAF策略的集群数量
func (this *NodeClusterService) CountAllEnabledNodeClustersWithHTTPFirewallPolicyId(ctx context.Context, req *pb.CountAllEnabledNodeClustersWithHTTPFirewallPolicyIdRequest) (*pb.RPCCountResponse, error) {
2022-07-22 14:35:17 +08:00
_, err := this.ValidateAdmin(ctx)
if err != nil {
return nil, err
}
2022-07-22 15:05:30 +08:00
var tx = this.NullTx()
count, err := models.SharedNodeClusterDAO.CountAllEnabledNodeClustersWithHTTPFirewallPolicyId(tx, req.HttpFirewallPolicyId)
if err != nil {
return nil, err
}
return this.SuccessCount(count)
}
2021-05-03 11:32:43 +08:00
// FindAllEnabledNodeClustersWithHTTPFirewallPolicyId 查找使用WAF策略的所有集群
func (this *NodeClusterService) FindAllEnabledNodeClustersWithHTTPFirewallPolicyId(ctx context.Context, req *pb.FindAllEnabledNodeClustersWithHTTPFirewallPolicyIdRequest) (*pb.FindAllEnabledNodeClustersWithHTTPFirewallPolicyIdResponse, error) {
2022-07-22 14:35:17 +08:00
_, err := this.ValidateAdmin(ctx)
if err != nil {
return nil, err
}
2022-07-22 15:05:30 +08:00
var tx = this.NullTx()
result := []*pb.NodeCluster{}
clusters, err := models.SharedNodeClusterDAO.FindAllEnabledNodeClustersWithHTTPFirewallPolicyId(tx, req.HttpFirewallPolicyId)
if err != nil {
return nil, err
}
for _, cluster := range clusters {
result = append(result, &pb.NodeCluster{
Id: int64(cluster.Id),
Name: cluster.Name,
2022-03-22 21:45:07 +08:00
IsOn: cluster.IsOn,
})
}
return &pb.FindAllEnabledNodeClustersWithHTTPFirewallPolicyIdResponse{
NodeClusters: result,
}, nil
}
2021-05-03 11:32:43 +08:00
// UpdateNodeClusterHTTPCachePolicyId 修改集群的缓存策略
func (this *NodeClusterService) UpdateNodeClusterHTTPCachePolicyId(ctx context.Context, req *pb.UpdateNodeClusterHTTPCachePolicyIdRequest) (*pb.RPCSuccess, error) {
2022-07-22 14:35:17 +08:00
_, err := this.ValidateAdmin(ctx)
if err != nil {
return nil, err
}
2022-07-22 15:05:30 +08:00
var tx = this.NullTx()
err = models.SharedNodeClusterDAO.UpdateNodeClusterHTTPCachePolicyId(tx, req.NodeClusterId, req.HttpCachePolicyId)
if err != nil {
return nil, err
}
2020-12-17 17:36:20 +08:00
return this.Success()
}
2021-05-03 11:32:43 +08:00
// UpdateNodeClusterHTTPFirewallPolicyId 修改集群的WAF策略
func (this *NodeClusterService) UpdateNodeClusterHTTPFirewallPolicyId(ctx context.Context, req *pb.UpdateNodeClusterHTTPFirewallPolicyIdRequest) (*pb.RPCSuccess, error) {
2022-07-22 14:35:17 +08:00
_, err := this.ValidateAdmin(ctx)
if err != nil {
return nil, err
}
2022-07-22 15:05:30 +08:00
var tx = this.NullTx()
err = models.SharedNodeClusterDAO.UpdateNodeClusterHTTPFirewallPolicyId(tx, req.NodeClusterId, req.HttpFirewallPolicyId)
if err != nil {
return nil, err
}
2020-12-17 17:36:20 +08:00
return this.Success()
}
2021-01-11 18:16:04 +08:00
2021-05-03 11:32:43 +08:00
// UpdateNodeClusterSystemService 修改集群的系统服务设置
2021-01-11 18:16:04 +08:00
func (this *NodeClusterService) UpdateNodeClusterSystemService(ctx context.Context, req *pb.UpdateNodeClusterSystemServiceRequest) (*pb.RPCSuccess, error) {
2022-07-22 14:35:17 +08:00
_, err := this.ValidateAdmin(ctx)
2021-01-11 18:16:04 +08:00
if err != nil {
return nil, err
}
params := maps.Map{}
if len(req.ParamsJSON) > 0 {
err = json.Unmarshal(req.ParamsJSON, &params)
if err != nil {
return nil, err
}
}
2022-07-22 15:05:30 +08:00
var tx = this.NullTx()
2021-01-11 18:16:04 +08:00
err = models.SharedNodeClusterDAO.UpdateNodeClusterSystemService(tx, req.NodeClusterId, req.Type, params)
if err != nil {
return nil, err
}
return this.Success()
}
2021-05-03 11:32:43 +08:00
// FindNodeClusterSystemService 查找集群的系统服务设置
2021-01-11 18:16:04 +08:00
func (this *NodeClusterService) FindNodeClusterSystemService(ctx context.Context, req *pb.FindNodeClusterSystemServiceRequest) (*pb.FindNodeClusterSystemServiceResponse, error) {
2022-07-22 14:35:17 +08:00
_, err := this.ValidateAdmin(ctx)
2021-01-11 18:16:04 +08:00
if err != nil {
return nil, err
}
2022-07-22 15:05:30 +08:00
var tx = this.NullTx()
2021-01-11 18:16:04 +08:00
params, err := models.SharedNodeClusterDAO.FindNodeClusterSystemServiceParams(tx, req.NodeClusterId, req.Type)
if err != nil {
return nil, err
}
if params == nil {
params = maps.Map{}
}
paramsJSON, err := json.Marshal(params)
if err != nil {
return nil, err
}
return &pb.FindNodeClusterSystemServiceResponse{ParamsJSON: paramsJSON}, nil
}
2021-05-03 11:32:43 +08:00
// FindFreePortInNodeCluster 获取集群中可以使用的端口
func (this *NodeClusterService) FindFreePortInNodeCluster(ctx context.Context, req *pb.FindFreePortInNodeClusterRequest) (*pb.FindFreePortInNodeClusterResponse, error) {
2022-09-17 16:07:37 +08:00
_, _, err := this.ValidateAdminAndUser(ctx, false)
if err != nil {
return nil, err
}
var tx = this.NullTx()
globalConfig, err := models.SharedSysSettingDAO.ReadGlobalConfig(tx)
if err != nil {
return nil, err
}
// 检查端口
portMin := globalConfig.TCPAll.PortRangeMin
portMax := globalConfig.TCPAll.PortRangeMax
denyPorts := globalConfig.TCPAll.DenyPorts
if portMin == 0 && portMax == 0 {
portMin = 10_000
portMax = 40_000
}
if portMin < 1024 {
portMin = 10_000
}
if portMin > 65534 {
portMin = 65534
}
if portMax < 1024 {
portMax = 30_000
}
if portMax > 65534 {
portMax = 65534
}
if portMin > portMax {
portMax, portMin = portMin, portMax
}
// 最多尝试N次
for i := 0; i < 60; i++ {
port := rands.Int(portMin, portMax)
if len(denyPorts) > 0 && lists.ContainsInt(denyPorts, port) {
continue
}
isUsing, err := models.SharedServerDAO.CheckPortIsUsing(tx, req.NodeClusterId, req.ProtocolFamily, port, 0, "")
if err != nil {
return nil, err
}
if !isUsing {
return &pb.FindFreePortInNodeClusterResponse{Port: int32(port)}, nil
}
}
return nil, errors.New("can not find random port")
}
2021-05-03 11:32:43 +08:00
// CheckPortIsUsingInNodeCluster 检查端口是否已经被使用
func (this *NodeClusterService) CheckPortIsUsingInNodeCluster(ctx context.Context, req *pb.CheckPortIsUsingInNodeClusterRequest) (*pb.CheckPortIsUsingInNodeClusterResponse, error) {
2022-09-17 16:07:37 +08:00
_, _, err := this.ValidateAdminAndUser(ctx, true)
if err != nil {
return nil, err
}
var tx = this.NullTx()
isUsing, err := models.SharedServerDAO.CheckPortIsUsing(tx, req.NodeClusterId, req.ProtocolFamily, int(req.Port), req.ExcludeServerId, req.ExcludeProtocol)
if err != nil {
return nil, err
}
return &pb.CheckPortIsUsingInNodeClusterResponse{IsUsing: isUsing}, nil
}
2021-05-03 11:32:43 +08:00
// FindLatestNodeClusters 查找最近访问的集群
func (this *NodeClusterService) FindLatestNodeClusters(ctx context.Context, req *pb.FindLatestNodeClustersRequest) (*pb.FindLatestNodeClustersResponse, error) {
2022-07-22 14:35:17 +08:00
_, err := this.ValidateAdmin(ctx)
2021-05-03 11:32:43 +08:00
if err != nil {
return nil, err
}
var tx = this.NullTx()
clusters, err := models.SharedNodeClusterDAO.FindLatestNodeClusters(tx, req.Size)
if err != nil {
return nil, err
}
pbClusters := []*pb.NodeCluster{}
for _, cluster := range clusters {
pbClusters = append(pbClusters, &pb.NodeCluster{
Id: int64(cluster.Id),
Name: cluster.Name,
2022-03-22 21:45:07 +08:00
IsOn: cluster.IsOn,
2021-05-03 11:32:43 +08:00
})
}
return &pb.FindLatestNodeClustersResponse{NodeClusters: pbClusters}, nil
}
2021-06-27 21:59:37 +08:00
// FindEnabledNodeClusterConfigInfo 取得集群的配置概要信息
func (this *NodeClusterService) FindEnabledNodeClusterConfigInfo(ctx context.Context, req *pb.FindEnabledNodeClusterConfigInfoRequest) (*pb.FindEnabledNodeClusterConfigInfoResponse, error) {
2022-07-22 14:35:17 +08:00
_, err := this.ValidateAdmin(ctx)
2021-06-27 21:59:37 +08:00
if err != nil {
return nil, err
}
var tx = this.NullTx()
cluster, err := models.SharedNodeClusterDAO.FindEnabledNodeCluster(tx, req.NodeClusterId)
if err != nil {
return nil, err
}
if cluster == nil {
return &pb.FindEnabledNodeClusterConfigInfoResponse{}, nil
}
var result = &pb.FindEnabledNodeClusterConfigInfoResponse{}
// health check
if models.IsNotNull(cluster.HealthCheck) {
2021-06-27 21:59:37 +08:00
healthCheckConfig := &serverconfigs.HealthCheckConfig{}
err = json.Unmarshal(cluster.HealthCheck, healthCheckConfig)
2021-06-27 21:59:37 +08:00
if err != nil {
return nil, err
}
result.HealthCheckIsOn = healthCheckConfig.IsOn
}
// firewall actions
countFirewallActions, err := models.SharedNodeClusterFirewallActionDAO.CountAllEnabledFirewallActions(tx, req.NodeClusterId)
if err != nil {
return nil, err
}
result.HasFirewallActions = countFirewallActions > 0
// thresholds
countThresholds, err := models.SharedNodeThresholdDAO.CountAllEnabledThresholds(tx, "node", req.NodeClusterId, 0)
if err != nil {
return nil, err
}
result.HasThresholds = countThresholds > 0
// message receivers
countReceivers, err := models.SharedMessageReceiverDAO.CountAllEnabledReceivers(tx, nodeconfigs.NodeRoleNode, req.NodeClusterId, 0, 0, "")
2021-06-27 21:59:37 +08:00
if err != nil {
return nil, err
}
result.HasMessageReceivers = countReceivers > 0
// toa
if models.IsNotNull(cluster.Toa) {
2023-07-24 15:33:44 +08:00
var toaConfig = nodeconfigs.NewTOAConfig()
err = json.Unmarshal(cluster.Toa, toaConfig)
2021-06-27 21:59:37 +08:00
if err != nil {
return nil, err
}
result.IsTOAEnabled = toaConfig.IsOn
}
// metric items
countMetricItems, err := models.SharedNodeClusterMetricItemDAO.CountAllClusterItems(tx, req.NodeClusterId)
if err != nil {
return nil, err
}
result.HasMetricItems = countMetricItems > 0
2022-04-01 16:18:54 +08:00
// webp
if models.IsNotNull(cluster.Webp) {
var webpPolicy = &nodeconfigs.WebPImagePolicy{}
err = json.Unmarshal(cluster.Webp, webpPolicy)
if err != nil {
return nil, err
}
result.WebpIsOn = webpPolicy.IsOn
} else {
result.WebpIsOn = nodeconfigs.DefaultWebPImagePolicy.IsOn
}
2022-07-03 22:10:46 +08:00
// UAM
2023-07-03 17:12:24 +08:00
var uamPolicy = nodeconfigs.NewUAMPolicy()
2022-07-03 22:10:46 +08:00
if models.IsNotNull(cluster.Uam) {
err = json.Unmarshal(cluster.Uam, uamPolicy)
if err != nil {
return nil, err
}
result.UamIsOn = uamPolicy.IsOn
} else {
2023-07-03 17:12:24 +08:00
result.UamIsOn = uamPolicy.IsOn
2022-07-03 22:10:46 +08:00
}
2023-05-23 19:16:30 +08:00
// HTTP CC
if models.IsNotNull(cluster.Cc) {
var httpCCPolicy = nodeconfigs.NewHTTPCCPolicy()
err = json.Unmarshal(cluster.Cc, httpCCPolicy)
if err != nil {
return nil, err
}
result.HttpCCIsOn = httpCCPolicy.IsOn
} else {
result.HttpCCIsOn = nodeconfigs.NewHTTPCCPolicy().IsOn
}
// system service
if models.IsNotNull(cluster.SystemServices) {
var servicesMap = map[string]maps.Map{}
err = json.Unmarshal(cluster.SystemServices, &servicesMap)
if err != nil {
return nil, err
}
for _, serviceMap := range servicesMap {
if serviceMap.GetBool("isOn") {
result.HasSystemServices = true
}
}
}
2022-05-18 21:02:53 +08:00
// ddos
result.HasDDoSProtection = cluster.HasDDoSProtection()
2023-05-22 17:30:33 +08:00
// HTTP Pages
if models.IsNotNull(cluster.HttpPages) {
var pagesPolicy = nodeconfigs.NewHTTPPagesPolicy()
err = json.Unmarshal(cluster.HttpPages, pagesPolicy)
if err != nil {
return nil, err
}
result.HasHTTPPagesPolicy = pagesPolicy.IsOn && len(pagesPolicy.Pages) > 0
}
2023-06-01 17:46:10 +08:00
// HTTP/3
if models.IsNotNull(cluster.Http3) {
var http3Policy = nodeconfigs.NewHTTP3Policy()
err = json.Unmarshal(cluster.Http3, http3Policy)
if err != nil {
return nil, err
}
result.Http3IsOn = http3Policy.IsOn
}
2021-06-27 21:59:37 +08:00
return result, nil
}
2022-03-17 11:12:46 +08:00
// UpdateNodeClusterPinned 设置集群是否置顶
func (this *NodeClusterService) UpdateNodeClusterPinned(ctx context.Context, req *pb.UpdateNodeClusterPinnedRequest) (*pb.RPCSuccess, error) {
2022-07-22 14:35:17 +08:00
_, err := this.ValidateAdmin(ctx)
2022-03-17 11:12:46 +08:00
if err != nil {
return nil, err
}
var tx = this.NullTx()
err = models.SharedNodeClusterDAO.UpdateClusterIsPinned(tx, req.NodeClusterId, req.IsPinned)
if err != nil {
return nil, err
}
return this.Success()
}
2022-04-01 16:18:54 +08:00
// FindEnabledNodeClusterWebPPolicy 读取集群WebP策略
func (this *NodeClusterService) FindEnabledNodeClusterWebPPolicy(ctx context.Context, req *pb.FindEnabledNodeClusterWebPPolicyRequest) (*pb.FindEnabledNodeClusterWebPPolicyResponse, error) {
2022-09-17 16:07:37 +08:00
_, _, err := this.ValidateAdminAndUser(ctx, false)
2022-04-01 16:18:54 +08:00
if err != nil {
return nil, err
}
var tx = this.NullTx()
webpPolicy, err := models.SharedNodeClusterDAO.FindClusterWebPPolicy(tx, req.NodeClusterId, nil)
if err != nil {
return nil, err
}
webpPolicyJSON, err := json.Marshal(webpPolicy)
if err != nil {
return nil, err
}
return &pb.FindEnabledNodeClusterWebPPolicyResponse{
WebpPolicyJSON: webpPolicyJSON,
}, nil
}
// UpdateNodeClusterWebPPolicy 设置集群WebP策略
func (this *NodeClusterService) UpdateNodeClusterWebPPolicy(ctx context.Context, req *pb.UpdateNodeClusterWebPPolicyRequest) (*pb.RPCSuccess, error) {
2022-07-22 14:35:17 +08:00
_, err := this.ValidateAdmin(ctx)
2022-04-01 16:18:54 +08:00
if err != nil {
return nil, err
}
var webpPolicy = &nodeconfigs.WebPImagePolicy{}
err = json.Unmarshal(req.WebpPolicyJSON, webpPolicy)
if err != nil {
return nil, err
}
2022-07-03 22:10:46 +08:00
err = webpPolicy.Init()
if err != nil {
return nil, errors.New("validate webp policy failed: " + err.Error())
}
var tx = this.NullTx()
2022-04-01 16:18:54 +08:00
err = models.SharedNodeClusterDAO.UpdateClusterWebPPolicy(tx, req.NodeClusterId, webpPolicy)
if err != nil {
return nil, err
}
return this.Success()
}
2022-05-18 21:02:53 +08:00
2022-07-03 22:10:46 +08:00
// FindEnabledNodeClusterUAMPolicy 读取集群UAM策略
func (this *NodeClusterService) FindEnabledNodeClusterUAMPolicy(ctx context.Context, req *pb.FindEnabledNodeClusterUAMPolicyRequest) (*pb.FindEnabledNodeClusterUAMPolicyResponse, error) {
2023-05-23 19:16:30 +08:00
if !teaconst.IsPlus {
return nil, this.NotImplementedYet()
}
2022-09-17 16:07:37 +08:00
_, _, err := this.ValidateAdminAndUser(ctx, false)
2022-07-03 22:10:46 +08:00
if err != nil {
return nil, err
}
var tx = this.NullTx()
uamPolicy, err := models.SharedNodeClusterDAO.FindClusterUAMPolicy(tx, req.NodeClusterId, nil)
if err != nil {
return nil, err
}
uamPolicyJSON, err := json.Marshal(uamPolicy)
if err != nil {
return nil, err
}
return &pb.FindEnabledNodeClusterUAMPolicyResponse{
UamPolicyJSON: uamPolicyJSON,
}, nil
}
// UpdateNodeClusterUAMPolicy 设置集群的UAM策略
func (this *NodeClusterService) UpdateNodeClusterUAMPolicy(ctx context.Context, req *pb.UpdateNodeClusterUAMPolicyRequest) (*pb.RPCSuccess, error) {
2023-05-23 19:16:30 +08:00
if !teaconst.IsPlus {
return nil, this.NotImplementedYet()
}
2022-07-22 14:35:17 +08:00
_, err := this.ValidateAdmin(ctx)
2022-07-03 22:10:46 +08:00
if err != nil {
return nil, err
}
2023-07-03 17:12:24 +08:00
var uamPolicy = nodeconfigs.NewUAMPolicy()
2022-07-03 22:10:46 +08:00
err = json.Unmarshal(req.UamPolicyJSON, uamPolicy)
if err != nil {
return nil, err
}
err = uamPolicy.Init()
if err != nil {
return nil, errors.New("validate uam policy failed: " + err.Error())
}
var tx = this.NullTx()
err = models.SharedNodeClusterDAO.UpdateClusterUAMPolicy(tx, req.NodeClusterId, uamPolicy)
if err != nil {
return nil, err
}
return this.Success()
}
2023-05-23 19:16:30 +08:00
// FindEnabledNodeClusterHTTPCCPolicy 读取集群HTTP CC策略
func (this *NodeClusterService) FindEnabledNodeClusterHTTPCCPolicy(ctx context.Context, req *pb.FindEnabledNodeClusterHTTPCCPolicyRequest) (*pb.FindEnabledNodeClusterHTTPCCPolicyResponse, error) {
if !teaconst.IsPlus {
return nil, this.NotImplementedYet()
}
_, _, err := this.ValidateAdminAndUser(ctx, false)
if err != nil {
return nil, err
}
var tx = this.NullTx()
httpCCPolicy, err := models.SharedNodeClusterDAO.FindClusterHTTPCCPolicy(tx, req.NodeClusterId, nil)
if err != nil {
return nil, err
}
httpCCPolicyJSON, err := json.Marshal(httpCCPolicy)
if err != nil {
return nil, err
}
return &pb.FindEnabledNodeClusterHTTPCCPolicyResponse{
HttpCCPolicyJSON: httpCCPolicyJSON,
}, nil
}
// UpdateNodeClusterHTTPCCPolicy 设置集群的HTTP CC策略
func (this *NodeClusterService) UpdateNodeClusterHTTPCCPolicy(ctx context.Context, req *pb.UpdateNodeClusterHTTPCCPolicyRequest) (*pb.RPCSuccess, error) {
if !teaconst.IsPlus {
return nil, this.NotImplementedYet()
}
_, err := this.ValidateAdmin(ctx)
if err != nil {
return nil, err
}
var httpCCPolicy = nodeconfigs.NewHTTPCCPolicy()
err = json.Unmarshal(req.HttpCCPolicyJSON, httpCCPolicy)
if err != nil {
return nil, err
}
err = httpCCPolicy.Init()
if err != nil {
return nil, errors.New("validate http cc policy failed: " + err.Error())
}
var tx = this.NullTx()
err = models.SharedNodeClusterDAO.UpdateClusterHTTPCCPolicy(tx, req.NodeClusterId, httpCCPolicy)
if err != nil {
return nil, err
}
return this.Success()
}
2022-08-22 15:11:22 +08:00
// FindNodeClusterDDoSProtection 获取集群的DDoS设置
2022-05-18 21:02:53 +08:00
func (this *NodeClusterService) FindNodeClusterDDoSProtection(ctx context.Context, req *pb.FindNodeClusterDDoSProtectionRequest) (*pb.FindNodeClusterDDoSProtectionResponse, error) {
2022-07-22 14:35:17 +08:00
_, err := this.ValidateAdmin(ctx)
2022-05-18 21:02:53 +08:00
if err != nil {
return nil, err
}
var tx *dbs.Tx
ddosProtection, err := models.SharedNodeClusterDAO.FindClusterDDoSProtection(tx, req.NodeClusterId)
if err != nil {
return nil, err
}
if ddosProtection == nil {
ddosProtection = ddosconfigs.DefaultProtectionConfig()
}
ddosProtectionJSON, err := json.Marshal(ddosProtection)
if err != nil {
return nil, err
}
var result = &pb.FindNodeClusterDDoSProtectionResponse{
DdosProtectionJSON: ddosProtectionJSON,
}
return result, nil
}
2022-08-22 15:11:22 +08:00
// UpdateNodeClusterDDoSProtection 修改集群的DDoS设置
2022-05-18 21:02:53 +08:00
func (this *NodeClusterService) UpdateNodeClusterDDoSProtection(ctx context.Context, req *pb.UpdateNodeClusterDDoSProtectionRequest) (*pb.RPCSuccess, error) {
2022-07-22 14:35:17 +08:00
_, err := this.ValidateAdmin(ctx)
2022-05-18 21:02:53 +08:00
if err != nil {
return nil, err
}
var ddosProtection = &ddosconfigs.ProtectionConfig{}
err = json.Unmarshal(req.DdosProtectionJSON, ddosProtection)
if err != nil {
return nil, err
}
var tx *dbs.Tx
err = models.SharedNodeClusterDAO.UpdateClusterDDoSProtection(tx, req.NodeClusterId, ddosProtection)
if err != nil {
return nil, err
}
return this.Success()
}
2022-09-16 18:42:14 +08:00
// FindNodeClusterGlobalServerConfig 获取集群的全局服务设置
func (this *NodeClusterService) FindNodeClusterGlobalServerConfig(ctx context.Context, req *pb.FindNodeClusterGlobalServerConfigRequest) (*pb.FindNodeClusterGlobalServerConfigResponse, error) {
_, err := this.ValidateAdmin(ctx)
if err != nil {
return nil, err
}
var tx = this.NullTx()
config, err := models.SharedNodeClusterDAO.FindClusterGlobalServerConfig(tx, req.NodeClusterId)
if err != nil {
return nil, err
}
configJSON, err := json.Marshal(config)
if err != nil {
return nil, err
}
return &pb.FindNodeClusterGlobalServerConfigResponse{
GlobalServerConfigJSON: configJSON,
}, nil
}
// UpdateNodeClusterGlobalServerConfig 修改集群的全局服务设置
func (this *NodeClusterService) UpdateNodeClusterGlobalServerConfig(ctx context.Context, req *pb.UpdateNodeClusterGlobalServerConfigRequest) (*pb.RPCSuccess, error) {
_, err := this.ValidateAdmin(ctx)
if err != nil {
return nil, err
}
var config = serverconfigs.NewGlobalServerConfig()
2022-09-16 18:42:14 +08:00
err = json.Unmarshal(req.GlobalServerConfigJSON, config)
if err != nil {
return nil, err
}
err = config.Init()
if err != nil {
return nil, errors.New("validate config failed: " + err.Error())
}
var tx = this.NullTx()
err = models.SharedNodeClusterDAO.UpdateClusterGlobalServerConfig(tx, req.NodeClusterId, config)
if err != nil {
return nil, err
}
return this.Success()
}
2023-05-22 17:30:33 +08:00
// FindNodeClusterHTTPPagesPolicy 获取集群的自定义页面设置
func (this *NodeClusterService) FindNodeClusterHTTPPagesPolicy(ctx context.Context, req *pb.FindNodeClusterHTTPPagesPolicyRequest) (*pb.FindNodeClusterHTTPPagesPolicyResponse, error) {
_, err := this.ValidateAdmin(ctx)
if err != nil {
return nil, err
}
var tx = this.NullTx()
httpPagesPolicy, err := models.SharedNodeClusterDAO.FindClusterHTTPPagesPolicy(tx, req.NodeClusterId, nil)
if err != nil {
return nil, err
}
httpPagesPolicyJSON, err := json.Marshal(httpPagesPolicy)
if err != nil {
return nil, err
}
return &pb.FindNodeClusterHTTPPagesPolicyResponse{
HttpPagesPolicyJSON: httpPagesPolicyJSON,
}, nil
}
// UpdateNodeClusterHTTPPagesPolicy 修改集群的自定义页面设置
func (this *NodeClusterService) UpdateNodeClusterHTTPPagesPolicy(ctx context.Context, req *pb.UpdateNodeClusterHTTPPagesPolicyRequest) (*pb.RPCSuccess, error) {
_, err := this.ValidateAdmin(ctx)
if err != nil {
return nil, err
}
var tx = this.NullTx()
var policy = nodeconfigs.NewHTTPPagesPolicy()
err = json.Unmarshal(req.HttpPagesPolicyJSON, policy)
if err != nil {
return nil, err
}
err = policy.Init()
if err != nil {
return nil, errors.New("validate policy failed: " + err.Error())
}
err = models.SharedNodeClusterDAO.UpdateClusterHTTPPagesPolicy(tx, req.NodeClusterId, policy)
if err != nil {
return nil, err
}
return this.Success()
}
2023-06-01 17:46:10 +08:00
// UpdateNodeClusterHTTP3Policy 修改集群的HTTP3设置
func (this *NodeClusterService) UpdateNodeClusterHTTP3Policy(ctx context.Context, req *pb.UpdateNodeClusterHTTP3PolicyRequest) (*pb.RPCSuccess, error) {
if !teaconst.IsPlus {
return nil, this.NotImplementedYet()
}
_, err := this.ValidateAdmin(ctx)
if err != nil {
return nil, err
}
var http3Policy = nodeconfigs.NewHTTP3Policy()
err = json.Unmarshal(req.Http3PolicyJSON, http3Policy)
if err != nil {
return nil, err
}
err = http3Policy.Init()
if err != nil {
return nil, errors.New("validate http3 policy failed: " + err.Error())
}
var tx = this.NullTx()
err = models.SharedNodeClusterDAO.UpdateClusterHTTP3Policy(tx, req.NodeClusterId, http3Policy)
if err != nil {
return nil, err
}
return this.Success()
}