citus/src/test/regress/spec/isolation_global_pid.spec

97 lines
2.7 KiB
Ruby

#include "isolation_mx_common.include.spec"
setup
{
SET citus.next_shard_id TO 12345000;
CREATE TABLE dist_table (a INT, b INT);
SELECT create_distributed_table('dist_table', 'a', shard_count:=4);
}
teardown
{
DROP TABLE dist_table;
SELECT citus_internal.restore_isolation_tester_func();
}
session "s1"
step "s1-coordinator-begin"
{
BEGIN;
}
step "s1-coordinator-select"
{
SET citus.enable_local_execution TO off;
SET citus.force_max_query_parallelization TO ON;
SELECT * FROM dist_table;
}
step "s1-coordinator-commit"
{
COMMIT;
}
step "s1-start-session-level-connection"
{
SELECT start_session_level_connection_to_node('localhost', 57637);
}
step "s1-worker-begin"
{
SELECT run_commands_on_session_level_connection_to_node('BEGIN');
}
step "s1-worker-select"
{
SELECT run_commands_on_session_level_connection_to_node('SET citus.enable_local_execution TO off; SET citus.force_max_query_parallelization TO ON; SELECT * FROM dist_table');
}
step "s1-worker-commit"
{
SELECT run_commands_on_session_level_connection_to_node('COMMIT');
}
step "s1-stop-session-level-connection"
{
SELECT stop_session_level_connection_to_node();
}
session "s2"
step "s2-coordinator-citus_dist_stat_activity"
{
SELECT global_pid != 0 FROM citus_dist_stat_activity() WHERE query LIKE '%SELECT * FROM dist\_table%' and query NOT ILIKE '%run_commands_on_session_level_connection_to_node%';
}
step "s2-coordinator-citus_worker_stat_activity"
{
SELECT query FROM citus_worker_stat_activity() WHERE global_pid IN (
SELECT global_pid FROM citus_dist_stat_activity() WHERE query LIKE '%SELECT * FROM dist\_table%'
)
ORDER BY 1;
}
step "s2-coordinator-get_all_active_transactions"
{
SELECT count(*) FROM get_all_active_transactions() WHERE global_pid IN (
SELECT global_pid FROM citus_dist_stat_activity() WHERE query LIKE '%SELECT * FROM dist\_table%'
);
}
step "s2-coordinator-get_global_active_transactions"
{
SELECT count(*) FROM get_global_active_transactions() WHERE global_pid IN (
SELECT global_pid FROM citus_dist_stat_activity() WHERE query LIKE '%SELECT * FROM dist\_table%'
)
AND transaction_number != 0;
}
// worker - coordinator
permutation "s1-start-session-level-connection" "s1-worker-begin" "s1-worker-select" "s2-coordinator-citus_dist_stat_activity" "s2-coordinator-citus_worker_stat_activity" "s1-worker-commit" "s1-stop-session-level-connection"
// coordinator - coordinator
permutation "s1-coordinator-begin" "s1-coordinator-select" "s2-coordinator-citus_dist_stat_activity" "s2-coordinator-citus_worker_stat_activity" "s2-coordinator-get_all_active_transactions" "s2-coordinator-get_global_active_transactions" "s1-coordinator-commit"