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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
|
if not modules then modules = { } end modules ['libs-imp-postgress'] = {
version = 1.001,
comment = "companion to util-sql.lua",
author = "Hans Hagen, PRAGMA-ADE, Hasselt NL",
copyright = "PRAGMA ADE / ConTeXt Development Team",
license = "see context related readme files"
}
-- Not yet untested !
-- c:/data/develop/tex-context/tex/texmf-win64/bin/lib/luametatex/lua/copies/postgress/libpq.dll
-- we can now share the code between mysql, postgress, sqlite ... todo
local libname = "postgress"
local libfile = "libpq"
local postgresslib = resolvers.libraries.validoptional(libname)
if not postgresslib then return end
local function okay()
if resolvers.libraries.optionalloaded(libname,libfile) then
okay = function() return true end
else
okay = function() return false end
end
return okay()
end
local lpegmatch = lpeg.match
local setmetatable = setmetatable
local sql = utilities.sql or require("util-sql")
local report = logs.reporter(libname)
local trace_sql = false trackers.register("sql.trace", function(v) trace_sql = v end)
local trace_queries = false trackers.register("sql.queries",function(v) trace_queries = v end)
local postgress_open = postgresslib.open
local postgress_close = postgresslib.close
local postgress_execute = postgresslib.execute
local postgress_getmessage = postgresslib.getmessage
local helpers = sql.helpers
local methods = sql.methods
local validspecification = helpers.validspecification
local preparetemplate = helpers.preparetemplate
local querysplitter = helpers.querysplitter -- not needed
local cache = { }
local function connect(specification)
return postgress_open(
specification.database or "",
specification.username or "",
specification.password or "",
specification.host or "",
specification.port
)
end
local function execute_once(specification,retry)
if okay() then
if trace_sql then
report("executing postgress")
end
if not validspecification(specification) then
report("error in specification")
end
local query = preparetemplate(specification)
if not query then
report("error in preparation")
return
else
query = lpegmatch(querysplitter,query) -- not needed
end
local base = specification.database -- or specification.presets and specification.presets.database
if not base then
report("no database specified")
return
end
local result = { }
local keys = { }
local id = specification.id
local db = nil
if id then
local session = cache[id]
if session then
db = session.db
else
db = connect(specification)
if not db then
report("no session database specified")
else
cache[id] = {
specification = specification,
db = db,
}
end
end
else
db = connect(specification)
end
if not db then
report("no database opened")
else
local converter = specification.converter
local nofrows = 0
local callback = nil
if converter then
local convert = converter.postgress
callback = function(nofcolumns,values,fields)
nofrows = nofrows + 1
result[nofrows] = convert(values)
end
else
callback = function(nofcolumns,values,fields)
local column = { }
for i=1,nofcolumns do
local field
if fields then
field = fields[i]
keys[i] = field
else
field = keys[i]
end
if field then
column[field] = values[i]
end
end
nofrows = nofrows + 1
result[nofrows] = column
end
end
for i=1,#query do
local okay = postgress_execute(db,query[i],callback)
if not okay then
if id and retry and i == 1 then
report("error: %s, retrying to connect",postgress_getmessage(db))
postgress_close(db)
cache[id] = nil
return execute_once(specification,false)
else
report("error: %s",postgress_getmessage(db))
end
end
end
end
if db and not id then
postgress_close(db)
end
-- bonus
local one = result[1]
if one then
setmetatable(result,{ __index = one } )
end
--
return result, keys
else
report("error: ","no library loaded")
end
end
local function execute(specification)
return execute_once(specification,true)
end
-- Here we build the dataset stepwise so we don't use the data hack that
-- is used in the client variant.
local wraptemplate = [[
local converters = utilities.sql.converters
local deserialize = utilities.sql.deserialize
local tostring = tostring
local tonumber = tonumber
local booleanstring = string.booleanstring
%s
return function(cells)
-- %s (not needed)
-- %s (not needed)
return {
%s
}
end
]]
local celltemplate = "cells[%s]"
methods.postgress = {
execute = execute,
usesfiles = false,
wraptemplate = wraptemplate,
celltemplate = celltemplate,
}
package.loaded["util-sql-imp-postgress"] = methods.postgress
package.loaded[libname] = methods.postgress
|