pg_dump: Dump comments and security labels for publication and subscriptions

diff --git a/src/bin/pg_dump/pg_dump.c b/src/bin/pg_dump/pg_dump.c
index 14dc1b242320bea1f1324897181057b133aba1d1..58566f423f57fe8cd6ebdf28a2e8d4b560385c52 100644 (file)
--- a/src/bin/pg_dump/pg_dump.c
+++ b/src/bin/pg_dump/pg_dump.c
@@ -3456,12 +3456,14 @@ dumpPublication(Archive *fout, PublicationInfo *pubinfo)
 {
    PQExpBuffer delq;
    PQExpBuffer query;
+   PQExpBuffer labelq;
 
    if (!(pubinfo->dobj.dump & DUMP_COMPONENT_DEFINITION))
        return;
 
    delq = createPQExpBuffer();
    query = createPQExpBuffer();
+   labelq = createPQExpBuffer();
 
    appendPQExpBuffer(delq, "DROP PUBLICATION %s;\n",
                      fmtId(pubinfo->dobj.name));
@@ -3469,6 +3471,8 @@ dumpPublication(Archive *fout, PublicationInfo *pubinfo)
    appendPQExpBuffer(query, "CREATE PUBLICATION %s",
                      fmtId(pubinfo->dobj.name));
 
+   appendPQExpBuffer(labelq, "PUBLICATION %s", fmtId(pubinfo->dobj.name));
+
    if (pubinfo->puballtables)
        appendPQExpBufferStr(query, " FOR ALL TABLES");
 
@@ -3500,6 +3504,16 @@ dumpPublication(Archive *fout, PublicationInfo *pubinfo)
                 NULL, 0,
                 NULL, NULL);
 
+   if (pubinfo->dobj.dump & DUMP_COMPONENT_COMMENT)
+       dumpComment(fout, labelq->data,
+                   NULL, pubinfo->rolname,
+                   pubinfo->dobj.catId, 0, pubinfo->dobj.dumpId);
+
+   if (pubinfo->dobj.dump & DUMP_COMPONENT_SECLABEL)
+       dumpSecLabel(fout, labelq->data,
+                    NULL, pubinfo->rolname,
+                    pubinfo->dobj.catId, 0, pubinfo->dobj.dumpId);
+
    destroyPQExpBuffer(delq);
    destroyPQExpBuffer(query);
 }
