mirrors/prosody
1
0
Fork 0
mirror of https://github.com/bjc/prosody.git synced 2025-04-03 21:27:38 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

mod_blocklist: Use util.cache to manage how many users blocklists are kept in memory

Browse source
This commit is contained in:
Kim Alvefur 2015-11-25 21:06:01 +01:00
parent dd9e60beba
commit 13d34d918f
1 changed files with 13 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

20
plugins/mod_blocklist.lua
View file

@ -19,12 +19,13 @@ local jid_split = require"util.jid".split;
local storage = module:open_store();
local sessions = prosody.hosts[module.host].sessions;
-- Cache of blocklists used since module was loaded
local cache = {};
if module:get_option_boolean("blocklist_weak_cache") then
-- Lower memory usage, more IO and latency
setmetatable(cache, { __mode = "v" });
end
-- Cache of blocklists by username may randomly expire at any time
local cache = setmetatable({}, { __mode = "v" });
-- Second level of caching, keeps a fixed number of items,
-- also anchors items in the above cache
local cache_size = module:get_option_number("blocklist_cache_size", 64);
local cache2 = require"util.cache".new(cache_size);
local null_blocklist = {};
@ -36,6 +37,7 @@ local function set_blocklist(username, blocklist)
return ok, err;
end
-- Successful save, update the cache
cache2:set(username, blocklist);
cache[username] = blocklist;
return true;
end
@ -71,6 +73,9 @@ end
local function get_blocklist(username)
local blocklist = cache[username];
if not blocklist then
blocklist = cache2:get(username);
end
if not blocklist then
if not user_exists(username, module.host) then
return null_blocklist;
@ -79,8 +84,9 @@ local function get_blocklist(username)
if not blocklist then
blocklist = migrate_privacy_list(username);
end
cache[username] = blocklist;
cache2:set(username, blocklist);
end
cache[username] = blocklist;
return blocklist;
end