mirror of
				https://github.com/TeaOSLab/EdgeAPI.git
				synced 2025-11-05 00:11:55 +08:00 
			
		
		
		
	
		
			
				
	
	
		
			701 lines
		
	
	
		
			23 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			701 lines
		
	
	
		
			23 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
// Copyright 2021 Liuxiangchao iwind.liu@gmail.com. All rights reserved.
 | 
						|
 | 
						|
package services
 | 
						|
 | 
						|
import (
 | 
						|
	"context"
 | 
						|
	"encoding/json"
 | 
						|
	"github.com/TeaOSLab/EdgeAPI/internal/db/models"
 | 
						|
	"github.com/TeaOSLab/EdgeAPI/internal/db/models/stats"
 | 
						|
	"github.com/TeaOSLab/EdgeAPI/internal/errors"
 | 
						|
	"github.com/TeaOSLab/EdgeCommon/pkg/configutils"
 | 
						|
	"github.com/TeaOSLab/EdgeCommon/pkg/nodeconfigs"
 | 
						|
	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
 | 
						|
	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs"
 | 
						|
	"github.com/iwind/TeaGo/dbs"
 | 
						|
	"github.com/iwind/TeaGo/lists"
 | 
						|
	"github.com/iwind/TeaGo/types"
 | 
						|
	timeutil "github.com/iwind/TeaGo/utils/time"
 | 
						|
	"time"
 | 
						|
)
 | 
						|
 | 
						|
// ServerStatBoardService 统计看板条目
 | 
						|
type ServerStatBoardService struct {
 | 
						|
	BaseService
 | 
						|
}
 | 
						|
 | 
						|
// FindAllEnabledServerStatBoards 读取所有看板
 | 
						|
func (this *ServerStatBoardService) FindAllEnabledServerStatBoards(ctx context.Context, req *pb.FindAllEnabledServerStatBoardsRequest) (*pb.FindAllEnabledServerStatBoardsResponse, error) {
 | 
						|
	_, err := this.ValidateAdmin(ctx, 0)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	var tx = this.NullTx()
 | 
						|
	boards, err := models.SharedServerStatBoardDAO.FindAllEnabledBoards(tx, req.NodeClusterId)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	var pbBoards = []*pb.ServerStatBoard{}
 | 
						|
	for _, board := range boards {
 | 
						|
		pbBoards = append(pbBoards, &pb.ServerStatBoard{
 | 
						|
			Id:   int64(board.Id),
 | 
						|
			Name: board.Name,
 | 
						|
			IsOn: board.IsOn == 1,
 | 
						|
		})
 | 
						|
	}
 | 
						|
 | 
						|
	return &pb.FindAllEnabledServerStatBoardsResponse{
 | 
						|
		ServerStatBoards: pbBoards,
 | 
						|
	}, nil
 | 
						|
}
 | 
						|
 | 
						|
// ComposeServerStatNodeClusterBoard 组合看板数据
 | 
						|
