aboutsummaryrefslogtreecommitdiffstats
path: root/lualib/lua_bayes_redis.lua
diff options
context:
space:
mode:
Diffstat (limited to 'lualib/lua_bayes_redis.lua')
-rw-r--r--lualib/lua_bayes_redis.lua61
1 files changed, 32 insertions, 29 deletions
diff --git a/lualib/lua_bayes_redis.lua b/lualib/lua_bayes_redis.lua
index 753399705..782e6fc47 100644
--- a/lualib/lua_bayes_redis.lua
+++ b/lualib/lua_bayes_redis.lua
@@ -134,40 +134,43 @@ exports.lua_bayes_init_statfile = function(classifier_ucl, statfile_ucl, symbol,
revision = 0, -- number of learns
}
local cursor = 0
- rspamd_config:add_periodic(ev_base, 0.0, function(cfg, _)
- local function stat_redis_cb(err, data)
- lua_util.debugm(N, cfg, 'stat redis cb: %s, %s', err, data)
+ if ev_base then
+ rspamd_config:add_periodic(ev_base, 0.0, function(cfg, _)
- if err then
- logger.warn(cfg, 'cannot get bayes statistics for %s: %s', symbol, err)
- else
- local new_cursor = data[1]
- current_data.users = current_data.users + data[2]
- current_data.revision = current_data.revision + data[3]
- if new_cursor == 0 then
- -- Done iteration
- final_data = lua_util.shallowcopy(current_data)
- current_data = {
- users = 0,
- revision = 0,
- }
- lua_util.debugm(N, cfg, 'final data: %s', final_data)
- stat_periodic_cb(cfg, final_data)
- end
+ local function stat_redis_cb(err, data)
+ lua_util.debugm(N, cfg, 'stat redis cb: %s, %s', err, data)
- cursor = new_cursor
+ if err then
+ logger.warn(cfg, 'cannot get bayes statistics for %s: %s', symbol, err)
+ else
+ local new_cursor = data[1]
+ current_data.users = current_data.users + data[2]
+ current_data.revision = current_data.revision + data[3]
+ if new_cursor == 0 then
+ -- Done iteration
+ final_data = lua_util.shallowcopy(current_data)
+ current_data = {
+ users = 0,
+ revision = 0,
+ }
+ lua_util.debugm(N, cfg, 'final data: %s', final_data)
+ stat_periodic_cb(cfg, final_data)
+ end
+
+ cursor = new_cursor
+ end
end
- end
- lua_redis.exec_redis_script(stat_script_id,
- { ev_base = ev_base, cfg = cfg, is_write = false },
- stat_redis_cb, { tostring(cursor),
- symbol,
- is_spam and "learns_spam" or "learns_ham",
- tostring(max_users) })
- return statfile_ucl.monitor_timeout or classifier_ucl.monitor_timeout or 30.0
- end)
+ lua_redis.exec_redis_script(stat_script_id,
+ { ev_base = ev_base, cfg = cfg, is_write = false },
+ stat_redis_cb, { tostring(cursor),
+ symbol,
+ is_spam and "learns_spam" or "learns_ham",
+ tostring(max_users) })
+ return statfile_ucl.monitor_timeout or classifier_ucl.monitor_timeout or 30.0
+ end)
+ end
return gen_classify_functor(redis_params, classify_script_id), gen_learn_functor(redis_params, learn_script_id)
end