adam@1295
|
1 (* Copyright (c) 2009-2010, 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
|
adam@1682
|
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@1011
|
39 | Char => "char"
|
adamc@873
|
40 | Bool => "bool"
|
adamc@873
|
41 | Time => "timestamp"
|
adamc@873
|
42 | Blob => "longblob"
|
adamc@873
|
43 | Channel => "bigint"
|
adamc@873
|
44 | Client => "int"
|
adamc@873
|
45 | Nullable t => p_sql_type t
|
adamc@873
|
46
|
adamc@873
|
47 fun p_buffer_type t =
|
adamc@873
|
48 case t of
|
adamc@873
|
49 Int => "MYSQL_TYPE_LONGLONG"
|
adamc@873
|
50 | Float => "MYSQL_TYPE_DOUBLE"
|
adamc@873
|
51 | String => "MYSQL_TYPE_STRING"
|
adamc@1013
|
52 | Char => "MYSQL_TYPE_STRING"
|
adamc@873
|
53 | Bool => "MYSQL_TYPE_LONG"
|
adamc@938
|
54 | Time => "MYSQL_TYPE_TIMESTAMP"
|
adamc@873
|
55 | Blob => "MYSQL_TYPE_BLOB"
|
adamc@873
|
56 | Channel => "MYSQL_TYPE_LONGLONG"
|
adamc@873
|
57 | Client => "MYSQL_TYPE_LONG"
|
adamc@873
|
58 | Nullable t => p_buffer_type t
|
adamc@873
|
59
|
adamc@874
|
60 fun p_sql_type_base t =
|
adamc@874
|
61 case t of
|
adamc@874
|
62 Int => "bigint"
|
adamc@874
|
63 | Float => "double"
|
adamc@874
|
64 | String => "longtext"
|
adamc@1011
|
65 | Char => "char"
|
adamc@874
|
66 | Bool => "tinyint"
|
adamc@874
|
67 | Time => "timestamp"
|
adamc@874
|
68 | Blob => "longblob"
|
adamc@874
|
69 | Channel => "bigint"
|
adamc@874
|
70 | Client => "int"
|
adamc@874
|
71 | Nullable t => p_sql_type_base t
|
adamc@874
|
72
|
adamc@874
|
73 val ident = String.translate (fn #"'" => "PRIME"
|
adamc@874
|
74 | ch => str ch)
|
adamc@874
|
75
|
adamc@874
|
76 fun checkRel (table, checkNullable) (s, xts) =
|
adamc@874
|
77 let
|
adamc@874
|
78 val sl = CharVector.map Char.toLower s
|
adam@1953
|
79 val sl = if size sl > 1 andalso String.sub (sl, 0) = #"\"" then
|
adam@1953
|
80 String.substring (sl, 1, size sl - 2)
|
adam@1953
|
81 else
|
adam@1953
|
82 sl
|
adam@1953
|
83 val both = "LOWER(table_name) = ('" ^ sl ^ "')"
|
adamc@874
|
84
|
adamc@998
|
85 val q = "SELECT COUNT(*) FROM information_schema." ^ table ^ " WHERE " ^ both
|
adamc@874
|
86
|
adamc@998
|
87 val q' = String.concat ["SELECT COUNT(*) FROM information_schema.columns WHERE ",
|
adamc@998
|
88 both,
|
adamc@998
|
89 " AND (",
|
adam@1600
|
90 case String.concatWith " OR "
|
adam@1600
|
91 (map (fn (x, t) =>
|
adam@1953
|
92 String.concat ["(LOWER(column_name) = '",
|
adam@1953
|
93 Settings.mangleSqlCatalog
|
adam@1953
|
94 (CharVector.map
|
adam@1953
|
95 Char.toLower (ident x)),
|
adam@1953
|
96 "' AND data_type ",
|
adam@1953
|
97 case p_sql_type_base t of
|
adam@1953
|
98 "bigint" =>
|
adam@1953
|
99 "IN ('bigint', 'int')"
|
adam@1953
|
100 | "longtext" =>
|
adam@1953
|
101 "IN ('longtext', 'varchar')"
|
adam@1953
|
102 | s => "= '" ^ s ^ "'",
|
adam@1600
|
103 if checkNullable then
|
adam@1600
|
104 (" AND is_nullable = '"
|
adam@1600
|
105 ^ (if isNotNull t then
|
adam@1600
|
106 "NO"
|
adam@1600
|
107 else
|
adam@1600
|
108 "YES")
|
adam@1600
|
109 ^ "'")
|
adam@1600
|
110 else
|
adam@1600
|
111 "",
|
adam@1600
|
112 ")"]) xts) of
|
adam@1600
|
113 "" => "FALSE"
|
adam@1600
|
114 | s => s,
|
adamc@874
|
115 ")"]
|
adamc@874
|
116
|
adamc@998
|
117 val q'' = String.concat ["SELECT COUNT(*) FROM information_schema.columns WHERE ",
|
adamc@998
|
118 both,
|
adam@1953
|
119 " AND LOWER(column_name) LIKE '", Settings.mangleSqlCatalog "%'"]
|
adamc@874
|
120 in
|
adamc@874
|
121 box [string "if (mysql_query(conn->conn, \"",
|
adamc@874
|
122 string q,
|
adamc@874
|
123 string "\")) {",
|
adamc@874
|
124 newline,
|
adamc@874
|
125 box [string "mysql_close(conn->conn);",
|
adamc@874
|
126 newline,
|
adamc@874
|
127 string "uw_error(ctx, FATAL, \"Query 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 ((res = mysql_store_result(conn->conn)) == NULL) {",
|
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, \"Result store failed:\\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 (mysql_num_fields(res) != 1) {",
|
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, \"Bad column count:\\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 ((row = mysql_fetch_row(res)) == NULL) {",
|
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, \"Row fetch failed:\\n",
|
adamc@874
|
170 string q,
|
adamc@874
|
171 string "\");",
|
adamc@874
|
172 newline],
|
adamc@874
|
173 string "}",
|
adamc@874
|
174 newline,
|
adamc@874
|
175 newline,
|
adamc@874
|
176
|
adamc@874
|
177 string "if (strcmp(row[0], \"1\")) {",
|
adamc@874
|
178 newline,
|
adamc@874
|
179 box [string "mysql_free_result(res);",
|
adamc@874
|
180 newline,
|
adamc@874
|
181 string "mysql_close(conn->conn);",
|
adamc@874
|
182 newline,
|
adamc@874
|
183 string "uw_error(ctx, FATAL, \"Table '",
|
adam@1953
|
184 string sl,
|
adamc@874
|
185 string "' does not exist.\");",
|
adamc@874
|
186 newline],
|
adamc@874
|
187 string "}",
|
adamc@874
|
188 newline,
|
adamc@874
|
189 newline,
|
adamc@874
|
190 string "mysql_free_result(res);",
|
adamc@874
|
191 newline,
|
adamc@874
|
192 newline,
|
adamc@874
|
193
|
adamc@874
|
194 string "if (mysql_query(conn->conn, \"",
|
adamc@874
|
195 string q',
|
adamc@874
|
196 string "\")) {",
|
adamc@874
|
197 newline,
|
adamc@874
|
198 box [string "mysql_close(conn->conn);",
|
adamc@874
|
199 newline,
|
adamc@874
|
200 string "uw_error(ctx, FATAL, \"Query 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 ((res = mysql_store_result(conn->conn)) == NULL) {",
|
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, \"Result store failed:\\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 (mysql_num_fields(res) != 1) {",
|
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, \"Bad column count:\\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 ((row = mysql_fetch_row(res)) == NULL) {",
|
adamc@874
|
237 newline,
|
adamc@874
|
238 box [string "mysql_free_result(res);",
|
adamc@874
|
239 newline,
|
adamc@874
|
240 string "mysql_close(conn->conn);",
|
adamc@874
|
241 newline,
|
adamc@874
|
242 string "uw_error(ctx, FATAL, \"Row fetch failed:\\n",
|
adamc@874
|
243 string q',
|
adamc@874
|
244 string "\");",
|
adamc@874
|
245 newline],
|
adamc@874
|
246 string "}",
|
adamc@874
|
247 newline,
|
adamc@874
|
248 newline,
|
adamc@874
|
249
|
adamc@874
|
250 string "if (strcmp(row[0], \"",
|
adamc@874
|
251 string (Int.toString (length xts)),
|
adamc@874
|
252 string "\")) {",
|
adamc@874
|
253 newline,
|
adamc@874
|
254 box [string "mysql_free_result(res);",
|
adamc@874
|
255 newline,
|
adamc@874
|
256 string "mysql_close(conn->conn);",
|
adamc@874
|
257 newline,
|
adamc@874
|
258 string "uw_error(ctx, FATAL, \"Table '",
|
adam@1953
|
259 string sl,
|
adamc@874
|
260 string "' has the wrong column types.\");",
|
adamc@874
|
261 newline],
|
adamc@874
|
262 string "}",
|
adamc@874
|
263 newline,
|
adamc@874
|
264 newline,
|
adamc@874
|
265 string "mysql_free_result(res);",
|
adamc@874
|
266 newline,
|
adamc@874
|
267 newline,
|
adam@1682
|
268
|
adamc@874
|
269 string "if (mysql_query(conn->conn, \"",
|
adamc@874
|
270 string q'',
|
adamc@874
|
271 string "\")) {",
|
adamc@874
|
272 newline,
|
adamc@874
|
273 box [string "mysql_close(conn->conn);",
|
adamc@874
|
274 newline,
|
adamc@874
|
275 string "uw_error(ctx, FATAL, \"Query 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 ((res = mysql_store_result(conn->conn)) == NULL) {",
|
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, \"Result store failed:\\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 (mysql_num_fields(res) != 1) {",
|
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, \"Bad column count:\\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 ((row = mysql_fetch_row(res)) == NULL) {",
|
adamc@874
|
312 newline,
|
adamc@874
|
313 box [string "mysql_free_result(res);",
|
adamc@874
|
314 newline,
|
adamc@874
|
315 string "mysql_close(conn->conn);",
|
adamc@874
|
316 newline,
|
adamc@874
|
317 string "uw_error(ctx, FATAL, \"Row fetch failed:\\n",
|
adamc@874
|
318 string q'',
|
adamc@874
|
319 string "\");",
|
adamc@874
|
320 newline],
|
adamc@874
|
321 string "}",
|
adamc@874
|
322 newline,
|
adamc@874
|
323 newline,
|
adamc@874
|
324
|
adamc@874
|
325 string "if (strcmp(row[0], \"",
|
adamc@874
|
326 string (Int.toString (length xts)),
|
adamc@874
|
327 string "\")) {",
|
adamc@874
|
328 newline,
|
adamc@874
|
329 box [string "mysql_free_result(res);",
|
adamc@874
|
330 newline,
|
adamc@874
|
331 string "mysql_close(conn->conn);",
|
adamc@874
|
332 newline,
|
adamc@874
|
333 string "uw_error(ctx, FATAL, \"Table '",
|
adam@1953
|
334 string sl,
|
adamc@874
|
335 string "' has extra columns.\");",
|
adamc@874
|
336 newline],
|
adamc@874
|
337 string "}",
|
adamc@874
|
338 newline,
|
adamc@874
|
339 newline,
|
adamc@874
|
340 string "mysql_free_result(res);",
|
adamc@874
|
341 newline]
|
adamc@874
|
342 end
|
adamc@874
|
343
|
adamc@872
|
344 fun init {dbstring, prepared = ss, tables, views, sequences} =
|
adamc@866
|
345 let
|
adamc@866
|
346 val host = ref NONE
|
adamc@866
|
347 val user = ref NONE
|
adamc@866
|
348 val passwd = ref NONE
|
adamc@866
|
349 val db = ref NONE
|
adamc@866
|
350 val port = ref NONE
|
adamc@866
|
351 val unix_socket = ref NONE
|
adamc@866
|
352
|
adamc@866
|
353 fun stringOf r = case !r of
|
adamc@866
|
354 NONE => string "NULL"
|
adamc@866
|
355 | SOME s => box [string "\"",
|
adam@1656
|
356 string (Prim.toCString s),
|
adamc@866
|
357 string "\""]
|
adamc@866
|
358 in
|
adamc@866
|
359 app (fn s =>
|
adamc@866
|
360 case String.fields (fn ch => ch = #"=") s of
|
adamc@866
|
361 [name, value] =>
|
adamc@866
|
362 (case name of
|
adamc@866
|
363 "host" =>
|
adamc@866
|
364 if size value > 0 andalso String.sub (value, 0) = #"/" then
|
adamc@866
|
365 unix_socket := SOME value
|
adamc@866
|
366 else
|
adamc@866
|
367 host := SOME value
|
adamc@866
|
368 | "hostaddr" => host := SOME value
|
adamc@866
|
369 | "port" => port := Int.fromString value
|
adamc@866
|
370 | "dbname" => db := SOME value
|
adamc@866
|
371 | "user" => user := SOME value
|
adamc@866
|
372 | "password" => passwd := SOME value
|
adamc@866
|
373 | _ => ())
|
adamc@866
|
374 | _ => ()) (String.tokens Char.isSpace dbstring);
|
adamc@866
|
375
|
adamc@866
|
376 box [string "typedef struct {",
|
adamc@866
|
377 newline,
|
adamc@866
|
378 box [string "MYSQL *conn;",
|
adamc@866
|
379 newline,
|
adamc@866
|
380 p_list_sepi (box [])
|
adamc@866
|
381 (fn i => fn _ =>
|
adamc@866
|
382 box [string "MYSQL_STMT *p",
|
adamc@866
|
383 string (Int.toString i),
|
adamc@866
|
384 string ";",
|
adamc@866
|
385 newline])
|
adamc@866
|
386 ss],
|
adamc@866
|
387 string "} uw_conn;",
|
adamc@866
|
388 newline,
|
adamc@866
|
389 newline,
|
adamc@866
|
390
|
adamc@1094
|
391 string "static void uw_client_init(void) {",
|
adamc@874
|
392 newline,
|
adamc@879
|
393 box [string "uw_sqlfmtInt = \"%lld%n\";",
|
adamc@879
|
394 newline,
|
adam@1920
|
395 string "uw_sqlfmtFloat = \"%.16g%n\";",
|
adamc@879
|
396 newline,
|
adamc@879
|
397 string "uw_Estrings = 0;",
|
adamc@879
|
398 newline,
|
adam@1834
|
399 string "uw_sql_type_annotations = 0;",
|
adam@1834
|
400 newline,
|
adamc@879
|
401 string "uw_sqlsuffixString = \"\";",
|
adamc@879
|
402 newline,
|
adamc@1011
|
403 string "uw_sqlsuffixChar = \"\";",
|
adamc@1011
|
404 newline,
|
adamc@879
|
405 string "uw_sqlsuffixBlob = \"\";",
|
adamc@879
|
406 newline,
|
adamc@879
|
407 string "uw_sqlfmtUint4 = \"%u%n\";",
|
adamc@879
|
408 newline,
|
adamc@879
|
409 newline,
|
adamc@879
|
410
|
adamc@879
|
411 string "if (mysql_library_init(0, NULL, NULL)) {",
|
adamc@874
|
412 newline,
|
adamc@874
|
413 box [string "fprintf(stderr, \"Could not initialize MySQL library\\n\");",
|
adamc@874
|
414 newline,
|
adamc@874
|
415 string "exit(1);",
|
adamc@874
|
416 newline],
|
adamc@874
|
417 string "}",
|
adamc@874
|
418 newline],
|
adamc@874
|
419 string "}",
|
adamc@874
|
420 newline,
|
adamc@874
|
421 newline,
|
adamc@874
|
422
|
adamc@866
|
423 if #persistent (currentProtocol ()) then
|
adamc@874
|
424 box [string "static void uw_db_validate(uw_context ctx) {",
|
adamc@874
|
425 newline,
|
adamc@874
|
426 string "uw_conn *conn = uw_get_db(ctx);",
|
adamc@874
|
427 newline,
|
adamc@874
|
428 string "MYSQL_RES *res;",
|
adamc@874
|
429 newline,
|
adamc@874
|
430 string "MYSQL_ROW row;",
|
adamc@874
|
431 newline,
|
adamc@874
|
432 newline,
|
adamc@874
|
433 p_list_sep newline (checkRel ("tables", true)) tables,
|
adamc@884
|
434 p_list_sep newline (fn name => checkRel ("tables", true)
|
adamc@884
|
435 (name, [("id", Settings.Client)])) sequences,
|
adamc@874
|
436 p_list_sep newline (checkRel ("views", false)) views,
|
adamc@874
|
437 string "}",
|
adamc@874
|
438 newline,
|
adamc@874
|
439 newline,
|
adamc@874
|
440
|
adamc@874
|
441 string "static void uw_db_prepare(uw_context ctx) {",
|
adamc@866
|
442 newline,
|
adamc@866
|
443 string "uw_conn *conn = uw_get_db(ctx);",
|
adamc@866
|
444 newline,
|
adamc@866
|
445 string "MYSQL_STMT *stmt;",
|
adamc@866
|
446 newline,
|
adamc@866
|
447 newline,
|
adamc@866
|
448
|
adamc@866
|
449 p_list_sepi newline (fn i => fn (s, n) =>
|
adamc@866
|
450 let
|
adamc@866
|
451 fun uhoh this s args =
|
adamc@866
|
452 box [p_list_sepi (box [])
|
adamc@866
|
453 (fn j => fn () =>
|
adamc@866
|
454 box [string
|
adamc@866
|
455 "mysql_stmt_close(conn->p",
|
adamc@866
|
456 string (Int.toString j),
|
adamc@866
|
457 string ");",
|
adamc@866
|
458 newline])
|
adamc@866
|
459 (List.tabulate (i, fn _ => ())),
|
adamc@866
|
460 box (if this then
|
adamc@866
|
461 [string
|
adamc@866
|
462 "mysql_stmt_close(conn->p",
|
adamc@866
|
463 string (Int.toString i),
|
adamc@866
|
464 string ");",
|
adamc@866
|
465 newline]
|
adamc@866
|
466 else
|
adamc@866
|
467 []),
|
adamc@866
|
468 string "mysql_close(conn->conn);",
|
adamc@866
|
469 newline,
|
adamc@866
|
470 string "uw_error(ctx, FATAL, \"",
|
adamc@866
|
471 string s,
|
adamc@866
|
472 string "\"",
|
adamc@866
|
473 p_list_sep (box []) (fn s => box [string ", ",
|
adamc@866
|
474 string s]) args,
|
adamc@866
|
475 string ");",
|
adamc@866
|
476 newline]
|
adamc@866
|
477 in
|
adamc@866
|
478 box [string "stmt = mysql_stmt_init(conn->conn);",
|
adamc@866
|
479 newline,
|
adamc@866
|
480 string "if (stmt == NULL) {",
|
adamc@866
|
481 newline,
|
adamc@866
|
482 uhoh false "Out of memory allocating prepared statement" [],
|
adamc@866
|
483 string "}",
|
adamc@866
|
484 newline,
|
adamc@874
|
485 string "conn->p",
|
adamc@874
|
486 string (Int.toString i),
|
adamc@874
|
487 string " = stmt;",
|
adamc@874
|
488 newline,
|
adamc@866
|
489
|
adamc@866
|
490 string "if (mysql_stmt_prepare(stmt, \"",
|
adam@1656
|
491 string (Prim.toCString s),
|
adamc@866
|
492 string "\", ",
|
adamc@866
|
493 string (Int.toString (size s)),
|
adamc@866
|
494 string ")) {",
|
adamc@866
|
495 newline,
|
adamc@866
|
496 box [string "char msg[1024];",
|
adamc@866
|
497 newline,
|
adamc@866
|
498 string "strncpy(msg, mysql_stmt_error(stmt), 1024);",
|
adamc@866
|
499 newline,
|
adamc@866
|
500 string "msg[1023] = 0;",
|
adamc@866
|
501 newline,
|
adamc@866
|
502 uhoh true "Error preparing statement: %s" ["msg"]],
|
adamc@866
|
503 string "}",
|
adamc@866
|
504 newline]
|
adamc@866
|
505 end)
|
adamc@866
|
506 ss,
|
adamc@866
|
507
|
adamc@866
|
508 string "}"]
|
adamc@866
|
509 else
|
adamc@882
|
510 box [string "static void uw_db_prepare(uw_context ctx) { }",
|
adamc@882
|
511 newline,
|
adamc@882
|
512 string "static void uw_db_validate(uw_context ctx) { }"],
|
adamc@866
|
513 newline,
|
adamc@866
|
514 newline,
|
adam@1682
|
515
|
adamc@1094
|
516 string "static void uw_db_init(uw_context ctx) {",
|
adamc@866
|
517 newline,
|
adamc@866
|
518 string "MYSQL *mysql = mysql_init(NULL);",
|
adamc@866
|
519 newline,
|
adamc@866
|
520 string "uw_conn *conn;",
|
adamc@866
|
521 newline,
|
adamc@866
|
522 string "if (mysql == NULL) uw_error(ctx, FATAL, ",
|
adamc@866
|
523 string "\"libmysqlclient can't allocate a connection.\");",
|
adamc@866
|
524 newline,
|
adamc@866
|
525 string "if (mysql_real_connect(mysql, ",
|
adamc@866
|
526 stringOf host,
|
adamc@866
|
527 string ", ",
|
adamc@866
|
528 stringOf user,
|
adamc@866
|
529 string ", ",
|
adamc@866
|
530 stringOf passwd,
|
adamc@866
|
531 string ", ",
|
adamc@866
|
532 stringOf db,
|
adamc@866
|
533 string ", ",
|
adamc@866
|
534 case !port of
|
adamc@866
|
535 NONE => string "0"
|
adamc@866
|
536 | SOME n => string (Int.toString n),
|
adamc@866
|
537 string ", ",
|
adamc@866
|
538 stringOf unix_socket,
|
adam@1955
|
539 string ", CLIENT_MULTI_STATEMENTS) == NULL) {",
|
adamc@866
|
540 newline,
|
adamc@866
|
541 box [string "char msg[1024];",
|
adamc@866
|
542 newline,
|
adamc@866
|
543 string "strncpy(msg, mysql_error(mysql), 1024);",
|
adamc@866
|
544 newline,
|
adamc@866
|
545 string "msg[1023] = 0;",
|
adamc@866
|
546 newline,
|
adamc@866
|
547 string "mysql_close(mysql);",
|
adamc@866
|
548 newline,
|
adamc@866
|
549 string "uw_error(ctx, BOUNDED_RETRY, ",
|
adamc@866
|
550 string "\"Connection to MySQL server failed: %s\", msg);"],
|
adamc@866
|
551 newline,
|
adamc@866
|
552 string "}",
|
adamc@866
|
553 newline,
|
adamc@874
|
554 string "conn = calloc(1, sizeof(uw_conn));",
|
adamc@866
|
555 newline,
|
adamc@866
|
556 string "conn->conn = mysql;",
|
adamc@866
|
557 newline,
|
adamc@866
|
558 string "uw_set_db(ctx, conn);",
|
adamc@866
|
559 newline,
|
adamc@866
|
560 string "uw_db_validate(ctx);",
|
adamc@866
|
561 newline,
|
adamc@866
|
562 string "uw_db_prepare(ctx);",
|
adamc@866
|
563 newline,
|
adamc@866
|
564 string "}",
|
adamc@866
|
565 newline,
|
adamc@866
|
566 newline,
|
adamc@866
|
567
|
adamc@1094
|
568 string "static void uw_db_close(uw_context ctx) {",
|
adamc@866
|
569 newline,
|
adamc@866
|
570 string "uw_conn *conn = uw_get_db(ctx);",
|
adamc@866
|
571 newline,
|
adamc@866
|
572 p_list_sepi (box [])
|
adamc@866
|
573 (fn i => fn _ =>
|
adamc@866
|
574 box [string "if (conn->p",
|
adamc@866
|
575 string (Int.toString i),
|
adamc@866
|
576 string ") mysql_stmt_close(conn->p",
|
adamc@866
|
577 string (Int.toString i),
|
adamc@866
|
578 string ");",
|
adamc@866
|
579 newline])
|
adamc@866
|
580 ss,
|
adamc@866
|
581 string "mysql_close(conn->conn);",
|
adamc@866
|
582 newline,
|
adamc@866
|
583 string "}",
|
adamc@866
|
584 newline,
|
adamc@866
|
585 newline,
|
adamc@866
|
586
|
adam@1936
|
587 string "static int uw_db_begin(uw_context ctx, int could_write) {",
|
adamc@866
|
588 newline,
|
adamc@866
|
589 string "uw_conn *conn = uw_get_db(ctx);",
|
adamc@866
|
590 newline,
|
adamc@866
|
591 newline,
|
adam@1955
|
592 string "return mysql_query(conn->conn, \"SET TRANSACTION ISOLATION LEVEL SERIALIZABLE; BEGIN\") ? 1 : (mysql_next_result(conn->conn), 0);",
|
adamc@866
|
593 newline,
|
adamc@866
|
594 string "}",
|
adamc@866
|
595 newline,
|
adamc@866
|
596 newline,
|
adamc@866
|
597
|
adamc@1094
|
598 string "static int uw_db_commit(uw_context ctx) {",
|
adamc@866
|
599 newline,
|
adamc@866
|
600 string "uw_conn *conn = uw_get_db(ctx);",
|
adamc@866
|
601 newline,
|
adamc@866
|
602 string "return mysql_commit(conn->conn);",
|
adamc@866
|
603 newline,
|
adamc@866
|
604 string "}",
|
adamc@866
|
605 newline,
|
adamc@866
|
606 newline,
|
adamc@866
|
607
|
adamc@1094
|
608 string "static int uw_db_rollback(uw_context ctx) {",
|
adamc@866
|
609 newline,
|
adamc@866
|
610 string "uw_conn *conn = uw_get_db(ctx);",
|
adamc@866
|
611 newline,
|
adamc@866
|
612 string "return mysql_rollback(conn->conn);",
|
adamc@866
|
613 newline,
|
adamc@866
|
614 string "}",
|
adamc@866
|
615 newline,
|
adamc@866
|
616 newline]
|
adamc@866
|
617 end
|
adamc@866
|
618
|
adamc@880
|
619 fun p_getcol {loc, wontLeakStrings = _, col = i, typ = t} =
|
adamc@873
|
620 let
|
adamc@873
|
621 fun getter t =
|
adamc@873
|
622 case t of
|
adamc@873
|
623 String => box [string "({",
|
adamc@873
|
624 newline,
|
adamc@873
|
625 string "uw_Basis_string s = uw_malloc(ctx, length",
|
adamc@873
|
626 string (Int.toString i),
|
adamc@873
|
627 string " + 1);",
|
adamc@873
|
628 newline,
|
adamc@873
|
629 string "out[",
|
adamc@873
|
630 string (Int.toString i),
|
adamc@873
|
631 string "].buffer = s;",
|
adamc@873
|
632 newline,
|
adamc@873
|
633 string "out[",
|
adamc@873
|
634 string (Int.toString i),
|
adamc@873
|
635 string "].buffer_length = length",
|
adamc@873
|
636 string (Int.toString i),
|
adamc@873
|
637 string " + 1;",
|
adamc@873
|
638 newline,
|
adamc@873
|
639 string "mysql_stmt_fetch_column(stmt, &out[",
|
adamc@873
|
640 string (Int.toString i),
|
adamc@873
|
641 string "], ",
|
adamc@873
|
642 string (Int.toString i),
|
adamc@873
|
643 string ", 0);",
|
adamc@873
|
644 newline,
|
adamc@873
|
645 string "s[length",
|
adamc@873
|
646 string (Int.toString i),
|
adamc@873
|
647 string "] = 0;",
|
adamc@873
|
648 newline,
|
adamc@873
|
649 string "s;",
|
adamc@873
|
650 newline,
|
adamc@873
|
651 string "})"]
|
adamc@873
|
652 | Blob => box [string "({",
|
adamc@873
|
653 newline,
|
adamc@873
|
654 string "uw_Basis_blob b = {length",
|
adamc@873
|
655 string (Int.toString i),
|
adamc@873
|
656 string ", uw_malloc(ctx, length",
|
adamc@873
|
657 string (Int.toString i),
|
adamc@873
|
658 string ")};",
|
adamc@873
|
659 newline,
|
adamc@873
|
660 string "out[",
|
adamc@873
|
661 string (Int.toString i),
|
adamc@873
|
662 string "].buffer = b.data;",
|
adamc@873
|
663 newline,
|
adamc@873
|
664 string "out[",
|
adamc@873
|
665 string (Int.toString i),
|
adamc@873
|
666 string "].buffer_length = length",
|
adamc@873
|
667 string (Int.toString i),
|
adamc@873
|
668 string ";",
|
adamc@873
|
669 newline,
|
adamc@873
|
670 string "mysql_stmt_fetch_column(stmt, &out[",
|
adamc@873
|
671 string (Int.toString i),
|
adamc@873
|
672 string "], ",
|
adamc@873
|
673 string (Int.toString i),
|
adamc@873
|
674 string ", 0);",
|
adamc@873
|
675 newline,
|
adamc@873
|
676 string "b;",
|
adamc@873
|
677 newline,
|
adamc@873
|
678 string "})"]
|
adamc@873
|
679 | Time => box [string "({",
|
adamc@876
|
680 string "MYSQL_TIME *mt = &buffer",
|
adamc@873
|
681 string (Int.toString i),
|
adamc@873
|
682 string ";",
|
adamc@873
|
683 newline,
|
adamc@873
|
684 newline,
|
adamc@938
|
685 string "struct tm t = {mt->second, mt->minute, mt->hour, mt->day, mt->month-1, mt->year - 1900, 0, 0, -1};",
|
adamc@873
|
686 newline,
|
adam@1443
|
687 string "uw_Basis_time res = {mktime(&t), 0};",
|
adam@1443
|
688 newline,
|
adam@1443
|
689 string "res;",
|
adamc@873
|
690 newline,
|
adamc@873
|
691 string "})"]
|
adamc@889
|
692 | Channel => box [string "({",
|
adamc@889
|
693 string "uw_Basis_channel ch = {buffer",
|
adamc@889
|
694 string (Int.toString i),
|
adamc@889
|
695 string " >> 32, buffer",
|
adamc@889
|
696 string (Int.toString i),
|
adamc@889
|
697 string " & 0xFFFFFFFF};",
|
adamc@889
|
698 newline,
|
adamc@889
|
699 string "ch;",
|
adamc@889
|
700 newline,
|
adamc@889
|
701 string "})"]
|
adamc@873
|
702 | _ => box [string "buffer",
|
adamc@873
|
703 string (Int.toString i)]
|
adamc@873
|
704 in
|
adamc@873
|
705 case t of
|
adamc@873
|
706 Nullable t => box [string "(is_null",
|
adamc@873
|
707 string (Int.toString i),
|
adamc@873
|
708 string " ? NULL : ",
|
adamc@873
|
709 case t of
|
adamc@873
|
710 String => getter t
|
adamc@873
|
711 | _ => box [string "({",
|
adamc@873
|
712 newline,
|
adamc@873
|
713 string (p_sql_ctype t),
|
adamc@873
|
714 space,
|
adamc@873
|
715 string "*tmp = uw_malloc(ctx, sizeof(",
|
adamc@873
|
716 string (p_sql_ctype t),
|
adamc@873
|
717 string "));",
|
adamc@873
|
718 newline,
|
adamc@873
|
719 string "*tmp = ",
|
adamc@873
|
720 getter t,
|
adamc@873
|
721 string ";",
|
adamc@873
|
722 newline,
|
adamc@873
|
723 string "tmp;",
|
adamc@873
|
724 newline,
|
adamc@873
|
725 string "})"],
|
adamc@873
|
726 string ")"]
|
adamc@873
|
727 | _ => box [string "(is_null",
|
adamc@873
|
728 string (Int.toString i),
|
adamc@873
|
729 string " ? ",
|
adamc@873
|
730 box [string "({",
|
adamc@873
|
731 string (p_sql_ctype t),
|
adamc@873
|
732 space,
|
adamc@873
|
733 string "tmp;",
|
adamc@873
|
734 newline,
|
adamc@873
|
735 string "uw_error(ctx, FATAL, \"Unexpectedly NULL field #",
|
adamc@873
|
736 string (Int.toString i),
|
adamc@873
|
737 string "\");",
|
adamc@873
|
738 newline,
|
adamc@873
|
739 string "tmp;",
|
adamc@873
|
740 newline,
|
adamc@873
|
741 string "})"],
|
adamc@873
|
742 string " : ",
|
adamc@873
|
743 getter t,
|
adamc@873
|
744 string ")"]
|
adamc@873
|
745 end
|
adamc@873
|
746
|
adamc@873
|
747 fun queryCommon {loc, query, cols, doCols} =
|
adamc@873
|
748 box [string "int n, r;",
|
adamc@873
|
749 newline,
|
adamc@873
|
750 string "MYSQL_BIND out[",
|
adamc@873
|
751 string (Int.toString (length cols)),
|
adamc@873
|
752 string "];",
|
adamc@873
|
753 newline,
|
adamc@873
|
754 p_list_sepi (box []) (fn i => fn t =>
|
adamc@873
|
755 let
|
adamc@873
|
756 fun buffers t =
|
adamc@873
|
757 case t of
|
adamc@873
|
758 String => box [string "unsigned long length",
|
adamc@873
|
759 string (Int.toString i),
|
adamc@873
|
760 string ";",
|
adamc@873
|
761 newline]
|
adamc@873
|
762 | Blob => box [string "unsigned long length",
|
adamc@873
|
763 string (Int.toString i),
|
adamc@873
|
764 string ";",
|
adamc@873
|
765 newline]
|
adamc@876
|
766 | Time => box [string "MYSQL_TIME buffer",
|
adamc@876
|
767 string (Int.toString i),
|
adamc@876
|
768 string ";",
|
adamc@876
|
769 newline]
|
adamc@889
|
770 | Channel => box [string "unsigned long long buffer",
|
adamc@889
|
771 string (Int.toString i),
|
adamc@889
|
772 string ";",
|
adamc@889
|
773 newline]
|
adamc@873
|
774 | _ => box [string (p_sql_ctype t),
|
adamc@873
|
775 space,
|
adamc@873
|
776 string "buffer",
|
adamc@873
|
777 string (Int.toString i),
|
adamc@873
|
778 string ";",
|
adamc@873
|
779 newline]
|
adamc@873
|
780 in
|
adamc@873
|
781 box [string "my_bool is_null",
|
adamc@873
|
782 string (Int.toString i),
|
adamc@873
|
783 string ";",
|
adamc@873
|
784 newline,
|
adamc@873
|
785 case t of
|
adamc@873
|
786 Nullable t => buffers t
|
adamc@873
|
787 | _ => buffers t,
|
adamc@873
|
788 newline]
|
adamc@873
|
789 end) cols,
|
adamc@873
|
790 newline,
|
adamc@873
|
791
|
adamc@873
|
792 string "memset(out, 0, sizeof out);",
|
adamc@873
|
793 newline,
|
adamc@873
|
794 p_list_sepi (box []) (fn i => fn t =>
|
adamc@873
|
795 let
|
adamc@873
|
796 fun buffers t =
|
adamc@873
|
797 case t of
|
adamc@875
|
798 String => box [string "out[",
|
adamc@875
|
799 string (Int.toString i),
|
adamc@875
|
800 string "].length = &length",
|
adamc@875
|
801 string (Int.toString i),
|
adamc@875
|
802 string ";",
|
adamc@875
|
803 newline]
|
adamc@1013
|
804 | Char => box [string "out[",
|
adamc@1013
|
805 string (Int.toString i),
|
adamc@1013
|
806 string "].buffer_length = 1;",
|
adamc@1013
|
807 newline,
|
adamc@1013
|
808 string "out[",
|
adamc@1013
|
809 string (Int.toString i),
|
adamc@1013
|
810 string "].buffer = &buffer",
|
adamc@1013
|
811 string (Int.toString i),
|
adamc@1013
|
812 string ";",
|
adamc@1013
|
813 newline]
|
adamc@875
|
814 | Blob => box [string "out[",
|
adamc@875
|
815 string (Int.toString i),
|
adamc@875
|
816 string "].length = &length",
|
adamc@875
|
817 string (Int.toString i),
|
adamc@875
|
818 string ";",
|
adamc@875
|
819 newline]
|
adamc@873
|
820 | _ => box [string "out[",
|
adamc@873
|
821 string (Int.toString i),
|
adamc@873
|
822 string "].buffer = &buffer",
|
adamc@873
|
823 string (Int.toString i),
|
adamc@873
|
824 string ";",
|
adamc@873
|
825 newline]
|
adamc@873
|
826 in
|
adamc@873
|
827 box [string "out[",
|
adamc@873
|
828 string (Int.toString i),
|
adamc@873
|
829 string "].buffer_type = ",
|
adamc@873
|
830 string (p_buffer_type t),
|
adamc@873
|
831 string ";",
|
adamc@873
|
832 newline,
|
adamc@873
|
833 string "out[",
|
adamc@873
|
834 string (Int.toString i),
|
adamc@873
|
835 string "].is_null = &is_null",
|
adamc@873
|
836 string (Int.toString i),
|
adamc@873
|
837 string ";",
|
adamc@873
|
838 newline,
|
adam@1682
|
839
|
adamc@873
|
840 case t of
|
adamc@873
|
841 Nullable t => buffers t
|
adamc@873
|
842 | _ => buffers t,
|
adamc@873
|
843 newline]
|
adamc@873
|
844 end) cols,
|
adamc@873
|
845 newline,
|
adamc@873
|
846
|
adamc@875
|
847 string "if (mysql_stmt_reset(stmt)) uw_error(ctx, FATAL, \"",
|
adamc@875
|
848 string (ErrorMsg.spanToString loc),
|
adamc@875
|
849 string ": Error reseting statement: %s\\n%s\", ",
|
adamc@875
|
850 query,
|
adamc@875
|
851 string ", mysql_error(conn->conn));",
|
adamc@875
|
852 newline,
|
adamc@875
|
853 newline,
|
adamc@875
|
854
|
adam@1954
|
855 string "if (mysql_stmt_execute(stmt)) {",
|
adam@1954
|
856 newline,
|
adam@1954
|
857 box [string "if (mysql_errno(conn->conn) == 1213)",
|
adam@1954
|
858 newline,
|
adam@1954
|
859 box [string "uw_error(ctx, UNLIMITED_RETRY, \"Deadlock detected\");",
|
adam@1954
|
860 newline],
|
adam@1954
|
861 newline,
|
adam@1954
|
862 string "uw_error(ctx, FATAL, \"",
|
adam@1954
|
863 string (ErrorMsg.spanToString loc),
|
adam@1954
|
864 string ": Error executing query: %s\\n%s\", ",
|
adam@1954
|
865 query,
|
adam@1954
|
866 string ", mysql_error(conn->conn));",
|
adam@1954
|
867 newline],
|
adam@1954
|
868 string "}",
|
adamc@875
|
869 newline,
|
adamc@875
|
870 newline,
|
adamc@875
|
871
|
adamc@875
|
872 string "if (mysql_stmt_bind_result(stmt, out)) uw_error(ctx, FATAL, \"",
|
adamc@875
|
873 string (ErrorMsg.spanToString loc),
|
adamc@875
|
874 string ": Error binding query result: %s\\n%s\", ",
|
adamc@875
|
875 query,
|
adamc@875
|
876 string ", mysql_error(conn->conn));",
|
adamc@873
|
877 newline,
|
adamc@873
|
878 newline,
|
adamc@873
|
879
|
adamc@873
|
880 string "if (mysql_stmt_store_result(stmt)) uw_error(ctx, FATAL, \"",
|
adamc@873
|
881 string (ErrorMsg.spanToString loc),
|
adamc@875
|
882 string ": Error storing query result: %s\\n%s\", ",
|
adamc@875
|
883 query,
|
adamc@875
|
884 string ", mysql_error(conn->conn));",
|
adamc@873
|
885 newline,
|
adamc@873
|
886 newline,
|
adamc@873
|
887
|
adamc@873
|
888 string "uw_end_region(ctx);",
|
adamc@873
|
889 newline,
|
adamc@875
|
890 string "while (1) {",
|
adamc@875
|
891 newline,
|
adamc@875
|
892 string "r = mysql_stmt_fetch(stmt);",
|
adamc@875
|
893 newline,
|
adamc@875
|
894 string "if (r != 0 && r != MYSQL_DATA_TRUNCATED) break;",
|
adamc@873
|
895 newline,
|
adamc@873
|
896 doCols p_getcol,
|
adamc@873
|
897 string "}",
|
adamc@873
|
898 newline,
|
adamc@873
|
899 newline,
|
adamc@873
|
900
|
adamc@874
|
901 string "if (r == 1) uw_error(ctx, FATAL, \"",
|
adamc@873
|
902 string (ErrorMsg.spanToString loc),
|
adamc@875
|
903 string ": query result fetching failed: %s\\n%s\", ",
|
adamc@875
|
904 query,
|
adamc@875
|
905 string ", mysql_error(conn->conn));",
|
adamc@875
|
906 newline,
|
adamc@875
|
907 newline,
|
adamc@875
|
908
|
adamc@875
|
909 string "if (mysql_stmt_reset(stmt)) uw_error(ctx, FATAL, \"",
|
adamc@875
|
910 string (ErrorMsg.spanToString loc),
|
adamc@875
|
911 string ": Error reseting statement: %s\\n%s\", ",
|
adamc@875
|
912 query,
|
adamc@875
|
913 string ", mysql_error(conn->conn));",
|
adamc@875
|
914 newline,
|
adamc@875
|
915 newline]
|
adamc@873
|
916
|
adamc@873
|
917 fun query {loc, cols, doCols} =
|
adamc@873
|
918 box [string "uw_conn *conn = uw_get_db(ctx);",
|
adamc@873
|
919 newline,
|
adamc@876
|
920 string "MYSQL_STMT *stmt = mysql_stmt_init(conn->conn);",
|
adamc@873
|
921 newline,
|
adamc@875
|
922 string "if (stmt == NULL) uw_error(ctx, FATAL, \"",
|
adamc@873
|
923 string (ErrorMsg.spanToString loc),
|
adamc@873
|
924 string ": can't allocate temporary prepared statement\");",
|
adamc@873
|
925 newline,
|
adamc@873
|
926 string "uw_push_cleanup(ctx, (void (*)(void *))mysql_stmt_close, stmt);",
|
adamc@873
|
927 newline,
|
adamc@873
|
928 string "if (mysql_stmt_prepare(stmt, query, strlen(query))) uw_error(ctx, FATAL, \"",
|
adamc@873
|
929 string (ErrorMsg.spanToString loc),
|
adamc@875
|
930 string ": error preparing statement: %s\\n%s\", query, mysql_error(conn->conn));",
|
adamc@873
|
931 newline,
|
adamc@873
|
932 newline,
|
adamc@873
|
933
|
adamc@873
|
934 queryCommon {loc = loc, cols = cols, doCols = doCols, query = string "query"},
|
adamc@873
|
935
|
adamc@873
|
936 string "uw_pop_cleanup(ctx);",
|
adamc@873
|
937 newline]
|
adamc@873
|
938
|
adamc@879
|
939 fun queryPrepared {loc, id, query, inputs, cols, doCols, nested} =
|
adamc@873
|
940 box [string "uw_conn *conn = uw_get_db(ctx);",
|
adamc@873
|
941 newline,
|
adamc@873
|
942 string "MYSQL_BIND in[",
|
adamc@873
|
943 string (Int.toString (length inputs)),
|
adamc@873
|
944 string "];",
|
adamc@873
|
945 newline,
|
adamc@873
|
946 p_list_sepi (box []) (fn i => fn t =>
|
adamc@873
|
947 let
|
adamc@873
|
948 fun buffers t =
|
adamc@873
|
949 case t of
|
adamc@873
|
950 String => box [string "unsigned long in_length",
|
adamc@873
|
951 string (Int.toString i),
|
adamc@873
|
952 string ";",
|
adamc@873
|
953 newline]
|
adamc@873
|
954 | Blob => box [string "unsigned long in_length",
|
adamc@873
|
955 string (Int.toString i),
|
adamc@873
|
956 string ";",
|
adamc@873
|
957 newline]
|
adamc@876
|
958 | Time => box [string "MYSQL_TIME in_buffer",
|
adamc@873
|
959 string (Int.toString i),
|
adamc@1013
|
960 string ";",
|
adamc@1013
|
961 newline]
|
adamc@873
|
962 | _ => box []
|
adamc@873
|
963 in
|
adamc@873
|
964 box [case t of
|
adamc@873
|
965 Nullable t => box [string "my_bool in_is_null",
|
adamc@873
|
966 string (Int.toString i),
|
adamc@873
|
967 string ";",
|
adamc@873
|
968 newline,
|
adamc@873
|
969 buffers t]
|
adamc@873
|
970 | _ => buffers t,
|
adamc@873
|
971 newline]
|
adamc@873
|
972 end) inputs,
|
adamc@873
|
973
|
adamc@879
|
974 if nested then
|
adamc@879
|
975 box [string "MYSQL_STMT *stmt;",
|
adamc@879
|
976 newline]
|
adamc@879
|
977 else
|
adamc@879
|
978 box [string "MYSQL_STMT *stmt = conn->p",
|
adamc@879
|
979 string (Int.toString id),
|
adamc@879
|
980 string ";",
|
adamc@879
|
981 newline,
|
adamc@879
|
982 newline,
|
adamc@879
|
983
|
adamc@879
|
984 string "if (stmt == NULL) {",
|
adamc@879
|
985 newline],
|
adamc@879
|
986
|
adamc@878
|
987 box [string "stmt = mysql_stmt_init(conn->conn);",
|
adamc@878
|
988 newline,
|
adamc@878
|
989 string "if (stmt == NULL) uw_error(ctx, FATAL, \"Out of memory allocating prepared statement\");",
|
adamc@878
|
990 newline,
|
adamc@880
|
991 if nested then
|
adamc@880
|
992 box [string "uw_push_cleanup(ctx, (void (*)(void *))mysql_stmt_close, stmt);",
|
adamc@880
|
993 newline]
|
adamc@880
|
994 else
|
adamc@880
|
995 box [],
|
adamc@878
|
996 string "if (mysql_stmt_prepare(stmt, \"",
|
adam@1656
|
997 string (Prim.toCString query),
|
adamc@878
|
998 string "\", ",
|
adamc@878
|
999 string (Int.toString (size query)),
|
adamc@878
|
1000 string ")) {",
|
adamc@878
|
1001 newline,
|
adamc@878
|
1002 box [string "char msg[1024];",
|
adamc@878
|
1003 newline,
|
adamc@878
|
1004 string "strncpy(msg, mysql_stmt_error(stmt), 1024);",
|
adamc@878
|
1005 newline,
|
adamc@878
|
1006 string "msg[1023] = 0;",
|
adamc@878
|
1007 newline,
|
adamc@880
|
1008 if nested then
|
adamc@880
|
1009 box []
|
adamc@880
|
1010 else
|
adamc@880
|
1011 box [string "mysql_stmt_close(stmt);",
|
adamc@880
|
1012 newline],
|
adamc@878
|
1013 string "uw_error(ctx, FATAL, \"Error preparing statement: %s\", msg);",
|
adamc@878
|
1014 newline],
|
adamc@878
|
1015 string "}",
|
adamc@878
|
1016 newline,
|
adamc@879
|
1017 if nested then
|
adamc@879
|
1018 box []
|
adamc@879
|
1019 else
|
adamc@879
|
1020 box [string "conn->p",
|
adamc@879
|
1021 string (Int.toString id),
|
adamc@879
|
1022 string " = stmt;",
|
adamc@879
|
1023 newline]],
|
adamc@879
|
1024 if nested then
|
adamc@879
|
1025 box []
|
adamc@879
|
1026 else
|
adamc@879
|
1027 box [string "}",
|
adamc@879
|
1028 newline],
|
adamc@878
|
1029 newline,
|
adamc@878
|
1030
|
adamc@873
|
1031 string "memset(in, 0, sizeof in);",
|
adamc@873
|
1032 newline,
|
adamc@873
|
1033 p_list_sepi (box []) (fn i => fn t =>
|
adamc@873
|
1034 let
|
adamc@873
|
1035 fun buffers t =
|
adamc@873
|
1036 case t of
|
adamc@873
|
1037 String => box [string "in[",
|
adamc@873
|
1038 string (Int.toString i),
|
adamc@873
|
1039 string "].buffer = arg",
|
adamc@873
|
1040 string (Int.toString (i + 1)),
|
adamc@873
|
1041 string ";",
|
adamc@873
|
1042 newline,
|
adamc@873
|
1043 string "in_length",
|
adamc@873
|
1044 string (Int.toString i),
|
adamc@873
|
1045 string "= in[",
|
adamc@873
|
1046 string (Int.toString i),
|
adamc@873
|
1047 string "].buffer_length = strlen(arg",
|
adamc@873
|
1048 string (Int.toString (i + 1)),
|
adamc@873
|
1049 string ");",
|
adamc@873
|
1050 newline,
|
adamc@873
|
1051 string "in[",
|
adamc@873
|
1052 string (Int.toString i),
|
adamc@873
|
1053 string "].length = &in_length",
|
adamc@873
|
1054 string (Int.toString i),
|
adamc@873
|
1055 string ";",
|
adamc@873
|
1056 newline]
|
adamc@1013
|
1057 | Char => box [string "in[",
|
adamc@1013
|
1058 string (Int.toString i),
|
adamc@1013
|
1059 string "].buffer = &arg",
|
adamc@1013
|
1060 string (Int.toString (i + 1)),
|
adamc@1013
|
1061 string ";",
|
adamc@1013
|
1062 newline,
|
adamc@1013
|
1063 string "in[",
|
adamc@1013
|
1064 string (Int.toString i),
|
adamc@1013
|
1065 string "].buffer_length = 1;",
|
adamc@1013
|
1066 newline]
|
adamc@873
|
1067 | Blob => box [string "in[",
|
adamc@873
|
1068 string (Int.toString i),
|
adamc@873
|
1069 string "].buffer = arg",
|
adamc@873
|
1070 string (Int.toString (i + 1)),
|
adamc@873
|
1071 string ".data;",
|
adamc@873
|
1072 newline,
|
adamc@873
|
1073 string "in_length",
|
adamc@873
|
1074 string (Int.toString i),
|
adamc@873
|
1075 string "= in[",
|
adamc@873
|
1076 string (Int.toString i),
|
adamc@873
|
1077 string "].buffer_length = arg",
|
adamc@873
|
1078 string (Int.toString (i + 1)),
|
adamc@873
|
1079 string ".size;",
|
adamc@873
|
1080 newline,
|
adamc@873
|
1081 string "in[",
|
adamc@873
|
1082 string (Int.toString i),
|
adamc@873
|
1083 string "].length = &in_length",
|
adamc@873
|
1084 string (Int.toString i),
|
adamc@873
|
1085 string ";",
|
adamc@873
|
1086 newline]
|
adamc@873
|
1087 | Time =>
|
adamc@873
|
1088 let
|
adamc@873
|
1089 fun oneField dst src =
|
adamc@873
|
1090 box [string "in_buffer",
|
adamc@873
|
1091 string (Int.toString i),
|
adamc@873
|
1092 string ".",
|
adamc@873
|
1093 string dst,
|
adamc@873
|
1094 string " = tms.tm_",
|
adamc@873
|
1095 string src,
|
adamc@873
|
1096 string ";",
|
adamc@873
|
1097 newline]
|
adamc@873
|
1098 in
|
adamc@873
|
1099 box [string "({",
|
adamc@873
|
1100 newline,
|
adamc@873
|
1101 string "struct tm tms;",
|
adamc@873
|
1102 newline,
|
adamc@873
|
1103 string "if (localtime_r(&arg",
|
adamc@873
|
1104 string (Int.toString (i + 1)),
|
adam@1443
|
1105 string ".seconds, &tms) == NULL) uw_error(ctx, FATAL, \"",
|
adamc@873
|
1106 string (ErrorMsg.spanToString loc),
|
adamc@873
|
1107 string ": error converting to MySQL time\");",
|
adamc@873
|
1108 newline,
|
adamc@938
|
1109 oneField "year" "year + 1900",
|
adamc@888
|
1110 box [string "in_buffer",
|
adamc@888
|
1111 string (Int.toString i),
|
adamc@888
|
1112 string ".month = tms.tm_mon + 1;",
|
adamc@888
|
1113 newline],
|
adamc@873
|
1114 oneField "day" "mday",
|
adam@1839
|
1115 oneField "hour" "hour",
|
adamc@873
|
1116 oneField "minute" "min",
|
adamc@873
|
1117 oneField "second" "sec",
|
adamc@873
|
1118 newline,
|
adamc@873
|
1119 string "in[",
|
adamc@873
|
1120 string (Int.toString i),
|
adamc@873
|
1121 string "].buffer = &in_buffer",
|
adamc@873
|
1122 string (Int.toString i),
|
adamc@873
|
1123 string ";",
|
adamc@933
|
1124 newline,
|
adamc@933
|
1125 string "});",
|
adamc@873
|
1126 newline]
|
adamc@873
|
1127 end
|
adamc@889
|
1128 | Channel => box [string "in_buffer",
|
adamc@889
|
1129 string (Int.toString i),
|
adamc@889
|
1130 string " = ((unsigned long long)arg",
|
adamc@889
|
1131 string (Int.toString (i + 1)),
|
adamc@889
|
1132 string ".cli << 32) | arg",
|
adamc@889
|
1133 string (Int.toString (i + 1)),
|
adamc@889
|
1134 string ".chn;",
|
adamc@889
|
1135 newline,
|
adamc@889
|
1136 string "in[",
|
adamc@889
|
1137 string (Int.toString i),
|
adamc@889
|
1138 string "].buffer = &in_buffer",
|
adamc@889
|
1139 string (Int.toString i),
|
adamc@889
|
1140 string ";",
|
adamc@889
|
1141 newline]
|
adam@1682
|
1142
|
adamc@873
|
1143 | _ => box [string "in[",
|
adamc@873
|
1144 string (Int.toString i),
|
adamc@873
|
1145 string "].buffer = &arg",
|
adamc@873
|
1146 string (Int.toString (i + 1)),
|
adamc@873
|
1147 string ";",
|
adamc@873
|
1148 newline]
|
adamc@873
|
1149 in
|
adamc@873
|
1150 box [string "in[",
|
adamc@873
|
1151 string (Int.toString i),
|
adamc@873
|
1152 string "].buffer_type = ",
|
adamc@873
|
1153 string (p_buffer_type t),
|
adamc@873
|
1154 string ";",
|
adamc@873
|
1155 newline,
|
adam@1682
|
1156
|
adamc@873
|
1157 case t of
|
adamc@873
|
1158 Nullable t => box [string "in[",
|
adamc@873
|
1159 string (Int.toString i),
|
adamc@873
|
1160 string "].is_null = &in_is_null",
|
adamc@873
|
1161 string (Int.toString i),
|
adamc@873
|
1162 string ";",
|
adamc@873
|
1163 newline,
|
adamc@873
|
1164 string "if (arg",
|
adamc@873
|
1165 string (Int.toString (i + 1)),
|
adamc@873
|
1166 string " == NULL) {",
|
adamc@873
|
1167 newline,
|
adamc@873
|
1168 box [string "in_is_null",
|
adamc@873
|
1169 string (Int.toString i),
|
adamc@873
|
1170 string " = 1;",
|
adamc@873
|
1171 newline],
|
adamc@873
|
1172 string "} else {",
|
adamc@873
|
1173 box [case t of
|
adamc@873
|
1174 String => box []
|
adamc@873
|
1175 | _ =>
|
adamc@873
|
1176 box [string (p_sql_ctype t),
|
adamc@873
|
1177 space,
|
adamc@876
|
1178 string "tmp = *arg",
|
adamc@876
|
1179 string (Int.toString (i + 1)),
|
adamc@876
|
1180 string ";",
|
adamc@876
|
1181 newline,
|
adamc@876
|
1182 string (p_sql_ctype t),
|
adamc@876
|
1183 space,
|
adamc@873
|
1184 string "arg",
|
adamc@873
|
1185 string (Int.toString (i + 1)),
|
adamc@876
|
1186 string " = tmp;",
|
adamc@873
|
1187 newline],
|
adamc@873
|
1188 string "in_is_null",
|
adamc@873
|
1189 string (Int.toString i),
|
adamc@873
|
1190 string " = 0;",
|
adamc@873
|
1191 newline,
|
adamc@873
|
1192 buffers t,
|
adamc@876
|
1193 newline],
|
adamc@876
|
1194 string "}",
|
adamc@876
|
1195 newline]
|
adam@1682
|
1196
|
adamc@873
|
1197 | _ => buffers t,
|
adamc@873
|
1198 newline]
|
adamc@873
|
1199 end) inputs,
|
adamc@873
|
1200 newline,
|
adamc@873
|
1201
|
adamc@875
|
1202 string "if (mysql_stmt_bind_param(stmt, in)) uw_error(ctx, FATAL, \"",
|
adamc@875
|
1203 string (ErrorMsg.spanToString loc),
|
adamc@875
|
1204 string ": error binding parameters\");",
|
adamc@875
|
1205 newline,
|
adamc@875
|
1206
|
adamc@873
|
1207 queryCommon {loc = loc, cols = cols, doCols = doCols, query = box [string "\"",
|
adam@1656
|
1208 string (Prim.toCString query),
|
adamc@879
|
1209 string "\""]},
|
adamc@879
|
1210
|
adamc@879
|
1211 if nested then
|
adamc@879
|
1212 box [string "uw_pop_cleanup(ctx);",
|
adamc@879
|
1213 newline]
|
adamc@879
|
1214 else
|
adamc@879
|
1215 box []]
|
adamc@873
|
1216
|
adam@1293
|
1217 fun dmlCommon {loc, dml, mode} =
|
adam@1953
|
1218 box [string "if (mysql_stmt_execute(stmt)) {",
|
adam@1953
|
1219 box [string "if (mysql_errno(conn->conn) == 1213)",
|
adam@1953
|
1220 newline,
|
adam@1953
|
1221 box [string "uw_error(ctx, UNLIMITED_RETRY, \"Deadlock detected\");",
|
adam@1953
|
1222 newline],
|
adam@1953
|
1223 newline,
|
adam@1953
|
1224 case mode of
|
adam@1953
|
1225 Settings.Error => box [string "uw_error(ctx, FATAL, \"",
|
adam@1953
|
1226 string (ErrorMsg.spanToString loc),
|
adam@1953
|
1227 string ": Error executing DML: %s\\n%s\", ",
|
adam@1953
|
1228 dml,
|
adam@1953
|
1229 string ", mysql_error(conn->conn));"]
|
adam@1953
|
1230 | Settings.None => string "uw_set_error_message(ctx, mysql_error(conn->conn));",
|
adam@1953
|
1231 newline],
|
adam@1953
|
1232 string "}",
|
adamc@875
|
1233 newline]
|
adamc@875
|
1234
|
adam@1293
|
1235 fun dml (loc, mode) =
|
adamc@875
|
1236 box [string "uw_conn *conn = uw_get_db(ctx);",
|
adamc@875
|
1237 newline,
|
adamc@1013
|
1238 string "MYSQL_STMT *stmt = mysql_stmt_init(conn->conn);",
|
adamc@875
|
1239 newline,
|
adamc@933
|
1240 string "if (stmt == NULL) uw_error(ctx, FATAL, \"",
|
adamc@875
|
1241 string (ErrorMsg.spanToString loc),
|
adamc@875
|
1242 string ": can't allocate temporary prepared statement\");",
|
adamc@875
|
1243 newline,
|
adamc@875
|
1244 string "uw_push_cleanup(ctx, (void (*)(void *))mysql_stmt_close, stmt);",
|
adamc@875
|
1245 newline,
|
adamc@875
|
1246 string "if (mysql_stmt_prepare(stmt, dml, strlen(dml))) uw_error(ctx, FATAL, \"",
|
adamc@875
|
1247 string (ErrorMsg.spanToString loc),
|
adamc@875
|
1248 string ": error preparing statement: %s\\n%s\", dml, mysql_error(conn->conn));",
|
adamc@875
|
1249 newline,
|
adamc@875
|
1250 newline,
|
adamc@875
|
1251
|
adam@1293
|
1252 dmlCommon {loc = loc, dml = string "dml", mode = mode},
|
adamc@875
|
1253
|
adamc@875
|
1254 string "uw_pop_cleanup(ctx);",
|
adamc@875
|
1255 newline]
|
adamc@875
|
1256
|
adam@1293
|
1257 fun dmlPrepared {loc, id, dml, inputs, mode} =
|
adamc@875
|
1258 box [string "uw_conn *conn = uw_get_db(ctx);",
|
adamc@875
|
1259 newline,
|
adamc@875
|
1260 string "MYSQL_BIND in[",
|
adamc@875
|
1261 string (Int.toString (length inputs)),
|
adamc@875
|
1262 string "];",
|
adamc@875
|
1263 newline,
|
adamc@875
|
1264 p_list_sepi (box []) (fn i => fn t =>
|
adamc@875
|
1265 let
|
adamc@875
|
1266 fun buffers t =
|
adamc@875
|
1267 case t of
|
adamc@875
|
1268 String => box [string "unsigned long in_length",
|
adamc@875
|
1269 string (Int.toString i),
|
adamc@875
|
1270 string ";",
|
adamc@875
|
1271 newline]
|
adamc@875
|
1272 | Blob => box [string "unsigned long in_length",
|
adamc@875
|
1273 string (Int.toString i),
|
adamc@875
|
1274 string ";",
|
adamc@875
|
1275 newline]
|
adamc@876
|
1276 | Time => box [string "MYSQL_TIME in_buffer",
|
adamc@875
|
1277 string (Int.toString i),
|
adamc@875
|
1278 string ";",
|
adamc@875
|
1279 newline]
|
adamc@889
|
1280 | Channel => box [string "unsigned long long in_buffer",
|
adamc@889
|
1281 string (Int.toString i),
|
adamc@889
|
1282 string ";",
|
adamc@889
|
1283 newline]
|
adamc@875
|
1284 | _ => box []
|
adamc@875
|
1285 in
|
adamc@875
|
1286 box [case t of
|
adamc@875
|
1287 Nullable t => box [string "my_bool in_is_null",
|
adamc@875
|
1288 string (Int.toString i),
|
adamc@875
|
1289 string ";",
|
adamc@875
|
1290 newline,
|
adamc@875
|
1291 buffers t]
|
adamc@875
|
1292 | _ => buffers t,
|
adamc@875
|
1293 newline]
|
adamc@875
|
1294 end) inputs,
|
adamc@875
|
1295 string "MYSQL_STMT *stmt = conn->p",
|
adamc@875
|
1296 string (Int.toString id),
|
adamc@875
|
1297 string ";",
|
adamc@875
|
1298 newline,
|
adamc@875
|
1299 newline,
|
adamc@875
|
1300
|
adamc@878
|
1301 string "if (stmt == NULL) {",
|
adamc@878
|
1302 newline,
|
adamc@878
|
1303 box [string "stmt = mysql_stmt_init(conn->conn);",
|
adamc@878
|
1304 newline,
|
adamc@878
|
1305 string "if (stmt == NULL) uw_error(ctx, FATAL, \"Out of memory allocating prepared statement\");",
|
adamc@878
|
1306 newline,
|
adamc@878
|
1307 string "if (mysql_stmt_prepare(stmt, \"",
|
adam@1656
|
1308 string (Prim.toCString dml),
|
adamc@878
|
1309 string "\", ",
|
adamc@878
|
1310 string (Int.toString (size dml)),
|
adamc@878
|
1311 string ")) {",
|
adamc@878
|
1312 newline,
|
adamc@878
|
1313 box [string "char msg[1024];",
|
adamc@878
|
1314 newline,
|
adamc@878
|
1315 string "strncpy(msg, mysql_stmt_error(stmt), 1024);",
|
adamc@878
|
1316 newline,
|
adamc@878
|
1317 string "msg[1023] = 0;",
|
adamc@878
|
1318 newline,
|
adamc@878
|
1319 string "uw_error(ctx, FATAL, \"Error preparing statement: %s\", msg);",
|
adamc@878
|
1320 newline],
|
adamc@878
|
1321 string "}",
|
adamc@878
|
1322 newline,
|
adamc@878
|
1323 string "conn->p",
|
adamc@878
|
1324 string (Int.toString id),
|
adamc@878
|
1325 string " = stmt;",
|
adamc@878
|
1326 newline],
|
adamc@878
|
1327 string "}",
|
adamc@878
|
1328 newline,
|
adamc@878
|
1329 newline,
|
adamc@878
|
1330
|
adamc@875
|
1331 string "memset(in, 0, sizeof in);",
|
adamc@875
|
1332 newline,
|
adamc@875
|
1333 p_list_sepi (box []) (fn i => fn t =>
|
adamc@875
|
1334 let
|
adamc@875
|
1335 fun buffers t =
|
adamc@875
|
1336 case t of
|
adamc@875
|
1337 String => box [string "in[",
|
adamc@875
|
1338 string (Int.toString i),
|
adamc@875
|
1339 string "].buffer = arg",
|
adamc@875
|
1340 string (Int.toString (i + 1)),
|
adamc@875
|
1341 string ";",
|
adamc@875
|
1342 newline,
|
adamc@875
|
1343 string "in_length",
|
adamc@875
|
1344 string (Int.toString i),
|
adamc@875
|
1345 string "= in[",
|
adamc@875
|
1346 string (Int.toString i),
|
adamc@875
|
1347 string "].buffer_length = strlen(arg",
|
adamc@875
|
1348 string (Int.toString (i + 1)),
|
adamc@875
|
1349 string ");",
|
adamc@875
|
1350 newline,
|
adamc@875
|
1351 string "in[",
|
adamc@875
|
1352 string (Int.toString i),
|
adamc@875
|
1353 string "].length = &in_length",
|
adamc@875
|
1354 string (Int.toString i),
|
adamc@875
|
1355 string ";",
|
adamc@875
|
1356 newline]
|
adamc@875
|
1357 | Blob => box [string "in[",
|
adamc@875
|
1358 string (Int.toString i),
|
adamc@875
|
1359 string "].buffer = arg",
|
adamc@875
|
1360 string (Int.toString (i + 1)),
|
adamc@875
|
1361 string ".data;",
|
adamc@875
|
1362 newline,
|
adamc@875
|
1363 string "in_length",
|
adamc@875
|
1364 string (Int.toString i),
|
adamc@875
|
1365 string "= in[",
|
adamc@875
|
1366 string (Int.toString i),
|
adamc@875
|
1367 string "].buffer_length = arg",
|
adamc@875
|
1368 string (Int.toString (i + 1)),
|
adamc@875
|
1369 string ".size;",
|
adamc@875
|
1370 newline,
|
adamc@875
|
1371 string "in[",
|
adamc@875
|
1372 string (Int.toString i),
|
adamc@875
|
1373 string "].length = &in_length",
|
adamc@875
|
1374 string (Int.toString i),
|
adamc@875
|
1375 string ";",
|
adamc@875
|
1376 newline]
|
adamc@875
|
1377 | Time =>
|
adamc@875
|
1378 let
|
adamc@875
|
1379 fun oneField dst src =
|
adamc@875
|
1380 box [string "in_buffer",
|
adamc@875
|
1381 string (Int.toString i),
|
adamc@875
|
1382 string ".",
|
adamc@875
|
1383 string dst,
|
adamc@875
|
1384 string " = tms.tm_",
|
adamc@875
|
1385 string src,
|
adamc@875
|
1386 string ";",
|
adamc@875
|
1387 newline]
|
adamc@875
|
1388 in
|
adamc@875
|
1389 box [string "({",
|
adamc@875
|
1390 newline,
|
adamc@875
|
1391 string "struct tm tms;",
|
adamc@875
|
1392 newline,
|
adamc@875
|
1393 string "if (localtime_r(&arg",
|
adamc@875
|
1394 string (Int.toString (i + 1)),
|
adam@1443
|
1395 string ".seconds, &tms) == NULL) uw_error(ctx, FATAL, \"",
|
adamc@875
|
1396 string (ErrorMsg.spanToString loc),
|
adamc@875
|
1397 string ": error converting to MySQL time\");",
|
adamc@875
|
1398 newline,
|
adamc@938
|
1399 oneField "year" "year + 1900",
|
adamc@938
|
1400 oneField "month" "mon + 1",
|
adamc@875
|
1401 oneField "day" "mday",
|
adam@1839
|
1402 oneField "hour" "hour",
|
adamc@875
|
1403 oneField "minute" "min",
|
adamc@875
|
1404 oneField "second" "sec",
|
adamc@875
|
1405 newline,
|
adamc@875
|
1406 string "in[",
|
adamc@875
|
1407 string (Int.toString i),
|
adamc@875
|
1408 string "].buffer = &in_buffer",
|
adamc@875
|
1409 string (Int.toString i),
|
adamc@875
|
1410 string ";",
|
adamc@933
|
1411 newline,
|
adamc@933
|
1412 string "});",
|
adamc@875
|
1413 newline]
|
adamc@875
|
1414 end
|
adamc@889
|
1415 | Channel => box [string "in_buffer",
|
adamc@889
|
1416 string (Int.toString i),
|
adamc@889
|
1417 string " = ((unsigned long long)arg",
|
adamc@889
|
1418 string (Int.toString (i + 1)),
|
adamc@889
|
1419 string ".cli << 32) | arg",
|
adamc@889
|
1420 string (Int.toString (i + 1)),
|
adamc@889
|
1421 string ".chn;",
|
adamc@889
|
1422 newline,
|
adamc@889
|
1423 string "in[",
|
adamc@889
|
1424 string (Int.toString i),
|
adamc@889
|
1425 string "].buffer = &in_buffer",
|
adamc@889
|
1426 string (Int.toString i),
|
adamc@889
|
1427 string ";",
|
adamc@889
|
1428 newline]
|
adam@1682
|
1429
|
adamc@875
|
1430 | _ => box [string "in[",
|
adamc@875
|
1431 string (Int.toString i),
|
adamc@875
|
1432 string "].buffer = &arg",
|
adamc@875
|
1433 string (Int.toString (i + 1)),
|
adamc@875
|
1434 string ";",
|
adamc@875
|
1435 newline]
|
adamc@875
|
1436 in
|
adamc@875
|
1437 box [string "in[",
|
adamc@875
|
1438 string (Int.toString i),
|
adamc@875
|
1439 string "].buffer_type = ",
|
adamc@875
|
1440 string (p_buffer_type t),
|
adamc@875
|
1441 string ";",
|
adamc@875
|
1442 newline,
|
adamc@889
|
1443
|
adamc@889
|
1444 case t of
|
adamc@889
|
1445 Channel => box [string "in[",
|
adamc@889
|
1446 string (Int.toString i),
|
adamc@889
|
1447 string "].is_unsigned = 1;",
|
adamc@889
|
1448 newline]
|
adamc@889
|
1449 | _ => box [],
|
adam@1682
|
1450
|
adamc@875
|
1451 case t of
|
adamc@875
|
1452 Nullable t => box [string "in[",
|
adamc@875
|
1453 string (Int.toString i),
|
adamc@875
|
1454 string "].is_null = &in_is_null",
|
adamc@875
|
1455 string (Int.toString i),
|
adamc@875
|
1456 string ";",
|
adamc@875
|
1457 newline,
|
adamc@875
|
1458 string "if (arg",
|
adamc@875
|
1459 string (Int.toString (i + 1)),
|
adamc@875
|
1460 string " == NULL) {",
|
adamc@875
|
1461 newline,
|
adamc@875
|
1462 box [string "in_is_null",
|
adamc@875
|
1463 string (Int.toString i),
|
adamc@875
|
1464 string " = 1;",
|
adamc@875
|
1465 newline],
|
adamc@875
|
1466 string "} else {",
|
adamc@875
|
1467 box [case t of
|
adamc@875
|
1468 String => box []
|
adamc@875
|
1469 | _ =>
|
adamc@875
|
1470 box [string (p_sql_ctype t),
|
adamc@875
|
1471 space,
|
adamc@876
|
1472 string "tmp = *arg",
|
adamc@876
|
1473 string (Int.toString (i + 1)),
|
adamc@876
|
1474 string ";",
|
adamc@876
|
1475 newline,
|
adamc@876
|
1476 string (p_sql_ctype t),
|
adamc@876
|
1477 space,
|
adamc@875
|
1478 string "arg",
|
adamc@875
|
1479 string (Int.toString (i + 1)),
|
adamc@876
|
1480 string " = tmp;",
|
adamc@875
|
1481 newline],
|
adamc@875
|
1482 string "in_is_null",
|
adamc@875
|
1483 string (Int.toString i),
|
adamc@875
|
1484 string " = 0;",
|
adamc@875
|
1485 newline,
|
adamc@875
|
1486 buffers t,
|
adamc@876
|
1487 newline],
|
adamc@876
|
1488 string "}",
|
adamc@876
|
1489 newline]
|
adam@1682
|
1490
|
adamc@875
|
1491 | _ => buffers t,
|
adamc@875
|
1492 newline]
|
adamc@875
|
1493 end) inputs,
|
adamc@875
|
1494 newline,
|
adamc@875
|
1495
|
adamc@875
|
1496 string "if (mysql_stmt_bind_param(stmt, in)) uw_error(ctx, FATAL, \"",
|
adamc@875
|
1497 string (ErrorMsg.spanToString loc),
|
adamc@875
|
1498 string ": error binding parameters\");",
|
adamc@875
|
1499 newline,
|
adamc@875
|
1500
|
adamc@875
|
1501 dmlCommon {loc = loc, dml = box [string "\"",
|
adam@1656
|
1502 string (Prim.toCString dml),
|
adam@1293
|
1503 string "\""], mode = mode}]
|
adamc@875
|
1504
|
adamc@878
|
1505 fun nextval {loc, seqE, seqName} =
|
adamc@878
|
1506 box [string "uw_conn *conn = uw_get_db(ctx);",
|
adamc@878
|
1507 newline,
|
adamc@878
|
1508 string "char *insert = ",
|
adamc@878
|
1509 case seqName of
|
adamc@878
|
1510 SOME s => string ("\"INSERT INTO " ^ s ^ " VALUES ()\"")
|
adamc@878
|
1511 | NONE => box [string "uw_Basis_strcat(ctx, \"INSERT INTO \", uw_Basis_strcat(ctx, ",
|
adamc@878
|
1512 seqE,
|
adamc@878
|
1513 string ", \" VALUES ()\"))"],
|
adamc@878
|
1514 string ";",
|
adamc@878
|
1515 newline,
|
adamc@878
|
1516 string "char *delete = ",
|
adamc@878
|
1517 case seqName of
|
adamc@878
|
1518 SOME s => string ("\"DELETE FROM " ^ s ^ "\"")
|
adamc@878
|
1519 | NONE => box [string "uw_Basis_strcat(ctx, \"DELETE FROM \", ",
|
adamc@878
|
1520 seqE,
|
adamc@878
|
1521 string ")"],
|
adamc@878
|
1522 string ";",
|
adamc@878
|
1523 newline,
|
adamc@878
|
1524 newline,
|
adamc@878
|
1525
|
adamc@878
|
1526 string "if (mysql_query(conn->conn, insert)) uw_error(ctx, FATAL, \"'nextval' INSERT failed\");",
|
adamc@878
|
1527 newline,
|
adamc@878
|
1528 string "n = mysql_insert_id(conn->conn);",
|
adamc@878
|
1529 newline,
|
adamc@878
|
1530 string "if (mysql_query(conn->conn, delete)) uw_error(ctx, FATAL, \"'nextval' DELETE failed\");",
|
adamc@878
|
1531 newline]
|
adamc@878
|
1532
|
adamc@878
|
1533 fun nextvalPrepared _ = raise Fail "MySQL.nextvalPrepared called"
|
adamc@867
|
1534
|
adamc@1073
|
1535 fun setval _ = raise Fail "MySQL.setval called"
|
adamc@1073
|
1536
|
adamc@877
|
1537 fun sqlifyString s = "'" ^ String.translate (fn #"'" => "\\'"
|
adamc@877
|
1538 | #"\\" => "\\\\"
|
adamc@877
|
1539 | ch =>
|
adamc@877
|
1540 if Char.isPrint ch then
|
adamc@877
|
1541 str ch
|
adamc@877
|
1542 else
|
adamc@877
|
1543 (ErrorMsg.error
|
adamc@877
|
1544 "Non-printing character found in SQL string literal";
|
adamc@877
|
1545 ""))
|
adam@1656
|
1546 (Prim.toCString s) ^ "'"
|
adamc@874
|
1547
|
adamc@877
|
1548 fun p_cast (s, _) = s
|
adamc@874
|
1549
|
adamc@874
|
1550 fun p_blank _ = "?"
|
adamc@874
|
1551
|
adamc@866
|
1552 val () = addDbms {name = "mysql",
|
adam@1464
|
1553 header = Config.msheader,
|
adam@1682
|
1554 randomFunction = "RAND",
|
adamc@866
|
1555 link = "-lmysqlclient",
|
adamc@867
|
1556 init = init,
|
adamc@873
|
1557 p_sql_type = p_sql_type,
|
adamc@867
|
1558 query = query,
|
adamc@868
|
1559 queryPrepared = queryPrepared,
|
adamc@868
|
1560 dml = dml,
|
adamc@869
|
1561 dmlPrepared = dmlPrepared,
|
adamc@869
|
1562 nextval = nextval,
|
adamc@874
|
1563 nextvalPrepared = nextvalPrepared,
|
adamc@1073
|
1564 setval = setval,
|
adamc@874
|
1565 sqlifyString = sqlifyString,
|
adamc@874
|
1566 p_cast = p_cast,
|
adamc@874
|
1567 p_blank = p_blank,
|
adamc@877
|
1568 supportsDeleteAs = false,
|
adamc@886
|
1569 supportsUpdateAs = false,
|
adamc@884
|
1570 createSequence = fn s => "CREATE TABLE " ^ s ^ " (uw_id INTEGER PRIMARY KEY AUTO_INCREMENT)",
|
adamc@878
|
1571 textKeysNeedLengths = true,
|
adamc@879
|
1572 supportsNextval = false,
|
adamc@882
|
1573 supportsNestedPrepared = false,
|
adamc@890
|
1574 sqlPrefix = "SET storage_engine=InnoDB;\n\n",
|
adamc@1014
|
1575 supportsOctetLength = true,
|
adamc@1014
|
1576 trueString = "TRUE",
|
adamc@1196
|
1577 falseString = "FALSE",
|
adamc@1196
|
1578 onlyUnion = true,
|
adam@1777
|
1579 nestedRelops = false,
|
adam@1777
|
1580 windowFunctions = false}
|
adamc@866
|
1581
|
adamc@866
|
1582 end
|