Failed BEGIN leaves Connection._top_xact set; every later transaction becomes a SAVEPOINT and fails with NoActiveSQLTransactionError
维护者通常 1 天内回复
评估
- 难度
- 2/5
- 预计耗时
- 1-3 小时
- 新手友好度
- 35/100
调研方向
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.
由索引模型根据 Issue 内容生成。
描述
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
- 主要语言
- Python
- 星标
- 8.1k
- 派生
- 474
- 平均合并
- 1 天 17 小时
- 30 天内合并 PR
- 24
环境准备
这个项目没有提供开发容器、Dockerfile 或贡献指南,环境需要你自己搭建:先看它的 README,通用步骤见我们的新手贡献指南。
从这里开始
- 先读完整个 Issue,再读项目的贡献指南。
- 在 Issue 下留言说明你要接手 —— 这能避免两个人做同样的事。
- Fork 仓库,在一个分支上完成修改。
- 提交 Pull Request,并在描述里引用这个 Issue 编号。
MagicStack/asyncpg 的其他 Issue
-
Connect call failed error doesn't distinguish port mismatch from "server not running"可能已有人在做 @aryansk 于 54 天前认领。 未关闭
难度 3/5 1-2 天 新手友好度 72/100
MagicStack/asyncpg#1342 ·
维护者通常 1 天内回复
-
TypeError in asyncpg.connect() for specific parameters when values are not str enough可能已有人在做 @pranjalm37 于 57 天前认领。 未关闭
难度 3/5 1-2 天 新手友好度 56/100
MagicStack/asyncpg#1340 · 1 条评论 ·
维护者通常 1 天内回复
-
setup.py relies on deprecated pkg_resources可能已有人在做 @jasonwbarnett 于 85 天前认领。 未关闭
难度 2/5 1-3 小时 新手友好度 28/100
MagicStack/asyncpg#1337 · 1 条评论 ·
维护者通常 1 天内回复
-
难度 4/5 3-5 天 新手友好度 42/100
MagicStack/asyncpg#1330 · 1 条评论 ·
维护者通常 1 天内回复
-
难度 5/5 一周以上 新手友好度 38/100
MagicStack/asyncpg#1322 · 1 条评论 ·
维护者通常 1 天内回复
查看 MagicStack/asyncpg 的全部 Issue
相似的 Issue
-
难度 1/5 1 小时以内 新手友好度 85/100
MystenLabs/MemWal#1163 · 2 条评论 ·
维护者通常 1 天内回复
-
infertopics leaves new nodes without a topic when untopiced neighbours outnumber topiced ones可能已有人在做 @moneebullah25 今天认领。 未关闭
难度 2/5 1-3 小时 新手友好度 72/100
维护者通常 1 天内回复
-
难度 2/5 1-3 小时 新手友好度 70/100
FinanceFlash/unvibecode#218 ·
维护者通常 1 天内回复
-
难度 2/5 1-3 小时 新手友好度 75/100
维护者通常 1 天内回复
-
难度 2/5 1-3 小时 新手友好度 70/100
NVIDIA/earth2studio#1241 ·
维护者通常 3 天内回复