func (this *ServerStatBoardService) ComposeServerStatNodeClusterBoard(ctx context.Context, req *pb.ComposeServerStatNodeClusterBoardRequest) (*pb.ComposeServerStatNodeClusterBoardResponse, error) {
 | 
						|
	_, err := this.ValidateAdmin(ctx, 0)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	var tx = this.NullTx()
 | 
						|
	var result = &pb.ComposeServerStatNodeClusterBoardResponse{}
 | 
						|
 | 
						|
	// 统计数字
 | 
						|
	countActiveNodes, err := models.SharedNodeDAO.CountAllEnabledNodesMatch(tx, req.NodeClusterId, configutils.BoolStateAll, configutils.BoolStateYes, "", 0, 0)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	result.CountActiveNodes = countActiveNodes
 | 
						|
 | 
						|
	countInactiveNodes, err := models.SharedNodeDAO.CountAllEnabledNodesMatch(tx, req.NodeClusterId, configutils.BoolStateAll, configutils.BoolStateNo, "", 0, 0)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	result.CountInactiveNodes = countInactiveNodes
 | 
						|
 | 
						|
	countUsers, err := models.SharedUserDAO.CountAllEnabledUsers(tx, req.NodeClusterId, "")
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	result.CountUsers = countUsers
 | 
						|
 | 
						|
	countServers, err := models.SharedServerDAO.CountAllEnabledServersWithNodeClusterId(tx, req.NodeClusterId)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	result.CountServers = countServers
 | 
						|
 | 
						|
	// 按日流量统计
 | 
						|
	dayFrom := timeutil.Format("Ymd", time.Now().AddDate(0, 0, -14))
 | 
						|
	dailyTrafficStats, err := stats.SharedNodeClusterTrafficDailyStatDAO.FindDailyStats(tx, req.NodeClusterId, dayFrom, timeutil.Format("Ymd"))
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	for _, stat := range dailyTrafficStats {
 | 
						|
		result.DailyTrafficStats = append(result.DailyTrafficStats, &pb.ComposeServerStatNodeClusterBoardResponse_DailyTrafficStat{
 | 
						|
			Day:                 stat.Day,
 | 
						|
			Bytes:               int64(stat.Bytes),
 | 
						|
			CachedBytes:         int64(stat.CachedBytes),
 | 
						|
			CountRequests:       int64(stat.CountRequests),
 | 
						|
			CountCachedRequests: int64(stat.CountCachedRequests),
 | 
						|
			CountAttackRequests: int64(stat.CountAttackRequests),
 | 
						|
			AttackBytes:         int64(stat.AttackBytes),
 | 
						|
		})
 | 
						|
	}
 | 
						|
 | 
						|
	// 小时流量统计
 | 
						|
	hourFrom := timeutil.Format("YmdH", time.Now().Add(-23*time.Hour))
 | 
						|
	hourTo := timeutil.Format("YmdH")
 | 
						|
	hourlyTrafficStats, err := stats.SharedNodeTrafficHourlyStatDAO.FindHourlyStatsWithClusterId(tx, req.NodeClusterId, hourFrom, hourTo)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	for _, stat := range hourlyTrafficStats {
 | 
						|
		result.HourlyTrafficStats = append(result.HourlyTrafficStats, &pb.ComposeServerStatNodeClusterBoardResponse_HourlyTrafficStat{
 | 
						|
			Hour:                stat.Hour,
 | 
						|
			Bytes:               int64(stat.Bytes),
 | 
						|
			CachedBytes:         int64(stat.CachedBytes),
 | 
						|
			CountRequests:       int64(stat.CountRequests),
 | 
						|
			CountCachedRequests: int64(stat.CountCachedRequests),
 | 
						|
			CountAttackRequests: int64(stat.CountAttackRequests),
 | 
						|
			AttackBytes:         int64(stat.AttackBytes),
 | 
						|
		})
 | 
						|
	}
 | 
						|
 | 
						|
	// 节点排行
 | 
						|
	topNodeStats, err := stats.SharedNodeTrafficHourlyStatDAO.FindTopNodeStatsWithClusterId(tx, "node", req.NodeClusterId, hourFrom, hourTo)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	for _, stat := range topNodeStats {
 | 
						|
		nodeName, err := models.SharedNodeDAO.FindNodeName(tx, int64(stat.NodeId))
 | 
						|
		if err != nil {
 | 
						|
			return nil, err
 | 
						|
		}
 | 
						|
		if len(nodeName) == 0 {
 | 
						|
			continue
 | 
						|
		}
 | 
						|
		result.TopNodeStats = append(result.TopNodeStats, &pb.ComposeServerStatNodeClusterBoardResponse_NodeStat{
 | 
						|
			NodeId:              int64(stat.NodeId),
 | 
						|
			NodeName:            nodeName,
 | 
						|
			CountRequests:       int64(stat.CountRequests),
 | 
						|
			Bytes:               int64(stat.Bytes),
 | 
						|
			CountAttackRequests: int64(stat.CountAttackRequests),
 | 
						|
			AttackBytes:         int64(stat.AttackBytes),
 | 
						|
		})
 | 
						|
	}
 | 
						|
 | 
						|
	// 域名排行
 | 
						|
	topDomainStats, err := stats.SharedServerDomainHourlyStatDAO.FindTopDomainStatsWithClusterId(tx, req.NodeClusterId, hourFrom, hourTo, 10)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	for _, stat := range topDomainStats {
 | 
						|
		result.TopDomainStats = append(result.TopDomainStats, &pb.ComposeServerStatNodeClusterBoardResponse_DomainStat{
 | 
						|
			ServerId:            int64(stat.ServerId),
 | 
						|
			Domain:              stat.Domain,
 | 
						|
			CountRequests:       int64(stat.CountRequests),
 | 
						|
			Bytes:               int64(stat.Bytes),
 | 
						|
			CountAttackRequests: int64(stat.CountAttackRequests),
 | 
						|
			AttackBytes:         int64(stat.AttackBytes),
 | 
						|
		})
 | 
						|
	}
 | 
						|
 | 
						|
	// CPU、内存、负载
 | 
						|
	cpuValues, err := models.SharedNodeValueDAO.ListValuesWithClusterId(tx, "node", req.NodeClusterId, nodeconfigs.NodeValueItemCPU, "usage", nodeconfigs.NodeValueRangeMinute)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	for _, v := range cpuValues {
 | 
						|
		valueJSON, err := json.Marshal(types.Float32(v.Value))
 | 
						|
		if err != nil {
 | 
						|
			return nil, err
 | 
						|
		}
 | 
						|
		result.CpuNodeValues = append(result.CpuNodeValues, &pb.NodeValue{
 | 
						|
			ValueJSON: valueJSON,
 | 
						|
			CreatedAt: int64(v.CreatedAt),
 | 
						|
		})
 | 
						|
	}
 | 
						|
 | 
						|
	memoryValues, err := models.SharedNodeValueDAO.ListValuesWithClusterId(tx, "node", req.NodeClusterId, nodeconfigs.NodeValueItemMemory, "usage", nodeconfigs.NodeValueRangeMinute)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	for _, v := range memoryValues {
 | 
						|
		valueJSON, err := json.Marshal(types.Float32(v.Value))
 | 
						|
		if err != nil {
 | 
						|
			return nil, err
 | 
						|
		}
 | 
						|
		result.MemoryNodeValues = append(result.MemoryNodeValues, &pb.NodeValue{
 | 
						|
			ValueJSON: valueJSON,
 | 
						|
			CreatedAt: int64(v.CreatedAt),
 | 
						|
		})
 | 
						|
	}
 | 
						|
 | 
						|
	loadValues, err := models.SharedNodeValueDAO.ListValuesWithClusterId(tx, "node", req.NodeClusterId, nodeconfigs.NodeValueItemLoad, "load5m", nodeconfigs.NodeValueRangeMinute)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	for _, v := range loadValues {
 | 
						|
		valueJSON, err := json.Marshal(types.Float32(v.Value))
 | 
						|
		if err != nil {
 | 
						|
			return nil, err
 | 
						|
		}
 | 
						|
		result.LoadNodeValues = append(result.LoadNodeValues, &pb.NodeValue{
 | 
						|
			ValueJSON: valueJSON,
 | 
						|
			CreatedAt: int64(v.CreatedAt),
 | 
						|
		})
 | 
						|
	}
 | 
						|
 | 
						|
	charts, err := this.findNodeClusterMetricDataCharts(tx, req.NodeClusterId, 0, 0, serverconfigs.MetricItemCategoryHTTP)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	result.MetricDataCharts = charts
 | 
						|
 | 
						|
	return result, nil
 | 
						|
}
 | 
						|
 | 
						|
