diff options
| author | Kim Alvefur <zash@zash.se> | 2017-04-01 19:39:20 +0200 |
|---|---|---|
| committer | Kim Alvefur <zash@zash.se> | 2017-04-01 19:39:20 +0200 |
| commit | 119d7900b549af8006f6bb9f245d7082e7b75cfc (patch) | |
| tree | dbeac2bb0cefb030eab5dca2aa42d274c02f6d59 | |
| parent | df4e99bee70d82972ce485f5039973227df48d8f (diff) | |
| download | prosody-119d7900b549af8006f6bb9f245d7082e7b75cfc.tar.gz prosody-119d7900b549af8006f6bb9f245d7082e7b75cfc.zip | |
mod_storage_sql: Pass SQL engine as an argument to upgrade and table creator functions
| -rw-r--r-- | plugins/mod_storage_sql.lua | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/plugins/mod_storage_sql.lua b/plugins/mod_storage_sql.lua index 40a59911..b6b71138 100644 --- a/plugins/mod_storage_sql.lua +++ b/plugins/mod_storage_sql.lua @@ -423,7 +423,7 @@ end --- Initialization -local function create_table(name) +local function create_table(engine, name) -- luacheck: ignore 431/engine local Table, Column, Index = sql.Table, sql.Column, sql.Index; local ProsodyTable = Table { @@ -458,7 +458,7 @@ local function create_table(name) end); end -local function upgrade_table(params, apply_changes) +local function upgrade_table(engine, params, apply_changes) -- luacheck: ignore 431/engine local changes = false; if params.driver == "MySQL" then local success,err = engine:transaction(function() @@ -540,13 +540,13 @@ function module.load() engine = engines[sql.db2uri(params)]; if not engine then module:log("debug", "Creating new engine"); - engine = sql:create_engine(params, function (engine) + engine = sql:create_engine(params, function (engine) -- luacheck: ignore 431/engine if module:get_option("sql_manage_tables", true) then -- Automatically create table, ignore failure (table probably already exists) -- FIXME: we should check in information_schema, etc. - create_table(); + create_table(engine); -- Check whether the table needs upgrading - if upgrade_table(params, false) then + if upgrade_table(engine, params, false) then module:log("error", "Old database format detected. Please run: prosodyctl mod_%s upgrade", module.name); return false, "database upgrade needed"; end @@ -583,7 +583,7 @@ function module.command(arg) for _, params in pairs(uris) do print("Checking "..params.database.."..."); engine = sql:create_engine(params); - upgrade_table(params, true); + upgrade_table(engine, params, true); end print("All done!"); elseif command then |