1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
### Author: Dmitry Fedin <dmitry.fedin@gmail.com>
class dstat_plugin(dstat):
ibdirname = '/sys/class/infiniband'
"""
Bytes received or sent through infiniband/RoCE interfaces
Usage:
dstat --ib -N <adapter name>:<port>,total
default dstat --ib is the same as
dstat --ib -N total
example for Mellanox adapter, transfering data via port 2
dstat --ib -Nmlx4_0:2
"""
def __init__(self):
self.nick = ('recv', 'send')
self.type = 'd'
self.cols = 2
self.width = 6
def discover(self, *objlist):
ret = []
for subdirname in os.listdir(self.ibdirname):
if not os.path.isdir(os.path.join(self.ibdirname,subdirname)) : continue
device_dir = os.path.join(self.ibdirname, subdirname, 'ports')
for subdirname2 in os.listdir(device_dir) :
if not os.path.isdir(os.path.join(device_dir,subdirname2)): continue
name = subdirname + ":" + subdirname2
ret.append(name)
ret.sort()
for item in objlist: ret.append(item)
return ret
def vars(self):
ret = []
if op.netlist:
varlist = op.netlist
elif not op.full:
varlist = ('total',)
else:
varlist = self.discover
varlist.sort()
for name in varlist:
if name in self.discover + ['total']:
ret.append(name)
if not ret:
raise Exception('No suitable network interfaces found to monitor')
return ret
def name(self):
return ['ib/'+name for name in self.vars]
def extract(self):
self.set2['total'] = [0, 0]
ifaces = self.discover
for name in self.vars: self.set2[name] = [0, 0]
for name in ifaces:
l=name.split(':');
if len(l) < 2:
continue
rcv_counter_name=os.path.join('/sys/class/infiniband', l[0], 'ports', l[1], 'counters_ext/port_rcv_data_64')
xmit_counter_name=os.path.join('/sys/class/infiniband', l[0], 'ports', l[1], 'counters_ext/port_xmit_data_64')
rcv_lines = dopen(rcv_counter_name).readlines()
xmit_lines = dopen(xmit_counter_name).readlines()
if len(rcv_lines) < 1 or len(xmit_lines) < 1:
continue
rcv_value = int(rcv_lines[0])
xmit_value = int(xmit_lines[0])
if name in self.vars :
self.set2[name] = (rcv_value, xmit_value)
self.set2['total'] = ( self.set2['total'][0] + rcv_value, self.set2['total'][1] + xmit_value)
if update:
for name in self.set2:
self.val[name] = [
(self.set2[name][0] - self.set1[name][0]) * 4.0 / elapsed,
(self.set2[name][1] - self.set1[name][1]) * 4.0/ elapsed,
]
if self.val[name][0] < 0: self.val[name][0] += maxint + 1
if self.val[name][1] < 0: self.val[name][1] += maxint + 1
if step == op.delay:
self.set1.update(self.set2)
|
