adamc@879
|
1 (* Copyright (c) 2009, Adam Chlipala
|
adamc@866
|
2 * All rights reserved.
|
adamc@866
|
3 *
|
adamc@866
|
4 * Redistribution and use in source and binary forms, with or without
|
adamc@866
|
5 * modification, are permitted provided that the following conditions are met:
|
adamc@866
|
6 *
|
adamc@866
|
7 * - Redistributions of source code must retain the above copyright notice,
|
adamc@866
|
8 * this list of conditions and the following disclaimer.
|
adamc@866
|
9 * - Redistributions in binary form must reproduce the above copyright notice,
|
adamc@866
|
10 * this list of conditions and the following disclaimer in the documentation
|
adamc@866
|
11 * and/or other materials provided with the distribution.
|
adamc@866
|
12 * - The names of contributors may not be used to endorse or promote products
|
adamc@866
|
13 * derived from this software without specific prior written permission.
|
adamc@866
|
14 *
|
adamc@866
|
15 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
adamc@866
|
16 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
adamc@866
|
17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
adamc@866
|
18 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
|
adamc@866
|
19 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
adamc@866
|
20 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
adamc@866
|
21 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
adamc@866
|
22 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
adamc@866
|
23 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
adamc@866
|
24 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
adamc@866
|
25 * POSSIBILITY OF SUCH DAMAGE.
|
adamc@866
|
26 *)
|
adamc@866
|
27
|
adamc@866
|
28 structure MySQL :> MYSQL = struct
|
adamc@866
|
29
|
adamc@866
|
30 open Settings
|
adamc@866
|
31 open Print.PD
|
adamc@866
|
32 open Print
|
adamc@866
|
33
|
adamc@873
|
34 fun p_sql_type t =
|
adamc@873
|
35 case t of
|
adamc@873
|
36 Int => "bigint"
|
adamc@873
|
37 | Float => "double"
|
adamc@873
|
38 | String => "longtext"
|
adamc@873
|
39 | Bool => "bool"
|
adamc@873
|
40 | Time => "timestamp"
|
adamc@873
|
41 | Blob => "longblob"
|
adamc@873
|
42 | Channel => "bigint"
|
adamc@873
|
43 | Client => "int"
|
adamc@873
|
44 | Nullable t => p_sql_type t
|
adamc@873
|
45
|
adamc@873
|
46 fun p_buffer_type t =
|
adamc@873
|
47 case t of
|
adamc@873
|
48 Int => "MYSQL_TYPE_LONGLONG"
|
adamc@873
|
49 | Float => "MYSQL_TYPE_DOUBLE"
|
adamc@873
|
50 | String => "MYSQL_TYPE_STRING"
|
adamc@873
|
51 | Bool => "MYSQL_TYPE_LONG"
|
adamc@873
|
52 | Time => "MYSQL_TYPE_TIME"
|
adamc@873
|
53 | Blob => "MYSQL_TYPE_BLOB"
|
adamc@873
|
54 | Channel => "MYSQL_TYPE_LONGLONG"
|
adamc@873
|
55 | Client => "MYSQL_TYPE_LONG"
|
adamc@873
|
56 | Nullable t => p_buffer_type t
|
adamc@873
|
57
|
adamc@874
|
58 fun p_sql_type_base t =
|
adamc@874
|
59 case t of
|
adamc@874
|
60 Int => "bigint"
|
adamc@874
|
61 | Float => "double"
|
adamc@874
|
62 | String => "longtext"
|
adamc@874
|
63 | Bool => "tinyint"
|
adamc@874
|
64 | Time => "timestamp"
|
adamc@874
|
65 | Blob => "longblob"
|
adamc@874
|
66 | Channel => "bigint"
|
adamc@874
|
67 | Client => "int"
|
adamc@874
|
68 | Nullable t => p_sql_type_base t
|
adamc@874
|
69
|
adamc@874
|
70 val ident = String.translate (fn #"'" => "PRIME"
|
adamc@874
|
71 | ch => str ch)
|
adamc@874
|
72
|
adamc@874
|
73 fun checkRel (table, checkNullable) (s, xts) =
|
adamc@874
|
74 let
|
adamc@874
|
75 val sl = CharVector.map Char.toLower s
|
adamc@874
|
76
|
adamc@874
|
77 val q = "SELECT COUNT(*) FROM information_schema." ^ table ^ " WHERE table_name = '"
|
adamc@874
|
78 ^ sl ^ "'"
|
adamc@874
|
79
|
adamc@874
|
80 val q' = String.concat ["SELECT COUNT(*) FROM information_schema.columns WHERE table_name = '",
|
adamc@874
|
81 sl,
|
adamc@874
|
82 "' AND (",
|
adamc@874
|
83 String.concatWith " OR "
|
adamc@874
|
84 (map (fn (x, t) =>
|
adamc@874
|
85 String.concat ["(column_name = 'uw_",
|
adamc@874
|
86 CharVector.map
|
adamc@874
|
87 Char.toLower (ident x),
|
adamc@874
|
88 "' AND data_type = '",
|
adamc@874
|
89 p_sql_type_base t,
|
adamc@874
|
90 "'",
|
adamc@874
|
91 if checkNullable then
|
adamc@874
|
92 (" AND is_nullable = '"
|
adamc@874
|
93 ^ (if isNotNull t then
|
adamc@874
|
94 "NO"
|
adamc@874
|
95 else
|
adamc@874
|
96 "YES")
|
adamc@874
|
97 ^ "'")
|
adamc@874
|
98 else
|
adamc@874
|
99 "",
|
adamc@874
|
100 ")"]) xts),
|
adamc@874
|
101 ")"]
|
adamc@874
|
102
|
adamc@874
|
103 val q'' = String.concat ["SELECT COUNT(*) FROM information_schema.columns WHERE table_name = '",
|
adamc@874
|
104 sl,
|
adamc@874
|
105 "' AND column_name LIKE 'uw_%'"]
|
adamc@874
|
106 in
|
adamc@874
|
107 box [string "if (mysql_query(conn->conn, \"",
|
adamc@874
|
108 string q,
|
adamc@874
|
109 string "\")) {",
|
adamc@874
|
110 newline,
|
adamc@874
|
111 box [string "mysql_close(conn->conn);",
|
adamc@874
|
112 newline,
|
adamc@874
|
113 string "uw_error(ctx, FATAL, \"Query failed:\\n",
|
adamc@874
|
114 string q,
|
adamc@874
|
115 string "\");",
|
adamc@874
|
116 newline],
|
adamc@874
|
117 string "}",
|
adamc@874
|
118 newline,
|
adamc@874
|
119 newline,
|
adamc@874
|
120
|
adamc@874
|
121 string "if ((res = mysql_store_result(conn->conn)) == NULL) {",
|
adamc@874
|
122 newline,
|
adamc@874
|
123 box [string "mysql_free_result(res);",
|
adamc@874
|
124 newline,
|
adamc@874
|
125 string "mysql_close(conn->conn);",
|
adamc@874
|
126 newline,
|
adamc@874
|
127 string "uw_error(ctx, FATAL, \"Result store failed:\\n",
|
adamc@874
|
128 string q,
|
adamc@874
|
129 string "\");",
|
adamc@874
|
130 newline],
|
adamc@874
|
131 string "}",
|
adamc@874
|
132 newline,
|
adamc@874
|
133 newline,
|
adamc@874
|
134
|
adamc@874
|
135 string "if (mysql_num_fields(res) != 1) {",
|
adamc@874
|
136 newline,
|
adamc@874
|
137 box [string "mysql_free_result(res);",
|
adamc@874
|
138 newline,
|
adamc@874
|
139 string "mysql_close(conn->conn);",
|
adamc@874
|
140 newline,
|
adamc@874
|
141 string "uw_error(ctx, FATAL, \"Bad column count:\\n",
|
adamc@874
|
142 string q,
|
adamc@874
|
143 string "\");",
|
adamc@874
|
144 newline],
|
adamc@874
|
145 string "}",
|
adamc@874
|
146 newline,
|
adamc@874
|
147 newline,
|
adamc@874
|
148
|
adamc@874
|
149 string "if ((row = mysql_fetch_row(res)) == NULL) {",
|
adamc@874
|
150 newline,
|
adamc@874
|
151 box [string "mysql_free_result(res);",
|
adamc@874
|
152 newline,
|
adamc@874
|
153 string "mysql_close(conn->conn);",
|
adamc@874
|
154 newline,
|
adamc@874
|
155 string "uw_error(ctx, FATAL, \"Row fetch failed:\\n",
|
adamc@874
|
156 string q,
|
adamc@874
|
157 string "\");",
|
adamc@874
|
158 newline],
|
adamc@874
|
159 string "}",
|
adamc@874
|
160 newline,
|
adamc@874
|
161 newline,
|
adamc@874
|
162
|
adamc@874
|
163 string "if (strcmp(row[0], \"1\")) {",
|
adamc@874
|
164 newline,
|
adamc@874
|
165 box [string "mysql_free_result(res);",
|
adamc@874
|
166 newline,
|
adamc@874
|
167 string "mysql_close(conn->conn);",
|
adamc@874
|
168 newline,
|
adamc@874
|
169 string "uw_error(ctx, FATAL, \"Table '",
|
adamc@874
|
170 string s,
|
adamc@874
|
171 string "' does not exist.\");",
|
adamc@874
|
172 newline],
|
adamc@874
|
173 string "}",
|
adamc@874
|
174 newline,
|
adamc@874
|
175 newline,
|
adamc@874
|
176 string "mysql_free_result(res);",
|
adamc@874
|
177 newline,
|
adamc@874
|
178 newline,
|
adamc@874
|
179
|
adamc@874
|
180 string "if (mysql_query(conn->conn, \"",
|
adamc@874
|
181 string q',
|
adamc@874
|
182 string "\")) {",
|
adamc@874
|
183 newline,
|
adamc@874
|
184 box [string "mysql_close(conn->conn);",
|
adamc@874
|
185 newline,
|
adamc@874
|
186 string "uw_error(ctx, FATAL, \"Query failed:\\n",
|
adamc@874
|
187 string q',
|
adamc@874
|
188 string "\");",
|
adamc@874
|
189 newline],
|
adamc@874
|
190 string "}",
|
adamc@874
|
191 newline,
|
adamc@874
|
192 newline,
|
adamc@874
|
193
|
adamc@874
|
194 string "if ((res = mysql_store_result(conn->conn)) == NULL) {",
|
adamc@874
|
195 newline,
|
adamc@874
|
196 box [string "mysql_free_result(res);",
|
adamc@874
|
197 newline,
|
adamc@874
|
198 string "mysql_close(conn->conn);",
|
adamc@874
|
199 newline,
|
adamc@874
|
200 string "uw_error(ctx, FATAL, \"Result store failed:\\n",
|
adamc@874
|
201 string q',
|
adamc@874
|
202 string "\");",
|
adamc@874
|
203 newline],
|
adamc@874
|
204 string "}",
|
adamc@874
|
205 newline,
|
adamc@874
|
206 newline,
|
adamc@874
|
207
|
adamc@874
|
208 string "if (mysql_num_fields(res) != 1) {",
|
adamc@874
|
209 newline,
|
adamc@874
|
210 box [string "mysql_free_result(res);",
|
adamc@874
|
211 newline,
|
adamc@874
|
212 string "mysql_close(conn->conn);",
|
adamc@874
|
213 newline,
|
adamc@874
|
214 string "uw_error(ctx, FATAL, \"Bad column count:\\n",
|
adamc@874
|
215 string q',
|
adamc@874
|
216 string "\");",
|
adamc@874
|
217 newline],
|
adamc@874
|
218 string "}",
|
adamc@874
|
219 newline,
|
adamc@874
|
220 newline,
|
adamc@874
|
221
|
adamc@874
|
222 string "if ((row = mysql_fetch_row(res)) == NULL) {",
|
adamc@874
|
223 newline,
|
adamc@874
|
224 box [string "mysql_free_result(res);",
|
adamc@874
|
225 newline,
|
adamc@874
|
226 string "mysql_close(conn->conn);",
|
adamc@874
|
227 newline,
|
adamc@874
|
228 string "uw_error(ctx, FATAL, \"Row fetch failed:\\n",
|
adamc@874
|
229 string q',
|
adamc@874
|
230 string "\");",
|
adamc@874
|
231 newline],
|
adamc@874
|
232 string "}",
|
adamc@874
|
233 newline,
|
adamc@874
|
234 newline,
|
adamc@874
|
235
|
adamc@874
|
236 string "if (strcmp(row[0], \"",
|
adamc@874
|
237 string (Int.toString (length xts)),
|
adamc@874
|
238 string "\")) {",
|
adamc@874
|
239 newline,
|
adamc@874
|
240 box [string "mysql_free_result(res);",
|
adamc@874
|
241 newline,
|
adamc@874
|
242 string "mysql_close(conn->conn);",
|
adamc@874
|
243 newline,
|
adamc@874
|
244 string "uw_error(ctx, FATAL, \"Table '",
|
adamc@874
|
245 string s,
|
adamc@874
|
246 string "' has the wrong column types.\");",
|
adamc@874
|
247 newline],
|
adamc@874
|
248 string "}",
|
adamc@874
|
249 newline,
|
adamc@874
|
250 newline,
|
adamc@874
|
251 string "mysql_free_result(res);",
|
adamc@874
|
252 newline,
|
adamc@874
|
253 newline,
|
adamc@874
|
254
|
adamc@874
|
255 string "if (mysql_query(conn->conn, \"",
|
adamc@874
|
256 string q'',
|
adamc@874
|
257 string "\")) {",
|
adamc@874
|
258 newline,
|
adamc@874
|
259 box [string "mysql_close(conn->conn);",
|
adamc@874
|
260 newline,
|
adamc@874
|
261 string "uw_error(ctx, FATAL, \"Query failed:\\n",
|
adamc@874
|
262 string q'',
|
adamc@874
|
263 string "\");",
|
adamc@874
|
264 newline],
|
adamc@874
|
265 string "}",
|
adamc@874
|
266 newline,
|
adamc@874
|
267 newline,
|
adamc@874
|
268
|
adamc@874
|
269 string "if ((res = mysql_store_result(conn->conn)) == NULL) {",
|
adamc@874
|
270 newline,
|
adamc@874
|
271 box [string "mysql_free_result(res);",
|
adamc@874
|
272 newline,
|
adamc@874
|
273 string "mysql_close(conn->conn);",
|
adamc@874
|
274 newline,
|
adamc@874
|
275 string "uw_error(ctx, FATAL, \"Result store failed:\\n",
|
adamc@874
|
276 string q'',
|
adamc@874
|
277 string "\");",
|
adamc@874
|
278 newline],
|
adamc@874
|
279 string "}",
|
adamc@874
|
280 newline,
|
adamc@874
|
281 newline,
|
adamc@874
|
282
|
adamc@874
|
283 string "if (mysql_num_fields(res) != 1) {",
|
adamc@874
|
284 newline,
|
adamc@874
|
285 box [string "mysql_free_result(res);",
|
adamc@874
|
286 newline,
|
adamc@874
|
287 string "mysql_close(conn->conn);",
|
adamc@874
|
288 newline,
|
adamc@874
|
289 string "uw_error(ctx, FATAL, \"Bad column count:\\n",
|
adamc@874
|
290 string q'',
|
adamc@874
|
291 string "\");",
|
adamc@874
|
292 newline],
|
adamc@874
|
293 string "}",
|
adamc@874
|
294 newline,
|
adamc@874
|
295 newline,
|
adamc@874
|
296
|
adamc@874
|
297 string "if ((row = mysql_fetch_row(res)) == NULL) {",
|
adamc@874
|
298 newline,
|
adamc@874
|
299 box [string "mysql_free_result(res);",
|
adamc@874
|
300 newline,
|
adamc@874
|
301 string "mysql_close(conn->conn);",
|
adamc@874
|
302 newline,
|
adamc@874
|
303 string "uw_error(ctx, FATAL, \"Row fetch failed:\\n",
|
adamc@874
|
304 string q'',
|
adamc@874
|
305 string "\");",
|
adamc@874
|
306 newline],
|
adamc@874
|
307 string "}",
|
adamc@874
|
308 newline,
|
adamc@874
|
309 newline,
|
adamc@874
|
310
|
adamc@874
|
311 string "if (strcmp(row[0], \"",
|
adamc@874
|
312 string (Int.toString (length xts)),
|
adamc@874
|
313 string "\")) {",
|
adamc@874
|
314 newline,
|
adamc@874
|
315 box [string "mysql_free_result(res);",
|
adamc@874
|
316 newline,
|
adamc@874
|
317 string "mysql_close(conn->conn);",
|
adamc@874
|
318 newline,
|
adamc@874
|
319 string "uw_error(ctx, FATAL, \"Table '",
|
adamc@874
|
320 string s,
|
adamc@874
|
321 string "' has extra columns.\");",
|
adamc@874
|
322 newline],
|
adamc@874
|
323 string "}",
|
adamc@874
|
324 newline,
|
adamc@874
|
325 newline,
|
adamc@874
|
326 string "mysql_free_result(res);",
|
adamc@874
|
327 newline]
|
adamc@874
|
328 end
|
adamc@874
|
329
|
adamc@872
|
330 fun init {dbstring, prepared = ss, tables, views, sequences} =
|
adamc@866
|
331 let
|
adamc@866
|
332 val host = ref NONE
|
adamc@866
|
333 val user = ref NONE
|
adamc@866
|
334 val passwd = ref NONE
|
adamc@866
|
335 val db = ref NONE
|
adamc@866
|
336 val port = ref NONE
|
adamc@866
|
337 val unix_socket = ref NONE
|
adamc@866
|
338
|
adamc@866
|
339 fun stringOf r = case !r of
|
adamc@866
|
340 NONE => string "NULL"
|
adamc@866
|
341 | SOME s => box [string "\"",
|
adamc@866
|
342 string (String.toString s),
|
adamc@866
|
343 string "\""]
|
adamc@866
|
344 in
|
adamc@866
|
345 app (fn s =>
|
adamc@866
|
346 case String.fields (fn ch => ch = #"=") s of
|
adamc@866
|
347 [name, value] =>
|
adamc@866
|
348 (case name of
|
adamc@866
|
349 "host" =>
|
adamc@866
|
350 if size value > 0 andalso String.sub (value, 0) = #"/" then
|
adamc@866
|
351 unix_socket := SOME value
|
adamc@866
|
352 else
|
adamc@866
|
353 host := SOME value
|
adamc@866
|
354 | "hostaddr" => host := SOME value
|
adamc@866
|
355 | "port" => port := Int.fromString value
|
adamc@866
|
356 | "dbname" => db := SOME value
|
adamc@866
|
357 | "user" => user := SOME value
|
adamc@866
|
358 | "password" => passwd := SOME value
|
adamc@866
|
359 | _ => ())
|
adamc@866
|
360 | _ => ()) (String.tokens Char.isSpace dbstring);
|
adamc@866
|
361
|
adamc@866
|
362 box [string "typedef struct {",
|
adamc@866
|
363 newline,
|
adamc@866
|
364 box [string "MYSQL *conn;",
|
adamc@866
|
365 newline,
|
adamc@866
|
366 p_list_sepi (box [])
|
adamc@866
|
367 (fn i => fn _ =>
|
adamc@866
|
368 box [string "MYSQL_STMT *p",
|
adamc@866
|
369 string (Int.toString i),
|
adamc@866
|
370 string ";",
|
adamc@866
|
371 newline])
|
adamc@866
|
372 ss],
|
adamc@866
|
373 string "} uw_conn;",
|
adamc@866
|
374 newline,
|
adamc@866
|
375 newline,
|
adamc@866
|
376
|
adamc@874
|
377 string "void uw_client_init(void) {",
|
adamc@874
|
378 newline,
|
adamc@879
|
379 box [string "uw_sqlfmtInt = \"%lld%n\";",
|
adamc@879
|
380 newline,
|
adamc@879
|
381 string "uw_sqlfmtFloat = \"%g%n\";",
|
adamc@879
|
382 newline,
|
adamc@879
|
383 string "uw_Estrings = 0;",
|
adamc@879
|
384 newline,
|
adamc@879
|
385 string "uw_sqlsuffixString = \"\";",
|
adamc@879
|
386 newline,
|
adamc@879
|
387 string "uw_sqlsuffixBlob = \"\";",
|
adamc@879
|
388 newline,
|
adamc@879
|
389 string "uw_sqlfmtUint4 = \"%u%n\";",
|
adamc@879
|
390 newline,
|
adamc@879
|
391 newline,
|
adamc@879
|
392
|
adamc@879
|
393 string "if (mysql_library_init(0, NULL, NULL)) {",
|
adamc@874
|
394 newline,
|
adamc@874
|
395 box [string "fprintf(stderr, \"Could not initialize MySQL library\\n\");",
|
adamc@874
|
396 newline,
|
adamc@874
|
397 string "exit(1);",
|
adamc@874
|
398 newline],
|
adamc@874
|
399 string "}",
|
adamc@874
|
400 newline],
|
adamc@874
|
401 string "}",
|
adamc@874
|
402 newline,
|
adamc@874
|
403 newline,
|
adamc@874
|
404
|
adamc@866
|
405 if #persistent (currentProtocol ()) then
|
adamc@874
|
406 box [string "static void uw_db_validate(uw_context ctx) {",
|
adamc@874
|
407 newline,
|
adamc@874
|
408 string "uw_conn *conn = uw_get_db(ctx);",
|
adamc@874
|
409 newline,
|
adamc@874
|
410 string "MYSQL_RES *res;",
|
adamc@874
|
411 newline,
|
adamc@874
|
412 string "MYSQL_ROW row;",
|
adamc@874
|
413 newline,
|
adamc@874
|
414 newline,
|
adamc@874
|
415 p_list_sep newline (checkRel ("tables", true)) tables,
|
adamc@884
|
416 p_list_sep newline (fn name => checkRel ("tables", true)
|
adamc@884
|
417 (name, [("id", Settings.Client)])) sequences,
|
adamc@874
|
418 p_list_sep newline (checkRel ("views", false)) views,
|
adamc@874
|
419 string "}",
|
adamc@874
|
420 newline,
|
adamc@874
|
421 newline,
|
adamc@874
|
422
|
adamc@874
|
423 string "static void uw_db_prepare(uw_context ctx) {",
|
adamc@866
|
424 newline,
|
adamc@866
|
425 string "uw_conn *conn = uw_get_db(ctx);",
|
adamc@866
|
426 newline,
|
adamc@866
|
427 string "MYSQL_STMT *stmt;",
|
adamc@866
|
428 newline,
|
adamc@866
|
429 newline,
|
adamc@866
|
430
|
adamc@866
|
431 p_list_sepi newline (fn i => fn (s, n) =>
|
adamc@866
|
432 let
|
adamc@866
|
433 fun uhoh this s args =
|
adamc@866
|
434 box [p_list_sepi (box [])
|
adamc@866
|
435 (fn j => fn () =>
|
adamc@866
|
436 box [string
|
adamc@866
|
437 "mysql_stmt_close(conn->p",
|
adamc@866
|
438 string (Int.toString j),
|
adamc@866
|
439 string ");",
|
adamc@866
|
440 newline])
|
adamc@866
|
441 (List.tabulate (i, fn _ => ())),
|
adamc@866
|
442 box (if this then
|
adamc@866
|
443 [string
|
adamc@866
|
444 "mysql_stmt_close(conn->p",
|
adamc@866
|
445 string (Int.toString i),
|
adamc@866
|
446 string ");",
|
adamc@866
|
447 newline]
|
adamc@866
|
448 else
|
adamc@866
|
449 []),
|
adamc@866
|
450 string "mysql_close(conn->conn);",
|
adamc@866
|
451 newline,
|
adamc@866
|
452 string "uw_error(ctx, FATAL, \"",
|
adamc@866
|
453 string s,
|
adamc@866
|
454 string "\"",
|
adamc@866
|
455 p_list_sep (box []) (fn s => box [string ", ",
|
adamc@866
|
456 string s]) args,
|
adamc@866
|
457 string ");",
|
adamc@866
|
458 newline]
|
adamc@866
|
459 in
|
adamc@866
|
460 box [string "stmt = mysql_stmt_init(conn->conn);",
|
adamc@866
|
461 newline,
|
adamc@866
|
462 string "if (stmt == NULL) {",
|
adamc@866
|
463 newline,
|
adamc@866
|
464 uhoh false "Out of memory allocating prepared statement" [],
|
adamc@866
|
465 string "}",
|
adamc@866
|
466 newline,
|
adamc@874
|
467 string "conn->p",
|
adamc@874
|
468 string (Int.toString i),
|
adamc@874
|
469 string " = stmt;",
|
adamc@874
|
470 newline,
|
adamc@866
|
471
|
adamc@866
|
472 string "if (mysql_stmt_prepare(stmt, \"",
|
adamc@866
|
473 string (String.toString s),
|
adamc@866
|
474 string "\", ",
|
adamc@866
|
475 string (Int.toString (size s)),
|
adamc@866
|
476 string ")) {",
|
adamc@866
|
477 newline,
|
adamc@866
|
478 box [string "char msg[1024];",
|
adamc@866
|
479 newline,
|
adamc@866
|
480 string "strncpy(msg, mysql_stmt_error(stmt), 1024);",
|
adamc@866
|
481 newline,
|
adamc@866
|
482 string "msg[1023] = 0;",
|
adamc@866
|
483 newline,
|
adamc@866
|
484 uhoh true "Error preparing statement: %s" ["msg"]],
|
adamc@866
|
485 string "}",
|
adamc@866
|
486 newline]
|
adamc@866
|
487 end)
|
adamc@866
|
488 ss,
|
adamc@866
|
489
|
adamc@866
|
490 string "}"]
|
adamc@866
|
491 else
|
adamc@882
|
492 box [string "static void uw_db_prepare(uw_context ctx) { }",
|
adamc@882
|
493 newline,
|
adamc@882
|
494 string "static void uw_db_validate(uw_context ctx) { }"],
|
adamc@866
|
495 newline,
|
adamc@866
|
496 newline,
|
adamc@866
|
497
|
adamc@866
|
498 string "void uw_db_init(uw_context ctx) {",
|
adamc@866
|
499 newline,
|
adamc@866
|
500 string "MYSQL *mysql = mysql_init(NULL);",
|
adamc@866
|
501 newline,
|
adamc@866
|
502 string "uw_conn *conn;",
|
adamc@866
|
503 newline,
|
adamc@866
|
504 string "if (mysql == NULL) uw_error(ctx, FATAL, ",
|
adamc@866
|
505 string "\"libmysqlclient can't allocate a connection.\");",
|
adamc@866
|
506 newline,
|
adamc@866
|
507 string "if (mysql_real_connect(mysql, ",
|
adamc@866
|
508 stringOf host,
|
adamc@866
|
509 string ", ",
|
adamc@866
|
510 stringOf user,
|
adamc@866
|
511 string ", ",
|
adamc@866
|
512 stringOf passwd,
|
adamc@866
|
513 string ", ",
|
adamc@866
|
514 stringOf db,
|
adamc@866
|
515 string ", ",
|
adamc@866
|
516 case !port of
|
adamc@866
|
517 NONE => string "0"
|
adamc@866
|
518 | SOME n => string (Int.toString n),
|
adamc@866
|
519 string ", ",
|
adamc@866
|
520 stringOf unix_socket,
|
adamc@874
|
521 string ", 0) == NULL) {",
|
adamc@866
|
522 newline,
|
adamc@866
|
523 box [string "char msg[1024];",
|
adamc@866
|
524 newline,
|
adamc@866
|
525 string "strncpy(msg, mysql_error(mysql), 1024);",
|
adamc@866
|
526 newline,
|
adamc@866
|
527 string "msg[1023] = 0;",
|
adamc@866
|
528 newline,
|
adamc@866
|
529 string "mysql_close(mysql);",
|
adamc@866
|
530 newline,
|
adamc@866
|
531 string "uw_error(ctx, BOUNDED_RETRY, ",
|
adamc@866
|
532 string "\"Connection to MySQL server failed: %s\", msg);"],
|
adamc@866
|
533 newline,
|
adamc@866
|
534 string "}",
|
adamc@866
|
535 newline,
|
adamc@874
|
536 string "conn = calloc(1, sizeof(uw_conn));",
|
adamc@866
|
537 newline,
|
adamc@866
|
538 string "conn->conn = mysql;",
|
adamc@866
|
539 newline,
|
adamc@866
|
540 string "uw_set_db(ctx, conn);",
|
adamc@866
|
541 newline,
|
adamc@866
|
542 string "uw_db_validate(ctx);",
|
adamc@866
|
543 newline,
|
adamc@866
|
544 string "uw_db_prepare(ctx);",
|
adamc@866
|
545 newline,
|
adamc@866
|
546 string "}",
|
adamc@866
|
547 newline,
|
adamc@866
|
548 newline,
|
adamc@866
|
549
|
adamc@866
|
550 string "void uw_db_close(uw_context ctx) {",
|
adamc@866
|
551 newline,
|
adamc@866
|
552 string "uw_conn *conn = uw_get_db(ctx);",
|
adamc@866
|
553 newline,
|
adamc@866
|
554 p_list_sepi (box [])
|
adamc@866
|
555 (fn i => fn _ =>
|
adamc@866
|
556 box [string "if (conn->p",
|
adamc@866
|
557 string (Int.toString i),
|
adamc@866
|
558 string ") mysql_stmt_close(conn->p",
|
adamc@866
|
559 string (Int.toString i),
|
adamc@866
|
560 string ");",
|
adamc@866
|
561 newline])
|
adamc@866
|
562 ss,
|
adamc@866
|
563 string "mysql_close(conn->conn);",
|
adamc@866
|
564 newline,
|
adamc@866
|
565 string "}",
|
adamc@866
|
566 newline,
|
adamc@866
|
567 newline,
|
adamc@866
|
568
|
adamc@866
|
569 string "int uw_db_begin(uw_context ctx) {",
|
adamc@866
|
570 newline,
|
adamc@866
|
571 string "uw_conn *conn = uw_get_db(ctx);",
|
adamc@866
|
572 newline,
|
adamc@866
|
573 newline,
|
adamc@866
|
574 string "return mysql_query(conn->conn, \"SET TRANSACTION ISOLATION LEVEL SERIALIZABLE\")",
|
adamc@866
|
575 newline,
|
adamc@866
|
576 string " || mysql_query(conn->conn, \"BEGIN\");",
|
adamc@866
|
577 newline,
|
adamc@866
|
578 string "}",
|
adamc@866
|
579 newline,
|
adamc@866
|
580 newline,
|
adamc@866
|
581
|
adamc@866
|
582 string "int uw_db_commit(uw_context ctx) {",
|
adamc@866
|
583 newline,
|
adamc@866
|
584 string "uw_conn *conn = uw_get_db(ctx);",
|
adamc@866
|
585 newline,
|
adamc@866
|
586 string "return mysql_commit(conn->conn);",
|
adamc@866
|
587 newline,
|
adamc@866
|
588 string "}",
|
adamc@866
|
589 newline,
|
adamc@866
|
590 newline,
|
adamc@866
|
591
|
adamc@866
|
592 string "int uw_db_rollback(uw_context ctx) {",
|
adamc@866
|
593 newline,
|
adamc@866
|
594 string "uw_conn *conn = uw_get_db(ctx);",
|
adamc@866
|
595 newline,
|
adamc@866
|
596 string "return mysql_rollback(conn->conn);",
|
adamc@866
|
597 newline,
|
adamc@866
|
598 string "}",
|
adamc@866
|
599 newline,
|
adamc@866
|
600 newline]
|
adamc@866
|
601 end
|
adamc@866
|
602
|
adamc@880
|
603 fun p_getcol {loc, wontLeakStrings = _, col = i, typ = t} =
|
adamc@873
|
604 let
|
adamc@873
|
605 fun getter t =
|
adamc@873
|
606 case t of
|
adamc@873
|
607 String => box [string "({",
|
adamc@873
|
608 newline,
|
adamc@873
|
609 string "uw_Basis_string s = uw_malloc(ctx, length",
|
adamc@873
|
610 string (Int.toString i),
|
adamc@873
|
611 string " + 1);",
|
adamc@873
|
612 newline,
|
adamc@873
|
613 string "out[",
|
adamc@873
|
614 string (Int.toString i),
|
adamc@873
|
615 string "].buffer = s;",
|
adamc@873
|
616 newline,
|
adamc@873
|
617 string "out[",
|
adamc@873
|
618 string (Int.toString i),
|
adamc@873
|
619 string "].buffer_length = length",
|
adamc@873
|
620 string (Int.toString i),
|
adamc@873
|
621 string " + 1;",
|
adamc@873
|
622 newline,
|
adamc@873
|
623 string "mysql_stmt_fetch_column(stmt, &out[",
|
adamc@873
|
624 string (Int.toString i),
|
adamc@873
|
625 string "], ",
|
adamc@873
|
626 string (Int.toString i),
|
adamc@873
|
627 string ", 0);",
|
adamc@873
|
628 newline,
|
adamc@873
|
629 string "s[length",
|
adamc@873
|
630 string (Int.toString i),
|
adamc@873
|
631 string "] = 0;",
|
adamc@873
|
632 newline,
|
adamc@873
|
633 string "s;",
|
adamc@873
|
634 newline,
|
adamc@873
|
635 string "})"]
|
adamc@873
|
636 | Blob => box [string "({",
|
adamc@873
|
637 newline,
|
adamc@873
|
638 string "uw_Basis_blob b = {length",
|
adamc@873
|
639 string (Int.toString i),
|
adamc@873
|
640 string ", uw_malloc(ctx, length",
|
adamc@873
|
641 string (Int.toString i),
|
adamc@873
|
642 string ")};",
|
adamc@873
|
643 newline,
|
adamc@873
|
644 string "out[",
|
adamc@873
|
645 string (Int.toString i),
|
adamc@873
|
646 string "].buffer = b.data;",
|
adamc@873
|
647 newline,
|
adamc@873
|
648 string "out[",
|
adamc@873
|
649 string (Int.toString i),
|
adamc@873
|
650 string "].buffer_length = length",
|
adamc@873
|
651 string (Int.toString i),
|
adamc@873
|
652 string ";",
|
adamc@873
|
653 newline,
|
adamc@873
|
654 string "mysql_stmt_fetch_column(stmt, &out[",
|
adamc@873
|
655 string (Int.toString i),
|
adamc@873
|
656 string "], ",
|
adamc@873
|
657 string (Int.toString i),
|
adamc@873
|
658 string ", 0);",
|
adamc@873
|
659 newline,
|
adamc@873
|
660 string "b;",
|
adamc@873
|
661 newline,
|
adamc@873
|
662 string "})"]
|
adamc@873
|
663 | Time => box [string "({",
|
adamc@876
|
664 string "MYSQL_TIME *mt = &buffer",
|
adamc@873
|
665 string (Int.toString i),
|
adamc@873
|
666 string ";",
|
adamc@873
|
667 newline,
|
adamc@873
|
668 newline,
|
adamc@873
|
669 string "struct tm t = {mt->second, mt->minute, mt->hour, mt->day, mt->month, mt->year, 0, 0, -1};",
|
adamc@873
|
670 newline,
|
adamc@876
|
671 string "mktime(&t);",
|
adamc@873
|
672 newline,
|
adamc@873
|
673 string "})"]
|
adamc@873
|
674 | _ => box [string "buffer",
|
adamc@873
|
675 string (Int.toString i)]
|
adamc@873
|
676 in
|
adamc@873
|
677 case t of
|
adamc@873
|
678 Nullable t => box [string "(is_null",
|
adamc@873
|
679 string (Int.toString i),
|
adamc@873
|
680 string " ? NULL : ",
|
adamc@873
|
681 case t of
|
adamc@873
|
682 String => getter t
|
adamc@873
|
683 | _ => box [string "({",
|
adamc@873
|
684 newline,
|
adamc@873
|
685 string (p_sql_ctype t),
|
adamc@873
|
686 space,
|
adamc@873
|
687 string "*tmp = uw_malloc(ctx, sizeof(",
|
adamc@873
|
688 string (p_sql_ctype t),
|
adamc@873
|
689 string "));",
|
adamc@873
|
690 newline,
|
adamc@873
|
691 string "*tmp = ",
|
adamc@873
|
692 getter t,
|
adamc@873
|
693 string ";",
|
adamc@873
|
694 newline,
|
adamc@873
|
695 string "tmp;",
|
adamc@873
|
696 newline,
|
adamc@873
|
697 string "})"],
|
adamc@873
|
698 string ")"]
|
adamc@873
|
699 | _ => box [string "(is_null",
|
adamc@873
|
700 string (Int.toString i),
|
adamc@873
|
701 string " ? ",
|
adamc@873
|
702 box [string "({",
|
adamc@873
|
703 string (p_sql_ctype t),
|
adamc@873
|
704 space,
|
adamc@873
|
705 string "tmp;",
|
adamc@873
|
706 newline,
|
adamc@873
|
707 string "uw_error(ctx, FATAL, \"Unexpectedly NULL field #",
|
adamc@873
|
708 string (Int.toString i),
|
adamc@873
|
709 string "\");",
|
adamc@873
|
710 newline,
|
adamc@873
|
711 string "tmp;",
|
adamc@873
|
712 newline,
|
adamc@873
|
713 string "})"],
|
adamc@873
|
714 string " : ",
|
adamc@873
|
715 getter t,
|
adamc@873
|
716 string ")"]
|
adamc@873
|
717 end
|
adamc@873
|
718
|
adamc@873
|
719 fun queryCommon {loc, query, cols, doCols} =
|
adamc@873
|
720 box [string "int n, r;",
|
adamc@873
|
721 newline,
|
adamc@873
|
722 string "MYSQL_BIND out[",
|
adamc@873
|
723 string (Int.toString (length cols)),
|
adamc@873
|
724 string "];",
|
adamc@873
|
725 newline,
|
adamc@873
|
726 p_list_sepi (box []) (fn i => fn t =>
|
adamc@873
|
727 let
|
adamc@873
|
728 fun buffers t =
|
adamc@873
|
729 case t of
|
adamc@873
|
730 String => box [string "unsigned long length",
|
adamc@873
|
731 string (Int.toString i),
|
adamc@873
|
732 string ";",
|
adamc@873
|
733 newline]
|
adamc@873
|
734 | Blob => box [string "unsigned long length",
|
adamc@873
|
735 string (Int.toString i),
|
adamc@873
|
736 string ";",
|
adamc@873
|
737 newline]
|
adamc@876
|
738 | Time => box [string "MYSQL_TIME buffer",
|
adamc@876
|
739 string (Int.toString i),
|
adamc@876
|
740 string ";",
|
adamc@876
|
741 newline]
|
adamc@873
|
742 | _ => box [string (p_sql_ctype t),
|
adamc@873
|
743 space,
|
adamc@873
|
744 string "buffer",
|
adamc@873
|
745 string (Int.toString i),
|
adamc@873
|
746 string ";",
|
adamc@873
|
747 newline]
|
adamc@873
|
748 in
|
adamc@873
|
749 box [string "my_bool is_null",
|
adamc@873
|
750 string (Int.toString i),
|
adamc@873
|
751 string ";",
|
adamc@873
|
752 newline,
|
adamc@873
|
753 case t of
|
adamc@873
|
754 Nullable t => buffers t
|
adamc@873
|
755 | _ => buffers t,
|
adamc@873
|
756 newline]
|
adamc@873
|
757 end) cols,
|
adamc@873
|
758 newline,
|
adamc@873
|
759
|
adamc@873
|
760 string "memset(out, 0, sizeof out);",
|
adamc@873
|
761 newline,
|
adamc@873
|
762 p_list_sepi (box []) (fn i => fn t =>
|
adamc@873
|
763 let
|
adamc@873
|
764 fun buffers t =
|
adamc@873
|
765 case t of
|
adamc@875
|
766 String => box [string "out[",
|
adamc@875
|
767 string (Int.toString i),
|
adamc@875
|
768 string "].length = &length",
|
adamc@875
|
769 string (Int.toString i),
|
adamc@875
|
770 string ";",
|
adamc@875
|
771 newline]
|
adamc@875
|
772 | Blob => box [string "out[",
|
adamc@875
|
773 string (Int.toString i),
|
adamc@875
|
774 string "].length = &length",
|
adamc@875
|
775 string (Int.toString i),
|
adamc@875
|
776 string ";",
|
adamc@875
|
777 newline]
|
adamc@873
|
778 | _ => box [string "out[",
|
adamc@873
|
779 string (Int.toString i),
|
adamc@873
|
780 string "].buffer = &buffer",
|
adamc@873
|
781 string (Int.toString i),
|
adamc@873
|
782 string ";",
|
adamc@873
|
783 newline]
|
adamc@873
|
784 in
|
adamc@873
|
785 box [string "out[",
|
adamc@873
|
786 string (Int.toString i),
|
adamc@873
|
787 string "].buffer_type = ",
|
adamc@873
|
788 string (p_buffer_type t),
|
adamc@873
|
789 string ";",
|
adamc@873
|
790 newline,
|
adamc@873
|
791 string "out[",
|
adamc@873
|
792 string (Int.toString i),
|
adamc@873
|
793 string "].is_null = &is_null",
|
adamc@873
|
794 string (Int.toString i),
|
adamc@873
|
795 string ";",
|
adamc@873
|
796 newline,
|
adamc@873
|
797
|
adamc@873
|
798 case t of
|
adamc@873
|
799 Nullable t => buffers t
|
adamc@873
|
800 | _ => buffers t,
|
adamc@873
|
801 newline]
|
adamc@873
|
802 end) cols,
|
adamc@873
|
803 newline,
|
adamc@873
|
804
|
adamc@875
|
805 string "if (mysql_stmt_reset(stmt)) uw_error(ctx, FATAL, \"",
|
adamc@875
|
806 string (ErrorMsg.spanToString loc),
|
adamc@875
|
807 string ": Error reseting statement: %s\\n%s\", ",
|
adamc@875
|
808 query,
|
adamc@875
|
809 string ", mysql_error(conn->conn));",
|
adamc@875
|
810 newline,
|
adamc@875
|
811 newline,
|
adamc@875
|
812
|
adamc@873
|
813 string "if (mysql_stmt_execute(stmt)) uw_error(ctx, FATAL, \"",
|
adamc@873
|
814 string (ErrorMsg.spanToString loc),
|
adamc@875
|
815 string ": Error executing query: %s\\n%s\", ",
|
adamc@875
|
816 query,
|
adamc@875
|
817 string ", mysql_error(conn->conn));",
|
adamc@875
|
818 newline,
|
adamc@875
|
819 newline,
|
adamc@875
|
820
|
adamc@875
|
821 string "if (mysql_stmt_bind_result(stmt, out)) uw_error(ctx, FATAL, \"",
|
adamc@875
|
822 string (ErrorMsg.spanToString loc),
|
adamc@875
|
823 string ": Error binding query result: %s\\n%s\", ",
|
adamc@875
|
824 query,
|
adamc@875
|
825 string ", mysql_error(conn->conn));",
|
adamc@873
|
826 newline,
|
adamc@873
|
827 newline,
|
adamc@873
|
828
|
adamc@873
|
829 string "if (mysql_stmt_store_result(stmt)) uw_error(ctx, FATAL, \"",
|
adamc@873
|
830 string (ErrorMsg.spanToString loc),
|
adamc@875
|
831 string ": Error storing query result: %s\\n%s\", ",
|
adamc@875
|
832 query,
|
adamc@875
|
833 string ", mysql_error(conn->conn));",
|
adamc@873
|
834 newline,
|
adamc@873
|
835 newline,
|
adamc@873
|
836
|
adamc@873
|
837 string "uw_end_region(ctx);",
|
adamc@873
|
838 newline,
|
adamc@875
|
839 string "while (1) {",
|
adamc@875
|
840 newline,
|
adamc@875
|
841 string "r = mysql_stmt_fetch(stmt);",
|
adamc@875
|
842 newline,
|
adamc@875
|
843 string "if (r != 0 && r != MYSQL_DATA_TRUNCATED) break;",
|
adamc@873
|
844 newline,
|
adamc@873
|
845 doCols p_getcol,
|
adamc@873
|
846 string "}",
|
adamc@873
|
847 newline,
|
adamc@873
|
848 newline,
|
adamc@873
|
849
|
adamc@874
|
850 string "if (r == 1) uw_error(ctx, FATAL, \"",
|
adamc@873
|
851 string (ErrorMsg.spanToString loc),
|
adamc@875
|
852 string ": query result fetching failed: %s\\n%s\", ",
|
adamc@875
|
853 query,
|
adamc@875
|
854 string ", mysql_error(conn->conn));",
|
adamc@875
|
855 newline,
|
adamc@875
|
856 newline,
|
adamc@875
|
857
|
adamc@875
|
858 string "if (mysql_stmt_reset(stmt)) uw_error(ctx, FATAL, \"",
|
adamc@875
|
859 string (ErrorMsg.spanToString loc),
|
adamc@875
|
860 string ": Error reseting statement: %s\\n%s\", ",
|
adamc@875
|
861 query,
|
adamc@875
|
862 string ", mysql_error(conn->conn));",
|
adamc@875
|
863 newline,
|
adamc@875
|
864 newline]
|
adamc@873
|
865
|
adamc@873
|
866 fun query {loc, cols, doCols} =
|
adamc@873
|
867 box [string "uw_conn *conn = uw_get_db(ctx);",
|
adamc@873
|
868 newline,
|
adamc@876
|
869 string "MYSQL_STMT *stmt = mysql_stmt_init(conn->conn);",
|
adamc@873
|
870 newline,
|
adamc@875
|
871 string "if (stmt == NULL) uw_error(ctx, FATAL, \"",
|
adamc@873
|
872 string (ErrorMsg.spanToString loc),
|
adamc@873
|
873 string ": can't allocate temporary prepared statement\");",
|
adamc@873
|
874 newline,
|
adamc@873
|
875 string "uw_push_cleanup(ctx, (void (*)(void *))mysql_stmt_close, stmt);",
|
adamc@873
|
876 newline,
|
adamc@873
|
877 string "if (mysql_stmt_prepare(stmt, query, strlen(query))) uw_error(ctx, FATAL, \"",
|
adamc@873
|
878 string (ErrorMsg.spanToString loc),
|
adamc@875
|
879 string ": error preparing statement: %s\\n%s\", query, mysql_error(conn->conn));",
|
adamc@873
|
880 newline,
|
adamc@873
|
881 newline,
|
adamc@873
|
882
|
adamc@873
|
883 queryCommon {loc = loc, cols = cols, doCols = doCols, query = string "query"},
|
adamc@873
|
884
|
adamc@873
|
885 string "uw_pop_cleanup(ctx);",
|
adamc@873
|
886 newline]
|
adamc@873
|
887
|
adamc@879
|
888 fun queryPrepared {loc, id, query, inputs, cols, doCols, nested} =
|
adamc@873
|
889 box [string "uw_conn *conn = uw_get_db(ctx);",
|
adamc@873
|
890 newline,
|
adamc@873
|
891 string "MYSQL_BIND in[",
|
adamc@873
|
892 string (Int.toString (length inputs)),
|
adamc@873
|
893 string "];",
|
adamc@873
|
894 newline,
|
adamc@873
|
895 p_list_sepi (box []) (fn i => fn t =>
|
adamc@873
|
896 let
|
adamc@873
|
897 fun buffers t =
|
adamc@873
|
898 case t of
|
adamc@873
|
899 String => box [string "unsigned long in_length",
|
adamc@873
|
900 string (Int.toString i),
|
adamc@873
|
901 string ";",
|
adamc@873
|
902 newline]
|
adamc@873
|
903 | Blob => box [string "unsigned long in_length",
|
adamc@873
|
904 string (Int.toString i),
|
adamc@873
|
905 string ";",
|
adamc@873
|
906 newline]
|
adamc@876
|
907 | Time => box [string "MYSQL_TIME in_buffer",
|
adamc@873
|
908 string (Int.toString i),
|
adamc@873
|
909 string ";",
|
adamc@873
|
910 newline]
|
adamc@873
|
911 | _ => box []
|
adamc@873
|
912 in
|
adamc@873
|
913 box [case t of
|
adamc@873
|
914 Nullable t => box [string "my_bool in_is_null",
|
adamc@873
|
915 string (Int.toString i),
|
adamc@873
|
916 string ";",
|
adamc@873
|
917 newline,
|
adamc@873
|
918 buffers t]
|
adamc@873
|
919 | _ => buffers t,
|
adamc@873
|
920 newline]
|
adamc@873
|
921 end) inputs,
|
adamc@873
|
922
|
adamc@879
|
923 if nested then
|
adamc@879
|
924 box [string "MYSQL_STMT *stmt;",
|
adamc@879
|
925 newline]
|
adamc@879
|
926 else
|
adamc@879
|
927 box [string "MYSQL_STMT *stmt = conn->p",
|
adamc@879
|
928 string (Int.toString id),
|
adamc@879
|
929 string ";",
|
adamc@879
|
930 newline,
|
adamc@879
|
931 newline,
|
adamc@879
|
932
|
adamc@879
|
933 string "if (stmt == NULL) {",
|
adamc@879
|
934 newline],
|
adamc@879
|
935
|
adamc@878
|
936 box [string "stmt = mysql_stmt_init(conn->conn);",
|
adamc@878
|
937 newline,
|
adamc@878
|
938 string "if (stmt == NULL) uw_error(ctx, FATAL, \"Out of memory allocating prepared statement\");",
|
adamc@878
|
939 newline,
|
adamc@880
|
940 if nested then
|
adamc@880
|
941 box [string "uw_push_cleanup(ctx, (void (*)(void *))mysql_stmt_close, stmt);",
|
adamc@880
|
942 newline]
|
adamc@880
|
943 else
|
adamc@880
|
944 box [],
|
adamc@878
|
945 string "if (mysql_stmt_prepare(stmt, \"",
|
adamc@878
|
946 string (String.toString query),
|
adamc@878
|
947 string "\", ",
|
adamc@878
|
948 string (Int.toString (size query)),
|
adamc@878
|
949 string ")) {",
|
adamc@878
|
950 newline,
|
adamc@878
|
951 box [string "char msg[1024];",
|
adamc@878
|
952 newline,
|
adamc@878
|
953 string "strncpy(msg, mysql_stmt_error(stmt), 1024);",
|
adamc@878
|
954 newline,
|
adamc@878
|
955 string "msg[1023] = 0;",
|
adamc@878
|
956 newline,
|
adamc@880
|
957 if nested then
|
adamc@880
|
958 box []
|
adamc@880
|
959 else
|
adamc@880
|
960 box [string "mysql_stmt_close(stmt);",
|
adamc@880
|
961 newline],
|
adamc@878
|
962 string "uw_error(ctx, FATAL, \"Error preparing statement: %s\", msg);",
|
adamc@878
|
963 newline],
|
adamc@878
|
964 string "}",
|
adamc@878
|
965 newline,
|
adamc@879
|
966 if nested then
|
adamc@879
|
967 box []
|
adamc@879
|
968 else
|
adamc@879
|
969 box [string "conn->p",
|
adamc@879
|
970 string (Int.toString id),
|
adamc@879
|
971 string " = stmt;",
|
adamc@879
|
972 newline]],
|
adamc@879
|
973 if nested then
|
adamc@879
|
974 box []
|
adamc@879
|
975 else
|
adamc@879
|
976 box [string "}",
|
adamc@879
|
977 newline],
|
adamc@878
|
978 newline,
|
adamc@878
|
979
|
adamc@873
|
980 string "memset(in, 0, sizeof in);",
|
adamc@873
|
981 newline,
|
adamc@873
|
982 p_list_sepi (box []) (fn i => fn t =>
|
adamc@873
|
983 let
|
adamc@873
|
984 fun buffers t =
|
adamc@873
|
985 case t of
|
adamc@873
|
986 String => box [string "in[",
|
adamc@873
|
987 string (Int.toString i),
|
adamc@873
|
988 string "].buffer = arg",
|
adamc@873
|
989 string (Int.toString (i + 1)),
|
adamc@873
|
990 string ";",
|
adamc@873
|
991 newline,
|
adamc@873
|
992 string "in_length",
|
adamc@873
|
993 string (Int.toString i),
|
adamc@873
|
994 string "= in[",
|
adamc@873
|
995 string (Int.toString i),
|
adamc@873
|
996 string "].buffer_length = strlen(arg",
|
adamc@873
|
997 string (Int.toString (i + 1)),
|
adamc@873
|
998 string ");",
|
adamc@873
|
999 newline,
|
adamc@873
|
1000 string "in[",
|
adamc@873
|
1001 string (Int.toString i),
|
adamc@873
|
1002 string "].length = &in_length",
|
adamc@873
|
1003 string (Int.toString i),
|
adamc@873
|
1004 string ";",
|
adamc@873
|
1005 newline]
|
adamc@873
|
1006 | Blob => box [string "in[",
|
adamc@873
|
1007 string (Int.toString i),
|
adamc@873
|
1008 string "].buffer = arg",
|
adamc@873
|
1009 string (Int.toString (i + 1)),
|
adamc@873
|
1010 string ".data;",
|
adamc@873
|
1011 newline,
|
adamc@873
|
1012 string "in_length",
|
adamc@873
|
1013 string (Int.toString i),
|
adamc@873
|
1014 string "= in[",
|
adamc@873
|
1015 string (Int.toString i),
|
adamc@873
|
1016 string "].buffer_length = arg",
|
adamc@873
|
1017 string (Int.toString (i + 1)),
|
adamc@873
|
1018 string ".size;",
|
adamc@873
|
1019 newline,
|
adamc@873
|
1020 string "in[",
|
adamc@873
|
1021 string (Int.toString i),
|
adamc@873
|
1022 string "].length = &in_length",
|
adamc@873
|
1023 string (Int.toString i),
|
adamc@873
|
1024 string ";",
|
adamc@873
|
1025 newline]
|
adamc@873
|
1026 | Time =>
|
adamc@873
|
1027 let
|
adamc@873
|
1028 fun oneField dst src =
|
adamc@873
|
1029 box [string "in_buffer",
|
adamc@873
|
1030 string (Int.toString i),
|
adamc@873
|
1031 string ".",
|
adamc@873
|
1032 string dst,
|
adamc@873
|
1033 string " = tms.tm_",
|
adamc@873
|
1034 string src,
|
adamc@873
|
1035 string ";",
|
adamc@873
|
1036 newline]
|
adamc@873
|
1037 in
|
adamc@873
|
1038 box [string "({",
|
adamc@873
|
1039 newline,
|
adamc@873
|
1040 string "struct tm tms;",
|
adamc@873
|
1041 newline,
|
adamc@873
|
1042 string "if (localtime_r(&arg",
|
adamc@873
|
1043 string (Int.toString (i + 1)),
|
adamc@873
|
1044 string ", &tm) == NULL) uw_error(\"",
|
adamc@873
|
1045 string (ErrorMsg.spanToString loc),
|
adamc@873
|
1046 string ": error converting to MySQL time\");",
|
adamc@873
|
1047 newline,
|
adamc@873
|
1048 oneField "year" "year",
|
adamc@873
|
1049 oneField "month" "mon",
|
adamc@873
|
1050 oneField "day" "mday",
|
adamc@873
|
1051 oneField "hour" "hour",
|
adamc@873
|
1052 oneField "minute" "min",
|
adamc@873
|
1053 oneField "second" "sec",
|
adamc@873
|
1054 newline,
|
adamc@873
|
1055 string "in[",
|
adamc@873
|
1056 string (Int.toString i),
|
adamc@873
|
1057 string "].buffer = &in_buffer",
|
adamc@873
|
1058 string (Int.toString i),
|
adamc@873
|
1059 string ";",
|
adamc@873
|
1060 newline]
|
adamc@873
|
1061 end
|
adamc@873
|
1062
|
adamc@873
|
1063 | _ => box [string "in[",
|
adamc@873
|
1064 string (Int.toString i),
|
adamc@873
|
1065 string "].buffer = &arg",
|
adamc@873
|
1066 string (Int.toString (i + 1)),
|
adamc@873
|
1067 string ";",
|
adamc@873
|
1068 newline]
|
adamc@873
|
1069 in
|
adamc@873
|
1070 box [string "in[",
|
adamc@873
|
1071 string (Int.toString i),
|
adamc@873
|
1072 string "].buffer_type = ",
|
adamc@873
|
1073 string (p_buffer_type t),
|
adamc@873
|
1074 string ";",
|
adamc@873
|
1075 newline,
|
adamc@873
|
1076
|
adamc@873
|
1077 case t of
|
adamc@873
|
1078 Nullable t => box [string "in[",
|
adamc@873
|
1079 string (Int.toString i),
|
adamc@873
|
1080 string "].is_null = &in_is_null",
|
adamc@873
|
1081 string (Int.toString i),
|
adamc@873
|
1082 string ";",
|
adamc@873
|
1083 newline,
|
adamc@873
|
1084 string "if (arg",
|
adamc@873
|
1085 string (Int.toString (i + 1)),
|
adamc@873
|
1086 string " == NULL) {",
|
adamc@873
|
1087 newline,
|
adamc@873
|
1088 box [string "in_is_null",
|
adamc@873
|
1089 string (Int.toString i),
|
adamc@873
|
1090 string " = 1;",
|
adamc@873
|
1091 newline],
|
adamc@873
|
1092 string "} else {",
|
adamc@873
|
1093 box [case t of
|
adamc@873
|
1094 String => box []
|
adamc@873
|
1095 | _ =>
|
adamc@873
|
1096 box [string (p_sql_ctype t),
|
adamc@873
|
1097 space,
|
adamc@876
|
1098 string "tmp = *arg",
|
adamc@876
|
1099 string (Int.toString (i + 1)),
|
adamc@876
|
1100 string ";",
|
adamc@876
|
1101 newline,
|
adamc@876
|
1102 string (p_sql_ctype t),
|
adamc@876
|
1103 space,
|
adamc@873
|
1104 string "arg",
|
adamc@873
|
1105 string (Int.toString (i + 1)),
|
adamc@876
|
1106 string " = tmp;",
|
adamc@873
|
1107 newline],
|
adamc@873
|
1108 string "in_is_null",
|
adamc@873
|
1109 string (Int.toString i),
|
adamc@873
|
1110 string " = 0;",
|
adamc@873
|
1111 newline,
|
adamc@873
|
1112 buffers t,
|
adamc@876
|
1113 newline],
|
adamc@876
|
1114 string "}",
|
adamc@876
|
1115 newline]
|
adamc@873
|
1116
|
adamc@873
|
1117 | _ => buffers t,
|
adamc@873
|
1118 newline]
|
adamc@873
|
1119 end) inputs,
|
adamc@873
|
1120 newline,
|
adamc@873
|
1121
|
adamc@875
|
1122 string "if (mysql_stmt_bind_param(stmt, in)) uw_error(ctx, FATAL, \"",
|
adamc@875
|
1123 string (ErrorMsg.spanToString loc),
|
adamc@875
|
1124 string ": error binding parameters\");",
|
adamc@875
|
1125 newline,
|
adamc@875
|
1126
|
adamc@873
|
1127 queryCommon {loc = loc, cols = cols, doCols = doCols, query = box [string "\"",
|
adamc@873
|
1128 string (String.toString query),
|
adamc@879
|
1129 string "\""]},
|
adamc@879
|
1130
|
adamc@879
|
1131 if nested then
|
adamc@879
|
1132 box [string "uw_pop_cleanup(ctx);",
|
adamc@879
|
1133 newline]
|
adamc@879
|
1134 else
|
adamc@879
|
1135 box []]
|
adamc@873
|
1136
|
adamc@875
|
1137 fun dmlCommon {loc, dml} =
|
adamc@875
|
1138 box [string "if (mysql_stmt_execute(stmt)) uw_error(ctx, FATAL, \"",
|
adamc@875
|
1139 string (ErrorMsg.spanToString loc),
|
adamc@875
|
1140 string ": Error executing DML: %s\\n%s\", ",
|
adamc@875
|
1141 dml,
|
adamc@875
|
1142 string ", mysql_error(conn->conn));",
|
adamc@875
|
1143 newline,
|
adamc@875
|
1144 newline]
|
adamc@875
|
1145
|
adamc@875
|
1146 fun dml loc =
|
adamc@875
|
1147 box [string "uw_conn *conn = uw_get_db(ctx);",
|
adamc@875
|
1148 newline,
|
adamc@875
|
1149 string "MYSQL_stmt *stmt = mysql_stmt_init(conn->conn);",
|
adamc@875
|
1150 newline,
|
adamc@875
|
1151 string "if (stmt == NULL) uw_error(ctx, \"",
|
adamc@875
|
1152 string (ErrorMsg.spanToString loc),
|
adamc@875
|
1153 string ": can't allocate temporary prepared statement\");",
|
adamc@875
|
1154 newline,
|
adamc@875
|
1155 string "uw_push_cleanup(ctx, (void (*)(void *))mysql_stmt_close, stmt);",
|
adamc@875
|
1156 newline,
|
adamc@875
|
1157 string "if (mysql_stmt_prepare(stmt, dml, strlen(dml))) uw_error(ctx, FATAL, \"",
|
adamc@875
|
1158 string (ErrorMsg.spanToString loc),
|
adamc@875
|
1159 string ": error preparing statement: %s\\n%s\", dml, mysql_error(conn->conn));",
|
adamc@875
|
1160 newline,
|
adamc@875
|
1161 newline,
|
adamc@875
|
1162
|
adamc@875
|
1163 dmlCommon {loc = loc, dml = string "dml"},
|
adamc@875
|
1164
|
adamc@875
|
1165 string "uw_pop_cleanup(ctx);",
|
adamc@875
|
1166 newline]
|
adamc@875
|
1167
|
adamc@875
|
1168 fun dmlPrepared {loc, id, dml, inputs} =
|
adamc@875
|
1169 box [string "uw_conn *conn = uw_get_db(ctx);",
|
adamc@875
|
1170 newline,
|
adamc@875
|
1171 string "MYSQL_BIND in[",
|
adamc@875
|
1172 string (Int.toString (length inputs)),
|
adamc@875
|
1173 string "];",
|
adamc@875
|
1174 newline,
|
adamc@875
|
1175 p_list_sepi (box []) (fn i => fn t =>
|
adamc@875
|
1176 let
|
adamc@875
|
1177 fun buffers t =
|
adamc@875
|
1178 case t of
|
adamc@875
|
1179 String => box [string "unsigned long in_length",
|
adamc@875
|
1180 string (Int.toString i),
|
adamc@875
|
1181 string ";",
|
adamc@875
|
1182 newline]
|
adamc@875
|
1183 | Blob => box [string "unsigned long in_length",
|
adamc@875
|
1184 string (Int.toString i),
|
adamc@875
|
1185 string ";",
|
adamc@875
|
1186 newline]
|
adamc@876
|
1187 | Time => box [string "MYSQL_TIME in_buffer",
|
adamc@875
|
1188 string (Int.toString i),
|
adamc@875
|
1189 string ";",
|
adamc@875
|
1190 newline]
|
adamc@875
|
1191 | _ => box []
|
adamc@875
|
1192 in
|
adamc@875
|
1193 box [case t of
|
adamc@875
|
1194 Nullable t => box [string "my_bool in_is_null",
|
adamc@875
|
1195 string (Int.toString i),
|
adamc@875
|
1196 string ";",
|
adamc@875
|
1197 newline,
|
adamc@875
|
1198 buffers t]
|
adamc@875
|
1199 | _ => buffers t,
|
adamc@875
|
1200 newline]
|
adamc@875
|
1201 end) inputs,
|
adamc@875
|
1202 string "MYSQL_STMT *stmt = conn->p",
|
adamc@875
|
1203 string (Int.toString id),
|
adamc@875
|
1204 string ";",
|
adamc@875
|
1205 newline,
|
adamc@875
|
1206 newline,
|
adamc@875
|
1207
|
adamc@878
|
1208 string "if (stmt == NULL) {",
|
adamc@878
|
1209 newline,
|
adamc@878
|
1210 box [string "stmt = mysql_stmt_init(conn->conn);",
|
adamc@878
|
1211 newline,
|
adamc@878
|
1212 string "if (stmt == NULL) uw_error(ctx, FATAL, \"Out of memory allocating prepared statement\");",
|
adamc@878
|
1213 newline,
|
adamc@878
|
1214 string "if (mysql_stmt_prepare(stmt, \"",
|
adamc@878
|
1215 string (String.toString dml),
|
adamc@878
|
1216 string "\", ",
|
adamc@878
|
1217 string (Int.toString (size dml)),
|
adamc@878
|
1218 string ")) {",
|
adamc@878
|
1219 newline,
|
adamc@878
|
1220 box [string "char msg[1024];",
|
adamc@878
|
1221 newline,
|
adamc@878
|
1222 string "strncpy(msg, mysql_stmt_error(stmt), 1024);",
|
adamc@878
|
1223 newline,
|
adamc@878
|
1224 string "msg[1023] = 0;",
|
adamc@878
|
1225 newline,
|
adamc@878
|
1226 string "uw_error(ctx, FATAL, \"Error preparing statement: %s\", msg);",
|
adamc@878
|
1227 newline],
|
adamc@878
|
1228 string "}",
|
adamc@878
|
1229 newline,
|
adamc@878
|
1230 string "conn->p",
|
adamc@878
|
1231 string (Int.toString id),
|
adamc@878
|
1232 string " = stmt;",
|
adamc@878
|
1233 newline],
|
adamc@878
|
1234 string "}",
|
adamc@878
|
1235 newline,
|
adamc@878
|
1236 newline,
|
adamc@878
|
1237
|
adamc@875
|
1238 string "memset(in, 0, sizeof in);",
|
adamc@875
|
1239 newline,
|
adamc@875
|
1240 p_list_sepi (box []) (fn i => fn t =>
|
adamc@875
|
1241 let
|
adamc@875
|
1242 fun buffers t =
|
adamc@875
|
1243 case t of
|
adamc@875
|
1244 String => box [string "in[",
|
adamc@875
|
1245 string (Int.toString i),
|
adamc@875
|
1246 string "].buffer = arg",
|
adamc@875
|
1247 string (Int.toString (i + 1)),
|
adamc@875
|
1248 string ";",
|
adamc@875
|
1249 newline,
|
adamc@875
|
1250 string "in_length",
|
adamc@875
|
1251 string (Int.toString i),
|
adamc@875
|
1252 string "= in[",
|
adamc@875
|
1253 string (Int.toString i),
|
adamc@875
|
1254 string "].buffer_length = strlen(arg",
|
adamc@875
|
1255 string (Int.toString (i + 1)),
|
adamc@875
|
1256 string ");",
|
adamc@875
|
1257 newline,
|
adamc@875
|
1258 string "in[",
|
adamc@875
|
1259 string (Int.toString i),
|
adamc@875
|
1260 string "].length = &in_length",
|
adamc@875
|
1261 string (Int.toString i),
|
adamc@875
|
1262 string ";",
|
adamc@875
|
1263 newline]
|
adamc@875
|
1264 | Blob => box [string "in[",
|
adamc@875
|
1265 string (Int.toString i),
|
adamc@875
|
1266 string "].buffer = arg",
|
adamc@875
|
1267 string (Int.toString (i + 1)),
|
adamc@875
|
1268 string ".data;",
|
adamc@875
|
1269 newline,
|
adamc@875
|
1270 string "in_length",
|
adamc@875
|
1271 string (Int.toString i),
|
adamc@875
|
1272 string "= in[",
|
adamc@875
|
1273 string (Int.toString i),
|
adamc@875
|
1274 string "].buffer_length = arg",
|
adamc@875
|
1275 string (Int.toString (i + 1)),
|
adamc@875
|
1276 string ".size;",
|
adamc@875
|
1277 newline,
|
adamc@875
|
1278 string "in[",
|
adamc@875
|
1279 string (Int.toString i),
|
adamc@875
|
1280 string "].length = &in_length",
|
adamc@875
|
1281 string (Int.toString i),
|
adamc@875
|
1282 string ";",
|
adamc@875
|
1283 newline]
|
adamc@875
|
1284 | Time =>
|
adamc@875
|
1285 let
|
adamc@875
|
1286 fun oneField dst src =
|
adamc@875
|
1287 box [string "in_buffer",
|
adamc@875
|
1288 string (Int.toString i),
|
adamc@875
|
1289 string ".",
|
adamc@875
|
1290 string dst,
|
adamc@875
|
1291 string " = tms.tm_",
|
adamc@875
|
1292 string src,
|
adamc@875
|
1293 string ";",
|
adamc@875
|
1294 newline]
|
adamc@875
|
1295 in
|
adamc@875
|
1296 box [string "({",
|
adamc@875
|
1297 newline,
|
adamc@875
|
1298 string "struct tm tms;",
|
adamc@875
|
1299 newline,
|
adamc@875
|
1300 string "if (localtime_r(&arg",
|
adamc@875
|
1301 string (Int.toString (i + 1)),
|
adamc@875
|
1302 string ", &tm) == NULL) uw_error(\"",
|
adamc@875
|
1303 string (ErrorMsg.spanToString loc),
|
adamc@875
|
1304 string ": error converting to MySQL time\");",
|
adamc@875
|
1305 newline,
|
adamc@875
|
1306 oneField "year" "year",
|
adamc@875
|
1307 oneField "month" "mon",
|
adamc@875
|
1308 oneField "day" "mday",
|
adamc@875
|
1309 oneField "hour" "hour",
|
adamc@875
|
1310 oneField "minute" "min",
|
adamc@875
|
1311 oneField "second" "sec",
|
adamc@875
|
1312 newline,
|
adamc@875
|
1313 string "in[",
|
adamc@875
|
1314 string (Int.toString i),
|
adamc@875
|
1315 string "].buffer = &in_buffer",
|
adamc@875
|
1316 string (Int.toString i),
|
adamc@875
|
1317 string ";",
|
adamc@875
|
1318 newline]
|
adamc@875
|
1319 end
|
adamc@875
|
1320
|
adamc@875
|
1321 | _ => box [string "in[",
|
adamc@875
|
1322 string (Int.toString i),
|
adamc@875
|
1323 string "].buffer = &arg",
|
adamc@875
|
1324 string (Int.toString (i + 1)),
|
adamc@875
|
1325 string ";",
|
adamc@875
|
1326 newline]
|
adamc@875
|
1327 in
|
adamc@875
|
1328 box [string "in[",
|
adamc@875
|
1329 string (Int.toString i),
|
adamc@875
|
1330 string "].buffer_type = ",
|
adamc@875
|
1331 string (p_buffer_type t),
|
adamc@875
|
1332 string ";",
|
adamc@875
|
1333 newline,
|
adamc@875
|
1334
|
adamc@875
|
1335 case t of
|
adamc@875
|
1336 Nullable t => box [string "in[",
|
adamc@875
|
1337 string (Int.toString i),
|
adamc@875
|
1338 string "].is_null = &in_is_null",
|
adamc@875
|
1339 string (Int.toString i),
|
adamc@875
|
1340 string ";",
|
adamc@875
|
1341 newline,
|
adamc@875
|
1342 string "if (arg",
|
adamc@875
|
1343 string (Int.toString (i + 1)),
|
adamc@875
|
1344 string " == NULL) {",
|
adamc@875
|
1345 newline,
|
adamc@875
|
1346 box [string "in_is_null",
|
adamc@875
|
1347 string (Int.toString i),
|
adamc@875
|
1348 string " = 1;",
|
adamc@875
|
1349 newline],
|
adamc@875
|
1350 string "} else {",
|
adamc@875
|
1351 box [case t of
|
adamc@875
|
1352 String => box []
|
adamc@875
|
1353 | _ =>
|
adamc@875
|
1354 box [string (p_sql_ctype t),
|
adamc@875
|
1355 space,
|
adamc@876
|
1356 string "tmp = *arg",
|
adamc@876
|
1357 string (Int.toString (i + 1)),
|
adamc@876
|
1358 string ";",
|
adamc@876
|
1359 newline,
|
adamc@876
|
1360 string (p_sql_ctype t),
|
adamc@876
|
1361 space,
|
adamc@875
|
1362 string "arg",
|
adamc@875
|
1363 string (Int.toString (i + 1)),
|
adamc@876
|
1364 string " = tmp;",
|
adamc@875
|
1365 newline],
|
adamc@875
|
1366 string "in_is_null",
|
adamc@875
|
1367 string (Int.toString i),
|
adamc@875
|
1368 string " = 0;",
|
adamc@875
|
1369 newline,
|
adamc@875
|
1370 buffers t,
|
adamc@876
|
1371 newline],
|
adamc@876
|
1372 string "}",
|
adamc@876
|
1373 newline]
|
adamc@875
|
1374
|
adamc@875
|
1375 | _ => buffers t,
|
adamc@875
|
1376 newline]
|
adamc@875
|
1377 end) inputs,
|
adamc@875
|
1378 newline,
|
adamc@875
|
1379
|
adamc@875
|
1380 string "if (mysql_stmt_bind_param(stmt, in)) uw_error(ctx, FATAL, \"",
|
adamc@875
|
1381 string (ErrorMsg.spanToString loc),
|
adamc@875
|
1382 string ": error binding parameters\");",
|
adamc@875
|
1383 newline,
|
adamc@875
|
1384
|
adamc@875
|
1385 dmlCommon {loc = loc, dml = box [string "\"",
|
adamc@875
|
1386 string (String.toString dml),
|
adamc@875
|
1387 string "\""]}]
|
adamc@875
|
1388
|
adamc@878
|
1389 fun nextval {loc, seqE, seqName} =
|
adamc@878
|
1390 box [string "uw_conn *conn = uw_get_db(ctx);",
|
adamc@878
|
1391 newline,
|
adamc@878
|
1392 string "char *insert = ",
|
adamc@878
|
1393 case seqName of
|
adamc@878
|
1394 SOME s => string ("\"INSERT INTO " ^ s ^ " VALUES ()\"")
|
adamc@878
|
1395 | NONE => box [string "uw_Basis_strcat(ctx, \"INSERT INTO \", uw_Basis_strcat(ctx, ",
|
adamc@878
|
1396 seqE,
|
adamc@878
|
1397 string ", \" VALUES ()\"))"],
|
adamc@878
|
1398 string ";",
|
adamc@878
|
1399 newline,
|
adamc@878
|
1400 string "char *delete = ",
|
adamc@878
|
1401 case seqName of
|
adamc@878
|
1402 SOME s => string ("\"DELETE FROM " ^ s ^ "\"")
|
adamc@878
|
1403 | NONE => box [string "uw_Basis_strcat(ctx, \"DELETE FROM \", ",
|
adamc@878
|
1404 seqE,
|
adamc@878
|
1405 string ")"],
|
adamc@878
|
1406 string ";",
|
adamc@878
|
1407 newline,
|
adamc@878
|
1408 newline,
|
adamc@878
|
1409
|
adamc@878
|
1410 string "if (mysql_query(conn->conn, insert)) uw_error(ctx, FATAL, \"'nextval' INSERT failed\");",
|
adamc@878
|
1411 newline,
|
adamc@878
|
1412 string "n = mysql_insert_id(conn->conn);",
|
adamc@878
|
1413 newline,
|
adamc@878
|
1414 string "if (mysql_query(conn->conn, delete)) uw_error(ctx, FATAL, \"'nextval' DELETE failed\");",
|
adamc@878
|
1415 newline]
|
adamc@878
|
1416
|
adamc@878
|
1417 fun nextvalPrepared _ = raise Fail "MySQL.nextvalPrepared called"
|
adamc@867
|
1418
|
adamc@877
|
1419 fun sqlifyString s = "'" ^ String.translate (fn #"'" => "\\'"
|
adamc@877
|
1420 | #"\\" => "\\\\"
|
adamc@877
|
1421 | ch =>
|
adamc@877
|
1422 if Char.isPrint ch then
|
adamc@877
|
1423 str ch
|
adamc@877
|
1424 else
|
adamc@877
|
1425 (ErrorMsg.error
|
adamc@877
|
1426 "Non-printing character found in SQL string literal";
|
adamc@877
|
1427 ""))
|
adamc@877
|
1428 (String.toString s) ^ "'"
|
adamc@874
|
1429
|
adamc@877
|
1430 fun p_cast (s, _) = s
|
adamc@874
|
1431
|
adamc@874
|
1432 fun p_blank _ = "?"
|
adamc@874
|
1433
|
adamc@866
|
1434 val () = addDbms {name = "mysql",
|
adamc@866
|
1435 header = "mysql/mysql.h",
|
adamc@866
|
1436 link = "-lmysqlclient",
|
adamc@867
|
1437 init = init,
|
adamc@873
|
1438 p_sql_type = p_sql_type,
|
adamc@867
|
1439 query = query,
|
adamc@868
|
1440 queryPrepared = queryPrepared,
|
adamc@868
|
1441 dml = dml,
|
adamc@869
|
1442 dmlPrepared = dmlPrepared,
|
adamc@869
|
1443 nextval = nextval,
|
adamc@874
|
1444 nextvalPrepared = nextvalPrepared,
|
adamc@874
|
1445 sqlifyString = sqlifyString,
|
adamc@874
|
1446 p_cast = p_cast,
|
adamc@874
|
1447 p_blank = p_blank,
|
adamc@877
|
1448 supportsDeleteAs = false,
|
adamc@886
|
1449 supportsUpdateAs = false,
|
adamc@884
|
1450 createSequence = fn s => "CREATE TABLE " ^ s ^ " (uw_id INTEGER PRIMARY KEY AUTO_INCREMENT)",
|
adamc@878
|
1451 textKeysNeedLengths = true,
|
adamc@879
|
1452 supportsNextval = false,
|
adamc@882
|
1453 supportsNestedPrepared = false,
|
adamc@885
|
1454 sqlPrefix = "SET storage_engine=InnoDB;\n\n"}
|
adamc@866
|
1455
|
adamc@866
|
1456 end
|