阶段性提交

internal/rpc/services/service_node_cluster_metric_item.go

@@ -0,0 +1,81 @@
+// Copyright 2021 Liuxiangchao iwind.liu@gmail.com. All rights reserved.
+
+package services
+
+import (
+	"context"
+	"github.com/TeaOSLab/EdgeAPI/internal/db/models"
+	"github.com/TeaOSLab/EdgeAPI/internal/db/models/metrics"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+	"github.com/iwind/TeaGo/types"
+)
+
+// NodeClusterMetricItemService 集群指标
+type NodeClusterMetricItemService struct {
+	BaseService
+}
+
+// EnableNodeClusterMetricItem 启用某个指标
+func (this *NodeClusterMetricItemService) EnableNodeClusterMetricItem(ctx context.Context, req *pb.EnableNodeClusterMetricItemRequest) (*pb.RPCSuccess, error) {
+	_, err := this.ValidateAdmin(ctx, 0)
+	if err != nil {
+		return nil, err
+	}
+
+	var tx = this.NullTx()
+	err = models.SharedNodeClusterMetricItemDAO.EnableClusterItem(tx, req.NodeClusterId, req.MetricItemId)
+	if err != nil {
+		return nil, err
+	}
+	return this.Success()
+}
+
+// DisableNodeClusterMetricItem 禁用某个指标
+func (this *NodeClusterMetricItemService) DisableNodeClusterMetricItem(ctx context.Context, req *pb.DisableNodeClusterMetricItemRequest) (*pb.RPCSuccess, error) {
+	_, err := this.ValidateAdmin(ctx, 0)
+	if err != nil {
+		return nil, err
+	}
+
+	var tx = this.NullTx()
+	err = models.SharedNodeClusterMetricItemDAO.DisableClusterItem(tx, req.NodeClusterId, req.MetricItemId)
+	if err != nil {
+		return nil, err
+	}
+	return this.Success()
+}
+
+// FindAllNodeClusterMetricItems 查找集群中所有指标
+func (this *NodeClusterMetricItemService) FindAllNodeClusterMetricItems(ctx context.Context, req *pb.FindAllNodeClusterMetricItemsRequest) (*pb.FindAllNodeClusterMetricItemsResponse, error) {
+	_, err := this.ValidateAdmin(ctx, 0)
+	if err != nil {
+		return nil, err
+	}
+
+	var tx = this.NullTx()
+	clusterItems, err := models.SharedNodeClusterMetricItemDAO.FindAllClusterItems(tx, req.NodeClusterId)
+	if err != nil {
+		return nil, err
+	}
+	var pbItems = []*pb.MetricItem{}
+	for _, clusterItem := range clusterItems {
+		item, err := metrics.SharedMetricItemDAO.FindEnabledMetricItem(tx, int64(clusterItem.ItemId))
+		if err != nil {
+			return nil, err
+		}
+		if item != nil {
+			pbItems = append(pbItems, &pb.MetricItem{
+				Id:         int64(item.Id),
+				IsOn:       item.IsOn == 1,
+				Code:       item.Code,
+				Category:   item.Category,
+				Name:       item.Name,
+				Keys:       item.DecodeKeys(),
+				Period:     types.Int32(item.Period),
+				PeriodUnit: item.PeriodUnit,
+				Value:      item.Value,
+			})
+		}
+	}
+	return &pb.FindAllNodeClusterMetricItemsResponse{MetricItems: pbItems}, nil
+}