aboutsummaryrefslogtreecommitdiffstats
path: root/src/plugins/lua/fuzzy_collect.lua
diff options
context:
space:
mode:
Diffstat (limited to 'src/plugins/lua/fuzzy_collect.lua')
-rw-r--r--src/plugins/lua/fuzzy_collect.lua26
1 files changed, 14 insertions, 12 deletions
diff --git a/src/plugins/lua/fuzzy_collect.lua b/src/plugins/lua/fuzzy_collect.lua
index d64830c6b..132ace90c 100644
--- a/src/plugins/lua/fuzzy_collect.lua
+++ b/src/plugins/lua/fuzzy_collect.lua
@@ -39,7 +39,7 @@ local function send_data_mirror(m, cfg, ev_base, body)
rspamd_logger.infox(cfg, 'saved data on %s(%s)', m.server, m.name)
end
end
- rspamd_http.request{
+ rspamd_http.request {
url = string.format('http://%s//update_v1/%s', m.server, m.name),
resolver = cfg:get_resolver(),
config = cfg,
@@ -58,8 +58,10 @@ local function collect_fuzzy_hashes(cfg, ev_base)
rspamd_logger.errx(cfg, 'cannot load data: %s', err)
else
-- Here, we actually copy body once for each mirror
- fun.each(function(_, v) send_data_mirror(v, cfg, ev_base, body) end,
- settings.mirrors)
+ fun.each(function(_, v)
+ send_data_mirror(v, cfg, ev_base, body)
+ end,
+ settings.mirrors)
end
end
@@ -70,13 +72,13 @@ local function collect_fuzzy_hashes(cfg, ev_base)
if settings.saved_cookie ~= tostring(body) then
settings.saved_cookie = tostring(body)
rspamd_logger.infox(cfg, 'received collection cookie %s',
- tostring(rspamd_util.encode_base32(settings.saved_cookie:sub(1, 6))))
+ tostring(rspamd_util.encode_base32(settings.saved_cookie:sub(1, 6))))
local sig = rspamd_cryptolib.sign_memory(settings.sign_keypair,
- settings.saved_cookie)
+ settings.saved_cookie)
if not sig then
rspamd_logger.info(cfg, 'cannot sign cookie')
else
- rspamd_http.request{
+ rspamd_http.request {
url = string.format('http://%s/data', settings.collect_server),
resolver = cfg:get_resolver(),
config = cfg,
@@ -96,8 +98,8 @@ local function collect_fuzzy_hashes(cfg, ev_base)
end
end
rspamd_logger.infox(cfg, 'start fuzzy collection, next sync in %s seconds',
- settings.sync_time)
- rspamd_http.request{
+ settings.sync_time)
+ rspamd_http.request {
url = string.format('http://%s/cookie', settings.collect_server),
resolver = cfg:get_resolver(),
config = cfg,
@@ -141,7 +143,7 @@ end
local opts = rspamd_config:get_all_opt('fuzzy_collect')
if opts and type(opts) == 'table' then
- for k,v in pairs(opts) do
+ for k, v in pairs(opts) do
settings[k] = v
end
local sane_config = true
@@ -180,9 +182,9 @@ if opts and type(opts) == 'table' then
rspamd_config:add_on_load(function(_, ev_base, worker)
if worker:is_primary_controller() then
rspamd_config:add_periodic(ev_base, 0.0,
- function(cfg, _)
- return collect_fuzzy_hashes(cfg, ev_base)
- end)
+ function(cfg, _)
+ return collect_fuzzy_hashes(cfg, ev_base)
+ end)
end
end)
else