in_ - sqlalchemy python postgres

SQLAlchemy sesión y relación de conexión

Question

¿Las consultas ejecutadas con el mismo objeto de session SQLAlchemy utilizan la misma conexión subyacente? Si no, ¿hay alguna manera de asegurar esto?

Algunos antecedentes: tengo la necesidad de utilizar la función de bloqueo con nombre de MySQL, es decir, las funciones GET_LOCK() y RELEASE_LOCK() . En lo que respecta al servidor MySQL, solo la conexión que obtuvo el bloqueo puede liberarlo, así que debo asegurarme de que ejecuto estos dos comandos dentro de la misma conexión o la conexión muere para asegurar que se libere el bloqueo.

Para hacer las cosas más agradables, he creado un contexto "bloqueado" así:

@contextmanager
def mysql_named_lock(session, name, timeout):
    """Get a named mysql lock on a session
    """
    lock = session.execute("SELECT GET_LOCK(:name, :timeout)",
                           name=name, timeout=timeout).scalar()
    if lock:
        try:
            yield session
        finally:
            session.execute("SELECT RELEASE_LOCK(:name)", name=name)
    else:
        e = "Count not obtain named lock {} within {} sections".format(
            name, timeout)
        raise RuntimeError(e)

def my_critical_section(session):
    with mysql_named_lock(session, __name__, 10) as lockedsession:
        thing = lockedsession.query(MyStuff).one()
    return thing

Quiero asegurarme de que las dos llamadas de execute en mysql_named_lock ocurran en la misma conexión subyacente o que la conexión esté cerrada.

¿Puedo asumir que esto "solo funcionará" o hay algo de lo que tenga que estar al tanto aquí?

Answer 1

"solo funcionará" si (a) su sesión es una scoped_session y (b) la está usando de forma no simultánea (el mismo pid / thread). Si eres demasiado paranoico, asegúrate de que estés utilizando la misma ID de conexión a través de

session.connection().connection.thread_id()

Además, no tiene sentido pasar la sesión como argumento. Inítelo una vez, en algún lugar del alcance global de su aplicación, luego llame a cualquier parte de un código, obtendrá la misma ID de conexión.