From 91d97462c96e07a6885261c9d46554b1c4ce834b Mon Sep 17 00:00:00 2001 From: Alvaro Herrera Date: Wed, 6 May 2020 12:29:41 -0400 Subject: [PATCH] Heed lock protocol in DROP OWNED BY MIME-Version: 1.0 Content-Type: text/plain; charset=utf8 Content-Transfer-Encoding: 8bit We were acquiring object locks then deleting objects one by one, instead of acquiring all object locks first, ignoring those that did not exist, and then deleting all objects together. The latter is the correct protocol to use, and what this commits changes to code to do. Failing to follow that leads to "cache lookup failed for relation XYZ" error reports when DROP OWNED runs concurrently with other DDL -- for example, a session termination that removes some temp tables. Author: Álvaro Herrera Reported-by: Mithun Chicklore Yogendra (Mithun CY) Reviewed-by: Ahsan Hadi, Tom Lane Discussion: https://postgr.es/m/CADq3xVZTbzK4ZLKq+dn_vB4QafXXbmMgDP3trY-GuLnib2Ai1w@mail.gmail.com --- src/backend/catalog/dependency.c | 15 +++++++++------ src/backend/catalog/pg_shdepend.c | 24 +++++++++++++++++++++++- src/include/catalog/dependency.h | 4 ++++ 3 files changed, 36 insertions(+), 7 deletions(-) diff --git a/src/backend/catalog/dependency.c b/src/backend/catalog/dependency.c index 6e1289c7ee8..fa9e11cde3c 100644 --- a/src/backend/catalog/dependency.c +++ b/src/backend/catalog/dependency.c @@ -176,8 +176,6 @@ static void reportDependentObjects(const ObjectAddresses *targetObjects, static void deleteOneObject(const ObjectAddress *object, Relation *depRel, int32 flags); static void doDeletion(const ObjectAddress *object, int flags); -static void AcquireDeletionLock(const ObjectAddress *object, int flags); -static void ReleaseDeletionLock(const ObjectAddress *object); static bool find_expr_references_walker(Node *node, find_expr_references_context *context); static void eliminate_duplicate_dependencies(ObjectAddresses *addrs); @@ -1279,11 +1277,14 @@ doDeletion(const ObjectAddress *object, int flags) /* * AcquireDeletionLock - acquire a suitable lock for deleting an object * + * Accepts the same flags as performDeletion (though currently only + * PERFORM_DELETION_CONCURRENTLY does anything). + * * We use LockRelation for relations, LockDatabaseObject for everything - * else. Note that dependency.c is not concerned with deleting any kind of - * shared-across-databases object, so we have no need for LockSharedObject. + * else. Shared-across-databases objects are not currently supported + * because no caller cares, but could be modified to use LockSharedObject. */ -static void +void AcquireDeletionLock(const ObjectAddress *object, int flags) { if (object->classId == RelationRelationId) @@ -1309,8 +1310,10 @@ AcquireDeletionLock(const ObjectAddress *object, int flags) /* * ReleaseDeletionLock - release an object deletion lock + * + * Companion to AcquireDeletionLock. */ -static void +void ReleaseDeletionLock(const ObjectAddress *object) { if (object->classId == RelationRelationId) diff --git a/src/backend/catalog/pg_shdepend.c b/src/backend/catalog/pg_shdepend.c index c3cce67a8b6..eed74b755b4 100644 --- a/src/backend/catalog/pg_shdepend.c +++ b/src/backend/catalog/pg_shdepend.c @@ -1248,7 +1248,10 @@ shdepDropOwned(List *roleids, DropBehavior behavior) sdepForm->objid); break; case SHARED_DEPENDENCY_POLICY: - /* If unable to remove role from policy, remove policy. */ + /* + * Try to remove role from policy; if unable to, remove + * policy. + */ if (!RemoveRoleFromObjectPolicy(roleid, sdepForm->classid, sdepForm->objid)) @@ -1256,6 +1259,18 @@ shdepDropOwned(List *roleids, DropBehavior behavior) obj.classId = sdepForm->classid; obj.objectId = sdepForm->objid; obj.objectSubId = sdepForm->objsubid; + /* + * Acquire lock on object, then verify this dependency + * is still relevant. If not, the object might have + * been dropped or the policy modified. Ignore the + * object in that case. + */ + AcquireDeletionLock(&obj, 0); + if (!systable_recheck_tuple(scan, tuple)) + { + ReleaseDeletionLock(&obj); + break; + } add_exact_object_address(&obj, deleteobjs); } break; @@ -1266,6 +1281,13 @@ shdepDropOwned(List *roleids, DropBehavior behavior) obj.classId = sdepForm->classid; obj.objectId = sdepForm->objid; obj.objectSubId = sdepForm->objsubid; + /* as above */ + AcquireDeletionLock(&obj, 0); + if (!systable_recheck_tuple(scan, tuple)) + { + ReleaseDeletionLock(&obj); + break; + } add_exact_object_address(&obj, deleteobjs); } break; diff --git a/src/include/catalog/dependency.h b/src/include/catalog/dependency.h index fbcf9044325..a690eac41c5 100644 --- a/src/include/catalog/dependency.h +++ b/src/include/catalog/dependency.h @@ -164,6 +164,10 @@ typedef enum ObjectClass #define PERFORM_DELETION_INTERNAL 0x0001 #define PERFORM_DELETION_CONCURRENTLY 0x0002 +extern void AcquireDeletionLock(const ObjectAddress *object, int flags); + +extern void ReleaseDeletionLock(const ObjectAddress *object); + extern void performDeletion(const ObjectAddress *object, DropBehavior behavior, int flags); -- 2.39.5