MagicStack / MagicStack/asyncpg

`get_settings()` can get out of sync after a transaction with an error is rolled back

Aperta
#1,215 0 commenti 0 reazioni 0 assegnatari Vedi su GitHub

Nessuno ha ancora preso questa issue.

Lingua principale
Python
Stelle
8.1k
Fork
468
Metriche di merge delle PR
Nessuna PR unita negli ultimi 30g

Descrizione

* **asyncpg version**: 0.30.0
* **PostgreSQL version**: 17.2
* **Do you use a PostgreSQL SaaS? If so, which? Can you reproduce
the issue with a local PostgreSQL install?**: No
* **Python version**: 3.12.2
* **Platform**: Linux
* **Do you use pgbouncer?**: No
* **Did you install asyncpg with pip?**: Yes
* **If you built asyncpg locally, which version of Cython did you use?**:
* **Can the issue be reproduced under both asyncio and
[uvloop](https://github.com/magicstack/uvloop)?**: Only checked asyncio but it doesn't seem likely to matter

```
#!/usr/bin/env python3

import asyncio
import asyncpg

async def _check_encoding(con):
current_encoding = (await con.fetch(
"select current_setting('client_encoding')")
)[0]['current_setting']
print(current_encoding)
print(con.get_settings().client_encoding)

async def test(dsn):
con = await asyncpg.connect(dsn)

print("Orig")
await _check_encoding(con)

tran = con.transaction()
await tran.start()

await con.execute("set client_encoding to 'latin1'")
print("In transaction:")
await _check_encoding(con)
try:
await con.fetch("select 1 + 'a'")
except Exception:
pass

await tran.rollback()
print("After rollback:")
await _check_encoding(con)

res = await con.fetch('select $1::text', '💩')
print(res)

asyncio.run(test(
'postgres:///main?user=edgedb&port=5656&host=localhost'
))
```

In a transaction, if I set `client_encoding` to something, and then the transaction has an error and is rolled back, the `client_encoding` that `get_settings()` returns is never reverted.
If the transaction is rolled back without an error occuring, the right thing happens.

Output:
```
Orig
UTF8
UTF_8
In transaction:
LATIN1
LATIN1
After rollback:
UTF8
LATIN1
Traceback (most recent call last):
File "asyncpg/protocol/prepared_stmt.pyx", line 175, in asyncpg.protocol.protocol.PreparedStatementState._encode_bind_msg
File "asyncpg/protocol/codecs/base.pyx", line 227, in asyncpg.protocol.protocol.Codec.encode
File "asyncpg/protocol/codecs/base.pyx", line 129, in asyncpg.protocol.protocol.Codec.encode_scalar
File "asyncpg/pgproto/./codecs/text.pyx", line 29, in asyncpg.pgproto.pgproto.text_encode
File "asyncpg/pgproto/./codecs/text.pyx", line 17, in asyncpg.pgproto.pgproto.as_pg_string_and_size
UnicodeEncodeError: 'latin-1' codec can't encode character '\U0001f4a9' in position 0: ordinal not in range(256)
```

I would expect
```
After rollback:
UTF8
UTF8
```
and no encoding error.

Guida per i contributori

Nessuna guida per i contributori indicizzata per questo repository

Come iniziare

  1. Leggi tutta la issue e poi la guida ai contributi del progetto.
  2. Commenta sulla issue per dire che te ne occupi tu — evita che due persone facciano lo stesso lavoro.
  3. Fai un fork del repository e lavora su un branch.
  4. Apri una pull request che faccia riferimento al numero della issue.

Direzione di ricerca

Inizia con il riproduttore Python fornito, concentrandoti sul percorso di rollback della transazione e sul punto di ingresso get_settings() della connessione. Conferma la differenza tra un rollback dopo un errore della transazione e un rollback pulito, quindi traccia come viene aggiornato lo stato di client_encoding; il lavoro è terminato quando get_settings() restituisce UTF8 dopo il rollback e la query successiva con un parametro di testo ha esito positivo.

Scritto dal modello di indicizzazione a partire dal testo della issue.

Valutazione

Stack tecnologico
postgresql, python
Ambito
backend, databases
Tipo di issue
Bug
Difficoltà
4/5
Tempo stimato
3-5 giorni
Stato di attività
Ferma
Chiarezza
Abbastanza chiara
Idoneità per principianti
42/100

Ricevi le nuove issue nella tua casella

Un breve riepilogo di issue GitHub adatte ai principianti.