¿Cómo cerrar la conexión sqlalchemy en MySQL

Question

Este es un ejemplo de código que me gustaría correr:

for i in range(1,2000): 
    db = create_engine('mysql://root@localhost/test_database') 
    conn = db.connect() 
    #some simple data operations 
    conn.close() 
    db.dispose() 

¿Hay una manera de ejecutar este sin conseguir "Demasiadas conexiones" errores de MySQL? Ya sé que puedo manejar la conexión de lo contrario o tengo un grupo de conexiones. Simplemente me gustaría saber cómo cerrar correctamente una conexión desde sqlalchemy. ¡Gracias de antemano!

Answer 1

Aquí es cómo escribir ese código correctamente:

db = create_engine('mysql://root@localhost/test_database') 
for i in range(1,2000): 
    conn = db.connect() 
    #some simple data operations 
    conn.close() 
db.dispose() 

Es decir, el Engine es una fábrica para las conexiones, así como una piscina de conexiones, no la propia conexión. Cuando dice conn.close(), la conexión es devuelta al grupo de conexiones dentro del motor, no realmente cerrada.

Si desea que la conexión sea realmente cerrada, es decir, no se reúnen, se deshabilita la agrupación a través de NullPool:

from sqlalchemy.pool import NullPool 
db = create_engine('mysql://root@localhost/test_database', poolclass=NullPool) 

Con la configuración anterior Engine, cada llamada a conn.close() cerrará la conexión DBAPI subyacente.

Si otoh que realmente desea conectarse diferentes bases de datos en cada llamada, es decir, su hardcoded "localhost/test_database" es solo un ejemplo y que en realidad tienen un montón de diferentes bases de datos, entonces el enfoque utilizando dispose() está bien; cerrará todas las conexiones que no estén desprotegidas del grupo.

En todos los casos anteriores, lo importante es que el objeto Connection se cierra a través de close(). Si está utilizando cualquier tipo de ejecución "sin conexión", es decir, engine.execute() o statement.execute(), el objeto ResultProxy devuelto por esa llamada de ejecución debe leerse completamente o cerrarse explícitamente de otra manera a través del close(). Un Connection o ResultProxy que aún está abierto prohibirá los enfoques NullPool o dispose() desde el cierre de cada conexión.

Answer 2

Trató de encontrar una solución para desconectarse de la base de datos por un problema no relacionado (debe desconectarse antes de bifurcar).

Necesita invalidate la conexión desde la piscina de conexión también.

En su ejemplo:

for i in range(1,2000): 
    db = create_engine('mysql://root@localhost/test_database') 
    conn = db.connect() 
    # some simple data operations 
    # session.close() if needed 
    conn.invalidate() 
    db.dispose()