88 lines
2.8 KiB
Perl
88 lines
2.8 KiB
Perl
|
|
# Copyright (c) 2021-2023, PostgreSQL Global Development Group
|
|
|
|
# Test streaming of transaction with subtransactions, DDLs, DMLs, and
|
|
# rollbacks
|
|
#
|
|
# This file is mainly to test the DDL/DML interaction of the publisher side,
|
|
# so we didn't add a parallel apply version for the tests in this file.
|
|
use strict;
|
|
use warnings;
|
|
use PostgreSQL::Test::Cluster;
|
|
use PostgreSQL::Test::Utils;
|
|
use Test::More;
|
|
|
|
# Create publisher node
|
|
my $node_publisher = PostgreSQL::Test::Cluster->new('publisher');
|
|
$node_publisher->init(allows_streaming => 'logical');
|
|
$node_publisher->append_conf('postgresql.conf',
|
|
'logical_replication_mode = immediate');
|
|
$node_publisher->start;
|
|
|
|
# Create subscriber node
|
|
my $node_subscriber = PostgreSQL::Test::Cluster->new('subscriber');
|
|
$node_subscriber->init(allows_streaming => 'logical');
|
|
$node_subscriber->start;
|
|
|
|
# Create some preexisting content on publisher
|
|
$node_publisher->safe_psql('postgres',
|
|
"CREATE TABLE test_tab (a int primary key, b bytea)");
|
|
$node_publisher->safe_psql('postgres',
|
|
"INSERT INTO test_tab VALUES (1, 'foo'), (2, 'bar')");
|
|
|
|
# Setup structure on subscriber
|
|
$node_subscriber->safe_psql('postgres',
|
|
"CREATE TABLE test_tab (a int primary key, b bytea, c INT, d INT, e INT)");
|
|
|
|
# Setup logical replication
|
|
my $publisher_connstr = $node_publisher->connstr . ' dbname=postgres';
|
|
$node_publisher->safe_psql('postgres',
|
|
"CREATE PUBLICATION tap_pub FOR TABLE test_tab");
|
|
|
|
my $appname = 'tap_sub';
|
|
$node_subscriber->safe_psql('postgres',
|
|
"CREATE SUBSCRIPTION tap_sub CONNECTION '$publisher_connstr application_name=$appname' PUBLICATION tap_pub WITH (streaming = on)"
|
|
);
|
|
|
|
# Wait for initial table sync to finish
|
|
$node_subscriber->wait_for_subscription_sync($node_publisher, $appname);
|
|
|
|
my $result =
|
|
$node_subscriber->safe_psql('postgres',
|
|
"SELECT count(*), count(c) FROM test_tab");
|
|
is($result, qq(2|0), 'check initial data was copied to subscriber');
|
|
|
|
# streamed transaction with DDL, DML and ROLLBACKs
|
|
$node_publisher->safe_psql(
|
|
'postgres', q{
|
|
BEGIN;
|
|
INSERT INTO test_tab VALUES (3, sha256(3::text::bytea));
|
|
ALTER TABLE test_tab ADD COLUMN c INT;
|
|
SAVEPOINT s1;
|
|
INSERT INTO test_tab VALUES (4, sha256(4::text::bytea), -4);
|
|
ALTER TABLE test_tab ADD COLUMN d INT;
|
|
SAVEPOINT s2;
|
|
INSERT INTO test_tab VALUES (5, sha256(5::text::bytea), -5, 5*2);
|
|
ALTER TABLE test_tab ADD COLUMN e INT;
|
|
SAVEPOINT s3;
|
|
INSERT INTO test_tab VALUES (6, sha256(6::text::bytea), -6, 6*2, -6*3);
|
|
ALTER TABLE test_tab DROP COLUMN c;
|
|
ROLLBACK TO s1;
|
|
INSERT INTO test_tab VALUES (4, sha256(4::text::bytea), 4);
|
|
COMMIT;
|
|
});
|
|
|
|
$node_publisher->wait_for_catchup($appname);
|
|
|
|
$result =
|
|
$node_subscriber->safe_psql('postgres',
|
|
"SELECT count(*), count(c) FROM test_tab");
|
|
is($result, qq(4|1),
|
|
'check rollback to savepoint was reflected on subscriber and extra columns contain local defaults'
|
|
);
|
|
|
|
$node_subscriber->stop;
|
|
$node_publisher->stop;
|
|
|
|
done_testing();
|