annotate src/mysql.sml @ 885:e6070333d8a8

demo/sql works with SQLite
author Adam Chlipala <adamc@hcoop.net>
date Fri, 17 Jul 2009 16:29:36 -0400
parents ced093080e17
children 5805fa825fe8
rev   line source
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@884 1449 createSequence = fn s => "CREATE TABLE " ^ s ^ " (uw_id INTEGER PRIMARY KEY AUTO_INCREMENT)",
adamc@878 1450 textKeysNeedLengths = true,
adamc@879 1451 supportsNextval = false,
adamc@882 1452 supportsNestedPrepared = false,
adamc@885 1453 sqlPrefix = "SET storage_engine=InnoDB;\n\n"}
adamc@866 1454
adamc@866 1455 end