isolation bench

Author: kelvich

contrib/postgres_fdw/Makefile

@@ -12,6 +12,8 @@ DATA = postgres_fdw--1.0.sql
 
 REGRESS = postgres_fdw
 
+EXTRA_INSTALL += contrib/pg_tsdtm
+
 ifdef USE_PGXS
 PG_CONFIG = pg_config
 PGXS := $(shell $(PG_CONFIG) --pgxs)
@@ -23,3 +25,9 @@ top_builddir = ../..
 include $(top_builddir)/src/Makefile.global
 include $(top_srcdir)/contrib/contrib-global.mk
 endif
+
+# Poor's man overload of already defined target: just copy it
+check: submake $(REGRESS_PREP)
+	$(pg_regress_check) $(REGRESS_OPTS) $(REGRESS)
+	env DESTDIR='$(abs_top_builddir)'/tmp_install make install
+	$(prove_check)

contrib/postgres_fdw/t/001_bank_check.pl

@@ -0,0 +1,86 @@
+use strict;
+use warnings;
+
+use PostgresNode;
+use TestLib;
+use Test::More tests => 1;
+
+my $master = get_new_node("master");
+$master->init;
+$master->append_conf('postgresql.conf', qq(
+	max_prepared_transactions = 10
+	log_checkpoints = true
+	postgres_fdw.use_tsdtm = on
+));
+$master->start;
+
+my $shard1 = get_new_node("shard1");
+$shard1->init;
+$shard1->append_conf('postgresql.conf', qq(
+	max_prepared_transactions = 10
+	log_checkpoints = true
+	shared_preload_libraries = 'pg_tsdtm'
+));
+$shard1->start;
+
+my $shard2 = get_new_node("shard2");
+$shard2->init;
+$shard2->append_conf('postgresql.conf', qq(
+	max_prepared_transactions = 10
+	log_checkpoints = true
+	shared_preload_libraries = 'pg_tsdtm'
+));
+$shard2->start;
+
+###############################################################################
+
+$master->psql('postgres', "CREATE EXTENSION postgres_fdw");
+$master->psql('postgres', "CREATE TABLE accounts(id integer primary key, amount integer)");
+
+foreach my $node ($shard1, $shard2)
+{
+	my $port = $node->port;
+	my $host = $node->host;
+
+	$node->psql('postgres', "CREATE EXTENSION pg_tsdtm");
+	$node->psql('postgres', "CREATE TABLE accounts(id integer primary key, amount integer)");
+
+	$master->psql('postgres', "CREATE SERVER shard_$port FOREIGN DATA WRAPPER postgres_fdw options(dbname 'postgres', host '$host', port '$port')");
+	$master->psql('postgres', "CREATE FOREIGN TABLE accounts_fdw_$port() inherits (accounts) server shard_$port options(table_name 'accounts')");
+	$master->psql('postgres', "CREATE USER MAPPING for stas SERVER shard_$port options (user 'stas')");
+
+	diag("done $host $port");
+}
+
+$shard1->psql('postgres', "insert into accounts select 2*id-1, 0 from generate_series(1, 1000) as id;");
+$shard2->psql('postgres', "insert into accounts select 2*id, 0 from generate_series(1, 1000) as id;");
+
+###############################################################################
+
+my ($err, $rc);
+my $seconds = 30;
+my $total = 0;
+my $oldtotal = 0;
+my $isolation_error = 0;
+
+my $pgb_handle = $master->pgbench_async(-n, -c => 1, -T => $seconds, -f => "$TestLib::log_path/../../t/bank.pgb", 'postgres' );
+
+my $started = time();
+while (time() - $started < $seconds)
+{
+	($rc, $total, $err) = $master->psql('postgres', "select sum(amount) from accounts");
+	if ($oldtotal != $total)
+	{
+		$isolation_error = 1;
+		$oldtotal = $total;
+		diag("Isolation error. Total = $total");
+	}
+}
+
+$master->pgbench_await($pgb_handle);
+
+is($isolation_error, 0, 'check proper isolation');
+
+$master->stop;
+$shard1->stop;
+$shard2->stop;

contrib/postgres_fdw/t/bank.pgb

@@ -0,0 +1,6 @@
+\set id random(1, 2000)
+
+BEGIN;
+UPDATE accounts SET amount = amount - 1 WHERE id = :id;
+UPDATE accounts SET amount = amount + 1 WHERE id = (:id + 1);
+COMMIT;

src/test/perl/PostgresNode.pm

@@ -1728,6 +1728,37 @@ sub pg_recvlogical_upto
 	}
 }
 
+sub pgbench()
+{
+	my ($self, $node, @args) = @_;
+	my $pgbench_handle = $self->pgbench_async($node, @args);
+	$self->pgbench_await($pgbench_handle);
+}
+
+sub pgbench_async()
+{
+	my ($self, @args) = @_;
+
+	my ($in, $out, $err, $rc);
+	$in = '';
+	$out = '';
+
+	my @pgbench_command = (
+		'pgbench',
+		-h => $self->host,
+		-p => $self->port,
+		@args
+	);
+	my $handle = IPC::Run::start(\@pgbench_command, $in, $out);
+	return $handle;
+}
+
+sub pgbench_await()
+{
+	my ($self, $pgbench_handle) = @_;
+	IPC::Run::finish($pgbench_handle) || BAIL_OUT("pgbench exited with $?");
+}
+
 =pod
 
 =back