Failed BEGIN leaves Connection._top_xact set; every later transaction becomes a SAVEPOINT and fails with NoActiveSQLTransactionError
I maintainer di solito rispondono entro 1 giorno
Valutazione
- Difficoltà
- 2/5
- Tempo stimato
- 1-3 ore
- Idoneità per principianti
- 35/100
Direzione di ricerca
Read Transaction.start() and the transaction tests in tests/test_transaction.py, focusing on the path where BEGIN raises. Check the existing cleanup in __commit() and __rollback() and add coverage for a failed BEGIN, including the nested-savepoint case. Done means the connection can start a later transaction after a failed BEGIN without leaving an outer transaction’s state corrupted; linked pull request #1387 is already open.
Scritto dal modello di indicizzazione a partire dal testo della issue.
Descrizione
Summary
Transaction.start() assigns con._top_xact = self before it sends BEGIN. If BEGIN raises, the transaction is marked FAILED, but _top_xact is never reset. From then on, every conn.transaction() on that connection is treated as nested and sends SAVEPOINT ..., which the server rejects:
asyncpg.exceptions.NoActiveSQLTransactionError: SAVEPOINT can only be used in transaction blocks
The connection is still open and idle (is_closed() == False, is_in_transaction() == False), but it cannot run a transaction again until it is closed or reset().
Reproduction
The concurrent query below is just a convenient way to make BEGIN raise. Any exception from BEGIN has the same effect (see "How we hit it").
import asyncio
import asyncpg
async def main():
conn = await asyncpg.connect("postgresql://postgres:postgres@localhost:5432/postgres")
# Make BEGIN fail: the connection is busy, so execute("BEGIN") raises before sending anything.
busy = asyncio.create_task(conn.execute("SELECT pg_sleep(0.5)"))
await asyncio.sleep(0.1)
try:
await conn.transaction().start()
except asyncpg.InterfaceError as e:
print("BEGIN failed:", e)
await busy
print("is_in_transaction:", conn.is_in_transaction(), "| _top_xact:", conn._top_xact)
async with conn.transaction(): # sent as SAVEPOINT -> NoActiveSQLTransactionError
await conn.execute("SELECT 1")
asyncio.run(main())
BEGIN failed: cannot perform operation: another operation is in progress
is_in_transaction: False | _top_xact: <asyncpg.Transaction state:failed 0x...>
asyncpg.exceptions.NoActiveSQLTransactionError: SAVEPOINT can only be used in transaction blocks
How we hit it
We run Aurora PostgreSQL behind Amazon RDS Proxy, with SQLAlchemy 2.0's asyncpg dialect. RDS Proxy could not borrow a backend connection in time, answered BEGIN with an error (SQLSTATE 08000) and kept the client socket open:
03:00:03 PostgresConnectionError: Timed-out waiting to acquire database connection.
03:00:05 NoActiveSQLTransactionError: SAVEPOINT can only be used in transaction blocks
03:00:07 NoActiveSQLTransactionError: SAVEPOINT can only be used in transaction blocks
... (every write until the process was restarted)
asyncpg.Pool clears it on release(), which calls Connection._reset() and resets _top_xact. Until then, retries within the same acquire() fail the same way. Connections reused without reset(), such as SQLAlchemy's pool or a long-lived connection, stay broken until they are closed.
Is this intentional? We couldn't tell whether leaving _top_xact set after a failed BEGIN is by design. The tests in tests/test_transaction.py assert _top_xact is None after every error case they cover, but none covers BEGIN itself failing.
Proposed fix
In Transaction.start():
try:
await self._connection.execute(query)
except BaseException:
self._state = TransactionState.FAILED
+ if con._top_xact is self:
+ con._top_xact = None
raise
This mirrors what __commit() and __rollback() already do. The is self guard leaves the outer transaction in place when a nested SAVEPOINT fails.
Why this is safe for any exception
Only __commit() and __rollback() clear _top_xact, and both raise for a FAILED transaction before reaching that line. So after a failed BEGIN, nothing can ever release it, whatever the exception was. Whether the server is actually in a transaction is tracked separately, through is_in_transaction(), and start() already checks that when _top_xact is None:
How BEGIN failed |
Server afterwards | Today | With the fix |
|---|---|---|---|
Error response (e.g. the proxy's 08000) |
not in a transaction | every later transaction fails with 25P01 |
next BEGIN works |
| Raised before sending (e.g. the repro above) | not in a transaction | same | next BEGIN works |
Cancelled after the server ran BEGIN |
in a transaction | next transaction becomes a SAVEPOINT in that open transaction, reports success, but nothing is committed |
if the reply to the cancelled BEGIN has been read: InterfaceError: cannot use Connection.transaction() in a manually started transaction; if not yet: BEGIN is sent again (the server only warns) and the transaction commits normally |
Environment
- asyncpg 0.32.0 (also 0.31.0)
- PostgreSQL 17 (repro); Aurora PostgreSQL behind RDS Proxy (production)
- Python 3.14
- Lingua principale
- Python
- Stelle
- 8.1k
- Fork
- 474
- Merge medio
- 1g 17h
- PR unite (30g)
- 24
Preparare l'ambiente
Questo progetto non fornisce container di sviluppo, Dockerfile né guida per i contributori, quindi l'ambiente è a tuo carico: parti dal suo README e consulta la nostra guida al primo contributo per i passaggi generali.
Come iniziare
- Leggi tutta la issue e poi la guida ai contributi del progetto.
- Commenta sulla issue per dire che te ne occupi tu — evita che due persone facciano lo stesso lavoro.
- Fai un fork del repository e lavora su un branch.
- Apri una pull request che faccia riferimento al numero della issue.
Altre issue di MagicStack/asyncpg
-
Connect call failed error doesn't distinguish port mismatch from "server not running"Forse già presa @aryansk l’ha presa 54 giorni fa. Aperta
Difficoltà 3/5 1-2 giorni Idoneità per principianti 72/100
MagicStack/asyncpg#1342 ·
I maintainer di solito rispondono entro 1 giorno
-
TypeError in asyncpg.connect() for specific parameters when values are not str enoughForse già presa @pranjalm37 l’ha presa 57 giorni fa. Aperta
Difficoltà 3/5 1-2 giorni Idoneità per principianti 56/100
MagicStack/asyncpg#1340 · 1 commento ·
I maintainer di solito rispondono entro 1 giorno
-
setup.py relies on deprecated pkg_resourcesForse già presa @jasonwbarnett l’ha presa 85 giorni fa. Aperta
Difficoltà 2/5 1-3 ore Idoneità per principianti 28/100
MagicStack/asyncpg#1337 · 1 commento ·
I maintainer di solito rispondono entro 1 giorno
-
Difficoltà 4/5 3-5 giorni Idoneità per principianti 42/100
MagicStack/asyncpg#1330 · 1 commento ·
I maintainer di solito rispondono entro 1 giorno
-
Difficoltà 5/5 Più di una settimana Idoneità per principianti 38/100
MagicStack/asyncpg#1322 · 1 commento ·
I maintainer di solito rispondono entro 1 giorno
Tutte le issue di MagicStack/asyncpg
Issue simili
-
Difficoltà 1/5 Meno di un'ora Idoneità per principianti 85/100
MystenLabs/MemWal#1163 · 2 commenti ·
I maintainer di solito rispondono entro 1 giorno
-
infertopics leaves new nodes without a topic when untopiced neighbours outnumber topiced onesForse già presa @moneebullah25 l’ha presa oggi. Aperta
Difficoltà 2/5 1-3 ore Idoneità per principianti 72/100
I maintainer di solito rispondono entro 1 giorno
-
bug
Difficoltà 2/5 1-3 ore Idoneità per principianti 62/100
ClanGenOfficial/clangen#6254 ·
I maintainer di solito rispondono entro 1 giorno
-
Difficoltà 2/5 1-3 ore Idoneità per principianti 70/100
FinanceFlash/unvibecode#218 ·
I maintainer di solito rispondono entro 1 giorno
-
Difficoltà 2/5 1-3 ore Idoneità per principianti 75/100
I maintainer di solito rispondono entro 1 giorno