external
/
citus
mirror of https://github.com/citusdata/citus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add simple test for background job

background-job-details
Nils Dijk 2022-07-18 14:18:48 +02:00 committed by Jelte Fennema
parent 12aec202d8
commit 5e7a4edd64
5 changed files with 48 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/backend/distributed/sql/udfs/citus_wait_for_rebalance_job/11.1-1.sql generated
View File

@ -2,7 +2,7 @@ CREATE FUNCTION pg_catalog.citus_wait_for_rebalance_job(jobid bigint)
RETURNS VOID
LANGUAGE C STRICT
AS 'MODULE_PATHNAME',$$citus_wait_for_rebalance_job$$;
COMMENT ON FUNCTION pg_catalog.citus_wait_for_rebalance_job()
COMMENT ON FUNCTION pg_catalog.citus_wait_for_rebalance_job(jobid bigint)
IS 'blocks till the job identified by jobid is at a terminal state, errors if no such job exists';
GRANT EXECUTE ON FUNCTION pg_catalog.citus_wait_for_rebalance_job() TO PUBLIC;
GRANT EXECUTE ON FUNCTION pg_catalog.citus_wait_for_rebalance_job(jobid bigint) TO PUBLIC;

4
src/backend/distributed/sql/udfs/citus_wait_for_rebalance_job/latest.sql
View File

@ -2,7 +2,7 @@ CREATE FUNCTION pg_catalog.citus_wait_for_rebalance_job(jobid bigint)
RETURNS VOID
LANGUAGE C STRICT
AS 'MODULE_PATHNAME',$$citus_wait_for_rebalance_job$$;
COMMENT ON FUNCTION pg_catalog.citus_wait_for_rebalance_job()
COMMENT ON FUNCTION pg_catalog.citus_wait_for_rebalance_job(jobid bigint)
IS 'blocks till the job identified by jobid is at a terminal state, errors if no such job exists';
GRANT EXECUTE ON FUNCTION pg_catalog.citus_wait_for_rebalance_job() TO PUBLIC;
GRANT EXECUTE ON FUNCTION pg_catalog.citus_wait_for_rebalance_job(jobid bigint) TO PUBLIC;

27
src/test/regress/expected/rebalance_job.out Normal file
View File

@ -0,0 +1,27 @@
CREATE SCHEMA rebalance_job;
SET search_path TO rebalance_job;
SET citus.shard_count TO 4;
SET citus.shard_replication_factor TO 1;
SET citus.next_shard_id TO 3536400;
CREATE TABLE results (a int);
-- simple job that inserts 1 into results to show that query runs
SELECT a FROM results WHERE a = 1; -- verify result is not in there
a
---------------------------------------------------------------------
(0 rows)
INSERT INTO pg_dist_rebalance_jobs (command) VALUES ($job$ INSERT INTO rebalance_job.results VALUES ( 1 ); $job$) RETURNING jobid \gset
SELECT citus_wait_for_rebalance_job(:jobid); -- wait for the job to be finished
citus_wait_for_rebalance_job
---------------------------------------------------------------------
(1 row)
SELECT a FROM results WHERE a = 1; -- verify result is there
a
---------------------------------------------------------------------
1
(1 row)
SET client_min_messages TO WARNING;
DROP SCHEMA rebalance_job CASCADE;

1
src/test/regress/multi_schedule
View File

@ -97,6 +97,7 @@ test: issue_5248 issue_5099
test: object_propagation_debug
test: undistribute_table
test: run_command_on_all_nodes
test: rebalance_job
# ---------

16
src/test/regress/sql/rebalance_job.sql Normal file
View File

@ -0,0 +1,16 @@
CREATE SCHEMA rebalance_job;
SET search_path TO rebalance_job;
SET citus.shard_count TO 4;
SET citus.shard_replication_factor TO 1;
SET citus.next_shard_id TO 3536400;
CREATE TABLE results (a int);
-- simple job that inserts 1 into results to show that query runs
SELECT a FROM results WHERE a = 1; -- verify result is not in there
INSERT INTO pg_dist_rebalance_jobs (command) VALUES ($job$ INSERT INTO rebalance_job.results VALUES ( 1 ); $job$) RETURNING jobid \gset
SELECT citus_wait_for_rebalance_job(:jobid); -- wait for the job to be finished
SELECT a FROM results WHERE a = 1; -- verify result is there
SET client_min_messages TO WARNING;
DROP SCHEMA rebalance_job CASCADE;