// ComposeServerStatNodeBoard 组合节点看板数据
 | 
						|
func (this *ServerStatBoardService) ComposeServerStatNodeBoard(ctx context.Context, req *pb.ComposeServerStatNodeBoardRequest) (*pb.ComposeServerStatNodeBoardResponse, error) {
 | 
						|
	_, err := this.ValidateAdmin(ctx, 0)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	var result = &pb.ComposeServerStatNodeBoardResponse{}
 | 
						|
 | 
						|
	// 在线状态
 | 
						|
	var tx = this.NullTx()
 | 
						|
	node, err := models.SharedNodeDAO.FindEnabledNode(tx, req.NodeId)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	if node == nil {
 | 
						|
		return nil, errors.New("node not found")
 | 
						|
	}
 | 
						|
 | 
						|
	status, err := node.DecodeStatus()
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	if status != nil && time.Now().Unix()-status.UpdatedAt < 60 {
 | 
						|
		result.IsActive = true
 | 
						|
		result.CacheDiskSize = status.CacheTotalDiskSize
 | 
						|
		result.CacheMemorySize = status.CacheTotalMemorySize
 | 
						|
 | 
						|
	}
 | 
						|
 | 
						|
	// 流量
 | 
						|
	{
 | 
						|
		value, err := models.SharedNodeValueDAO.FindLatestNodeValue(tx, "node", int64(node.Id), nodeconfigs.NodeValueItemTrafficIn)
 | 
						|
		if err != nil {
 | 
						|
			return nil, err
 | 
						|
		}
 | 
						|
		if value != nil && time.Now().Unix()-int64(value.CreatedAt) < 120 {
 | 
						|
			result.TrafficInBytes = value.DecodeMapValue().GetInt64("total")
 | 
						|
		}
 | 
						|
	}
 | 
						|
	{
 | 
						|
		value, err := models.SharedNodeValueDAO.FindLatestNodeValue(tx, "node", int64(node.Id), nodeconfigs.NodeValueItemTrafficOut)
 | 
						|
		if err != nil {
 | 
						|
			return nil, err
 | 
						|
		}
 | 
						|
		if value != nil && time.Now().Unix()-int64(value.CreatedAt) < 120 {
 | 
						|
			result.TrafficOutBytes = value.DecodeMapValue().GetInt64("total")
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	// 连接数
 | 
						|
	{
 | 
						|
		value, err := models.SharedNodeValueDAO.FindLatestNodeValue(tx, "node", int64(node.Id), nodeconfigs.NodeValueItemConnections)
 | 
						|
		if err != nil {
 | 
						|
			return nil, err
 | 
						|
		}
 | 
						|
		if value != nil && time.Now().Unix()-int64(value.CreatedAt) < 120 {
 | 
						|
			result.CountConnections = value.DecodeMapValue().GetInt64("total")
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	// 请求量
 | 
						|
	{
 | 
						|
		value, err := models.SharedNodeValueDAO.FindLatestNodeValue(tx, "node", int64(node.Id), nodeconfigs.NodeValueItemRequests)
 | 
						|
		if err != nil {
 | 
						|
			return nil, err
 | 
						|
		}
 | 
						|
		if value != nil && time.Now().Unix()-int64(value.CreatedAt) < 120 {
 | 
						|
			result.CountRequests = value.DecodeMapValue().GetInt64("total")
 | 
						|
		}
 | 
						|
	}
 | 
						|
	{
 | 
						|
		value, err := models.SharedNodeValueDAO.FindLatestNodeValue(tx, "node", int64(node.Id), nodeconfigs.NodeValueItemAttackRequests)
 | 
						|
		if err != nil {
 | 
						|
			return nil, err
 | 
						|
		}
 | 
						|
		if value != nil && time.Now().Unix()-int64(value.CreatedAt) < 120 {
 | 
						|
			result.CountAttackRequests = value.DecodeMapValue().GetInt64("total")
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	// CPU
 | 
						|
	{
 | 
						|
		value, err := models.SharedNodeValueDAO.FindLatestNodeValue(tx, "node", int64(node.Id), nodeconfigs.NodeValueItemCPU)
 | 
						|
		if err != nil {
 | 
						|
			return nil, err
 | 
						|
		}
 | 
						|
		if value != nil && time.Now().Unix()-int64(value.CreatedAt) < 120 {
 | 
						|
			result.CpuUsage = value.DecodeMapValue().GetFloat32("usage")
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	// 内存
 | 
						|
	{
 | 
						|
		value, err := models.SharedNodeValueDAO.FindLatestNodeValue(tx, "node", int64(node.Id), nodeconfigs.NodeValueItemMemory)
 | 
						|
		if err != nil {
 | 
						|
			return nil, err
 | 
						|
		}
 | 
						|
		if value != nil && time.Now().Unix()-int64(value.CreatedAt) < 120 {
 | 
						|
			m := value.DecodeMapValue()
 | 
						|
			result.MemoryUsage = m.GetFloat32("usage")
 | 
						|
			result.MemoryTotalSize = m.GetInt64("total")
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	// 负载
 | 
						|
	{
 | 
						|
		value, err := models.SharedNodeValueDAO.FindLatestNodeValue(tx, "node", int64(node.Id), nodeconfigs.NodeValueItemLoad)
 | 
						|
		if err != nil {
 | 
						|
			return nil, err
 | 
						|
		}
 | 
						|
		if value != nil && time.Now().Unix()-int64(value.CreatedAt) < 120 {
 | 
						|
			result.Load = value.DecodeMapValue().GetFloat32("load1m")
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	// 按日流量统计
 | 
						|
	dayFrom := timeutil.Format("Ymd", time.Now().AddDate(0, 0, -14))
 | 
						|
	dailyTrafficStats, err := stats.SharedNodeTrafficDailyStatDAO.FindDailyStats(tx, "node", req.NodeId, dayFrom, timeutil.Format("Ymd"))
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	for _, stat := range dailyTrafficStats {
 | 
						|
		result.DailyTrafficStats = append(result.DailyTrafficStats, &pb.ComposeServerStatNodeBoardResponse_DailyTrafficStat{
 | 
						|
			Day:                 stat.Day,
 | 
						|
			Bytes:               int64(stat.Bytes),
 | 
						|
			CachedBytes:         int64(stat.CachedBytes),
 | 
						|
			CountRequests:       int64(stat.CountRequests),
 | 
						|
			CountCachedRequests: int64(stat.CountCachedRequests),
 | 
						|
			CountAttackRequests: int64(stat.CountAttackRequests),
 | 
						|
			AttackBytes:         int64(stat.AttackBytes),
 | 
						|
		})
 | 
						|
	}
 | 
						|
 | 
						|
	// 小时流量统计
 | 
						|
	hourFrom := timeutil.Format("YmdH", time.Now().Add(-23*time.Hour))
 | 
						|
	hourTo := timeutil.Format("YmdH")
 | 
						|
	hourlyTrafficStats, err := stats.SharedNodeTrafficHourlyStatDAO.FindHourlyStatsWithNodeId(tx, "node", req.NodeId, hourFrom, hourTo)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	for _, stat := range hourlyTrafficStats {
 | 
						|
		result.HourlyTrafficStats = append(result.HourlyTrafficStats, &pb.ComposeServerStatNodeBoardResponse_HourlyTrafficStat{
 | 
						|
			Hour:                stat.Hour,
 | 
						|
			Bytes:               int64(stat.Bytes),
 | 
						|
			CachedBytes:         int64(stat.CachedBytes),
 | 
						|
			CountRequests:       int64(stat.CountRequests),
 | 
						|
			CountCachedRequests: int64(stat.CountCachedRequests),
 | 
						|
			CountAttackRequests: int64(stat.CountAttackRequests),
 | 
						|
			AttackBytes:         int64(stat.AttackBytes),
 | 
						|
		})
 | 
						|
	}
 | 
						|
 | 
						|
	// 域名排行
 | 
						|
	topDomainStats, err := stats.SharedServerDomainHourlyStatDAO.FindTopDomainStatsWithNodeId(tx, req.NodeId, hourFrom, hourTo, 10)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	for _, stat := range topDomainStats {
 | 
						|
		result.TopDomainStats = append(result.TopDomainStats, &pb.ComposeServerStatNodeBoardResponse_DomainStat{
 | 
						|
			ServerId:      int64(stat.ServerId),
 | 
						|
			Domain:        stat.Domain,
 | 
						|
			CountRequests: int64(stat.CountRequests),
 | 
						|
			Bytes:         int64(stat.Bytes),
 | 
						|
		})
 | 
						|
	}
 | 
						|
 | 
						|
	// CPU、内存、负载
 | 
						|
	cpuValues, err := models.SharedNodeValueDAO.ListValues(tx, "node", req.NodeId, nodeconfigs.NodeValueItemCPU, nodeconfigs.NodeValueRangeMinute)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	for _, v := range cpuValues {
 | 
						|
		valueJSON, err := json.Marshal(types.Float32(v.DecodeMapValue().GetFloat32("usage")))
 | 
						|
		if err != nil {
 | 
						|
			return nil, err
 | 
						|
		}
 | 
						|
		result.CpuNodeValues = append(result.CpuNodeValues, &pb.NodeValue{
 | 
						|
			ValueJSON: valueJSON,
 | 
						|
			CreatedAt: int64(v.CreatedAt),
 | 
						|
		})
 | 
						|
	}
 | 
						|
 | 
						|
	memoryValues, err := models.SharedNodeValueDAO.ListValues(tx, "node", req.NodeId, nodeconfigs.NodeValueItemMemory, nodeconfigs.NodeValueRangeMinute)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	for _, v := range memoryValues {
 | 
						|
		valueJSON, err := json.Marshal(types.Float32(v.DecodeMapValue().GetFloat32("usage")))
 | 
						|
		if err != nil {
 | 
						|
			return nil, err
 | 
						|
		}
 | 
						|
		result.MemoryNodeValues = append(result.MemoryNodeValues, &pb.NodeValue{
 | 
						|
			ValueJSON: valueJSON,
 | 
						|
			CreatedAt: int64(v.CreatedAt),
 | 
						|
		})
 | 
						|
	}
 | 
						|
 | 
						|
	loadValues, err := models.SharedNodeValueDAO.ListValues(tx, "node", req.NodeId, nodeconfigs.NodeValueItemLoad, nodeconfigs.NodeValueRangeMinute)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	for _, v := range loadValues {
 | 
						|
		valueJSON, err := json.Marshal(types.Float32(v.DecodeMapValue().GetFloat32("load5m")))
 | 
						|
		if err != nil {
 | 
						|
			return nil, err
 | 
						|
		}
 | 
						|
		result.LoadNodeValues = append(result.LoadNodeValues, &pb.NodeValue{
 | 
						|
			ValueJSON: valueJSON,
 | 
						|
			CreatedAt: int64(v.CreatedAt),
 | 
						|
		})
 | 
						|
	}
 | 
						|
 | 
						|
	cacheDirValues, err := models.SharedNodeValueDAO.ListValues(tx, "node", req.NodeId, nodeconfigs.NodeValueItemCacheDir, nodeconfigs.NodeValueRangeMinute)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	for _, v := range cacheDirValues {
 | 
						|
		result.CacheDirsValues = append(result.CacheDirsValues, &pb.NodeValue{
 | 
						|
			ValueJSON: []byte(v.Value),
 | 
						|
			CreatedAt: int64(v.CreatedAt),
 | 
						|
		})
 | 
						|
	}
 | 
						|
 | 
						|
	// 指标
 | 
						|
	var clusterId = int64(node.ClusterId)
 | 
						|
	charts, err := this.findNodeClusterMetricDataCharts(tx, clusterId, req.NodeId, 0, serverconfigs.MetricItemCategoryHTTP)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	result.MetricDataCharts = charts
 | 
						|
 | 
						|
	return result, nil
 | 
						|
}
 | 
						|
 | 
						|
// ComposeServerStatBoard 组合服务看板数据
 | 
						|
func (this *ServerStatBoardService) ComposeServerStatBoard(ctx context.Context, req *pb.ComposeServerStatBoardRequest) (*pb.ComposeServerStatBoardResponse, error) {
 | 
						|
	_, err := this.ValidateAdmin(ctx, 0)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	var result = &pb.ComposeServerStatBoardResponse{}
 | 
						|
	var tx = this.NullTx()
 | 
						|
 | 
						|
	// 按日流量统计
 | 
						|
	dayFrom := timeutil.Format("Ymd", time.Now().AddDate(0, 0, -14))
 | 
						|
	dailyTrafficStats, err := models.SharedServerDailyStatDAO.FindDailyStats(tx, req.ServerId, dayFrom, timeutil.Format("Ymd"))
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	for _, stat := range dailyTrafficStats {
 | 
						|
		result.DailyTrafficStats = append(result.DailyTrafficStats, &pb.ComposeServerStatBoardResponse_DailyTrafficStat{
 | 
						|
			Day:                 stat.Day,
 | 
						|
			Bytes:               int64(stat.Bytes),
 | 
						|
			CachedBytes:         int64(stat.CachedBytes),
 | 
						|
			CountRequests:       int64(stat.CountRequests),
 | 
						|
			CountCachedRequests: int64(stat.CountCachedRequests),
 | 
						|
			CountAttackRequests: int64(stat.CountAttackRequests),
 | 
						|
			AttackBytes:         int64(stat.AttackBytes),
 | 
						|
		})
 | 
						|
	}
 | 
						|
 | 
						|
	// 小时流量统计
 | 
						|
	hourFrom := timeutil.Format("YmdH", time.Now().Add(-23*time.Hour))
 | 
						|
	hourTo := timeutil.Format("YmdH")
 | 
						|
	hourlyTrafficStats, err := models.SharedServerDailyStatDAO.FindHourlyStats(tx, req.ServerId, hourFrom, hourTo)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	for _, stat := range hourlyTrafficStats {
 | 
						|
		result.HourlyTrafficStats = append(result.HourlyTrafficStats, &pb.ComposeServerStatBoardResponse_HourlyTrafficStat{
 | 
						|
			Hour:                stat.Hour,
 | 
						|
			Bytes:               int64(stat.Bytes),
 | 
						|
			CachedBytes:         int64(stat.CachedBytes),
 | 
						|
			CountRequests:       int64(stat.CountRequests),
 | 
						|
			CountCachedRequests: int64(stat.CountCachedRequests),
 | 
						|
			CountAttackRequests: int64(stat.CountAttackRequests),
 | 
						|
			AttackBytes:         int64(stat.AttackBytes),
 | 
						|
		})
 | 
						|
	}
 | 
						|
 | 
						|
	// 域名排行
 | 
						|
	topDomainStats, err := stats.SharedServerDomainHourlyStatDAO.FindTopDomainStatsWithServerId(tx, req.ServerId, hourFrom, hourTo, 10)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	for _, stat := range topDomainStats {
 | 
						|
		result.TopDomainStats = append(result.TopDomainStats, &pb.ComposeServerStatBoardResponse_DomainStat{
 | 
						|
			ServerId:            int64(stat.ServerId),
 | 
						|
			Domain:              stat.Domain,
 | 
						|
			CountRequests:       int64(stat.CountRequests),
 | 
						|
			Bytes:               int64(stat.Bytes),
 | 
						|
			CountAttackRequests: int64(stat.CountAttackRequests),
 | 
						|
			AttackBytes:         int64(stat.AttackBytes),
 | 
						|
		})
 | 
						|
	}
 | 
						|
 | 
						|
	// 指标
 | 
						|
	clusterId, err := models.SharedServerDAO.FindServerClusterId(tx, req.ServerId)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	charts, err := this.findNodeClusterMetricDataCharts(tx, clusterId, 0, req.ServerId, serverconfigs.MetricItemCategoryHTTP)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	result.MetricDataCharts = charts
 | 
						|
 | 
						|
	return result, nil
 | 
						|
}
 | 
						|
 | 
						|
// 查找集群、节点和服务的指标数据
 | 
						|
func (this *ServerStatBoardService) findNodeClusterMetricDataCharts(tx *dbs.Tx, clusterId int64, nodeId int64, serverId int64, category string) (result []*pb.MetricDataChart, err error) {
 | 
						|
	// 集群指标
 | 
						|
	clusterMetricItems, err := models.SharedNodeClusterMetricItemDAO.FindAllClusterItems(tx, clusterId, category)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	var pbMetricCharts = []*pb.MetricDataChart{}
 | 
						|
	var metricItemIds = []int64{}
 | 
						|
	var items = []*models.MetricItem{}
 | 
						|
	for _, clusterMetricItem := range clusterMetricItems {
 | 
						|
		if clusterMetricItem.IsOn != 1 {
 | 
						|
			continue
 | 
						|
		}
 | 
						|
		var itemId = int64(clusterMetricItem.ItemId)
 | 
						|
		item, err := models.SharedMetricItemDAO.FindEnabledMetricItem(tx, itemId)
 | 
						|
		if err != nil {
 | 
						|
			return nil, err
 | 
						|
		}
 | 
						|
		if item == nil || item.IsOn == 0 {
 | 
						|
			continue
 | 
						|
		}
 | 
						|
		items = append(items, item)
 | 
						|
		metricItemIds = append(metricItemIds, itemId)
 | 
						|
	}
 | 
						|
 | 
						|
	publicMetricItems, err := models.SharedMetricItemDAO.FindAllPublicItems(tx)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	for _, item := range publicMetricItems {
 | 
						|
		if item.IsOn != 1 {
 | 
						|
			continue
 | 
						|
		}
 | 
						|
		if lists.ContainsInt64(metricItemIds, int64(item.Id)) {
 | 
						|
			continue
 | 
						|
		}
 | 
						|
		items = append(items, item)
 | 
						|
	}
 | 
						|
 | 
						|
	for _, item := range items {
 | 
						|
		var itemId = int64(item.Id)
 | 
						|
		charts, err := models.SharedMetricChartDAO.FindAllEnabledCharts(tx, itemId)
 | 
						|
		if err != nil {
 | 
						|
			return nil, err
 | 
						|
		}
 | 
						|
 | 
						|
		for _, chart := range charts {
 | 
						|
			if chart.IsOn == 0 {
 | 
						|
				continue
 | 
						|
			}
 | 
						|
 | 
						|
			var pbChart = &pb.MetricChart{
 | 
						|
				Id:         int64(chart.Id),
 | 
						|
				Name:       chart.Name,
 | 
						|
				Type:       chart.Type,
 | 
						|
				WidthDiv:   chart.WidthDiv,
 | 
						|
				ParamsJSON: nil,
 | 
						|
				IsOn:       chart.IsOn == 1,
 | 
						|
				MaxItems:   types.Int32(chart.MaxItems),
 | 
						|
				MetricItem: &pb.MetricItem{
 | 
						|
					Id:         itemId,
 | 
						|
					PeriodUnit: item.PeriodUnit,
 | 
						|
					Period:     types.Int32(item.Period),
 | 
						|
					Name:       item.Name,
 | 
						|
					Value:      item.Value,
 | 
						|
					Category:   item.Category,
 | 
						|
					Keys:       item.DecodeKeys(),
 | 
						|
					Code:       item.Code,
 | 
						|
					IsOn:       item.IsOn == 1,
 | 
						|
				},
 | 
						|
			}
 | 
						|
			var pbStats = []*pb.MetricStat{}
 | 
						|
			switch chart.Type {
 | 
						|
			case serverconfigs.MetricChartTypeTimeLine:
 | 
						|
				var itemStats []*models.MetricStat
 | 
						|
				if serverId > 0 {
 | 
						|
					itemStats, err = models.SharedMetricStatDAO.FindLatestItemStatsWithServerId(tx, serverId, itemId, chart.IgnoreEmptyKeys == 1, chart.DecodeIgnoredKeys(), types.Int32(item.Version), 10)
 | 
						|
				} else if nodeId > 0 {
 | 
						|
					itemStats, err = models.SharedMetricStatDAO.FindLatestItemStatsWithNodeId(tx, nodeId, itemId, chart.IgnoreEmptyKeys == 1, chart.DecodeIgnoredKeys(), types.Int32(item.Version), 10)
 | 
						|
				} else {
 | 
						|
					itemStats, err = models.SharedMetricStatDAO.FindLatestItemStatsWithClusterId(tx, clusterId, itemId, chart.IgnoreEmptyKeys == 1, chart.DecodeIgnoredKeys(), types.Int32(item.Version), 10)
 | 
						|
				}
 | 
						|
				if err != nil {
 | 
						|
					return nil, err
 | 
						|
				}
 | 
						|
 | 
						|
				for _, stat := range itemStats {
 | 
						|
					// 当前时间总和
 | 
						|
					var count int64
 | 
						|
					var total float32
 | 
						|
					if serverId > 0 {
 | 
						|
						count, total, err = models.SharedMetricSumStatDAO.FindServerSum(tx, serverId, stat.Time, itemId, types.Int32(item.Version))
 | 
						|
					} else if nodeId > 0 {
 | 
						|
						count, total, err = models.SharedMetricSumStatDAO.FindNodeSum(tx, nodeId, stat.Time, itemId, types.Int32(item.Version))
 | 
						|
					} else {
 | 
						|
						count, total, err = models.SharedMetricSumStatDAO.FindClusterSum(tx, clusterId, stat.Time, itemId, types.Int32(item.Version))
 | 
						|
					}
 | 
						|
					if err != nil {
 | 
						|
						return nil, err
 | 
						|
					}
 | 
						|
 | 
						|
					pbStats = append(pbStats, &pb.MetricStat{
 | 
						|
						Id:          int64(stat.Id),
 | 
						|
						Hash:        stat.Hash,
 | 
						|
						ServerId:    0,
 | 
						|
						ItemId:      0,
 | 
						|
						Keys:        stat.DecodeKeys(),
 | 
						|
						Value:       types.Float32(stat.Value),
 | 
						|
						Time:        stat.Time,
 | 
						|
						Version:     0,
 | 
						|
						NodeCluster: nil,
 | 
						|
						Node:        nil,
 | 
						|
						Server:      nil,
 | 
						|
						SumCount:    count,
 | 
						|
						SumTotal:    total,
 | 
						|
					})
 | 
						|
				}
 | 
						|
			default:
 | 
						|
				var itemStats []*models.MetricStat
 | 
						|
				if serverId > 0 {
 | 
						|
					itemStats, err = models.SharedMetricStatDAO.FindItemStatsWithServerIdAndLastTime(tx, serverId, itemId, chart.IgnoreEmptyKeys == 1, chart.DecodeIgnoredKeys(), types.Int32(item.Version), 10)
 | 
						|
				} else if nodeId > 0 {
 | 
						|
					itemStats, err = models.SharedMetricStatDAO.FindItemStatsWithNodeIdAndLastTime(tx, nodeId, itemId, chart.IgnoreEmptyKeys == 1, chart.DecodeIgnoredKeys(), types.Int32(item.Version), 10)
 | 
						|
				} else {
 | 
						|
					itemStats, err = models.SharedMetricStatDAO.FindItemStatsWithClusterIdAndLastTime(tx, clusterId, itemId, chart.IgnoreEmptyKeys == 1, chart.DecodeIgnoredKeys(), types.Int32(item.Version), 10)
 | 
						|
				}
 | 
						|
				if err != nil {
 | 
						|
					return nil, err
 | 
						|
				}
 | 
						|
				for _, stat := range itemStats {
 | 
						|
					// 当前时间总和
 | 
						|
					var count int64
 | 
						|
					var total float32
 | 
						|
					if serverId > 0 {
 | 
						|
						count, total, err = models.SharedMetricSumStatDAO.FindServerSum(tx, serverId, stat.Time, itemId, types.Int32(item.Version))
 | 
						|
					} else if nodeId > 0 {
 | 
						|
						count, total, err = models.SharedMetricSumStatDAO.FindNodeSum(tx, nodeId, stat.Time, itemId, types.Int32(item.Version))
 | 
						|
					} else {
 | 
						|
						count, total, err = models.SharedMetricSumStatDAO.FindClusterSum(tx, clusterId, stat.Time, itemId, types.Int32(item.Version))
 | 
						|
					}
 | 
						|
					if err != nil {
 | 
						|
						return nil, err
 | 
						|
					}
 | 
						|
 | 
						|
					pbStats = append(pbStats, &pb.MetricStat{
 | 
						|
						Id:          int64(stat.Id),
 | 
						|
						Hash:        stat.Hash,
 | 
						|
						ServerId:    0,
 | 
						|
						ItemId:      0,
 | 
						|
						Keys:        stat.DecodeKeys(),
 | 
						|
						Value:       types.Float32(stat.Value),
 | 
						|
						Time:        stat.Time,
 | 
						|
						Version:     0,
 | 
						|
						NodeCluster: nil,
 | 
						|
						Node:        nil,
 | 
						|
						Server:      nil,
 | 
						|
						SumCount:    count,
 | 
						|
						SumTotal:    total,
 | 
						|
					})
 | 
						|
				}
 | 
						|
			}
 | 
						|
			pbMetricCharts = append(pbMetricCharts, &pb.MetricDataChart{
 | 
						|
				MetricChart: pbChart,
 | 
						|
				MetricStats: pbStats,
 | 
						|
			})
 | 
						|
		}
 | 
						|
	}
 | 
						|
	return pbMetricCharts, nil
 | 
						|
}
 |