examples/nodestats.py

   1 #!/usr/bin/env python
   2 # Print some host NUMA node statistics
   3 #
   4 # Authors:
   5 #   Michal Privoznik <mprivozn@redhat.com>
   6
   7 import libvirt
   8 import sys
   9 from xml.dom import minidom
  10 import libxml2
  11
  12 def xpath_eval(ctxt, path):
  13     res = ctxt.xpathEval(path)
  14     if res is None or len(res) == 0:
  15         value = None
  16     else:
  17         value = res[0].content
  18     return value
  19
  20 try:
  21     conn = libvirt.openReadOnly(None)
  22 except libvirt.libvirtError:
  23     print("Failed to connect to the hypervisor")
  24     sys.exit(1)
  25
  26 try:
  27     capsXML = conn.getCapabilities()
  28 except libvirt.libvirtError:
  29     print("Failed to request capabilities")
  30     sys.exit(1)
  31
  32 caps = minidom.parseString(capsXML)
  33 cells = caps.getElementsByTagName("cells")[0]
  34
  35 nodesIDs = [ int(proc.getAttribute("id"))
  36              for proc in cells.getElementsByTagName("cell") ]
  37
  38 nodesMem = [ conn.getMemoryStats(int(proc))
  39              for proc in nodesIDs]
  40
  41 doms = conn.listAllDomains(libvirt.VIR_CONNECT_LIST_DOMAINS_ACTIVE)
  42
  43 domsStrict = [ proc
  44                for proc in doms
  45                if proc.numaParameters()["numa_mode"] == libvirt.VIR_DOMAIN_NUMATUNE_MEM_STRICT ]
  46
  47 domsStrictCfg = {}
  48 for dom in domsStrict:
  49     xmlStr = dom.XMLDesc()
  50     doc = libxml2.parseDoc(xmlStr)
  51     ctxt = doc.xpathNewContext()
  52
  53     domsStrictCfg[dom] = {}
  54
  55     pin = ctxt.xpathEval("string(/domain/numatune/memory/@nodeset)")
  56     memsize = ctxt.xpathEval("string(/domain/memory)")
  57     domsStrictCfg[dom]["memory"] = {"size": int(memsize), "pin": pin}
  58
  59     for memnode in ctxt.xpathEval("/domain/numatune/memnode"):
  60         ctxt.setContextNode(memnode)
  61         cellid = xpath_eval(ctxt, "@cellid")
  62         nodeset = xpath_eval(ctxt, "@nodeset")
  63
  64         nodesize = xpath_eval(ctxt, "/domain/cpu/numa/cell[@id='%s']/@memory" % cellid)
  65         domsStrictCfg[dom][cellid] = {"size": int(nodesize), "pin": nodeset}
  66
  67
  68 print("NUMA stats")
  69 print("NUMA nodes:\t" + "\t".join(str(node) for node in nodesIDs))
  70 print("MemTotal:\t" + "\t".join(str(i.get("total") // 1024) for i in nodesMem))
  71 print("MemFree:\t" + "\t".join(str(i.get("free") // 1024) for i in nodesMem))
  72
  73 for dom, v in domsStrictCfg.items():
  74     print("Domain '%s':\t" % dom.name())
  75
  76     toPrint = "\tOverall memory: %d MiB" % (v["memory"]["size"] // 1024)
  77     if v["memory"]["pin"] is not None and v["memory"]["pin"] is not "":
  78         toPrint = toPrint + " nodes %s" % v["memory"]["pin"]
  79     print(toPrint)
  80
  81     for k, node in sorted(v.items()):
  82         if k is "memory":
  83             continue
  84         toPrint = "\tNode %s:\t%d MiB" % (k, node["size"] // 1024)
  85         if node["pin"] is not None and node["pin"] is not "":
  86             toPrint = toPrint + " nodes %s" % node["pin"]
  87         print(toPrint)