pg_dump for domain constraints.

Rod Taylor

diff --git a/src/bin/pg_dump/pg_dump.c b/src/bin/pg_dump/pg_dump.c
index 2b111bdce5b4b4b26e9fb6735ac70108df5a6792..dc2bdd83d5bd3775457d4f8472372119d1ea00f0 100644 (file)
--- a/src/bin/pg_dump/pg_dump.c
+++ b/src/bin/pg_dump/pg_dump.c
@@ -22,7 +22,7 @@
  *
  *
  * IDENTIFICATION
- *       $Header: /cvsroot/pgsql/src/bin/pg_dump/pg_dump.c,v 1.306 2002/11/08 17:37:52 tgl Exp $
+ *       $Header: /cvsroot/pgsql/src/bin/pg_dump/pg_dump.c,v 1.307 2002/11/15 02:52:18 momjian Exp $
  *
  *-------------------------------------------------------------------------
  */
@@ -3157,8 +3157,10 @@ dumpOneDomain(Archive *fout, TypeInfo *tinfo)
        PQExpBuffer q = createPQExpBuffer();
        PQExpBuffer delq = createPQExpBuffer();
        PQExpBuffer query = createPQExpBuffer();
+       PQExpBuffer chkquery = createPQExpBuffer();
        PGresult   *res;
        int                     ntups;
+       int                     i;
        char       *typnotnull;
        char       *typdefn;
        char       *typdefault;
@@ -3228,6 +3230,34 @@ dumpOneDomain(Archive *fout, TypeInfo *tinfo)
        if (typdefault)
                appendPQExpBuffer(q, " DEFAULT %s", typdefault);
 
+       /* Fetch and process CHECK Constraints */
+       appendPQExpBuffer(chkquery, "SELECT conname, consrc "
+                                         "FROM pg_catalog.pg_constraint "
+                                         "WHERE contypid = '%s'::pg_catalog.oid",
+                                         tinfo->oid);
+
+       res = PQexec(g_conn, chkquery->data);
+       if (!res ||
+               PQresultStatus(res) != PGRES_TUPLES_OK)
+       {
+               write_msg(NULL, "query to obtain domain constraint information failed: %s",
+                                 PQerrorMessage(g_conn));
+               exit_nicely();
+       }
+
+       /* Expecting a single result only */
+       ntups = PQntuples(res);
+       for (i = 0; i < ntups; i++)
+       {
+               char       *conname;
+               char       *consrc;
+
+               conname = PQgetvalue(res, i, PQfnumber(res, "conname"));
+               consrc = PQgetvalue(res, i, PQfnumber(res, "consrc"));
+
+               appendPQExpBuffer(q, " CONSTRAINT %s CHECK %s", fmtId(conname), consrc);
+       }
+       
        appendPQExpBuffer(q, ";\n");
 
        (*deps)[depIdx++] = NULL;       /* End of List */