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
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
|
if not modules then modules = { } end modules ['libs-imp-mysql'] = {
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"
}
-- c:/data/develop/tex-context/tex/texmf-win64/bin/lib/luametatex/lua/copies/mysql/libmysql.dll
local libname = "mysql"
local libfile = "libmysql"
local mysqllib = resolvers.libraries.validoptional(libname)
if not mysqllib 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 mysql_open = mysqllib.open
local mysql_close = mysqllib.close
local mysql_execute = mysqllib.execute
local mysql_getmessage = mysqllib.getmessage
local helpers = sql.helpers
local methods = sql.methods
local validspecification = helpers.validspecification
local preparetemplate = helpers.preparetemplate
local querysplitter = helpers.querysplitter
local cache = { }
local timeout -- = 3600 -- to be tested
local function connect(specification)
local db = mysql_open(
specification.database or "",
specification.username or "",
specification.password or "",
specification.host or "",
specification.port
)
if db and timeout then
mysql_execute(db,formatters["SET SESSION connect_timeout=%s ;"](timeout))
end
return db
end
local function execute_once(specification,retry)
if okay() then
if trace_sql then
report("executing mysql")
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)
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.mysql
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 = mysql_execute(db,query[i],callback)
if not okay then
if id and retry and i == 1 then
report("error: %s, retrying to connect",mysql_getmessage(db))
mysql_close(db)
cache[id] = nil
return execute_once(specification,false)
else
report("error: %s",mysql_getmessage(db))
end
end
end
end
if db and not id then
mysql_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
]]
-- return function(result)
-- if not result then
-- return { }
-- end
-- local nofrows = #result
-- if nofrows == 0 then
-- return { }
-- end
-- local target = { } -- no %s needed here
-- for i=1,nofrows do
-- target[%s] = {
-- %s
-- }
-- end
-- return result
-- end
local celltemplate = "cells[%s]"
methods.mysql = {
execute = execute,
usesfiles = false,
wraptemplate = wraptemplate,
celltemplate = celltemplate,
}
package.loaded["util-sql-imp-ffi"] = methods.mysql
package.loaded["util-sql-imp-mysql"] = methods.mysql
package.loaded["util-sql-imp-library"] = methods.mysql
package.loaded[libname] = methods.mysql
|
