prosody
56bdfae9e4ea
util/datamanager.lua

Find changesets by keywords by author, files, the commit message, revision number or hash, or revset expression.

Software / code / prosody

Comparison

util/datamanager.lua @ 550:56bdfae9e4ea

Changed util.datamanager to use util.serialization
author Waqas Hussain <waqas20@gmail.com>
date Fri, 05 Dec 2008 00:16:38 +0500
parent 519:cccd610a0ef9
child 574:5963b780775d
comparison
equal deleted inserted replaced
549:c0947c0af398 550:56bdfae9e4ea
28 local os_remove = os.remove; 28 local os_remove = os.remove;
29 local tostring, tonumber = tostring, tonumber; 29 local tostring, tonumber = tostring, tonumber;
30 local error = error; 30 local error = error;
31 local next = next; 31 local next = next;
32 local t_insert = table.insert; 32 local t_insert = table.insert;
33 33 local append = require "util.serialization".append;
34 local indent = function(f, i)
35 for n = 1, i do
36 f:write("\t");
37 end
38 end
39
40 local data_path = "data";
41 34
42 module "datamanager" 35 module "datamanager"
43 36
44
45 ---- utils ----- 37 ---- utils -----
46 local encode, decode; 38 local encode, decode;
47
48 do 39 do
49 local urlcodes = setmetatable({}, { __index = function (t, k) t[k] = char(tonumber("0x"..k)); return t[k]; end }); 40 local urlcodes = setmetatable({}, { __index = function (t, k) t[k] = char(tonumber("0x"..k)); return t[k]; end });
50 41
51 decode = function (s) 42 decode = function (s)
52 return s and (s:gsub("+", " "):gsub("%%([a-fA-F0-9][a-fA-F0-9])", urlcodes)); 43 return s and (s:gsub("+", " "):gsub("%%([a-fA-F0-9][a-fA-F0-9])", urlcodes));
55 encode = function (s) 46 encode = function (s)
56 return s and (s:gsub("%W", function (c) return format("%%%x", c:byte()); end)); 47 return s and (s:gsub("%W", function (c) return format("%%%x", c:byte()); end));
57 end 48 end
58 end 49 end
59 50
60 local function basicSerialize (o)
61 if type(o) == "number" or type(o) == "boolean" then
62 return tostring(o);
63 else -- assume it is a string -- FIXME make sure it's a string. throw an error otherwise.
64 return (format("%q", tostring(o)):gsub("\\\n", "\\n"));
65 end
66 end
67
68
69 local function simplesave (f, o, ind)
70 if type(o) == "number" then
71 f:write(o)
72 elseif type(o) == "string" then
73 f:write((format("%q", o):gsub("\\\n", "\\n")))
74 elseif type(o) == "table" then
75 f:write("{\n")
76 for k,v in pairs(o) do
77 indent(f, ind);
78 f:write("[", basicSerialize(k), "] = ")
79 simplesave(f, v, ind+1)
80 f:write(",\n")
81 end
82 indent(f, ind-1);
83 f:write("}")
84 elseif type(o) == "boolean" then
85 f:write(o and "true" or "false");
86 else
87 error("cannot serialize a " .. type(o))
88 end
89 end
90
91 ------- API ------------- 51 ------- API -------------
92 52
53 local data_path = "data";
93 function set_data_path(path) 54 function set_data_path(path)
94 data_path = path; 55 data_path = path;
95 end 56 end
96 57
97 function getpath(username, host, datastore, ext) 58 function getpath(username, host, datastore, ext)
129 if not f then 90 if not f then
130 log("error", "Unable to write to "..datastore.." storage ('"..msg.."') for user: "..(username or "nil").."@"..(host or "nil")); 91 log("error", "Unable to write to "..datastore.." storage ('"..msg.."') for user: "..(username or "nil").."@"..(host or "nil"));
131 return; 92 return;
132 end 93 end
133 f:write("return "); 94 f:write("return ");
134 simplesave(f, data, 1); 95 append(f, data);
135 f:close(); 96 f:close();
136 if not next(data) then -- try to delete empty datastore 97 if not next(data) then -- try to delete empty datastore
137 os_remove(getpath(username, host, datastore)); 98 os_remove(getpath(username, host, datastore));
138 end 99 end
139 -- we write data even when we are deleting because lua doesn't have a 100 -- we write data even when we are deleting because lua doesn't have a
148 if not f then 109 if not f then
149 log("error", "Unable to write to "..datastore.." storage ('"..msg.."') for user: "..(username or "nil").."@"..(host or "nil")); 110 log("error", "Unable to write to "..datastore.." storage ('"..msg.."') for user: "..(username or "nil").."@"..(host or "nil"));
150 return; 111 return;
151 end 112 end
152 f:write("item("); 113 f:write("item(");
153 simplesave(f, data, 1); 114 append(f, data);
154 f:write(");\n"); 115 f:write(");\n");
155 f:close(); 116 f:close();
156 return true; 117 return true;
157 end 118 end
158 119
166 log("error", "Unable to write to "..datastore.." storage ('"..msg.."') for user: "..(username or "nil").."@"..(host or "nil")); 127 log("error", "Unable to write to "..datastore.." storage ('"..msg.."') for user: "..(username or "nil").."@"..(host or "nil"));
167 return; 128 return;
168 end 129 end
169 for _, d in ipairs(data) do 130 for _, d in ipairs(data) do
170 f:write("item("); 131 f:write("item(");
171 simplesave(f, d, 1); 132 append(f, d);
172 f:write(");\n"); 133 f:write(");\n");
173 end 134 end
174 f:close(); 135 f:close();
175 if not next(data) then -- try to delete empty datastore 136 if not next(data) then -- try to delete empty datastore
176 os_remove(getpath(username, host, datastore, "list")); 137 os_remove(getpath(username, host, datastore, "list"));