Software /
code /
prosody
Diff
plugins/mod_storage_sql.lua @ 7149:bb0fd02ae70f
mod_storage_sql: Support non-string keys in map store
author | Kim Alvefur <zash@zash.se> |
---|---|
date | Mon, 08 Feb 2016 20:23:12 +0100 |
parent | 7008:9beba2572e2b |
child | 7153:89fa66d4e502 |
line wrap: on
line diff
--- a/plugins/mod_storage_sql.lua Fri Feb 05 16:12:01 2016 +0100 +++ b/plugins/mod_storage_sql.lua Mon Feb 08 20:23:12 2016 +0100 @@ -134,7 +134,10 @@ return deserialize(row[1], row[2]); end else - error("TODO: non-string keys"); + for row in engine:select("SELECT `type`, `value` FROM `prosody` WHERE `host`=? AND `user`=? AND `store`=? AND `key`=?", host, username or "", self.store, "") do + local data = deserialize(row[1], row[2]); + return data and data[key] or nil; + end end end); if not ok then return nil, result; end @@ -150,7 +153,16 @@ engine:insert("INSERT INTO `prosody` (`host`,`user`,`store`,`key`,`type`,`value`) VALUES (?,?,?,?,?,?)", host, username or "", self.store, key, t, value); end else - error("TODO: non-string keys"); + local extradata = {}; + for row in engine:select("SELECT `type`, `value` FROM `prosody` WHERE `host`=? AND `user`=? AND `store`=? AND `key`=?", host, username or "", self.store, "") do + extradata = deserialize(row[1], row[2]); + break; + end + engine:delete("DELETE FROM `prosody` WHERE `host`=? AND `user`=? AND `store`=? AND `key`=?", + host, username or "", self.store, ""); + extradata[key] = data; + local t, value = assert(serialize(extradata)); + engine:insert("INSERT INTO `prosody` (`host`,`user`,`store`,`key`,`type`,`value`) VALUES (?,?,?,?,?,?)", host, username or "", self.store, "", t, value); end return true; end);