1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
--
-- This file is part of lem-postgres.
-- Copyright 2011 Emil Renner Berthing
--
-- lem-postgres is free software: you can redistribute it and/or
-- modify it under the terms of the GNU General Public License as
-- published by the Free Software Foundation, either version 3 of
-- the License, or (at your option) any later version.
--
-- lem-postgres is distributed in the hope that it will be useful,
-- but WITHOUT ANY WARRANTY; without even the implied warranty of
-- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
-- GNU General Public License for more details.
--
-- You should have received a copy of the GNU General Public License
-- along with lem-postgres. If not, see <http://www.gnu.org/licenses/>.
--
local utils = require 'lem.utils'
local postgres = require 'lem.postgres'
local remove = table.remove
local thisthread, suspend, resume = utils.thisthread, utils.suspend, utils.resume
local QConnection = {}
QConnection.__index = QConnection
function QConnection:close()
local ok, err = self.conn:close()
for i = 1, self.n - 1 do
resume(self[i])
self[i] = nil
end
return ok, err
end
QConnection.__gc = QConnection.close
local function lock(self)
local n = self.n
if n == 0 then
self.n = 1
else
self[n] = thisthread()
self.n = n + 1
suspend()
end
return self.conn
end
QConnection.lock = lock
local function unlock(self, ...)
local n = self.n - 1
self.n = n
if n > 0 then
resume(remove(self, 1))
end
return ...
end
QConnection.unlock = unlock
local function wrap(method)
return function(self, ...)
return unlock(self, method(lock(self), ...))
end
end
local Connection = postgres.Connection
QConnection.exec = wrap(Connection.exec)
QConnection.prepare = wrap(Connection.prepare)
QConnection.run = wrap(Connection.run)
local qconnect
do
local setmetatable = setmetatable
local connect = postgres.connect
function qconnect(...)
local conn, err = connect(...)
if not conn then return nil, err end
return setmetatable({
n = 0,
conn = conn,
}, QConnection)
end
end
return {
QConnection = QConnection,
connect = qconnect,
}
-- vim: ts=2 sw=2 noet:
|
