#!/bin/bash

port_primary=5432
port_secondary=5433
port_third=5434
port_secondary_2=5436
port_secondary_3=5437
port_subscriber=5434
port_subscriber2=5435

echo '=========='
echo '=Clean up='
echo '=========='

pg_ctl stop -D data_primary
pg_ctl stop -D data_secondary
pg_ctl stop -D data_third_
pg_ctl stop -D data_secondary_2
pg_ctl stop -D data_secondary_3
pg_ctl stop -D data_subscriber
pg_ctl stop -D data_subscriber2

rm -rf data_* *log
rm -rf /home/houzj/archivedir/*

echo '======================='
echo '=Set up primary server='
echo '======================='

initdb -D data_primary

cat << EOF >> data_primary/postgresql.conf
wal_level = logical 
port = $port_primary
#standby_slot_names = 'physical'
#log_replication_commands = 'on'
#max_slot_wal_keep_size = 64kB

max_wal_senders=550
max_worker_processes=1000 
max_replication_slots=550 
log_replication_commands = 'on' 
checkpoint_timeout = 1d 
shared_buffers = 6GB 
max_worker_processes = 32 
max_parallel_maintenance_workers = 24 
max_parallel_workers = 32 
synchronous_commit = on 
checkpoint_timeout = 1d 
max_wal_size = 24GB 
min_wal_size = 15GB 
autovacuum = off

wal_sender_timeout = 6000s
wal_receiver_timeout = 6000s

#log_min_messages = 'debug2'

#archive_mode = on
#archive_command = 'cp %p /home/houzj/archivedir/%f'
#restore_command = 'cp /home/houzj/archivedir/%f %p'

max_prepared_transactions = 10

EOF

pg_ctl -D data_primary start -w -l primary.log 

psql -d postgres -p $port_primary -c "SELECT * FROM pg_create_physical_replication_slot('physical');"

echo '========================='
echo '=Set up secondary server='
echo '========================='

psql -d postgres -p $port_primary -c "CHECKPOINT;"

pg_basebackup -D data_secondary -p $port_primary

cat << EOF >> data_secondary/postgresql.conf
port = $port_secondary
primary_conninfo = 'port=$port_primary application_name=secondary dbname=postgres'
#primary_conninfo = 'port=$port_primary application_name=secondary dbname=postgreis'
primary_slot_name = 'physical'
hot_standby = on 
hot_standby_feedback = on
#sync_replication_slots = on
#standby_slot_names = ''
EOF

cat << EOF >> data_secondary/standby.signal
EOF


pg_ctl -D data_secondary start -w 
psql -d postgres -p $port_secondary -c "SELECT 'init' FROM pg_create_logical_replication_slot('stuck', 'pgoutput');" &

sleep 1

psql -d postgres -p $port_primary -c "CHECKPOINT;"


echo '==================='
echo '=Set up subscirber='
echo '==================='

initdb -D data_subscriber

cat << EOF >> data_subscriber/postgresql.conf
port = $port_subscriber


checkpoint_timeout = 1h
shared_buffers = '8GB'
wal_buffers = '1GB'
max_connections = '5000'
max_wal_size = 20GB
min_wal_size = 10GB
max_wal_senders = 100
max_replication_slots = 101
autovacuum = off

wal_sender_timeout = 6000s
wal_receiver_timeout = 6000s


max_prepared_transactions = 10
EOF

pg_ctl start -D data_subscriber


psql -d postgres -p $port_primary -c "create table tab1 (a int);CREATE PUBLICATION pub FOR TABLE tab1;"
psql -d postgres -p $port_primary -c "SELECT 'init' FROM pg_create_logical_replication_slot('sub', 'pgoutput', false, false, true);"

psql -d postgres -p $port_subscriber -c "create table tab1 (a int);"
psql -d postgres -p $port_subscriber -c "CREATE SUBSCRIPTION sub CONNECTION 'dbname=postgres port=$port_primary' PUBLICATION pub WITH (create_slot=false, copy_data = false, two_phase=false, failover, enabled=false)"

sleep 1

psql -d postgres -p $port_primary -c "INSERT INTO tab1 VALUES(1);"
psql -d postgres -p $port_primary -c "SELECT pg_current_wal_lsn();"
psql -d postgres -p $port_primary -c "SELECT pg_log_standby_snapshot();"
psql -d postgres -p $port_primary -c "SELECT txid_current();"
psql -d postgres -p $port_primary -c "BEGIN; INSERT INTO tab1 VALUES(2); PREPARE TRANSACTION 'test';"
psql -d postgres -p $port_primary -c "SELECT pg_log_standby_snapshot();"
psql -d postgres -p $port_primary -c "INSERT INTO tab1 VALUES(3);"

psql -d postgres -p $port_subscriber -c "alter subscription sub enable;"

sleep 1

psql -d postgres -p $port_secondary -c "select pg_sync_replication_slots();"

#psql -d postgres -p $port_primary -c "INSERT INTO tab1 VALUES(4);"

psql -d postgres -p $port_primary -c "BEGIN; INSERT INTO tab1 VALUES(11); PREPARE TRANSACTION 'bug_twophase';"

sleep 1

psql -d postgres -p $port_subscriber -c "alter subscription sub disable ; ;"
sleep 1
psql -d postgres -p $port_subscriber -c "alter subscription sub set (two_phase =on); alter subscription sub enable ;"

sleep 1

psql -d postgres -p $port_secondary -c "select pg_sync_replication_slots();"

sleep 1

pg_ctl promote -D data_secondary

sleep 1

psql -d postgres -p $port_secondary -c "COMMIT PREPARED 'bug_twophase';"



psql -d postgres -p $port_subscriber -c "ALTER SUBSCRIPTION sub CONNECTION 'dbname=postgres port=$port_secondary'"



#psql -d postgres -p $port_secondary -c "SELECT * FROM pg_replication_slots;"

#echo 'Moving lsns on primary so that slot-creation on secondary is successful'
#psql -d postgres -p $port_primary -c "INSERT INTO tbl VALUES (2)"
#psql -d postgres -p $port_primary -c "INSERT INTO tbl VALUES (3)"
#psql -d postgres -p $port_primary -c "INSERT INTO tbl VALUES (4)"