mirror of
https://github.com/ntop/ntopng.git
synced 2026-04-30 07:59:35 +00:00
140 lines
3 KiB
Lua
140 lines
3 KiB
Lua
--
|
|
-- (C) 2020 - ntop.org
|
|
--
|
|
|
|
local datamodel = {}
|
|
datamodel.__index = datamodel
|
|
|
|
local datamodel_colors = {
|
|
'rgba(255, 127, 14, 1)',
|
|
'rgba(174, 199, 232, 1)',
|
|
'rgba(255, 187, 120, 1)',
|
|
'rgba(31, 119, 180, 1)',
|
|
'rgba(255, 99, 132, 1)',
|
|
'rgba(54, 162, 235, 1)',
|
|
'rgba(255, 206, 86, 1)',
|
|
'rgba(75, 192, 192, 1)',
|
|
'rgba(153, 102, 255, 1)',
|
|
'rgba(255, 159, 64, 1)'
|
|
}
|
|
|
|
-- ######################################
|
|
|
|
function datamodel:create(labels)
|
|
local ret = {}
|
|
|
|
setmetatable(ret,datamodel) -- Create the class
|
|
|
|
ret.column_labels = labels
|
|
ret.datasets = {}
|
|
|
|
return(ret)
|
|
end
|
|
|
|
-- ######################################
|
|
|
|
function datamodel:appendRow(when, dataset_name, row)
|
|
if(self.datasets[dataset_name] == nil) then
|
|
self.datasets[dataset_name] = {}
|
|
|
|
self.datasets[dataset_name].rows = {}
|
|
self.datasets[dataset_name].timestamps = {}
|
|
end
|
|
|
|
table.insert(self.datasets[dataset_name].timestamps, when)
|
|
table.insert(self.datasets[dataset_name].rows, row)
|
|
|
|
end
|
|
|
|
-- ######################################
|
|
|
|
-- Return the data formatted as expected by a table widget
|
|
function datamodel:getAsTable()
|
|
local ret = {}
|
|
local dataset_name
|
|
|
|
-- take the first dataset
|
|
for k,v in pairs(self.datasets) do
|
|
dataset_name = k
|
|
end
|
|
|
|
ret.header = self.column_labels
|
|
|
|
if(dataset_name == nil) then
|
|
ret.rows = {}
|
|
else
|
|
ret.rows = self.datasets[dataset_name].rows
|
|
end
|
|
|
|
return(ret)
|
|
end
|
|
|
|
-- ######################################
|
|
|
|
-- Return the data formatted as expected by a donut chart
|
|
function datamodel:getAsDonut()
|
|
local ret = { data = {}}
|
|
|
|
for k,v in pairs(self.datasets) do
|
|
local i = 1
|
|
|
|
for k1,v1 in pairs(v.rows) do
|
|
for k2,v2 in pairs(v1) do
|
|
table.insert(ret.data, { label = self.column_labels[i], value = v2 })
|
|
i = i + 1
|
|
end
|
|
|
|
ret.title = k
|
|
break -- We expect only one entry
|
|
end
|
|
|
|
end
|
|
|
|
return(ret)
|
|
end
|
|
|
|
-- ######################################
|
|
|
|
-- Return the data formatted as expected by a multibar chart
|
|
function datamodel:getAsMultibar()
|
|
local ret = { }
|
|
local i = 0
|
|
|
|
for k,v in pairs(self.datasets) do
|
|
local serie = { values = {} }
|
|
local label = self.column_labels[i+1]
|
|
|
|
for k1,v1 in pairs(v.rows) do
|
|
table.insert(serie.values, { x = v.timestamps[k1], y = v1, series = i, y0 = 0, y1 = v1, key = label })
|
|
end
|
|
|
|
serie.key = label
|
|
serie.nonStackable = false
|
|
|
|
table.insert(ret, serie)
|
|
i = i + 1
|
|
end
|
|
|
|
return(ret)
|
|
end
|
|
|
|
-- ######################################
|
|
|
|
-- Return the data
|
|
function datamodel:getData(transformation, dataset_name)
|
|
transformation = string.lower(transformation)
|
|
|
|
if(transformation == "table") then
|
|
return(self:getAsTable())
|
|
elseif(transformation == "donut") then
|
|
return(self:getAsDonut())
|
|
elseif(transformation == "multibar") then
|
|
return(self:getAsMultibar())
|
|
else
|
|
return({})
|
|
end
|
|
end
|
|
|
|
-- ######################################
|
|
|
|
return(datamodel)
|