diff src/cjr_print.sml @ 747:e42f08f96eb5

Extra checks on assumptions about SQL results
author Adam Chlipala <adamc@hcoop.net>
date Sun, 26 Apr 2009 12:47:53 -0400
parents cd67c3a942e3
children 8688e01ae469
line wrap: on
line diff
--- a/src/cjr_print.sml	Sun Apr 26 12:35:45 2009 -0400
+++ b/src/cjr_print.sml	Sun Apr 26 12:47:53 2009 -0400
@@ -418,7 +418,7 @@
 fun p_getcol wontLeakStrings env (tAll as (t, loc)) i =
     case t of
         TOption t =>
-        box [string "(PQgetisnull (res, i, ",
+        box [string "(PQgetisnull(res, i, ",
              string (Int.toString i),
              string ") ? NULL : ",
              case t of
@@ -440,13 +440,30 @@
                            string "})"],
              string ")"]
       | _ =>
-        p_unsql wontLeakStrings env tAll
-                (box [string "PQgetvalue(res, i, ",
-                      string (Int.toString i),
-                      string ")"])
-                (box [string "PQgetlength(res, i, ",
-                      string (Int.toString i),
-                      string ")"])
+        box [string "(PQgetisnull(res, i, ",
+             string (Int.toString i),
+             string ") ? ",
+             box [string "({",
+                  p_typ env tAll,
+                  space,
+                  string "tmp;",
+                  newline,
+                  string "uw_error(ctx, FATAL, \"Unexpectedly NULL field #",
+                  string (Int.toString i),
+                  string "\");",
+                  newline,
+                  string "tmp;",
+                  newline,
+                  string "})"],
+             string " : ",
+             p_unsql wontLeakStrings env tAll
+                     (box [string "PQgetvalue(res, i, ",
+                           string (Int.toString i),
+                           string ")"])
+                     (box [string "PQgetlength(res, i, ",
+                           string (Int.toString i),
+                           string ")"]),
+             string ")"]
 
 datatype sql_type =
          Int
@@ -1571,6 +1588,28 @@
                  newline,
                  newline,
 
+                 string "if (PQnfields(res) != ",
+                 string (Int.toString (length outputs)),
+                 string ") {",
+                 newline,
+                 box [string "int nf = PQnfields(res);",
+                      newline,
+                      string "PQclear(res);",
+                      newline,
+                      string "uw_error(ctx, FATAL, \"",
+                      string (ErrorMsg.spanToString loc),
+                      string ": Query returned %d columns instead of ",
+                      string (Int.toString (length outputs)),
+                      string ":\\n%s\\n%s\", ",
+                      case prepared of
+                          NONE => string "query"
+                        | SOME _ => p_exp env query,
+                      string ", nf, PQerrorMessage(conn));",
+                      newline],
+                 string "}",
+                 newline,
+                 newline,
+
                  string "uw_end_region(ctx);",
                  newline,
                  string "uw_push_cleanup(ctx, (void (*)(void *))PQclear, res);",