psql: Support identity columns in sequence display

Where the footer for an owned serial sequence would say "Owned by", put
something analogous for a sequence belonging to an identity column.

Reported-by: Vitaly Burovoy <vitaly.burovoy@gmail.com>

diff --git a/src/bin/psql/describe.c b/src/bin/psql/describe.c
index 0f9f497c66441a77deabdba5ceac2bea9ea438b4..dbfc7339e574a9690d4fcadcaf03b55f18400513 100644 (file)
--- a/src/bin/psql/describe.c
+++ b/src/bin/psql/describe.c
@@ -2006,7 +2006,8 @@ describeOneTableDetails(const char *schemaname,
                /* Get the column that owns this sequence */
                printfPQExpBuffer(&buf, "SELECT pg_catalog.quote_ident(nspname) || '.' ||"
                                                  "\n   pg_catalog.quote_ident(relname) || '.' ||"
-                                                 "\n   pg_catalog.quote_ident(attname)"
+                                                 "\n   pg_catalog.quote_ident(attname),"
+                                                 "\n   d.deptype"
                                                  "\nFROM pg_catalog.pg_class c"
                                        "\nINNER JOIN pg_catalog.pg_depend d ON c.oid=d.refobjid"
                         "\nINNER JOIN pg_catalog.pg_namespace n ON n.oid=c.relnamespace"
@@ -2016,7 +2017,7 @@ describeOneTableDetails(const char *schemaname,
                           "\nWHERE d.classid='pg_catalog.pg_class'::pg_catalog.regclass"
                         "\n AND d.refclassid='pg_catalog.pg_class'::pg_catalog.regclass"
                                                  "\n AND d.objid=%s"
-                                                 "\n AND d.deptype='a'",
+                                                 "\n AND d.deptype IN ('a', 'i')",
                                                  oid);
 
                result = PSQLexec(buf.data);
@@ -2024,9 +2025,19 @@ describeOneTableDetails(const char *schemaname,
                        goto error_return;
                else if (PQntuples(result) == 1)
                {
-                       printfPQExpBuffer(&buf, _("Owned by: %s"),
-                                                         PQgetvalue(result, 0, 0));
-                       printTableAddFooter(&cont, buf.data);
+                       switch (PQgetvalue(result, 0, 1)[0])
+                       {
+                               case 'a':
+                                       printfPQExpBuffer(&buf, _("Owned by: %s"),
+                                                                         PQgetvalue(result, 0, 0));
+                                       printTableAddFooter(&cont, buf.data);
+                                       break;
+                               case 'i':
+                                       printfPQExpBuffer(&buf, _("Sequence for identity column: %s"),
+                                                                         PQgetvalue(result, 0, 0));
+                                       printTableAddFooter(&cont, buf.data);
+                                       break;
+                       }
                }
 
                /*