feat(metrics): add /metrics read endpoints for vm and container

This commit is contained in:
2026-06-21 21:10:44 +05:00
parent e380ec1ab2
commit 6769e5b218
2 changed files with 150 additions and 0 deletions

View File

@@ -1845,6 +1845,58 @@ def get_container_rrddata(
raise HTTPException(status_code=500, detail=str(e))
# ==================== Metrics History (DB-only) ====================
def _resolve_metrics_window(timeframe, from_ts, to_ts):
from ...services.metrics_history import timeframe_to_range
from ...config import settings
now = int(time_lib.time())
if from_ts is not None and to_ts is not None:
f, t = int(from_ts), int(to_ts)
else:
f, t = timeframe_to_range(timeframe, now)
max_span = getattr(settings, "METRICS_RETENTION_DAYS", 30) * 86400
if t - f > max_span:
f = t - max_span
return f, t
@router.get("/api/{server_id}/vm/{vmid}/metrics")
def get_vm_metrics(
server_id: int,
vmid: int,
node: str = Query(None),
timeframe: str = Query("hour", regex="^(hour|day|week|month)$"),
from_ts: int = Query(None),
to_ts: int = Query(None),
nic: str = Query("all"),
db: Session = Depends(get_db),
current_user: User = Depends(PermissionChecker("vm:view")),
):
"""Retrieve VM metrics time-series from the local DB (no Proxmox call)."""
from ...services.metrics_history import query_instance_metrics
f, t = _resolve_metrics_window(timeframe, from_ts, to_ts)
return query_instance_metrics(db, server_id, vmid, f, t, nic)
@router.get("/api/{server_id}/container/{vmid}/metrics")
def get_container_metrics(
server_id: int,
vmid: int,
node: str = Query(None),
timeframe: str = Query("hour", regex="^(hour|day|week|month)$"),
from_ts: int = Query(None),
to_ts: int = Query(None),
nic: str = Query("all"),
db: Session = Depends(get_db),
current_user: User = Depends(PermissionChecker("vm:view")),
):
"""Retrieve container metrics time-series from the local DB (no Proxmox call)."""
from ...services.metrics_history import query_instance_metrics
f, t = _resolve_metrics_window(timeframe, from_ts, to_ts)
return query_instance_metrics(db, server_id, vmid, f, t, nic)
# ==================== VNC Console ====================
@router.get("/api/{server_id}/vm/{vmid}/vnc")
@@ -2827,6 +2879,7 @@ def set_vm_owner(
if not instance:
raise HTTPException(status_code=404, detail="VM not found in cache")
target_user = None
if body.user_id is not None:
target_user = db.query(User).filter(User.id == body.user_id, User.is_active == True).first()
if not target_user:
@@ -2835,6 +2888,29 @@ def set_vm_owner(
instance.owner_id = body.user_id
db.commit()
logger.info(f"Admin {current_user.username} set owner of VM {vmid} (server {server_id}) to user_id={body.user_id}")
# Push a real-time event so every connected client refreshes the owner
# column without a manual page reload.
try:
from ...websocket_manager import broadcast_event
owner_user = (
{
"username": target_user.username,
"email": target_user.email,
"full_name": target_user.full_name,
}
if target_user else None
)
broadcast_event(
"vm_owner_changed",
server_id=server_id,
vmid=vmid,
owner_id=body.user_id,
owner_user=owner_user,
)
except Exception as e:
logger.debug(f"Failed to broadcast vm_owner_changed: {e}")
return {"ok": True, "owner_id": body.user_id}

View File

@@ -0,0 +1,74 @@
"""Tests for /metrics read endpoints (Task 6)."""
from datetime import datetime, timezone
import time
def test_metrics_endpoint_returns_series(client, admin_headers, db_session):
"""Seeds InstanceMetric rows then reads them via the API.
The endpoint reads only the DB — no Proxmox call required."""
from app.models import InstanceMetric
now = int(time.time())
for i, cpu in enumerate((0.1, 0.3)):
db_session.add(InstanceMetric(
server_id=1, vmid=114, vm_type="qemu",
ts=datetime.fromtimestamp(now - 10 + i, tz=timezone.utc),
cpu=cpu, mem=100, maxmem=200, disk_used=10, disk_total=40,
netin_rate=1.0, netout_rate=2.0, diskread_rate=0.0, diskwrite_rate=0.0,
iops_read=5.0, iops_write=1.0))
db_session.commit()
r = client.get(
"/proxmox/api/1/vm/114/metrics?timeframe=hour&node=prod",
headers=admin_headers,
)
assert r.status_code == 200
body = r.json()
assert "data" in body and "meta" in body
assert body["data"], "expected at least one bucket"
assert "nics" in body["meta"]
def test_container_metrics_endpoint_returns_series(client, admin_headers, db_session):
"""Same flow for container endpoint."""
from app.models import InstanceMetric
now = int(time.time())
db_session.add(InstanceMetric(
server_id=1, vmid=200, vm_type="lxc",
ts=datetime.fromtimestamp(now - 5, tz=timezone.utc),
cpu=0.2, mem=512, maxmem=1024, disk_used=20, disk_total=80,
netin_rate=3.0, netout_rate=1.5, diskread_rate=0.0, diskwrite_rate=0.0,
iops_read=2.0, iops_write=0.5))
db_session.commit()
r = client.get(
"/proxmox/api/1/container/200/metrics?timeframe=hour&node=prod",
headers=admin_headers,
)
assert r.status_code == 200
body = r.json()
assert "data" in body and "meta" in body
assert "nics" in body["meta"]
def test_metrics_explicit_from_to(client, admin_headers, db_session):
"""Explicit from_ts/to_ts take precedence over timeframe."""
from app.models import InstanceMetric
now = int(time.time())
db_session.add(InstanceMetric(
server_id=1, vmid=300, vm_type="qemu",
ts=datetime.fromtimestamp(now - 30, tz=timezone.utc),
cpu=0.5, mem=256, maxmem=512, disk_used=5, disk_total=20,
netin_rate=0.5, netout_rate=0.5, diskread_rate=0.0, diskwrite_rate=0.0,
iops_read=1.0, iops_write=1.0))
db_session.commit()
r = client.get(
f"/proxmox/api/1/vm/300/metrics?from_ts={now - 60}&to_ts={now}",
headers=admin_headers,
)
assert r.status_code == 200
body = r.json()
assert "data" in body and "meta" in body