ntopng/scripts/lua/modules/storage_utils.lua
2018-12-26 16:31:09 +01:00

79 lines
2.2 KiB
Lua

--
-- (C) 2014-18 - ntop.org
--
local dirs = ntop.getDirs()
package.path = dirs.installdir .. "/scripts/lua/modules/?.lua;" .. package.path
if((dirs.scriptdir ~= nil) and (dirs.scriptdir ~= "")) then package.path = dirs.scriptdir .. "/lua/modules/?.lua;" .. package.path end
if ntop.isEnterprise() then
package.path = dirs.installdir .. "/pro/scripts/lua/modules/?.lua;" .. package.path
end
require "lua_utils"
local json = require("dkjson")
local rrd_utils = require "rrd_utils"
local recording_utils = require "recording_utils"
local storage_utils = {}
-- #################################
function storage_utils.interfaceStorageInfo(ifid)
local info = { total = 0}
local key = "ntopng.cache."..ifid..".storage_info"
local info_json = ntop.getCache(key)
if not isEmptyString(info_json) then
info = json.decode(info_json)
else
-- if ts_utils.getDriverName() == "rrd" then
local rrd_storage_info = rrd_utils.storageInfo(ifid)
info["rrd"] = rrd_storage_info.total
info["total"] = info["total"] + rrd_storage_info.total
-- end
if ntop.isEnterprise() and hasNindexSupport() then
local nindex_utils = require "nindex_utils"
local flows_storage_info = nindex_utils.storageInfo(ifid)
info["flows"] = flows_storage_info.total
info["total"] = info["total"] + flows_storage_info.total
end
-- if recording_utils.isAvailable() then
if not ntop.isWindows() then
local pcap_storage_info = recording_utils.storageInfo(ifid)
local total_pcap_dump_used = (pcap_storage_info.if_used + pcap_storage_info.extraction_used)
info["pcap"] = total_pcap_dump_used
info["total"] = info["total"] + total_pcap_dump_used
end
-- end
ntop.setCache(key, json.encode(info), 60)
end
return info
end
-- #################################
function storage_utils.storageInfo()
local ifnames = interface.getIfNames()
local info = { total = 0, interfaces = {} }
for id, name in pairs(ifnames) do
local ifid = tonumber(id)
local if_info = storage_utils.interfaceStorageInfo(ifid)
info.interfaces[ifid] = if_info
info.total = info.total + if_info.total
end
return info
end
-- #################################
return storage_utils