MagicStack / MagicStack/asyncpg

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

Abierto
#1,215 0 comentarios 0 reacciones 0 asignados Ver en GitHub

Nadie ha tomado este issue todavía.

Lenguaje dominante
Python
Estrellas
8.1k
Forks
468
Métricas de merge de PR
Sin PR fusionados en 30 d

Descripción

  • 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?
    : 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.

Guía de contribución

No hay ninguna guía de contribución indexada para este repositorio

Primeros pasos

  1. Lee el issue completo y luego la guía de contribución del proyecto.
  2. Comenta en el issue que vas a ocuparte — evita que dos personas hagan lo mismo.
  3. Haz un fork del repositorio y trabaja en una rama.
  4. Abre un pull request que haga referencia al número del issue.

Línea de trabajo

Comienza con el reproductor de Python proporcionado, centrándote en la ruta de rollback de la transacción y en el punto de entrada get_settings() de la conexión. Confirma la diferencia entre un rollback después de un error de transacción y un rollback limpio; el trabajo estará terminado cuando get_settings() informe UTF8 después del rollback y la consulta posterior con un parámetro de texto se complete correctamente.

Escrito por el modelo de indexación a partir del texto del issue.

Evaluación

Stack tecnológico
postgresql, python
Área
backend, databases
Tipo de issue
Error
Dificultad
4/5
Tiempo estimado
3-5 días
Estado de actividad
Estancado
Claridad
Bastante claro
Aptitud para principiantes
42/100

Recibe los nuevos issues en tu correo

Un resumen breve de issues de GitHub para principiantes.