mirror of https://github.com/citusdata/citus.git
169 lines
6.4 KiB
Ruby
169 lines
6.4 KiB
Ruby
# Create and use UDF to send commands from the same connection. Also make the cluster
|
|
# ready for testing MX functionalities.
|
|
setup
|
|
{
|
|
CREATE OR REPLACE FUNCTION start_session_level_connection_to_node(text, integer)
|
|
RETURNS void
|
|
LANGUAGE C STRICT VOLATILE
|
|
AS 'citus', $$start_session_level_connection_to_node$$;
|
|
|
|
CREATE OR REPLACE FUNCTION run_commands_on_session_level_connection_to_node(text)
|
|
RETURNS void
|
|
LANGUAGE C STRICT VOLATILE
|
|
AS 'citus', $$run_commands_on_session_level_connection_to_node$$;
|
|
|
|
CREATE OR REPLACE FUNCTION stop_session_level_connection_to_node()
|
|
RETURNS void
|
|
LANGUAGE C STRICT VOLATILE
|
|
AS 'citus', $$stop_session_level_connection_to_node$$;
|
|
|
|
SELECT citus.replace_isolation_tester_func();
|
|
SELECT citus.refresh_isolation_tester_prepared_statement();
|
|
|
|
-- start_metadata_sync_to_node can not be run inside a transaction block
|
|
-- following is a workaround to overcome that
|
|
-- port numbers are hard coded at the moment
|
|
SELECT master_run_on_worker(
|
|
ARRAY['localhost']::text[],
|
|
ARRAY[57636]::int[],
|
|
ARRAY[format('SELECT start_metadata_sync_to_node(''%s'', %s)', nodename, nodeport)]::text[],
|
|
false)
|
|
FROM pg_dist_node;
|
|
|
|
SET citus.replication_model to streaming;
|
|
|
|
CREATE TABLE ref_table(user_id int, value_1 int);
|
|
SELECT create_reference_table('ref_table');
|
|
INSERT INTO ref_table VALUES (1, 11), (2, 21), (3, 31), (4, 41), (5, 51), (6, 61), (7, 71);
|
|
}
|
|
|
|
# Create and use UDF to close the connection opened in the setup step. Also return the cluster
|
|
# back to the initial state.
|
|
teardown
|
|
{
|
|
DROP TABLE ref_table;
|
|
SELECT citus.restore_isolation_tester_func();
|
|
}
|
|
|
|
session "s1"
|
|
|
|
step "s1-begin"
|
|
{
|
|
BEGIN;
|
|
}
|
|
|
|
# We do not need to begin a transaction on coordinator, since it will be open on workers.
|
|
|
|
step "s1-start-session-level-connection"
|
|
{
|
|
SELECT start_session_level_connection_to_node('localhost', 57637);
|
|
}
|
|
|
|
step "s1-begin-on-worker"
|
|
{
|
|
SELECT run_commands_on_session_level_connection_to_node('BEGIN');
|
|
}
|
|
|
|
step "s1-update-ref-table"
|
|
{
|
|
SELECT run_commands_on_session_level_connection_to_node('UPDATE ref_table SET value_1 = 12 WHERE user_id = 1');
|
|
}
|
|
|
|
step "s1-delete-from-ref-table"
|
|
{
|
|
SELECT run_commands_on_session_level_connection_to_node('DELETE FROM ref_table WHERE user_id = 1');
|
|
}
|
|
|
|
step "s1-insert-into-ref-table"
|
|
{
|
|
SELECT run_commands_on_session_level_connection_to_node('INSERT INTO ref_table VALUES(8,81),(9,91)');
|
|
}
|
|
|
|
step "s1-copy-to-ref-table"
|
|
{
|
|
SELECT run_commands_on_session_level_connection_to_node('COPY ref_table FROM PROGRAM ''echo 10, 101 && echo 11, 111'' WITH CSV');
|
|
}
|
|
|
|
step "s1-select-for-update"
|
|
{
|
|
SELECT run_commands_on_session_level_connection_to_node('SELECT * FROM ref_table FOR UPDATE');
|
|
}
|
|
|
|
step "s1-commit-worker"
|
|
{
|
|
SELECT run_commands_on_session_level_connection_to_node('COMMIT');
|
|
}
|
|
|
|
step "s1-alter-table"
|
|
{
|
|
ALTER TABLE ref_table ADD CONSTRAINT rf_p_key PRIMARY KEY(user_id);
|
|
}
|
|
|
|
step "s1-stop-connection"
|
|
{
|
|
SELECT stop_session_level_connection_to_node();
|
|
}
|
|
|
|
step "s1-commit"
|
|
{
|
|
COMMIT;
|
|
}
|
|
|
|
session "s2"
|
|
|
|
step "s2-start-session-level-connection"
|
|
{
|
|
SELECT start_session_level_connection_to_node('localhost', 57638);
|
|
}
|
|
|
|
step "s2-begin-on-worker"
|
|
{
|
|
SELECT run_commands_on_session_level_connection_to_node('BEGIN');
|
|
}
|
|
|
|
step "s2-update-ref-table"
|
|
{
|
|
SELECT run_commands_on_session_level_connection_to_node('UPDATE ref_table SET value_1 = 12 WHERE user_id = 1');
|
|
}
|
|
|
|
step "s2-select-from-ref-table"
|
|
{
|
|
SELECT run_commands_on_session_level_connection_to_node('SELECT count(*) FROM ref_table');
|
|
}
|
|
|
|
step "s2-delete-from-ref-table"
|
|
{
|
|
SELECT run_commands_on_session_level_connection_to_node('DELETE FROM ref_table WHERE user_id = 2');
|
|
}
|
|
|
|
step "s2-insert-into-ref-table"
|
|
{
|
|
SELECT run_commands_on_session_level_connection_to_node('INSERT INTO ref_table VALUES(8,81),(9,91)');
|
|
}
|
|
|
|
step "s2-copy-to-ref-table"
|
|
{
|
|
SELECT run_commands_on_session_level_connection_to_node('COPY ref_table FROM PROGRAM ''echo 10, 101 && echo 11, 111'' WITH CSV');
|
|
}
|
|
|
|
step "s2-stop-connection"
|
|
{
|
|
SELECT stop_session_level_connection_to_node();
|
|
}
|
|
|
|
step "s2-commit-worker"
|
|
{
|
|
SELECT run_commands_on_session_level_connection_to_node('COMMIT');
|
|
}
|
|
|
|
permutation "s1-start-session-level-connection" "s1-begin-on-worker" "s1-update-ref-table" "s2-start-session-level-connection" "s2-begin-on-worker" "s2-update-ref-table" "s1-commit-worker" "s2-commit-worker" "s1-stop-connection" "s2-stop-connection"
|
|
permutation "s1-start-session-level-connection" "s1-begin-on-worker" "s1-delete-from-ref-table" "s2-start-session-level-connection" "s2-begin-on-worker" "s2-update-ref-table" "s1-commit-worker" "s2-commit-worker" "s1-stop-connection" "s2-stop-connection"
|
|
permutation "s1-start-session-level-connection" "s1-begin-on-worker" "s1-insert-into-ref-table" "s2-start-session-level-connection" "s2-begin-on-worker" "s2-update-ref-table" "s1-commit-worker" "s2-commit-worker" "s1-stop-connection" "s2-stop-connection"
|
|
permutation "s1-start-session-level-connection" "s1-begin-on-worker" "s1-insert-into-ref-table" "s2-start-session-level-connection" "s2-begin-on-worker" "s2-insert-into-ref-table" "s1-commit-worker" "s2-commit-worker" "s1-stop-connection" "s2-stop-connection"
|
|
permutation "s1-start-session-level-connection" "s1-begin-on-worker" "s1-copy-to-ref-table" "s2-start-session-level-connection" "s2-begin-on-worker" "s2-update-ref-table" "s1-commit-worker" "s2-commit-worker" "s1-stop-connection" "s2-stop-connection"
|
|
permutation "s1-start-session-level-connection" "s1-begin-on-worker" "s1-copy-to-ref-table" "s2-start-session-level-connection" "s2-begin-on-worker" "s2-insert-into-ref-table" "s1-commit-worker" "s2-commit-worker" "s1-stop-connection" "s2-stop-connection"
|
|
permutation "s1-start-session-level-connection" "s1-begin-on-worker" "s1-copy-to-ref-table" "s2-start-session-level-connection" "s2-begin-on-worker" "s2-copy-to-ref-table" "s1-commit-worker" "s2-commit-worker" "s1-stop-connection" "s2-stop-connection"
|
|
permutation "s1-start-session-level-connection" "s1-begin-on-worker" "s1-select-for-update" "s2-start-session-level-connection" "s2-begin-on-worker" "s2-update-ref-table" "s1-commit-worker" "s2-commit-worker" "s1-stop-connection" "s2-stop-connection"
|
|
permutation "s2-start-session-level-connection" "s2-begin-on-worker" "s2-insert-into-ref-table" "s1-begin" "s1-alter-table" "s2-commit-worker" "s1-commit" "s2-stop-connection"
|
|
permutation "s2-start-session-level-connection" "s2-begin-on-worker" "s2-select-from-ref-table" "s1-begin" "s1-alter-table" "s2-commit-worker" "s1-commit" "s2-stop-connection"
|