mirrors/prosody
1
0
Fork 0
mirror of https://github.com/bjc/prosody.git synced 2025-04-04 21:57:45 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

util.sql: Greedily read all rows so we can close queries early (fixes #391)

Browse source
This commit is contained in:
Matthew Wild 2017-08-04 18:52:15 +01:00
parent 15ee257b35
commit c5af3aee3e
1 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
util/sql.lua
View file

@ -175,7 +175,11 @@ function engine:execute_query(sql, ...)
sql = self:prepquery(sql); sql = self:prepquery(sql);
local stmt = assert(self.conn:prepare(sql)); local stmt = assert(self.conn:prepare(sql));
assert(stmt:execute(...)); assert(stmt:execute(...));
return stmt:rows(); local result = {};
for row in stmt:rows() do result[#result + 1] = row; end
stmt:close();
local i = 0;
return function() i=i+1; return result[i]; end;
end end
function engine:execute_update(sql, ...) function engine:execute_update(sql, ...)
sql = self:prepquery(sql); sql = self:prepquery(sql);