@@ -3754,6 +3768,7 @@ dumpSubscription(Archive *fout, SubscriptionInfo *subinfo)
 {
    PQExpBuffer delq;
    PQExpBuffer query;
+   PQExpBuffer labelq;
    PQExpBuffer publications;
    char      **pubnames = NULL;
    int         npubnames = 0;
@@ -3764,6 +3779,7 @@ dumpSubscription(Archive *fout, SubscriptionInfo *subinfo)
 
    delq = createPQExpBuffer();
    query = createPQExpBuffer();
+   labelq = createPQExpBuffer();
 
    appendPQExpBuffer(delq, "DROP SUBSCRIPTION %s;\n",
                      fmtId(subinfo->dobj.name));
@@ -3796,6 +3812,8 @@ dumpSubscription(Archive *fout, SubscriptionInfo *subinfo)
    appendStringLiteralAH(query, subinfo->subslotname, fout);
    appendPQExpBufferStr(query, ");\n");
 
+   appendPQExpBuffer(labelq, "SUBSCRIPTION %s", fmtId(subinfo->dobj.name));
+
    ArchiveEntry(fout, subinfo->dobj.catId, subinfo->dobj.dumpId,
                 subinfo->dobj.name,
                 NULL,
@@ -3806,6 +3824,16 @@ dumpSubscription(Archive *fout, SubscriptionInfo *subinfo)
                 NULL, 0,
                 NULL, NULL);
 
+   if (subinfo->dobj.dump & DUMP_COMPONENT_COMMENT)
+       dumpComment(fout, labelq->data,
+                   NULL, subinfo->rolname,
+                   subinfo->dobj.catId, 0, subinfo->dobj.dumpId);
+
+   if (subinfo->dobj.dump & DUMP_COMPONENT_SECLABEL)
+       dumpSecLabel(fout, labelq->data,
+                    NULL, subinfo->rolname,
+                    subinfo->dobj.catId, 0, subinfo->dobj.dumpId);
+
    destroyPQExpBuffer(publications);
    if (pubnames)
        free(pubnames);

diff --git a/src/bin/pg_dump/t/002_pg_dump.pl b/src/bin/pg_dump/t/002_pg_dump.pl
index e0d1ce62323df88b2e383be5927620b5eb3fb529..4dd208e8e1111978504d09a61b84b79106a54bc5 100644 (file)
--- a/src/bin/pg_dump/t/002_pg_dump.pl
+++ b/src/bin/pg_dump/t/002_pg_dump.pl
@@ -1517,6 +1517,7 @@ my %tests = (
            only_dump_test_schema  => 1,
            only_dump_test_table   => 1,
            role                   => 1,
+           section_post_data       => 1,
            test_schema_plus_blobs => 1, }, },
 
    'COMMENT ON EXTENSION plpgsql' => {
@@ -1545,6 +1546,7 @@ my %tests = (
            only_dump_test_schema  => 1,
            only_dump_test_table   => 1,
            role                   => 1,
+           section_post_data       => 1,
            test_schema_plus_blobs => 1, }, },
 
    'COMMENT ON TABLE dump_test.test_table' => {
@@ -1576,7 +1578,8 @@ my %tests = (
            data_only                => 1,
            exclude_dump_test_schema => 1,
            exclude_test_table       => 1,
-           role                     => 1, }, },
+           role                     => 1,
+           section_post_data        => 1, }, },
 
    'COMMENT ON COLUMN dump_test.test_table.col1' => {
        all_runs     => 1,
@@ -1609,7 +1612,8 @@ my %tests = (
            data_only                => 1,
            exclude_dump_test_schema => 1,
            exclude_test_table       => 1,
-           role                     => 1, }, },
+           role                     => 1,
+           section_post_data        => 1, }, },
 
    'COMMENT ON COLUMN dump_test.composite.f1' => {
        all_runs     => 1,
@@ -1642,7 +1646,8 @@ my %tests = (
            data_only                => 1,
            exclude_dump_test_schema => 1,
            only_dump_test_table     => 1,
-           role                     => 1, }, },
+           role                     => 1,
+           section_post_data        => 1, }, },
 
    'COMMENT ON COLUMN dump_test.test_second_table.col1' => {
        all_runs     => 1,
@@ -1675,7 +1680,8 @@ my %tests = (
            data_only                => 1,
            exclude_dump_test_schema => 1,
            only_dump_test_table     => 1,
-           role                     => 1, }, },
+           role                     => 1,
+           section_post_data        => 1, }, },
 
    'COMMENT ON COLUMN dump_test.test_second_table.col2' => {
        all_runs     => 1,
@@ -1708,7 +1714,8 @@ my %tests = (
            data_only                => 1,
            exclude_dump_test_schema => 1,
            only_dump_test_table     => 1,
-           role                     => 1, }, },
+           role                     => 1,
+           section_post_data        => 1, }, },
 
    'COMMENT ON CONVERSION dump_test.test_conversion' => {
        all_runs     => 1,
@@ -1739,7 +1746,8 @@ my %tests = (
            data_only                => 1,
            exclude_dump_test_schema => 1,
            only_dump_test_table     => 1,
-           role                     => 1, }, },
+           role                     => 1,
+           section_post_data        => 1, }, },
 
    'COMMENT ON COLLATION test0' => {
        all_runs     => 1,
@@ -1771,6 +1779,7 @@ my %tests = (
            only_dump_test_schema  => 1,
            only_dump_test_table   => 1,
            role                   => 1,
+           section_post_data      => 1,
            test_schema_plus_blobs => 1, }, },
 
    'COMMENT ON LARGE OBJECT ...' => {
@@ -1814,6 +1823,74 @@ my %tests = (
            section_data             => 1,
            section_post_data        => 1, }, },
 
