diff options
author | Kim Alvefur <zash@zash.se> | 2013-10-31 20:49:03 +0100 |
---|---|---|
committer | Kim Alvefur <zash@zash.se> | 2013-10-31 20:49:03 +0100 |
commit | d938c65ce63fcc22d915abcc5215efdc52b5cec4 (patch) | |
tree | fc4ccb38b95ca249345023b2a6b8dc0bf1054deb /util/sql.lua | |
parent | 9fde274e2174f63d3f50e38f2992b6019a106ec7 (diff) | |
parent | edc3624d3575688943cf8953d76f0c862189408f (diff) | |
download | prosody-d938c65ce63fcc22d915abcc5215efdc52b5cec4.tar.gz prosody-d938c65ce63fcc22d915abcc5215efdc52b5cec4.zip |
Merge 0.10 -> trunk
Diffstat (limited to 'util/sql.lua')
-rw-r--r-- | util/sql.lua | 41 |
1 files changed, 40 insertions, 1 deletions
diff --git a/util/sql.lua b/util/sql.lua index b8c16e27..735fbce8 100644 --- a/util/sql.lua +++ b/util/sql.lua @@ -251,14 +251,31 @@ function engine:_create_index(index) elseif self.params.driver == "MySQL" then sql = sql:gsub("`([,)])", "`(20)%1"); end + if index.unique then + sql = sql:gsub("^CREATE", "CREATE UNIQUE"); + end --print(sql); return self:execute(sql); end function engine:_create_table(table) local sql = "CREATE TABLE `"..table.name.."` ("; for i,col in ipairs(table.c) do - sql = sql.."`"..col.name.."` "..col.type; + local col_type = col.type; + if col_type == "MEDIUMTEXT" and self.params.driver ~= "MySQL" then + col_type = "TEXT"; -- MEDIUMTEXT is MySQL-specific + end + sql = sql.."`"..col.name.."` "..col_type; if col.nullable == false then sql = sql.." NOT NULL"; end + if col.primary_key == true then sql = sql.." PRIMARY KEY"; end + if col.auto_increment == true then + if self.params.driver == "PostgreSQL" then + sql = sql.." SERIAL"; + elseif self.params.driver == "MySQL" then + sql = sql.." AUTO_INCREMENT"; + elseif self.params.driver == "SQLite3" then + sql = sql.." AUTOINCREMENT"; + end + end if i ~= #table.c then sql = sql..", "; end end sql = sql.. ");" @@ -276,6 +293,28 @@ function engine:_create_table(table) end return success; end +function engine:set_encoding() -- to UTF-8 + local driver = self.params.driver; + if driver == "SQLite3" then + return self:transaction(function() + if self:select"PRAGMA encoding;"()[1] == "UTF-8" then + self.charset = "utf8"; + end + end); + end + local set_names_query = "SET NAMES '%s';" + local charset = "utf8"; + if driver == "MySQL" then + set_names_query = set_names_query:gsub(";$", " COLLATE 'utf8_bin';"); + local ok, charsets = self:transaction(function() + return self:select"SELECT `CHARACTER_SET_NAME` FROM `CHARACTER_SETS` WHERE `CHARACTER_SET_NAME` LIKE 'utf8%' ORDER BY MAXLEN DESC LIMIT 1;"; + end); + local row = ok and charsets(); + charset = row and row[1] or charset; + end + self.charset = charset; + return self:transaction(function() return engine:execute(set_names_query:format(charset)); end); +end local engine_mt = { __index = engine }; local function db2uri(params) |