Merged changes and fixed python xen bindings
[invirt/packages/xvm-munin-config.git] / host / usr / share / xvm-munin-host-config / plugins / xen_cpu
1 #!/usr/bin/python
2 import sys
3 sys.path.append('/usr/lib/xen-default/lib/python/')
4 from xen.xm.main import SERVER_LEGACY_XMLRPC, SERVER_XEN_API, parseServer, parseAuthentication
5 from xen.xend import sxp
6 import atexit
7 import sys
8
9 serverType, serverURI = parseServer()
10
11 if serverType != SERVER_LEGACY_XMLRPC:
12     print >>sys.stderr, "xen_cpu is untested with this Xen server type"
13     sys.exit(1)
14
15 if serverType == SERVER_XEN_API:
16     from xen.xm import XenAPI
17     server = XenAPI.Session(serverURI)
18     username, password = parseAuthentication()
19     server.login_with_password(username, password)
20     def logout():
21         try:
22             server.xenapi.session.logout()
23         except:
24             pass
25     atexit.register(logout)
26 else:
27     from xen.util.xmlrpcclient import ServerProxy
28     server = ServerProxy(serverURI)
29
30 if len(sys.argv) > 1:
31     cmd = sys.argv[1]
32 else:
33     cmd = None
34
35 def getDomains():
36     ret = {}
37     if serverType == SERVER_XEN_API:
38         domains = server.xenapi.VM.get_all_records()
39         metrics = server.xenapi.VM_metrics.get_all_records()
40         for d in domains.values():
41             ret['uuid_'+d['uuid'].replace('-', '_')] = {'name': d['name_label'],
42                                                         'cpu_time': sum(metrics[d['metrics']]['VCPUs_utilisation'].values()),
43                                                         'domid': d['domid'],
44                                                         'uuid': d['uuid'],
45                                                         # No equivalent
46                                                         }
47         return ret
48     else:
49         domains = server.xend.domains_with_state(True, 'all', True)
50         for d in domains:
51             data = {'name': sxp.child_value(d, 'name', 'UNKNOWN'),
52                     'cpu_time': sxp.child_value(d, 'cpu_time', 0.0),
53                     'domid': sxp.child_value(d, 'domid', -1),
54                     'uuid': sxp.child_value(d, 'uuid', 'NONE'),
55                     }
56             try:
57                 sched = server.xend.domain.sched_credit_get(data['name'])
58                 data['sched-credit'] = sched
59             except:
60                 data['sched-credit'] = None
61             ret['uuid_'+sxp.child_value(d, 'uuid', 'NONE').replace('-', '_')] = data
62         return ret
63
64 if cmd == 'config':
65     print """
66 graph_title Xen domain CPU usage
67 graph_args --base 1000 -r --lower-limit 0 --upper-limit 800
68 graph_vlabel %
69 graph_scale no
70 graph_info This graph shows how CPU time is spent by Xen domains.
71 graph_category system
72 graph_period second"""
73     domains = getDomains()
74     for d in sorted(domains):
75         name = domains[d]['name']
76         if name[0:2] == 'd_':
77             name = 'db domid %d' % domains[d]['domid']
78         print "%s.label %s" % (d, name)
79         if domains[d]['domid'] == 0:
80             print "%s.draw AREA" % d
81         else:
82             print "%s.draw STACK" % d
83         print "%s.max 19200000000" % d # 64x 100% CPU usage
84         print "%s.min 0" % d
85         print "%s.type DERIVE" % d
86         if domains[d].get('sched-credit'):
87             print "%s.info uuid %s CPU weight %d cap %d%%" % (d, domains[d]['uuid'], domains[d]['sched-credit']['weight'], domains[d]['sched-credit']['cap'])
88         else:
89             print "%s.info uuid %s" % (d, domains[d]['uuid'])
90         print "%s.cdef %s,10000,/" % (d, d)
91     sys.exit(0)
92
93 domains = getDomains()
94 for d in sorted(domains):
95     print "%s.value %s" % (d, long(domains[d]['cpu_time']*1000000))