Fix pg_dump's handling of event triggers.

tglsfdc · tglsfdc · commit 502ead3d6176 · 2017-07-22T20:20:10.000-04:00
pg_dump with the --clean option failed to emit DROP EVENT TRIGGER commands for event triggers. In a closely related oversight, it also did not emit ALTER OWNER commands for event triggers. Since only superusers can create event triggers, the latter oversight is of little practical consequence ... but if we're going to record an owner for event triggers, then surely pg_dump should preserve it. Per complaint from Greg Atkins. Back-patch to 9.3 where event triggers were introduced. Discussion: https://postgr.es/m/20170722191142.yi4e7tzcg3iacclg@gmail.com
diff --git a/src/bin/pg_dump/pg_backup_archiver.c b/src/bin/pg_dump/pg_backup_archiver.c
@@ -3093,6 +3093,7 @@ _getObjectDescription(PQExpBuffer buf, TocEntry *te, ArchiveHandle *AH)
 		strcmp(type, "DATABASE") == 0 ||
 		strcmp(type, "PROCEDURAL LANGUAGE") == 0 ||
 		strcmp(type, "SCHEMA") == 0 ||
+		strcmp(type, "EVENT TRIGGER") == 0 ||
 		strcmp(type, "FOREIGN DATA WRAPPER") == 0 ||
 		strcmp(type, "SERVER") == 0 ||
 		strcmp(type, "USER MAPPING") == 0)
@@ -3137,7 +3138,7 @@ _getObjectDescription(PQExpBuffer buf, TocEntry *te, ArchiveHandle *AH)
 		return;
 	}
 
-	write_msg(modulename, "WARNING: don't know how to set owner for object type %s\n",
+	write_msg(modulename, "WARNING: don't know how to set owner for object type \"%s\"\n",
 			  type);
 }
 
@@ -3294,6 +3295,7 @@ _printTocEntry(ArchiveHandle *AH, TocEntry *te, RestoreOptions *ropt, bool isDat
 			strcmp(te->desc, "OPERATOR FAMILY") == 0 ||
 			strcmp(te->desc, "PROCEDURAL LANGUAGE") == 0 ||
 			strcmp(te->desc, "SCHEMA") == 0 ||
+			strcmp(te->desc, "EVENT TRIGGER") == 0 ||
 			strcmp(te->desc, "TABLE") == 0 ||
 			strcmp(te->desc, "TYPE") == 0 ||
 			strcmp(te->desc, "VIEW") == 0 ||
@@ -3327,7 +3329,7 @@ _printTocEntry(ArchiveHandle *AH, TocEntry *te, RestoreOptions *ropt, bool isDat
 		}
 		else
 		{
-			write_msg(modulename, "WARNING: don't know how to set owner for object type %s\n",
+			write_msg(modulename, "WARNING: don't know how to set owner for object type \"%s\"\n",
 					  te->desc);
 		}
 	}
diff --git a/src/bin/pg_dump/pg_dump.c b/src/bin/pg_dump/pg_dump.c
@@ -14833,13 +14833,15 @@ static void
 dumpEventTrigger(Archive *fout, EventTriggerInfo *evtinfo)
 {
 	PQExpBuffer query;
+	PQExpBuffer delqry;
 	PQExpBuffer labelq;
 
 	/* Skip if not to be dumped */
 	if (!evtinfo->dobj.dump || dataOnly)
 		return;
 
 	query = createPQExpBuffer();
+	delqry = createPQExpBuffer();
 	labelq = createPQExpBuffer();
 
 	appendPQExpBufferStr(query, "CREATE EVENT TRIGGER ");
@@ -14879,19 +14881,27 @@ dumpEventTrigger(Archive *fout, EventTriggerInfo *evtinfo)
 		}
 		appendPQExpBufferStr(query, ";\n");
 	}
