ntopng/scripts/lua/get_ts.lua

111 lines
2.9 KiB
Lua

--
-- (C) 2013-18 - ntop.org
--
local dirs = ntop.getDirs()
package.path = dirs.installdir .. "/scripts/lua/modules/?.lua;" .. package.path
local nv_graph_utils
if ntop.isPro() then
package.path = dirs.installdir .. "/pro/scripts/lua/modules/?.lua;" .. package.path
nv_graph_utils = require "nv_graph_utils"
end
require "lua_utils"
require "graph_utils"
local ts_utils = require("ts_utils")
local json = require("dkjson")
local ts_schema = _GET["ts_schema"]
local query = _GET["ts_query"]
local tstart = tonumber(_GET["epoch_begin"]) or (os.time() - 3600)
local tend = tonumber(_GET["epoch_end"]) or os.time()
local compare_backward = _GET["ts_compare"]
local tags = tsQueryToTags(_GET["ts_query"])
local driver = ts_utils.getQueryDriver()
local latest_tstamp = driver:getLatestTimestamp(tags.ifid or -1)
local options = {
max_num_points = tonumber(_GET["limit"]),
initial_point = toboolean(_GET["initial_point"]),
no_timeout = _GET["no_timeout"],
with_series = true,
}
-- Check end time bound and realign if necessary
if tend > latest_tstamp then
local delta = tend - latest_tstamp
local alignment = (tend - tstart) / options.max_num_points
delta = delta + (alignment - delta % alignment)
tend = math.floor(tend - delta)
tstart = math.floor(tstart - delta)
end
if tags.ifid then
interface.select(tags.ifid)
end
sendHTTPHeader('application/json')
local function performQuery(tstart, tend, keep_total)
local res
if starts(ts_schema, "top:") then
local ts_schema = split(ts_schema, "top:")[2]
res = ts_utils.queryTopk(ts_schema, tags, tstart, tend, options)
else
res = ts_utils.query(ts_schema, tags, tstart, tend, options)
if(not keep_total) and (res) and (res.additional_series) then
-- no need for total serie in normal queries
res.additional_series.total = nil
end
end
return res
end
local res
if starts(ts_schema, "custom:") and ntop.isPro() then
res = performCustomQuery(ts_schema, tags, tstart, tend, options)
compare_backward = nil
else
res = performQuery(tstart, tend)
end
if res == nil then
res = {}
if(ts_utils.getLastError() ~= nil) then
res["tsLastError"] = ts_utils.getLastError()
res["error"] = ts_utils.getLastErrorMessage()
end
print(json.encode(res))
return
end
if not isEmptyString(compare_backward) and compare_backward ~= "1Y" then
local backward_sec = getZoomDuration(compare_backward)
local tstart_cmp = tstart - backward_sec
local tend_cmp = tend - backward_sec
local res_cmp = performQuery(tstart_cmp, tend_cmp, true)
if res_cmp and res_cmp.additional_series and res_cmp.additional_series.total then
res.additional_series = res.additional_series or {}
res.additional_series[compare_backward.. " " ..i18n("details.ago")] = res_cmp.additional_series.total
end
end
-- TODO make a script parameter?
local extend_labels = true
if extend_labels and ntop.isPro() then
extendLabels(res)
end
print(json.encode(res))