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

util.pubsub: Add support for limiting number of items to retrieve

Browse source 
Hopefully this will eventually be upgraded to RSM, which is why the
argument is called 'resultspec' and is a table.
This commit is contained in:
Kim Alvefur 2021-09-05 16:21:10 +02:00
parent e3c0a877bf
commit 1546e59310
2 changed files with 63 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
util/pubsub.lua
View file

@ -642,7 +642,7 @@ function service:purge(node, actor, notify) --> ok, err
return true
end
function service:get_items(node, actor, ids) --> (true, { id, [id] = node }) or (false, err)
function service:get_items(node, actor, ids, resultspec) --> (true, { id, [id] = node }) or (false, err)
-- Access checking
if not self:may(node, actor, "get_items") then
return false, "forbidden";
@ -660,18 +660,23 @@ function service:get_items(node, actor, ids) --> (true, { id, [id] = node }) or
ids = { ids };
end
local data = {};
local limit = resultspec and resultspec.max;
if ids then
for _, key in ipairs(ids) do
local value = self.data[node]:get(key);
if value then
data[#data+1] = key;
data[key] = value;
-- Limits and ids seem like a problematic combination.
if limit and #data >= limit then break end
end
end
else
for key, value in self.data[node]:items() do
data[#data+1] = key;
data[key] = value;
if limit and #data >= limit then break
end
end
end
return true, data;