diff options
Diffstat (limited to 'applications/luci-statistics/luasrc/statistics/rrdtool.lua')
-rw-r--r-- | applications/luci-statistics/luasrc/statistics/rrdtool.lua | 428 |
1 files changed, 298 insertions, 130 deletions
diff --git a/applications/luci-statistics/luasrc/statistics/rrdtool.lua b/applications/luci-statistics/luasrc/statistics/rrdtool.lua index 783314e8c..17ac48a5a 100644 --- a/applications/luci-statistics/luasrc/statistics/rrdtool.lua +++ b/applications/luci-statistics/luasrc/statistics/rrdtool.lua @@ -3,8 +3,8 @@ module("luci.statistics.rrdtool", package.seeall) require("luci.statistics.datatree") require("luci.statistics.rrdtool.colors") require("luci.statistics.rrdtool.definitions") +require("luci.i18n") require("luci.util") -require("luci.bits") require("luci.fs") @@ -13,22 +13,34 @@ Graph = luci.util.class() function Graph.__init__( self, timespan, opts ) opts = opts or { } - opts.width = opts.width or "400" self.colors = luci.statistics.rrdtool.colors.Instance() self.defs = luci.statistics.rrdtool.definitions.Instance() self.tree = luci.statistics.datatree.Instance() + self.i18n = luci.i18n - -- rrdtool defalt args - self.args = { + -- options + opts.rrasingle = opts.rrasingle or true -- XXX: fixme (uci) + opts.host = opts.host or "OpenWrt" -- XXX: fixme (uci) + opts.timespan = opts.timespan or 900 -- XXX: fixme (uci) + opts.width = opts.width or 400 -- XXX: fixme (uci) + + -- rrdtool default args + self.args = { "-a", "PNG", - "-s", "NOW-" .. ( timespan or 900 ), + "-s", "NOW-" .. opts.timespan, "-w", opts.width } + + -- store options + self.opts = opts + + -- load language file + self.i18n.loadc("statistics") end -function Graph.mktitle( self, host, plugin, plugin_instance, dtype, dtype_instance ) - local t = host .. "/" .. plugin +function Graph.mktitle( self, plugin, plugin_instance, dtype, dtype_instance ) + local t = self.opts.host .. "/" .. plugin if type(plugin_instance) == "string" and plugin_instance:len() > 0 then t = t .. "-" .. plugin_instance end @@ -47,25 +59,6 @@ function Graph.mkpngpath( self, ... ) return string.format( "/tmp/rrdimg/%s.png", self:mktitle( ... ) ) end -function Graph._push( self, elem ) - - if type(elem) == "string" then - table.insert( self.args, elem ) - else - for i, item in ipairs(elem) do - table.insert( self.args, item ) - end - end - - return( self.args ) -end - -function Graph._clearargs( self ) - for i = #self.args, 7, -1 do - table.remove( self.args, i ) - end -end - function Graph._forcelol( self, list ) if type(list[1]) ~= "table" then return( { list } ) @@ -73,17 +66,22 @@ function Graph._forcelol( self, list ) return( list ) end -function Graph._rrdtool( self, png, rrd ) +function Graph._rrdtool( self, def, rrd ) -- prepare directory - local dir = png:gsub("/[^/]+$","") + local dir = def[1]:gsub("/[^/]+$","") luci.fs.mkdir( dir, true ) -- construct commandline - local cmdline = "rrdtool graph " .. png + local cmdline = "rrdtool graph" + -- copy default arguments to def stack for i, opt in ipairs(self.args) do + table.insert( def, 1 + i, opt ) + end + -- construct commandline from def stack + for i, opt in ipairs(def) do opt = opt .. "" -- force string if rrd then @@ -102,12 +100,14 @@ function Graph._rrdtool( self, png, rrd ) rrdtool:close() end -function Graph._generic( self, opts ) +function Graph._generic( self, opts, plugin, plugin_instance, dtype, index ) - local images = { } - local rrasingle = false -- XXX: fixme + -- generated graph defs + local defs = { } -- internal state variables + local _args = { } + local _sources = { } local _stack_neg = { } local _stack_pos = { } local _longest_name = 0 @@ -117,6 +117,11 @@ function Graph._generic( self, opts ) local _ti = table.insert local _sf = string.format + -- local helper: append a string.format() formatted string to given table + function _tif( list, fmt, ... ) + table.insert( list, string.format( fmt, ... ) ) + end + -- local helper: create definitions for min, max, avg and create *_nnl (not null) variable from avg function __def(source) @@ -126,22 +131,19 @@ function Graph._generic( self, opts ) if not ds or ds:len() == 0 then ds = "value" end - local rv = { _sf( "DEF:%s_avg=%s:%s:AVERAGE", inst, rrd, ds ) } + _tif( _args, "DEF:%s_avg=%s:%s:AVERAGE", inst, rrd, ds ) - if not rrasingle then - _ti( rv, _sf( "DEF:%s_min=%s:%s:MIN", inst, rrd, ds ) ) - _ti( rv, _sf( "DEF:%s_max=%s:%s:MAX", inst, rrd, ds ) ) + if not self.opts.rrasingle then + _tif( _args, "DEF:%s_min=%s:%s:MIN", inst, rrd, ds ) + _tif( _args, "DEF:%s_max=%s:%s:MAX", inst, rrd, ds ) end - _ti( rv, _sf( "CDEF:%s_nnl=%s_avg,UN,0,%s_avg,IF", inst, inst, inst ) ) - - return rv + _tif( _args, "CDEF:%s_nnl=%s_avg,UN,0,%s_avg,IF", inst, inst, inst ) end -- local helper: create cdefs depending on source options like flip and overlay function __cdef(source) - local rv = { } local prev -- find previous source, choose stack depending on flip state @@ -154,21 +156,19 @@ function Graph._generic( self, opts ) -- is first source in stack or overlay source: source_stk = source_nnl if not prev or source.overlay then -- create cdef statement - _ti( rv, _sf( "CDEF:%s_stk=%s_nnl", source.sname, source.sname ) ) + _tif( _args, "CDEF:%s_stk=%s_nnl", source.sname, source.sname ) -- is subsequent source without overlay: source_stk = source_nnl + previous_stk else -- create cdef statement - _ti( rv, _sf( - "CDEF:%s_stk=%s_nnl,%s_stk,+", source.sname, source.sname, prev - ) ) + _tif( _args, "CDEF:%s_stk=%s_nnl,%s_stk,+", source.sname, source.sname, prev ) end -- create multiply by minus one cdef if flip is enabled if source.flip then -- create cdef statement: source_stk = source_stk * -1 - _ti( rv, _sf( "CDEF:%s_neg=%s_stk,-1,*", source.sname, source.sname ) ) + _tif( _args, "CDEF:%s_neg=%s_stk,-1,*", source.sname, source.sname ) -- push to negative stack if overlay is disabled if not source.overlay then @@ -184,36 +184,29 @@ function Graph._generic( self, opts ) -- calculate total amount of data if requested if source.total then - _ti( rv, _sf( + _tif( _args, "CDEF:%s_avg_sample=%s_avg,UN,0,%s_avg,IF,sample_len,*", source.sname, source.sname, source.sname - ) ) + ) - _ti( rv, _sf( + _tif( _args, "CDEF:%s_avg_sum=PREV,UN,0,PREV,IF,%s_avg_sample,+", source.sname, source.sname, source.sname - ) ) + ) end - - - return rv end -- local helper: create cdefs required for calculating total values function __cdef_totals() if _has_totals then - return { - _sf( "CDEF:mytime=%s_avg,TIME,TIME,IF", opts.sources[1].sname ), - "CDEF:sample_len_raw=mytime,PREV(mytime),-", - "CDEF:sample_len=sample_len_raw,UN,0,sample_len_raw,IF" - } - else - return { } + _tif( _args, "CDEF:mytime=%s_avg,TIME,TIME,IF", _sources[1].sname ) + _ti( _args, "CDEF:sample_len_raw=mytime,PREV(mytime),-" ) + _ti( _args, "CDEF:sample_len=sample_len_raw,UN,0,sample_len_raw,IF" ) end end -- local helper: create line and area statements - function __area(source) + function __line(source) local line_color local area_color @@ -243,96 +236,236 @@ function Graph._generic( self, opts ) end -- create legend - legend = _sf( "%-" .. _longest_name .. "s", source.name ) + legend = _sf( "%-" .. _longest_name .. "s", source.title ) + + -- create area if not disabled + if not source.noarea then + _tif( _args, "AREA:%s_%s#%s", source.sname, var, area_color ) + end - -- create area and line1 statement - return { - _sf( "AREA:%s_%s#%s", source.sname, var, area_color ), - _sf( "LINE1:%s_%s#%s:%s", source.sname, var, line_color, legend ) - } + -- create line1 statement + _tif( _args, "LINE1:%s_%s#%s:%s", source.sname, var, line_color, legend ) end -- local helper: create gprint statements function __gprint(source) - local rv = { } local numfmt = opts.number_format or "%6.1lf" local totfmt = opts.totals_format or "%5.1lf%s" -- don't include MIN if rrasingle is enabled - if not rrasingle then - _ti( rv, _sf( "GPRINT:%s_min:MIN:%s Min", source.sname, numfmt ) ) + if not self.opts.rrasingle then + _tif( _args, "GPRINT:%s_min:MIN:%s Min", source.sname, numfmt ) end -- always include AVERAGE - _ti( rv, _sf( "GPRINT:%s_avg:AVERAGE:%s Avg", source.sname, numfmt ) ) + _tif( _args, "GPRINT:%s_avg:AVERAGE:%s Avg", source.sname, numfmt ) -- don't include MAX if rrasingle is enabled - if not rrasingle then - _ti( rv, _sf( "GPRINT:%s_max:MAX:%s Max", source.sname, numfmt ) ) + if not self.opts.rrasingle then + _tif( _args, "GPRINT:%s_max:MAX:%s Max", source.sname, numfmt ) end -- include total count if requested else include LAST if source.total then - _ti( rv, _sf( "GPRINT:%s_avg_sum:LAST:(ca. %s Total)", source.sname, totfmt ) ) + _tif( _args, "GPRINT:%s_avg_sum:LAST:(ca. %s Total)\\l", source.sname, totfmt ) else - _ti( rv, _sf( "GPRINT:%s_avg:LAST:%s Last", source.sname, numfmt ) ) + _tif( _args, "GPRINT:%s_avg:LAST:%s Last\\l", source.sname, numfmt ) end + end + - -- end label line - rv[#rv] = rv[#rv] .. "\\l" + -- + -- find all data sources + -- + -- find data types + local data_types - return rv + if dtype then + data_types = { dtype } + else + data_types = opts.data.types or { } end + if not ( dtype or opts.data.types ) then + if opts.data.instances then + for k, v in pairs(opts.data.instances) do + _ti( data_types, k ) + end + elseif opts.data.sources then + for k, v in pairs(opts.data.sources) do + _ti( data_types, k ) + end + end + end - -- remember images - _ti( images, opts.image ) - -- insert provided addition rrd options - self:_push( { "-t", opts.title or "Unknown title" } ) - self:_push( opts.rrd ) + -- iterate over data types + for i, dtype in ipairs(data_types) do - -- store index and safe instance name within each source object, - -- find longest instance name - for i, source in ipairs(opts.sources) do + -- find instances - if source.name:len() > _longest_name then - _longest_name = source.name:len() - end + local data_instances - if source.total then - _has_totals = true + if not opts.per_instance then + if type(opts.data.instances) == "table" and type(opts.data.instances[dtype]) == "table" then + data_instances = opts.data.instances[dtype] + else + data_instances = self.tree:data_instances( plugin, plugin_instance, dtype ) + end end - source.index = i - source.sname = i .. source.name:gsub("[^A-Za-z0-9%-_]","_") - end + if type(data_instances) ~= "table" or #data_instances == 0 then data_instances = { "" } end + + + -- iterate over data instances + for i, dinst in ipairs(data_instances) do + + -- construct combined data type / instance name + local dname = dtype - -- create DEF statements for each instance, find longest instance name - for i, source in ipairs(opts.sources) do - self:_push( __def( source ) ) + if dinst:len() > 0 then + dname = dname .. "_" .. dinst + end + + + -- find sources + local data_sources = { "value" } + + if type(opts.data.sources) == "table" then + if type(opts.data.sources[dname]) == "table" then + data_sources = opts.data.sources[dname] + elseif type(opts.data.sources[dtype]) == "table" then + data_sources = opts.data.sources[dtype] + end + end + + + -- iterate over data sources + for i, dsource in ipairs(data_sources) do + + local dsname = dtype .. "_" .. dinst:gsub("[^%w]","_") .. "_" .. dsource + local altname = dtype .. "__" .. dsource + + -- find datasource options + local dopts = { } + + if type(opts.data.options) == "table" then + if type(opts.data.options[dsname]) == "table" then + dopts = opts.data.options[dsname] + elseif type(opts.data.options[altname]) == "table" then + dopts = opts.data.options[altname] + elseif type(opts.data.options[dname]) == "table" then + dopts = opts.data.options[dname] + elseif type(opts.data.options[dtype]) == "table" then + dopts = opts.data.options[dtype] + end + end + + + -- store values + _ti( _sources, { + title = dsname, -- XXX: fixme i18n (dopts.title || i18n || dname) + rrd = dopts.rrd or self:mkrrdpath( plugin, plugin_instance, dtype, dinst ), + color = dopts.color or self.colors:to_string( self.colors:random() ), + flip = dopts.flip or false, + total = dopts.total or false, + overlay = dopts.overlay or false, + noarea = dopts.noarea or false, + ds = dsource, + type = dtype, + instance = dinst, + index = #_sources + 1, + sname = ( #_sources + 1 ) .. dtype + } ) + + + -- find longest name ... + if _sources[#_sources].title:len() > _longest_name then + _longest_name = _sources[#_sources].title:len() + end + + + -- has totals? + if _sources[#_sources].total then + _has_totals = true + end + end + end end - -- create CDEF required for calculating totals - self:_push( __cdef_totals() ) - -- create CDEF statements for each instance in reversed order - for i, source in ipairs(opts.sources) do - self:_push( __cdef( opts.sources[1 + #opts.sources - i] ) ) + -- + -- construct diagrams + -- + + -- if per_instance is enabled then find all instances from the first datasource in diagram + -- if per_instance is disabled then use an empty pseudo instance and use model provided values + local instances = { "" } + + if opts.per_instance then + instances = self.tree:data_instances( plugin, plugin_instance, _sources[1].type ) end - -- create LINE1, AREA and GPRINT statements for each instance - for i, source in ipairs(opts.sources) do - self:_push( __area( source ) ) - self:_push( __gprint( source ) ) + + -- iterate over instances + for i, instance in ipairs(instances) do + + -- store title and vlabel + -- XXX: i18n + _ti( _args, "-t" ) + _ti( _args, opts.title ) + _ti( _args, "-v" ) + _ti( _args, opts.vlabel ) + + -- store additional rrd options + if opts.rrdopts then + for i, o in ipairs(opts.rrdopts) do _ti( _args, o ) end + end + + + -- create DEF statements for each instance + for i, source in ipairs(_sources) do + -- fixup properties for per instance mode... + if opts.per_instance then + source.instance = instance + source.rrd = self:mkrrdpath( plugin, plugin_instance, source.type, instance ) + end + + __def( source ) + end + + -- create CDEF required for calculating totals + __cdef_totals() + + -- create CDEF statements for each instance in reversed order + for i, source in ipairs(_sources) do + __cdef( _sources[1 + #_sources - i] ) + end + + -- create LINE1, AREA and GPRINT statements for each instance + for i, source in ipairs(_sources) do + __line( source ) + __gprint( source ) + end + + -- prepend image path to arg stack + _ti( _args, 1, self:mkpngpath( plugin, plugin_instance, index .. instance ) ) + + -- push arg stack to definition list + _ti( defs, _args ) + + -- reset stacks + _args = { } + _stack_pos = { } + _stack_neg = { } end - return images + return defs end -function Graph.render( self, host, plugin, plugin_instance ) +function Graph.render( self, plugin, plugin_instance ) dtype_instances = dtype_instances or { "" } local pngs = { } @@ -342,15 +475,33 @@ function Graph.render( self, host, plugin, plugin_instance ) local stat, def = pcall( require, plugin_def ) if stat and def and type(def.rrdargs) == "function" then - for i, opts in ipairs( self:_forcelol( def.rrdargs( self, host, plugin, plugin_instance, dtype ) ) ) do - for i, png in ipairs( self:_generic( opts ) ) do - table.insert( pngs, png ) + + -- temporary image matrix + local _images = { } + + -- get diagram definitions + for i, opts in ipairs( self:_forcelol( def.rrdargs( self, plugin, plugin_instance ) ) ) do + + _images[i] = { } + + -- get diagram definition instances + local diagrams = self:_generic( opts, plugin, plugin_instance, nil, i ) + + -- render all diagrams + for j, def in ipairs( diagrams ) do + + -- remember image + _images[i][j] = def[1] -- exec - self:_rrdtool( png ) + self:_rrdtool( def ) + end + end - -- clear args - self:_clearargs() + -- remember images - XXX: fixme (will cause probs with asymmetric data) + for y = 1, #_images[1] do + for x = 1, #_images do + table.insert( pngs, _images[x][y] ) end end else @@ -363,15 +514,33 @@ function Graph.render( self, host, plugin, plugin_instance ) local stat, def = pcall( require, dtype_def ) if stat and def and type(def.rrdargs) == "function" then - for i, opts in ipairs( self:_forcelol( def.rrdargs( self, host, plugin, plugin_instance, dtype ) ) ) do - for i, png in ipairs( self:_generic( opts ) ) do - table.insert( pngs, png ) + + -- temporary image matrix + local _images = { } + + -- get diagram definitions + for i, opts in ipairs( self:_forcelol( def.rrdargs( self, plugin, plugin_instance, dtype ) ) ) do + + _images[i] = { } + + -- get diagram definition instances + local diagrams = self:_generic( opts, plugin, plugin_instance, dtype, i ) + + -- render all diagrams + for j, def in ipairs( diagrams ) do + + -- remember image + _images[i][j] = def[1] -- exec - self:_rrdtool( png ) + self:_rrdtool( def ) + end + end - -- clear args - self:_clearargs() + -- remember images - XXX: fixme (will cause probs with asymmetric data) + for y = 1, #_images[1] do + for x = 1, #_images do + table.insert( pngs, _images[x][y] ) end end else @@ -382,20 +551,20 @@ function Graph.render( self, host, plugin, plugin_instance ) -- iterate over data type instances for i, inst in ipairs( self.tree:data_instances( plugin, plugin_instance, dtype ) ) do - local title = self:mktitle( host, plugin, plugin_instance, dtype, inst ) - local png = self:mkpngpath( host, plugin, plugin_instance, dtype, inst ) - local rrd = self:mkrrdpath( host, plugin, plugin_instance, dtype, inst ) - - self:_push( { "-t", title } ) - self:_push( self.defs.definitions[dtype] ) + local title = self:mktitle( plugin, plugin_instance, dtype, inst ) + local png = self:mkpngpath( plugin, plugin_instance, dtype, inst ) + local rrd = self:mkrrdpath( plugin, plugin_instance, dtype, inst ) + local args = { png, "-t", title } + + for i, o in ipairs(self.defs.definitions[dtype]) do + table.insert( args, o ) + end + -- remember image table.insert( pngs, png ) -- exec - self:_rrdtool( png, rrd ) - - -- clear args - self:_clearargs() + self:_rrdtool( args, rrd ) end end end @@ -404,4 +573,3 @@ function Graph.render( self, host, plugin, plugin_instance ) return pngs end - |