diff options
| author | Waqas Hussain <waqas20@gmail.com> | 2010-12-12 05:39:24 +0500 |
|---|---|---|
| committer | Waqas Hussain <waqas20@gmail.com> | 2010-12-12 05:39:24 +0500 |
| commit | d4a6582d20455a53120c70e98c16cde4f36ead3c (patch) | |
| tree | ff3df4b4c5d1567b912647569dbb360667c469db /plugins/mod_storage_sql.lua | |
| parent | 50fc30731e5a243c06ea4623c134de10baf1e901 (diff) | |
| download | prosody-d4a6582d20455a53120c70e98c16cde4f36ead3c.tar.gz prosody-d4a6582d20455a53120c70e98c16cde4f36ead3c.zip | |
mod_storage_sql: Auto-initialize SQLite3 database.
Diffstat (limited to 'plugins/mod_storage_sql.lua')
| -rw-r--r-- | plugins/mod_storage_sql.lua | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/plugins/mod_storage_sql.lua b/plugins/mod_storage_sql.lua index 68b38134..593acd31 100644 --- a/plugins/mod_storage_sql.lua +++ b/plugins/mod_storage_sql.lua @@ -47,6 +47,19 @@ do -- process options to get a db connection dbh:autocommit(false); -- don't commit automatically connection = dbh; + + if params.driver == "SQLite3" then -- auto initialize + local stmt = assert(connection:prepare("SELECT COUNT(*) FROM sqlite_master WHERE type='table' AND name='Prosody';")); + local ok = assert(stmt:execute()); + local count = stmt:fetch()[1]; + if count == 0 then + local stmt = assert(connection:prepare("CREATE TABLE Prosody (host TEXT, user TEXT, store TEXT, key TEXT, subkey TEXT, type TEXT, value TEXT);")); + assert(stmt:execute()); + assert(connection:commit()); + module:log("debug", "Initialized new SQLite3 database"); + end + --print("===", json.stringify()) + end end local function serialize(value) |