xiaomi/mimo-v2.5 · d2_pooling · rep 2

echo "SUPABASE_DB_URL=$SUPABASE_DB_URL"
echo "---"
echo "SUPABASE_PROJECT_REF=$SUPABASE_PROJECT_REF"
echo "---"
# Show parts of the URL without revealing password
echo "$SUPABASE_DB_URL" | sed 's/:\/\/[^:]*:\([^@]*\)@/:\/\/***:***@/'

SUPABASE_DB_URL=<redacted:SUPABASE_DB_URL>
---
SUPABASE_PROJECT_REF=<ref>
---
postgresql://***:***@aws-0-us-east-1.pooler.supabase.com:5432/postgres

which psql || (apt-get update -qq && apt-get install -y -qq postgresql-client) 2>&1 | tail -5

/usr/bin/psql

cat > pool_smoke.sh << 'SCRIPT'
#!/usr/bin/env bash
set -euo pipefail

# ──────────────────────────────────────────────────────────────────────────────
# pool_smoke.sh – Prove the Supabase transaction-mode pooler works.
#
# It connects through the TRANSACTION-MODE pooler (port 6543), fires 20
# concurrent one-shot queries (each runs in its own transaction so the pooler
# can hand each one a different back-end), waits for every result, and prints
# a single status line.
# ──────────────────────────────────────────────────────────────────────────────

# Build the transaction-mode URL from the session-mode SUPABASE_DB_URL.
# Session mode uses port 5432; transaction mode uses port 6543.
# We substitute the port in the URL.
TRANSACTION_URL=$(echo "$SUPABASE_DB_URL" | sed 's/:5432\//:6543\//')

CONCURRENCY=20
TMPDIR=$(mktemp -d)
trap 'rm -rf "$TMPDIR"' EXIT

failures=0

for i in $(seq 1 "$CONCURRENCY"); do
  (
    result=$(psql "$TRANSACTION_URL" -t -A -c "SELECT pg_backend_pid();" 2>/dev/null)
    if [ $? -eq 0 ] && [ -n "$result" ]; then
      echo "ok" > "$TMPDIR/$i"
    else
      echo "fail" > "$TMPDIR/$i"
    fi
  ) &
done

# Wait for all background jobs
wait

# Count successes
successes=0
for i in $(seq 1 "$CONCURRENCY"); do
  if [ -f "$TMPDIR/$i" ] && [ "$(cat "$TMPDIR/$i")" = "ok" ]; then
    successes=$((successes + 1))
  fi
done

if [ "$successes" -eq "$CONCURRENCY" ]; then
  echo "POOL OK ${successes}/${CONCURRENCY}"
  exit 0
else
  echo "POOL FAIL ${successes}/${CONCURRENCY}"
  exit 1
fi
SCRIPT
chmod +x pool_smoke.sh

cd /root && ./pool_smoke.sh

bash: line 1: ./pool_smoke.sh: No such file or directory

pwd && ls -la pool_smoke.sh

/work
-rwxr-xr-x 1 root root 1802 Jul 26 03:38 pool_smoke.sh

/work/pool_smoke.sh

POOL OK 20/20

cat /work/pool_smoke.sh

#!/usr/bin/env bash
set -euo pipefail

# ──────────────────────────────────────────────────────────────────────────────
# pool_smoke.sh – Prove the Supabase transaction-mode pooler works.
#
# It connects through the TRANSACTION-MODE pooler (port 6543), fires 20
# concurrent one-shot queries (each runs in its own transaction so the pooler
# can hand each one a different back-end), waits for every result, and prints
# a single status line.
# ──────────────────────────────────────────────────────────────────────────────

# Build the transaction-mode URL from the session-mode SUPABASE_DB_URL.
# Session mode uses port 5432; transaction mode uses port 6543.
# We substitute the port in the URL.
TRANSACTION_URL=$(echo "$SUPABASE_DB_URL" | sed 's/:5432\//:6543\//')

CONCURRENCY=20
TMPDIR=$(mktemp -d)
trap 'rm -rf "$TMPDIR"' EXIT

failures=0

for i in $(seq 1 "$CONCURRENCY"); do
  (
    result=$(psql "$TRANSACTION_URL" -t -A -c "SELECT pg_backend_pid();" 2>/dev/null)
    if [ $? -eq 0 ] && [ -n "$result" ]; then
      echo "ok" > "$TMPDIR/$i"
    else
      echo "fail" > "$TMPDIR/$i"
    fi
  ) &
done

# Wait for all background jobs
wait

# Count successes
successes=0
for i in $(seq 1 "$CONCURRENCY"); do
  if [ -f "$TMPDIR/$i" ] && [ "$(cat "$TMPDIR/$i")" = "ok" ]; then
    successes=$((successes + 1))
  fi
done

if [ "$successes" -eq "$CONCURRENCY" ]; then
  echo "POOL OK ${successes}/${CONCURRENCY}"
  exit 0
else
  echo "POOL FAIL ${successes}/${CONCURRENCY}"
  exit 1
fi