File: alert_store_utils.lua

package info (click to toggle)
ntopng 5.2.1%2Bdfsg1-2
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 121,832 kB
  • sloc: javascript: 143,431; cpp: 71,175; ansic: 11,108; sh: 4,687; makefile: 911; python: 587; sql: 512; pascal: 234; perl: 118; ruby: 52; exp: 4
file content (77 lines) | stat: -rw-r--r-- 2,229 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
--
-- (C) 2021-22 - ntop.org
--
local dirs = ntop.getDirs()
package.path = dirs.installdir .. "/scripts/lua/modules/alert_store/?.lua;" .. package.path

require "lua_utils"
local os_utils = require "os_utils"

-- ##############################################

local alert_store_utils = {}

-- ##############################################

-- @brief Returns an array of all available alert_store Lua class instances
function alert_store_utils.all_instances_factory()
   local alert_store_dir = os_utils.fixPath(dirs.installdir .. "/scripts/lua/modules/alert_store/")
   local res = {}

   for alert_store_file in pairs(ntop.readdir(alert_store_dir)) do
      
      -- Load all sub-classes of alert_store
      if alert_store_file:match("_alert_store%.lua$") then
	 if not ntop.isPro() then
	    if string.find(alert_store_file, "snmp") then
	       goto continue
	    end
	 end

	 -- Note: skip special instances without a family (e.g. 'all')
	 if alert_store_file:match("all_alert_store%.lua$") then
	    goto continue
	 end
	    
         local file_info = string.split(alert_store_file, "%.")
         local instance_name = file_info[1]

	 local instance = require(instance_name).new()

	 if instance then
            local family_name = instance:get_family()

            if family_name then
   	       res[family_name] = instance
            end
	 end

	 ::continue::
      end
   end

   return res
end

-- ##############################################

-- @brief Call instance:db_cleanup for every available alert_store instance
function alert_store_utils.housekeeping(ifid)
   -- NOTE: Clickhouse housekeeping is done daily in clickhouseDeleteOldPartitions
   if not ntop.isClickHouseEnabled() then
      local all_instances = alert_store_utils.all_instances_factory()
      for _, instance in pairs(all_instances) do
	 instance:housekeeping(ifid)
      end

      -- Reclaims unused disk space and defragments tables and indices.
      -- Should be called as disk space and defragmentation are not run
      -- automatically by sqlite.
      local q = string.format("VACUUM")
      local vacuum = interface.alert_store_query(q)
   end
end

-- ##############################################

return alert_store_utils