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

migrator: Refactor out individual item migrator for code deduplication

Browse source
This commit is contained in:
Kim Alvefur 2022-01-09 15:50:21 +01:00
parent 25a9fd3fdd
commit e9b2be9e44
1 changed files with 18 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

34
tools/migration/prosody-migrator.lua
View file

@ -168,6 +168,21 @@ local function get_driver(host, conf)
return assert(sm.load_driver(host, conf.type));
end
local migrate_once = {
keyval = function(origin, destination, user)
local data, err = origin:get(user);
assert(not err, err);
assert(destination:set(user, data));
end;
archive = function(origin, destination, user)
local iter, err = origin:find(user);
assert(iter, err);
for id, item, when, with in iter do
assert(destination:append(user, id, item, when, with));
end
end;
}
local migration_runner = async.runner(function (job)
for host, stores in pairs(job.input.hosts) do
prosody.hosts[host] = startup.make_host(host);
@ -186,26 +201,13 @@ local migration_runner = async.runner(function (job)
local origin = assert(input_driver:open(store, typ));
local destination = assert(output_driver:open(store, typ));
local migrate = assert(migrate_once[typ], "Unknown store type: "..typ);
if typ == "keyval" then -- host data
local data, err = origin:get(nil);
assert(not err, err);
assert(destination:set(nil, data));
migrate(origin, destination, nil);
end
for user in users(origin, host) do
if typ == "keyval" then
local data, err = origin:get(user);
assert(not err, err);
assert(destination:set(user, data));
elseif typ == "archive" then
local iter, err = origin:find(user);
assert(iter, err);
for id, item, when, with in iter do
assert(destination:append(user, id, item, when, with));
end
else
error("Don't know how to migrate data of type '"..typ.."'.");
end
migrate(origin, destination, user);
end
end
end