From 84beb10857accc55081c1f931076d33fc0b4158b Mon Sep 17 00:00:00 2001 From: Sanghun Lee Date: Wed, 23 Oct 2024 18:36:03 +0900 Subject: [PATCH 1/2] fix: Set readonly flag to false when begin new generic txn --- src/ai/backend/manager/models/utils.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/ai/backend/manager/models/utils.py b/src/ai/backend/manager/models/utils.py index 0dc596e8ea..2731b9fce4 100644 --- a/src/ai/backend/manager/models/utils.py +++ b/src/ai/backend/manager/models/utils.py @@ -103,7 +103,10 @@ async def _begin(self, connection: SAConnection) -> AsyncIterator[SAConnection]: """ Begin generic transaction within the given connection. """ - async with connection.begin(): + conn_with_exec_opts = await connection.execution_options( + postgresql_readonly=False, + ) + async with conn_with_exec_opts.begin(): self._generic_txn_count += 1 self._check_generic_txn_cnt() try: From e6537887a4cee0b26a38480d01915c6c6d20cb6b Mon Sep 17 00:00:00 2001 From: Sanghun Lee Date: Wed, 23 Oct 2024 18:43:47 +0900 Subject: [PATCH 2/2] Add news fragment --- changes/2946.fix.md | 1 + 1 file changed, 1 insertion(+) create mode 100644 changes/2946.fix.md diff --git a/changes/2946.fix.md b/changes/2946.fix.md new file mode 100644 index 0000000000..2176d57eb0 --- /dev/null +++ b/changes/2946.fix.md @@ -0,0 +1 @@ +Set the `postgres_readonly` flag to `false` when begin generic sessions