Mercurial > prosody-hg

comparison util/stanza.lua @ 1416:f916f0ff90e5

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
util.stanza: Rewrote stanza_mt.__tostring. 20-30% faster stanza serialization. - #optimization
author Waqas Hussain <waqas20@gmail.com>
date Thu, 25 Jun 2009 17:22:53 +0500
parents 957a81b72cb2
children 1576a5aa52f8
comparison
equal deleted inserted replaced
1415:957a81b72cb2 1416:f916f0ff90e5
21 local type = type; 21 local type = type;
22 local next = next; 22 local next = next;
23 local print = print; 23 local print = print;
24 local unpack = unpack; 24 local unpack = unpack;
25 local s_gsub = string.gsub; 25 local s_gsub = string.gsub;
26 local s_char = string.char;
26 local os = os; 27 local os = os;
27 28
28 local do_pretty_printing = not os.getenv("WINDIR"); 29 local do_pretty_printing = not os.getenv("WINDIR");
29 local getstyle, getstring = require "util.termcolours".getstyle, require "util.termcolours".getstring; 30 local getstyle, getstring = require "util.termcolours".getstyle, require "util.termcolours".getstring;
30 31
114 if v then return v; end 115 if v then return v; end
115 end, self.tags[1], i; 116 end, self.tags[1], i;
116 117
117 end 118 end
118 119
119 do 120 local xml_escape = (function()
120 local xml_entities = { ["'"] = "&apos;", ["\""] = "&quot;", ["<"] = "&lt;", [">"] = "&gt;", ["&"] = "&amp;" }; 121 local escape_table = { ["'"] = "&apos;", ["\""] = "&quot;", ["<"] = "&lt;", [">"] = "&gt;", ["&"] = "&amp;" };
121 function xml_escape(s) return s_gsub(s, "['&<>\"]", xml_entities); end 122 return function(str) return (s_gsub(str, "['&<>\"]", escape_table)); end
122 end 123 end)();
123 124 local function _dostring(t, buf, self, xml_escape)
124 local xml_escape = xml_escape;
125
126 local function dostring(t, buf, self, xml_escape)
127 local nsid, ns, attrk = 0; 125 local nsid, ns, attrk = 0;
128 t_insert(buf, "<"); 126 t_insert(buf, "<"..t.name);
129 t_insert(buf, t.name); 127 for k, v in pairs(t.attr) do
130 for k, v in pairs(t.attr) do if type(k) == "string" then
131 t_insert(buf, " ");
132 ns, attrk = s_match(k, "^([^|]+)|(.+)$"); 128 ns, attrk = s_match(k, "^([^|]+)|(.+)$");
133 if ns then 129 if ns then
134 nsid = (nsid or -1) + 1; 130 nsid = nsid + 1;
135 t_insert(buf, "xmlns:ns"..nsid); 131 t_insert(buf, " xmlns:ns"..nsid.."='"..xml_escape(ns).."' ".."ns"..nsid..":"..attrk.."='"..xml_escape(v).."'");
136 t_insert(buf, "='");
137 t_insert(buf, (xml_escape(tostring(ns))));
138 t_insert(buf, "' ");
139 t_insert(buf, "ns"..nsid..":"..attrk);
140 else 132 else
141 t_insert(buf, k); 133 t_insert(buf, " "..k.."='"..xml_escape(v).."'");
142 end 134 end
143 t_insert(buf, "='"); 135 end
144 t_insert(buf, (xml_escape(tostring(v))));
145 t_insert(buf, "'");
146 end end
147 t_insert(buf, ">"); 136 t_insert(buf, ">");
148 for n, child in ipairs(t) do 137 for n=1,#t do
149 if child.name then 138 local child = t[n];
139 if child.name then
150 self(child, buf, self, xml_escape); 140 self(child, buf, self, xml_escape);
151 else 141 else
152 t_insert(buf, (xml_escape(child))); 142 t_insert(buf, xml_escape(child));
153 end 143 end
154 end 144 end
155 t_insert(buf, "</"); 145 t_insert(buf, "</"..t.name..">");
156 t_insert(buf, t.name); 146 end
157 t_insert(buf, ">");
158 end
159
160 function stanza_mt.__tostring(t) 147 function stanza_mt.__tostring(t)
161 local buf = {}; 148 local buf = {};
162 dostring(t, buf, dostring, xml_escape); 149 _dostring(t, buf, _dostring, xml_escape);
163 return t_concat(buf); 150 return t_concat(buf);
164 end 151 end
165
166 152
167 function stanza_mt.top_tag(t) 153 function stanza_mt.top_tag(t)
168 local attr_string = ""; 154 local attr_string = "";
169 if t.attr then 155 if t.attr then
170 for k, v in pairs(t.attr) do if type(k) == "string" then attr_string = attr_string .. s_format(" %s='%s'", k, xml_escape(tostring(v))); end end 156 for k, v in pairs(t.attr) do if type(k) == "string" then attr_string = attr_string .. s_format(" %s='%s'", k, xml_escape(tostring(v))); end end