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

mod_storage_memory: Support query for set of IDs

Browse source
This commit is contained in:
Kim Alvefur 2021-01-12 18:06:33 +01:00
parent d5fb7e2510
commit 494239414f
1 changed files with 8 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

8
plugins/mod_storage_memory.lua
View file

@ -4,6 +4,7 @@ local envload = require "util.envload".envload;
local st = require "util.stanza"; local st = require "util.stanza";
local is_stanza = st.is_stanza or function (s) return getmetatable(s) == st.stanza_mt end local is_stanza = st.is_stanza or function (s) return getmetatable(s) == st.stanza_mt end
local new_id = require "util.id".medium; local new_id = require "util.id".medium;
local set = require "util.set";
local auto_purge_enabled = module:get_option_boolean("storage_memory_temporary", false); local auto_purge_enabled = module:get_option_boolean("storage_memory_temporary", false);
local auto_purge_stores = module:get_option_set("storage_memory_temporary_stores", {}); local auto_purge_stores = module:get_option_set("storage_memory_temporary_stores", {});
@ -58,6 +59,7 @@ archive_store.caps = {
quota = archive_item_limit; quota = archive_item_limit;
truncate = true; truncate = true;
full_id_range = true; full_id_range = true;
ids = true;
}; };
function archive_store:append(username, key, value, when, with) function archive_store:append(username, key, value, when, with)
@ -110,6 +112,12 @@ function archive_store:find(username, query)
return item.key == query.key; return item.key == query.key;
end); end);
end end
if query.ids then
local ids = set.new(query.ids);
items:filter(function (item)
return ids:contains(item.key);
end);
end
if query.with then if query.with then
items:filter(function (item) items:filter(function (item)
return item.with == query.with; return item.with == query.with;