From 1546e59310f4eb2888227a0f09a4554cc7da145d Mon Sep 17 00:00:00 2001 From: Kim Alvefur Date: Sun, 5 Sep 2021 16:21:10 +0200 Subject: util.pubsub: Add support for limiting number of items to retrieve Hopefully this will eventually be upgraded to RSM, which is why the argument is called 'resultspec' and is a table. --- util/pubsub.lua | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) (limited to 'util') diff --git a/util/pubsub.lua b/util/pubsub.lua index ae1d6353..e23818b8 100644 --- a/util/pubsub.lua +++ b/util/pubsub.lua @@ -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; -- cgit v1.2.3