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_storage_sql: Implement map-like API for archives

Browse source 
Used by mod_http_file_share, muc moderation, etc.

Tests tweaked because they failed on stanza internals that happen
becasue of re-serialization. Namespaces differ since inheritance is
implicit when building but explicit after parsing.
This commit is contained in:
Kim Alvefur 2021-02-01 12:47:05 +01:00
parent f307825a01
commit 0f0d32d335
2 changed files with 49 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

47
plugins/mod_storage_sql.lua
View file

@ -482,6 +482,53 @@ function archive_store:find(username, query)
end, total;
end
function archive_store:get(username, key)
local iter, err = self:find(username, { key = key })
if not iter then return iter, err; end
for _, stanza, when, with in iter do
return stanza, when, with;
end
return nil, "item-not-found";
end
function archive_store:set(username, key, new_value, new_when, new_with)
local user,store = username,self.store;
local ok, result = engine:transaction(function ()
local update_query = [[
UPDATE "prosodyarchive"
SET %s
WHERE %s
]];
local args = { host, user or "", store, key };
local setf = {};
local where = { "\"host\" = ?", "\"user\" = ?", "\"store\" = ?", "\"key\" = ?"};
if new_value then
table.insert(setf, '"type" = ?')
table.insert(setf, '"value" = ?')
local t, value = serialize(new_value);
table.insert(args, 1, t);
table.insert(args, 2, value);
end
if new_when then
table.insert(setf, 1, '"when" = ?')
table.insert(args, 1, new_when);
end
if new_with then
table.insert(setf, 1, '"with" = ?')
table.insert(args, 1, new_with);
end
update_query = update_query:format(t_concat(setf, ", "), t_concat(where, " AND "));
return engine:update(update_query, unpack(args));
end);
if not ok then return ok, result; end
return result:affected() == 1;
end
function archive_store:summary(username, query)
query = query or {};
local user,store = username,self.store;

4
spec/core_storagemanager_spec.lua
View file

@ -556,7 +556,7 @@ describe("storagemanager", function ()
local id = archive:append("mapuser", nil, test_stanza, test_time, "contact@example.com");
do
local stanza_roundtrip, when, with = archive:get("mapuser", id);
assert.same(test_stanza, stanza_roundtrip, "same stanza is returned");
assert.same(tostring(test_stanza), tostring(stanza_roundtrip), "same stanza is returned");
assert.equal(test_time, when, "same 'when' is returned");
assert.equal("contact@example.com", with, "same 'with' is returned");
end
@ -568,7 +568,7 @@ describe("storagemanager", function ()
do
local replaced, when, with = archive:get("mapuser", id);
assert.same(replacement_stanza, replaced, "replaced stanza is returned");
assert.same(tostring(replacement_stanza), tostring(replaced), "replaced stanza is returned");
assert.equal(test_time+1, when, "modified 'when' is returned");
assert.equal("contact@example.com", with, "original 'with' is returned");
end