Initial work towards REST-ful code rework

2026-05-01 00:19:33 +00:00 · 2019-01-30 10:54:09 +01:00 · 2019-01-30 10:54:09 +01:00 · 36b34389ac
commit 36b34389ac
parent 299ad0b5e2
7 changed files with 28 additions and 15 deletions
--- a/scripts/lua/rest/get/timeseries/ts.lua
+++ b/scripts/lua/rest/get/timeseries/ts.lua
@ -0,0 +1,121 @@
+--
+-- (C) 2013-18 - ntop.org
+--
+
+--
+-- Example of REST call
+-- 
+-- curl -u admin:admin -X POST -d '{"ts_schema":"host:traffic", "ts_query": "ifid:3,host:192.168.1.98", "epoch_begin": "1532180495", "epoch_end": "1548839346"}' -H "Content-Type: application/json" "http://127.0.0.1:3000/lua/rest/get/timeseries/ts.lua"
+--
+
+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    = _GET["epoch_begin"]
+local tend      = _GET["epoch_end"]
+local compare_backward = _GET["ts_compare"]
+local tags      = _GET["ts_query"]
+
+tstart = tonumber(tstart) or (os.time() - 3600)
+tend = tonumber(tend) or os.time()
+tags = tsQueryToTags(tags)
+
+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))