67 lines
2.7 KiB
Python
67 lines
2.7 KiB
Python
import datetime
|
||
from typing import Optional
|
||
|
||
from django.core.management.base import BaseCommand
|
||
from django.utils import timezone
|
||
|
||
from domains.models import Domain, DomainTrafficDaily
|
||
from core.goedge_client import GoEdgeClient
|
||
|
||
|
||
class Command(BaseCommand):
|
||
help = '从 GoEdge 拉取每日流量统计并写入 DomainTrafficDaily（默认处理昨天的数据）'
|
||
|
||
def add_arguments(self, parser):
|
||
parser.add_argument('--days', type=int, default=1, help='拉取最近N天（默认1）')
|
||
parser.add_argument('--server-id', type=int, default=0, help='仅处理指定的 serverId')
|
||
|
||
def handle(self, *args, **options):
|
||
days: int = options['days']
|
||
server_id_filter: int = options['server_id']
|
||
client = GoEdgeClient()
|
||
|
||
qs = Domain.objects.exclude(edge_server_id__isnull=True).exclude(edge_server_id__exact='')
|
||
if server_id_filter:
|
||
qs = qs.filter(edge_server_id=server_id_filter)
|
||
|
||
processed = 0
|
||
for domain in qs:
|
||
server_id = int(domain.edge_server_id)
|
||
try:
|
||
daily_stats = client.find_latest_server_daily_stats(server_id=server_id, days=days)
|
||
bw_stats = client.find_daily_bandwidth_stats(server_id=server_id, days=days, algo='avg')
|
||
except Exception as e:
|
||
self.stderr.write(self.style.ERROR(f'域名 {domain.name} 拉取失败: {e}'))
|
||
continue
|
||
|
||
# 将带宽按day映射
|
||
bw_by_day = {}
|
||
for s in bw_stats or []:
|
||
day = s.get('day')
|
||
bytes_per_sec = s.get('bytes') or 0
|
||
# 转换为 Mbps：bytes/sec * 8 / 1_000_000
|
||
mbps = (bytes_per_sec * 8) / 1_000_000 if bytes_per_sec else 0.0
|
||
bw_by_day[day] = mbps
|
||
|
||
for s in daily_stats or []:
|
||
day_str: Optional[str] = s.get('day') # YYYYMMDD
|
||
if not day_str:
|
||
continue
|
||
try:
|
||
day_date = datetime.datetime.strptime(day_str, '%Y%m%d').date()
|
||
except Exception:
|
||
continue
|
||
bytes_used = int(s.get('bytes') or 0)
|
||
peak_mbps = float(bw_by_day.get(day_str) or 0.0)
|
||
|
||
obj, created = DomainTrafficDaily.objects.update_or_create(
|
||
domain=domain,
|
||
day=day_date,
|
||
defaults={
|
||
'bytes': bytes_used,
|
||
'peak_bandwidth_mbps': peak_mbps,
|
||
}
|
||
)
|
||
processed += 1
|
||
|
||
self.stdout.write(self.style.SUCCESS(f'完成，写入/更新 {processed} 条日流量记录。')) |