#!/usr/bin/env bash
# ?Hace falta el injection point, o la carrera se gana sola?
#
# Con el punto de inyeccion el bug sale siempre, porque REPACK queda detenido
# justo antes de tomar el lock.  Sin el, hay que meter el UPDATE en la ventana
# que va desde que la transaccion vieja commitea hasta que REPACK copia la
# tabla.  Aca se intenta N veces, con el UPDATE disparado inmediatamente
# despues del COMMIT, sin esperas.
set -u
B=/home/manu/pgprog/i-serie
D=/home/manu/pgprog/data_carrera
P=55592
N=${N:-5}

"$B/bin/pg_ctl" -D "$D" -m immediate -w stop >/dev/null 2>&1
rm -rf "$D"
"$B/bin/initdb" -D "$D" -U postgres --no-sync -A trust >/dev/null 2>&1
cat >> "$D/postgresql.conf" <<'EOF'
wal_level = logical
max_replication_slots = 10
max_wal_senders = 10
EOF
"$B/bin/pg_ctl" -D "$D" -o "-p $P" -l /home/manu/pgprog/carrera.log -w start >/dev/null 2>&1

q() { "$B/bin/psql" -p $P -U postgres -qtAX -c "$1" 2>&1; }

ganadas=0
for i in $(seq 1 $N); do
	q "DROP TABLE IF EXISTS test" >/dev/null
	q "CREATE TABLE test (id int PRIMARY KEY, big text)" >/dev/null
	q "ALTER TABLE test ALTER COLUMN big SET STORAGE EXTERNAL" >/dev/null
	q "INSERT INTO test SELECT g, repeat('old', 3000) FROM generate_series(1,3) g" >/dev/null
	TOAST=$(q "SELECT 'pg_toast.' || c2.relname FROM pg_class c1 JOIN pg_class c2 ON c2.oid = c1.reltoastrelid WHERE c1.relname='test'")

	# A: transaccion abierta, se cierra sola a los 3s
	( "$B/bin/psql" -p $P -U postgres -qtAX \
		-c "BEGIN" -c "SELECT pg_current_xact_id()" -c "SELECT pg_sleep(3)" -c "COMMIT" >/dev/null 2>&1 ) &
	sleep 0.5

	# B: REPACK, que se va a quedar esperando a que A termine
	( q "REPACK (CONCURRENTLY) test" >/dev/null 2>&1 ) &

	# C: reescribir la toast mientras REPACK espera
	sleep 1
	q "VACUUM FULL $TOAST" >/dev/null 2>&1

	# esperar a que A commitee y disparar el UPDATE lo antes posible
	wait %1 2>/dev/null
	q "UPDATE test SET big = repeat('NEW', 4000) WHERE id IN (1,2,3)" >/dev/null 2>&1
	wait 2>/dev/null

	res=$(q "SELECT DISTINCT left(big,9) FROM test")
	if [ "$res" = "oldoldold" ]; then
		ganadas=$((ganadas+1)); echo "   intento $i: UPDATE PERDIDO"
	else
		echo "   intento $i: los updates sobrevivieron ($res)"
	fi
done

"$B/bin/pg_ctl" -D "$D" -m immediate -w stop >/dev/null 2>&1
echo
echo "sin injection point: $ganadas de $N intentos perdieron el update"