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
|
---------------------------------------------------
-- Licensed under the GNU General Public License v2
-- * (c) 2010, Adrian C. <anrxc@sysphere.org>
---------------------------------------------------
-- {{{ Grab environment
local ipairs = ipairs
local setmetatable = setmetatable
local table = { insert = table.insert }
local helpers = require("vicious.helpers")
local string = {
gmatch = string.gmatch,
format = string.format
}
-- }}}
-- Disk I/O: provides I/O statistics for requested storage devices
module("vicious.dio")
-- Initialise function tables
local disk_usage = {}
local disk_total = {}
-- {{{ Helper functions
local function uformat(array, key, value)
array["{"..key.."_s}"] = string.format("%.1f", value)
array["{"..key.."_kb}"] = string.format("%.1f", value/2)
array["{"..key.."_mb}"] = string.format("%.1f", value/2/1024)
return array
end
-- }}}
-- {{{ Disk I/O widget type
local function worker(format, disk)
local disk_lines = { disk = {} }
local disk_stats = helpers.pathtotable("/sys/block/" .. disk)
if disk_stats.stat then
local match = string.gmatch(disk_stats.stat, "[%s]+([%d]+)")
for i = 1, 11 do -- Store disk stats
table.insert(disk_lines.disk, match())
end
end
-- Ensure tables are initialized correctly
local diff_total = { disk = {} }
if not disk_total.disk then
disk_usage.disk = {}
disk_total.disk = {}
while #disk_total.disk < #disk_lines.disk do
table.insert(disk_total.disk, 0)
end
end
for i, v in ipairs(disk_lines.disk) do
-- Diskstats are absolute, substract our last reading
diff_total.disk[i] = v - disk_total.disk[i]
-- Store totals
disk_total.disk[i] = v
end
-- Calculate and store I/O
uformat(disk_usage.disk, "read", diff_total.disk[3])
uformat(disk_usage.disk, "write", diff_total.disk[7])
uformat(disk_usage.disk, "total", diff_total.disk[7] + diff_total.disk[3])
return disk_usage.disk
end
-- }}}
setmetatable(_M, { __call = function(_, ...) return worker(...) end })
|
