From b21bc2b374353ca562d95dfd574fed59206312b3 Mon Sep 17 00:00:00 2001 From: Kim Alvefur Date: Sun, 9 Jan 2022 16:08:15 +0100 Subject: migrator: Add option to keep going despite errors --- tools/migration/prosody-migrator.lua | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/tools/migration/prosody-migrator.lua b/tools/migration/prosody-migrator.lua index e50fcc76..57d87e20 100644 --- a/tools/migration/prosody-migrator.lua +++ b/tools/migration/prosody-migrator.lua @@ -37,6 +37,7 @@ local default_config = (CFG_CONFIGDIR or ".").."/migrator.cfg.lua"; local function usage() print("Usage: " .. arg[0] .. " [OPTIONS] FROM_STORE TO_STORE"); print(" --config FILE Specify config file") + print(" --keep-going Keep going in case of errors"); print(" -v, --verbose Incease log-level"); print(""); print("If no stores are specified, 'input' and 'output' are used."); @@ -183,6 +184,19 @@ local migrate_once = { end; } +if options["keep-going"] then + local xpcall = require "util.xpcall".xpcall; + local function log_err(err) + log("error", "Error migrating data: %s", err); + log("debug", "%s", debug.traceback()); + end + for t, f in pairs(migrate_once) do + migrate_once[t] = function (origin, destination, user) + xpcall(f, log_err, origin, destination, user); + end + 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); -- cgit v1.2.3