Add missing post create and alter hooks to policy objects.

jconway · jconway · commit 0bfbf14f93c3 · 2015-07-29T09:39:28.000-07:00
AlterPolicy() and CreatePolicy() lacked their respective hook invocations.
Noted by Noah Misch, review by Dean Rasheed. Back-patch to 9.5 where
RLS was introduced.
diff --git a/src/backend/commands/policy.c b/src/backend/commands/policy.c
@@ -629,6 +629,8 @@ CreatePolicy(CreatePolicyStmt *stmt)
 									 SHARED_DEPENDENCY_POLICY);
 	}
 
+	InvokeObjectPostCreateHook(PolicyRelationId, policy_id, 0);
+
 	/* Invalidate Relation Cache */
 	CacheInvalidateRelcache(target_table);
 
@@ -860,6 +862,8 @@ AlterPolicy(AlterPolicyStmt *stmt)
 									 SHARED_DEPENDENCY_POLICY);
 	}
 
+	InvokeObjectPostAlterHook(PolicyRelationId, policy_id, 0);
+
 	heap_freetuple(new_tuple);
 
 	/* Invalidate Relation Cache */

Original file line number	Diff line number	Diff line change
`@@ -629,6 +629,8 @@ CreatePolicy(CreatePolicyStmt *stmt)`
`629`	`629`	`SHARED_DEPENDENCY_POLICY);`
`630`	`630`	`}`
`631`	`631`
	`632`	`+ InvokeObjectPostCreateHook(PolicyRelationId, policy_id, 0);`
	`633`	`+`
`632`	`634`	`/* Invalidate Relation Cache */`
`633`	`635`	`CacheInvalidateRelcache(target_table);`
`634`	`636`
`@@ -860,6 +862,8 @@ AlterPolicy(AlterPolicyStmt *stmt)`
`860`	`862`	`SHARED_DEPENDENCY_POLICY);`
`861`	`863`	`}`
`862`	`864`
	`865`	`+ InvokeObjectPostAlterHook(PolicyRelationId, policy_id, 0);`
	`866`	`+`
`863`	`867`	`heap_freetuple(new_tuple);`
`864`	`868`
`865`	`869`	`/* Invalidate Relation Cache */`