优化看板打开速度

internal/rpc/services/service_server_domain_hourly_stat.go

@@ -0,0 +1,52 @@
+// Copyright 2022 Liuxiangchao iwind.liu@gmail.com. All rights reserved.
+
+package services
+
+import (
+	"context"
+	"github.com/TeaOSLab/EdgeAPI/internal/db/models/stats"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+)
+
+// ServerDomainHourlyStatService 服务域名按小时统计服务
+type ServerDomainHourlyStatService struct {
+	BaseService
+}
+
+// ListTopServerDomainStatsWithServerId 读取域名排行
+func (this *ServerDomainHourlyStatService) ListTopServerDomainStatsWithServerId(ctx context.Context, req *pb.ListTopServerDomainStatsWithServerIdRequest) (*pb.ListTopServerDomainStatsWithServerIdResponse, error) {
+	_, err := this.ValidateAdmin(ctx, 0)
+	if err != nil {
+		return nil, err
+	}
+
+	var tx = this.NullTx()
+	var topDomainStats []*stats.ServerDomainHourlyStat
+	if req.ServerId > 0 {
+		topDomainStats, err = stats.SharedServerDomainHourlyStatDAO.FindTopDomainStatsWithServerId(tx, req.ServerId, req.HourFrom, req.HourTo, req.Size)
+	} else if req.NodeId > 0 {
+		// 域名排行
+		topDomainStats, err = stats.SharedServerDomainHourlyStatDAO.FindTopDomainStatsWithNodeId(tx, req.NodeId, req.HourFrom, req.HourTo, 10)
+	} else if req.NodeClusterId > 0 {
+		topDomainStats, err = stats.SharedServerDomainHourlyStatDAO.FindTopDomainStatsWithClusterId(tx, req.NodeClusterId, req.HourFrom, req.HourTo, 10)
+	}
+	if err != nil {
+		return nil, err
+	}
+
+	var pbDomainStats = []*pb.ServerDomainHourlyStat{}
+	for _, stat := range topDomainStats {
+		pbDomainStats = append(pbDomainStats, &pb.ServerDomainHourlyStat{
+			ServerId:            int64(stat.ServerId),
+			Domain:              stat.Domain,
+			CountRequests:       int64(stat.CountRequests),
+			Bytes:               int64(stat.Bytes),
+			CountAttackRequests: int64(stat.CountAttackRequests),
+			AttackBytes:         int64(stat.AttackBytes),
+		})
+	}
+
+	return &pb.ListTopServerDomainStatsWithServerIdResponse{
+		DomainStats: pbDomainStats,
+	}, nil
+}