Fixes test issues

citusdata · Aug 17, 2023 · f641860 · f641860
1 parent 458edd8
commit f641860
Show file tree

Hide file tree

Showing 6 changed files with 355 additions and 241 deletions.
diff --git a/src/backend/distributed/sql/downgrades/citus--12.1-1--12.0-1.sql b/src/backend/distributed/sql/downgrades/citus--12.1-1--12.0-1.sql
@@ -1,7 +1,7 @@
 -- citus--12.1-1--12.0-1
 
 
-DROP FUNCTION pg_catalog.citus_pause_node_within_txn(int);
+DROP FUNCTION pg_catalog.citus_pause_node_within_txn(int,bool,int);
 
 
 -- we have modified the relevant upgrade script to include any_value changes

diff --git a/src/backend/distributed/sql/udfs/citus_pause_node_within_txn/12.1-1.sql b/src/backend/distributed/sql/udfs/citus_pause_node_within_txn/12.1-1.sql
diff --git a/src/backend/distributed/sql/udfs/citus_pause_node_within_txn/latest.sql b/src/backend/distributed/sql/udfs/citus_pause_node_within_txn/latest.sql
@@ -1,9 +1,11 @@
-CREATE FUNCTION pg_catalog.citus_pause_node_within_txn(node_id int)
+CREATE FUNCTION pg_catalog.citus_pause_node_within_txn(node_id int,force bool ,lock_cooldown int)
   RETURNS void
   LANGUAGE C STRICT
   AS 'MODULE_PATHNAME', $$citus_pause_node_within_txn$$;
 
-COMMENT ON FUNCTION pg_catalog.citus_pause_node_within_txn(node_id int)
+COMMENT ON FUNCTION pg_catalog.citus_pause_node_within_txn(node_id int,
+                                              force bool,
+                                              lock_cooldown int )
   IS 'pauses node with given id which leads to add lock in tables and prevent any queries to be executed on that node';
 
-REVOKE ALL ON FUNCTION pg_catalog.citus_pause_node_within_txn(int) FROM PUBLIC;
+REVOKE ALL ON FUNCTION pg_catalog.citus_pause_node_within_txn(int,bool,int) FROM PUBLIC;