+   'COMMENT ON PUBLICATION pub1' => {
+       all_runs     => 1,
+       create_order => 55,
+       create_sql   => 'COMMENT ON PUBLICATION pub1
+                      IS \'comment on publication\';',
+       regexp => qr/^COMMENT ON PUBLICATION pub1 IS 'comment on publication';/m,
+       like   => {
+           binary_upgrade           => 1,
+           clean                    => 1,
+           clean_if_exists          => 1,
+           createdb                 => 1,
+           defaults                 => 1,
+           exclude_dump_test_schema => 1,
+           exclude_test_table       => 1,
+           exclude_test_table_data  => 1,
+           no_blobs                 => 1,
+           no_privs                 => 1,
+           no_owner                 => 1,
+           pg_dumpall_dbprivs       => 1,
+           schema_only              => 1,
+           section_post_data        => 1,
+           with_oids                => 1, },
+       unlike => {
+           column_inserts           => 1,
+           data_only                => 1,
+           only_dump_test_table     => 1,
+           only_dump_test_schema    => 1,
+           pg_dumpall_globals       => 1,
+           pg_dumpall_globals_clean => 1,
+           role                     => 1,
+           section_data             => 1,
+           section_pre_data         => 1,
+           test_schema_plus_blobs   => 1, }, },
+
+   'COMMENT ON SUBSCRIPTION sub1' => {
+       all_runs     => 1,
+       create_order => 55,
+       create_sql   => 'COMMENT ON SUBSCRIPTION sub1
+                      IS \'comment on subscription\';',
+       regexp => qr/^COMMENT ON SUBSCRIPTION sub1 IS 'comment on subscription';/m,
+       like   => {
+           binary_upgrade           => 1,
+           clean                    => 1,
+           clean_if_exists          => 1,
+           createdb                 => 1,
+           defaults                 => 1,
+           exclude_dump_test_schema => 1,
+           exclude_test_table       => 1,
+           exclude_test_table_data  => 1,
+           no_blobs                 => 1,
+           no_privs                 => 1,
+           no_owner                 => 1,
+           pg_dumpall_dbprivs       => 1,
+           schema_only              => 1,
+           section_post_data        => 1,
+           with_oids                => 1, },
+       unlike => {
+           column_inserts           => 1,
+           data_only                => 1,
+           only_dump_test_table     => 1,
+           only_dump_test_schema    => 1,
+           pg_dumpall_globals       => 1,
+           pg_dumpall_globals_clean => 1,
+           role                     => 1,
+           section_data             => 1,
+           section_pre_data         => 1,
+           test_schema_plus_blobs   => 1, }, },
+
    'COMMENT ON TEXT SEARCH CONFIGURATION dump_test.alt_ts_conf1' => {
        all_runs     => 1,
        catch_all    => 'COMMENT commands',
@@ -1843,7 +1920,8 @@ my %tests = (
            data_only                => 1,
            exclude_dump_test_schema => 1,
            only_dump_test_table     => 1,
-           role                     => 1, }, },
+           role                     => 1,
+           section_post_data        => 1, }, },
 
    'COMMENT ON TEXT SEARCH DICTIONARY dump_test.alt_ts_dict1' => {
        all_runs     => 1,
@@ -1874,7 +1952,8 @@ my %tests = (
            data_only                => 1,
            exclude_dump_test_schema => 1,
            only_dump_test_table     => 1,
-           role                     => 1, }, },
+           role                     => 1,
+           section_post_data        => 1, }, },
 
    'COMMENT ON TEXT SEARCH PARSER dump_test.alt_ts_prs1' => {
        all_runs     => 1,
@@ -1905,7 +1984,8 @@ my %tests = (
            data_only                => 1,
            exclude_dump_test_schema => 1,
            only_dump_test_table     => 1,
-           role                     => 1, }, },
+           role                     => 1,
+           section_post_data        => 1, }, },
 
    'COMMENT ON TEXT SEARCH TEMPLATE dump_test.alt_ts_temp1' => {
        all_runs     => 1,
@@ -1936,7 +2016,8 @@ my %tests = (
            data_only                => 1,
            exclude_dump_test_schema => 1,
            only_dump_test_table     => 1,
-           role                     => 1, }, },
+           role                     => 1,
+           section_post_data        => 1, }, },
 
    'COMMENT ON TYPE dump_test.planets - ENUM' => {
        all_runs     => 1,
@@ -1967,7 +2048,8 @@ my %tests = (
            data_only                => 1,
            exclude_dump_test_schema => 1,
            only_dump_test_table     => 1,
-           role                     => 1, }, },
+           role                     => 1,
+           section_post_data        => 1, }, },
 
    'COMMENT ON TYPE dump_test.textrange - RANGE' => {
        all_runs     => 1,
@@ -1998,7 +2080,8 @@ my %tests = (
            data_only                => 1,
            exclude_dump_test_schema => 1,
            only_dump_test_table    => 1,
-           role                     => 1, }, },
+           role                     => 1,
+           section_post_data        => 1, }, },
 
    'COMMENT ON TYPE dump_test.int42 - Regular' => {
        all_runs     => 1,
@@ -2029,7 +2112,8 @@ my %tests = (
            data_only                => 1,
            exclude_dump_test_schema => 1,
            only_dump_test_table     => 1,
-           role                     => 1, }, },
+           role                     => 1,
+           section_post_data        => 1, }, },
 
    'COMMENT ON TYPE dump_test.undefined - Undefined' => {
        all_runs     => 1,
@@ -2060,7 +2144,8 @@ my %tests = (
            data_only                => 1,
            exclude_dump_test_schema => 1,
            only_dump_test_table     => 1,
-           role                     => 1, }, },
+           role                     => 1,
+           section_post_data        => 1, }, },
 
    # catch-all for COMMENTs
    'COMMENT commands' => {
@@ -2070,8 +2155,7 @@ my %tests = (
        unlike   => {
            pg_dumpall_globals       => 1,
            pg_dumpall_globals_clean => 1,
-           section_data             => 1,
-           section_post_data        => 1, }, },
+           section_data             => 1, }, },
 
    'COPY test_table' => {
        all_runs     => 1,