EdgeAPI/internal/rpc/services/service_traffic_daily_stat.go

// Copyright 2022 Liuxiangchao iwind.liu@gmail.com. All rights reserved.

package services

import (
	"context"
	"fmt"
	"github.com/TeaOSLab/EdgeAPI/internal/db/models/stats"
	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
	"github.com/iwind/TeaGo/types"
	"regexp"
)

// TrafficDailyStatService 按日统计服务
type TrafficDailyStatService struct {
	BaseService
}

// FindTrafficDailyStatWithDay 查找某日统计
func (this *TrafficDailyStatService) FindTrafficDailyStatWithDay(ctx context.Context, req *pb.FindTrafficDailyStatWithDayRequest) (*pb.FindTrafficDailyStatWithDayResponse, error) {
	_, err := this.ValidateAdmin(ctx, 0)
	if err != nil {
		return nil, err
	}

	var tx = this.NullTx()
	var day = req.Day
	var stat = &stats.TrafficDailyStat{
		Day: day,
	}
	if len(req.Minute) > 0 && regexp.MustCompile(`^\d{6}$`).MatchString(req.Minute) {
		var hourString = req.Minute[:2]
		var hourInt = types.Int(hourString)
		var lastHourInt = hourInt - 1

		// 过往小时
		if lastHourInt >= 0 {
			var hourFrom = day + "00"
			var hourTo = day + fmt.Sprintf("%02d", lastHourInt)
			sumStat, err := stats.SharedTrafficHourlyStatDAO.SumHourlyStats(tx, hourFrom, hourTo)
			if err != nil {
				return nil, err
			}
			if sumStat != nil {
				stat = &stats.TrafficDailyStat{
					Id:                  0,
					Day:                 day,
					CachedBytes:         sumStat.CachedBytes,
					Bytes:               sumStat.Bytes,
					CountRequests:       sumStat.CountRequests,
					CountCachedRequests: sumStat.CountCachedRequests,
					CountAttackRequests: sumStat.CountAttackRequests,
					AttackBytes:         sumStat.AttackBytes,
				}
			}
		}

		// 当前小时
		hourStat, err := stats.SharedTrafficHourlyStatDAO.FindHourlyStat(tx, day+hourString)
		if err != nil {
			return nil, err
		}
		if hourStat != nil {
			var seconds = types.Int(req.Minute[2:4])*60 + types.Int(req.Minute[4:]) + 1 /** 因为是0-59，所以+1 **/
			stat.Bytes += hourStat.Bytes * uint64(seconds) / 3600
			stat.CachedBytes += hourStat.CachedBytes * uint64(seconds) / 3600
			stat.CountRequests += hourStat.CountRequests * uint64(seconds) / 3600
			stat.CountCachedRequests += hourStat.CountCachedRequests * uint64(seconds) / 3600
			stat.CountAttackRequests += hourStat.CountAttackRequests * uint64(seconds) / 3600
			stat.AttackBytes += hourStat.AttackBytes * uint64(seconds) / 3600
		}
	} else {
		stat, err = stats.SharedTrafficDailyStatDAO.FindDailyStat(tx, day)
		if err != nil {
			return nil, err
		}
	}

	if stat == nil {
		return &pb.FindTrafficDailyStatWithDayResponse{
			TrafficDailyStat: nil,
		}, nil
	}

	return &pb.FindTrafficDailyStatWithDayResponse{
		TrafficDailyStat: &pb.TrafficDailyStat{
			Id:                  int64(stat.Id),
			Day:                 stat.Day,
			CachedBytes:         int64(stat.CachedBytes),
			Bytes:               int64(stat.Bytes),
			CountRequests:       int64(stat.CountRequests),
			CountCachedRequests: int64(stat.CountCachedRequests),
			CountAttackRequests: int64(stat.CountAttackRequests),
			AttackBytes:         int64(stat.AttackBytes),
		},
	}, nil
}