kvm-manager/backend/app/routers/monitor.py

"""资源监控路由"""
from fastapi import APIRouter, HTTPException, Query
from app.libvirt_conn import conn_pool
import libvirt
import time
import threading

router = APIRouter()

# 简易内存缓存
_stats_cache = {}
_cache_lock = threading.Lock()


@router.get("/overview")
async def monitor_overview(host_id: str = Query("local")):
    """宿主机总览监控"""
    conn = conn_pool.get_conn(host_id)

    host_info = conn.getInfo()
    hostname = conn.getHostname()

    cpu_stats = conn.getCPUStats(-1, 0)
    cpu_total = cpu_stats.get("user", 0) + cpu_stats.get("system", 0) + cpu_stats.get("idle", 0)
    cpu_used = cpu_stats.get("user", 0) + cpu_stats.get("system", 0)
    cpu_percent = round(cpu_used / cpu_total * 100, 1) if cpu_total > 0 else 0

    memory_total_kb = host_info[1]
    try:
        with open("/proc/meminfo", "r") as f:
            meminfo = {}
            for line in f:
                parts = line.split()
                if len(parts) >= 2:
                    meminfo[parts[0].rstrip(":")] = int(parts[1])

        mem_total_mb = meminfo.get("MemTotal", 0) // 1024
        mem_available_mb = meminfo.get("MemAvailable", 0) // 1024
        mem_used_mb = mem_total_mb - mem_available_mb
        mem_percent = round(mem_used_mb / mem_total_mb * 100, 1) if mem_total_mb > 0 else 0
    except Exception:
        mem_total_mb = memory_total_kb // 1024
        mem_used_mb = 0
        mem_available_mb = mem_total_mb
        mem_percent = 0

    domains = conn.listAllDomains(0)
    running = sum(1 for d in domains if d.isActive())
    stopped = len(domains) - running

    return {
        "hostname": hostname,
        "cpu": {
            "cores": host_info[2],
            "speed_mhz": host_info[3],
            "usage_percent": cpu_percent,
        },
        "memory": {
            "total_mb": mem_total_mb,
            "used_mb": mem_used_mb,
            "available_mb": mem_available_mb,
            "usage_percent": mem_percent,
        },
        "vms": {
            "total": len(domains),
            "running": running,
            "stopped": stopped,
        },
    }


@router.get("/vm/{name}")
async def monitor_vm(name: str, host_id: str = Query("local")):
    """获取虚拟机实时监控数据"""
    conn = conn_pool.get_conn(host_id)
    try:
        dom = conn.lookupByName(name)
    except libvirt.libvirtError:
        raise HTTPException(status_code=404, detail=f"虚拟机 '{name}' 不存在")

    if not dom.isActive():
        return {"name": name, "state": "stopped", "cpu_percent": 0, "memory": {}}

    cache_key = f"{host_id}_{name}"
    cpu_percent = _get_vm_cpu_percent(dom, cache_key)

    mem_stats = {}
    try:
        raw = dom.memoryStats()
        mem_stats = {
            "rss_mb": raw.get("rss", 0) // 1024 if "rss" in raw else 0,
            "actual_mb": raw.get("actual", 0) // 1024 if "actual" in raw else 0,
            "available_mb": raw.get("available", 0) // 1024 if "available" in raw else 0,
            "usage_percent": round(
                raw.get("rss", 0) / raw.get("actual", 1) * 100, 1
            ) if "rss" in raw and "actual" in raw else 0,
        }
    except Exception:
        pass

    disk_stats = _get_vm_disk_stats(dom)
    net_stats = _get_vm_net_stats(dom)

    return {
        "name": name,
        "state": "running",
        "cpu_percent": cpu_percent,
        "memory": mem_stats,
        "disk": disk_stats,
        "network": net_stats,
    }


def _get_vm_cpu_percent(dom, cache_key: str) -> float:
    """计算虚拟机 CPU 使用率"""
    try:
        info1 = dom.info()
        cpu_time1 = info1[2]
        t1 = time.time()

        with _cache_lock:
            prev = _stats_cache.get(cache_key)

        if prev:
            cpu_time0, t0 = prev
            elapsed = t1 - t0
            cpu_diff = cpu_time1 - cpu_time0
            cpu_percent = round((cpu_diff / 1e9) / elapsed * 100, 1)
            cpu_percent = min(cpu_percent, 100.0)
        else:
            cpu_percent = 0.0

        with _cache_lock:
            _stats_cache[cache_key] = (cpu_time1, t1)

        return cpu_percent
    except Exception:
        return 0.0


def _get_vm_disk_stats(dom) -> list:
    """获取虚拟机磁盘IO"""
    from lxml import etree
    stats = []
    try:
        xml = etree.fromstring(dom.XMLDesc(0).encode())
        for disk in xml.findall(".//disk[@device='disk']"):
            target = disk.find("target")
            if target is not None:
                dev = target.get("dev", "")
                try:
                    s = dom.blockStats(dev)
                    stats.append({
                        "dev": dev,
                        "read_bytes": s[1],
                        "write_bytes": s[3],
                        "read_requests": s[0],
                        "write_requests": s[2],
                    })
                except Exception:
                    pass
    except Exception:
        pass
    return stats


def _get_vm_net_stats(dom) -> list:
    """获取虚拟机网络IO"""
    from lxml import etree
    stats = []
    try:
        xml = etree.fromstring(dom.XMLDesc(0).encode())
        for iface in xml.findall(".//interface"):
            target = iface.find("target")
            if target is not None:
                dev = target.get("dev", "")
                try:
                    s = dom.interfaceStats(dev)
                    stats.append({
                        "dev": dev,
                        "rx_bytes": s[0],
                        "tx_bytes": s[4],
                        "rx_packets": s[1],
                        "tx_packets": s[5],
                        "rx_errors": s[2],
                        "tx_errors": s[6],
                    })
                except Exception:
                    pass
    except Exception:
        pass
    return stats