pgq_node.drop_node.sql: on local drop, delete all locations

diff --git a/sql/pgq_node/functions/pgq_node.drop_node.sql b/sql/pgq_node/functions/pgq_node.drop_node.sql
index 708f615229bba98d028e4bf4bdd31754752490de..fbf13ae7947de89061e0c9456ccebb2ed38133fc 100644 (file)
--- a/sql/pgq_node/functions/pgq_node.drop_node.sql
+++ b/sql/pgq_node/functions/pgq_node.drop_node.sql
@@ -56,11 +56,14 @@ begin
 
         perform pgq.drop_queue(queue_name, true)
            from pgq.queue where queue_name = i_queue_name;
-    end if;
 
-    delete from pgq_node.node_location
-     where queue_name = i_queue_name
-        and node_name = i_node_name;
+        delete from pgq_node.node_location
+         where queue_name = i_queue_name;
+    else
+        delete from pgq_node.node_location
+         where queue_name = i_queue_name
+            and node_name = i_node_name;
+    end if;
 
     select 200, 'Node dropped' into ret_code, ret_note;
     return;