+
+	appendPQExpBuffer(delqry, "DROP EVENT TRIGGER %s;\n",
+					  fmtId(evtinfo->dobj.name));
+
 	appendPQExpBuffer(labelq, "EVENT TRIGGER %s",
 					  fmtId(evtinfo->dobj.name));
 
 	ArchiveEntry(fout, evtinfo->dobj.catId, evtinfo->dobj.dumpId,
-				 evtinfo->dobj.name, NULL, NULL, evtinfo->evtowner, false,
+				 evtinfo->dobj.name, NULL, NULL,
+				 evtinfo->evtowner, false,
 				 "EVENT TRIGGER", SECTION_POST_DATA,
-				 query->data, "", NULL, NULL, 0, NULL, NULL);
+				 query->data, delqry->data, NULL,
+				 NULL, 0,
+				 NULL, NULL);
 
 	dumpComment(fout, labelq->data,
 				NULL, evtinfo->evtowner,
 				evtinfo->dobj.catId, 0, evtinfo->dobj.dumpId);
 
 	destroyPQExpBuffer(query);
+	destroyPQExpBuffer(delqry);
 	destroyPQExpBuffer(labelq);
 }
 

Original file line number	Diff line number	Diff line change
`@@ -3093,6 +3093,7 @@ _getObjectDescription(PQExpBuffer buf, TocEntry te, ArchiveHandle AH)`
`3093`	`3093`	`strcmp(type, "DATABASE") == 0 \|\|`
`3094`	`3094`	`strcmp(type, "PROCEDURAL LANGUAGE") == 0 \|\|`
`3095`	`3095`	`strcmp(type, "SCHEMA") == 0 \|\|`
	`3096`	`+ strcmp(type, "EVENT TRIGGER") == 0 \|\|`
`3096`	`3097`	`strcmp(type, "FOREIGN DATA WRAPPER") == 0 \|\|`
`3097`	`3098`	`strcmp(type, "SERVER") == 0 \|\|`
`3098`	`3099`	`strcmp(type, "USER MAPPING") == 0)`
`@@ -3137,7 +3138,7 @@ _getObjectDescription(PQExpBuffer buf, TocEntry te, ArchiveHandle AH)`
`3137`	`3138`	`return;`
`3138`	`3139`	`}`
`3139`	`3140`
`3140`		`- write_msg(modulename, "WARNING: don't know how to set owner for object type %s\n",`
	`3141`	`+ write_msg(modulename, "WARNING: don't know how to set owner for object type \"%s\"\n",`
`3141`	`3142`	`type);`
`3142`	`3143`	`}`
`3143`	`3144`
`@@ -3294,6 +3295,7 @@ _printTocEntry(ArchiveHandle AH, TocEntry te, RestoreOptions *ropt, bool isDat`
`3294`	`3295`	`strcmp(te->desc, "OPERATOR FAMILY") == 0 \|\|`
`3295`	`3296`	`strcmp(te->desc, "PROCEDURAL LANGUAGE") == 0 \|\|`
`3296`	`3297`	`strcmp(te->desc, "SCHEMA") == 0 \|\|`
	`3298`	`+ strcmp(te->desc, "EVENT TRIGGER") == 0 \|\|`
`3297`	`3299`	`strcmp(te->desc, "TABLE") == 0 \|\|`
`3298`	`3300`	`strcmp(te->desc, "TYPE") == 0 \|\|`
`3299`	`3301`	`strcmp(te->desc, "VIEW") == 0 \|\|`
`@@ -3327,7 +3329,7 @@ _printTocEntry(ArchiveHandle AH, TocEntry te, RestoreOptions *ropt, bool isDat`
`3327`	`3329`	`}`
`3328`	`3330`	`else`
`3329`	`3331`	`{`
`3330`		`- write_msg(modulename, "WARNING: don't know how to set owner for object type %s\n",`
	`3332`	`+ write_msg(modulename, "WARNING: don't know how to set owner for object type \"%s\"\n",`
`3331`	`3333`	`te->desc);`
`3332`	`3334`	`}`
`3333`	`3335`	`}`