Location via proxy:   [ UP ]  
[Report a bug]   [Manage cookies]                
Fix infelicities in describeOneTableDetails' partitioned-table handling.
authorTom Lane <tgl@sss.pgh.pa.us>
Wed, 24 Jul 2019 22:14:26 +0000 (18:14 -0400)
committerTom Lane <tgl@sss.pgh.pa.us>
Wed, 24 Jul 2019 22:14:45 +0000 (18:14 -0400)
describeOneTableDetails issued a partition-constraint-fetching query
for every table, even ones it knows perfectly well are not partitions.

To add insult to injury, it then proceeded to leak the empty PGresult
if the table wasn't a partition.  Doing that a lot of times might
amount to a meaningful leak, so this seems like a back-patchable bug.

Fix that, and also fix a related PGresult leak in the partition-parent
case (though that leak would occur only if we got no row, which is
unexpected).

Minor code beautification too, to make this code look more like the
pre-existing code around it.

Back-patch the whole change into v12.  However, the fact that we already
know whether the table is a partition dates only to commit 1af25ca0c;
back-patching the relevant changes from that is probably more churn
than is justified in released branches.  Hence, in v11 and v10, just
do the minimum to fix the PGresult leaks.

Noted while messing around with adjacent code for yesterday's \d
improvements.

src/bin/psql/describe.c

index 97167d2c4bdb2f29ceec491fe73062bd3a158db8..3a749d102926b725dad1c1bc1df72b30a5f8e50f 100644 (file)
@@ -2102,37 +2102,32 @@ describeOneTableDetails(const char *schemaname,
    }
 
    /* Make footers */
-   if (pset.sversion >= 100000)
+
+   if (tableinfo.ispartition)
    {
-       /* Get the partition information */
+       /* Footer information for a partition child table */
        PGresult   *result;
-       char       *parent_name;
-       char       *partdef;
-       char       *partconstraintdef = NULL;
 
        printfPQExpBuffer(&buf,
                          "SELECT inhparent::pg_catalog.regclass,\n"
-                         "  pg_catalog.pg_get_expr(c.relpartbound, inhrelid)");
+                         "  pg_catalog.pg_get_expr(c.relpartbound, c.oid)");
        /* If verbose, also request the partition constraint definition */
        if (verbose)
-           appendPQExpBuffer(&buf,
-                             ",\n  pg_catalog.pg_get_partition_constraintdef(inhrelid)");
+           appendPQExpBufferStr(&buf,
+                                ",\n  pg_catalog.pg_get_partition_constraintdef(c.oid)");
        appendPQExpBuffer(&buf,
                          "\nFROM pg_catalog.pg_class c"
                          " JOIN pg_catalog.pg_inherits i"
                          " ON c.oid = inhrelid"
-                         "\nWHERE c.oid = '%s' AND c.relispartition;", oid);
+                         "\nWHERE c.oid = '%s';", oid);
        result = PSQLexec(buf.data);
        if (!result)
            goto error_return;
 
        if (PQntuples(result) > 0)
        {
-           parent_name = PQgetvalue(result, 0, 0);
-           partdef = PQgetvalue(result, 0, 1);
-
-           if (PQnfields(result) == 3 && !PQgetisnull(result, 0, 2))
-               partconstraintdef = PQgetvalue(result, 0, 2);
+           char       *parent_name = PQgetvalue(result, 0, 0);
+           char       *partdef = PQgetvalue(result, 0, 1);
 
            printfPQExpBuffer(&tmpbuf, _("Partition of: %s %s"), parent_name,
                              partdef);
@@ -2140,6 +2135,10 @@ describeOneTableDetails(const char *schemaname,
 
            if (verbose)
            {
+               char       *partconstraintdef = NULL;
+
+               if (!PQgetisnull(result, 0, 2))
+                   partconstraintdef = PQgetvalue(result, 0, 2);
                /* If there isn't any constraint, show that explicitly */
                if (partconstraintdef == NULL || partconstraintdef[0] == '\0')
                    printfPQExpBuffer(&tmpbuf, _("No partition constraint"));
@@ -2148,27 +2147,29 @@ describeOneTableDetails(const char *schemaname,
                                      partconstraintdef);
                printTableAddFooter(&cont, tmpbuf.data);
            }
-
-           PQclear(result);
        }
+       PQclear(result);
    }
 
    if (tableinfo.relkind == RELKIND_PARTITIONED_TABLE)
    {
-       /* Get the partition key information  */
+       /* Footer information for a partitioned table (partitioning parent) */
        PGresult   *result;
-       char       *partkeydef;
 
        printfPQExpBuffer(&buf,
                          "SELECT pg_catalog.pg_get_partkeydef('%s'::pg_catalog.oid);",
                          oid);
        result = PSQLexec(buf.data);
-       if (!result || PQntuples(result) != 1)
+       if (!result)
            goto error_return;
 
-       partkeydef = PQgetvalue(result, 0, 0);
-       printfPQExpBuffer(&tmpbuf, _("Partition key: %s"), partkeydef);
-       printTableAddFooter(&cont, tmpbuf.data);
+       if (PQntuples(result) == 1)
+       {
+           char       *partkeydef = PQgetvalue(result, 0, 0);
+
+           printfPQExpBuffer(&tmpbuf, _("Partition key: %s"), partkeydef);
+           printTableAddFooter(&cont, tmpbuf.data);
+       }
        PQclear(result);
    }