MySQL如何关闭SQLAlchemy连接|极客笔记

MySQL如何关闭SQLAlchemy连接

在使用MySQL数据库和SQLAlchemy ORM框架时，我们需要在适当的时候关闭数据库连接以避免资源浪费。本文将介绍如何在MySQL中关闭SQLAlchemy连接。

阅读更多：MySQL 教程

关闭连接的方式

方法一：显式关闭连接

可以通过调用engine.dispose()或session.close_all()来显式关闭连接。代码示例如下：

from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker

# 创建MySQL引擎
engine = create_engine('mysql+pymysql://username:password@localhost:3306/test', pool_recycle=3600)

# 创建Session
Session = sessionmaker(bind=engine)
session = Session()

# 显式关闭连接
engine.dispose()  # 或者 session.close_all()

方法二：自动关闭连接

另一种关闭连接的方式是通过在创建引擎时指定pool_pre_ping=True，以确保在每次使用连接之前都会先ping一次数据库，如果连接已经失效，则会自动重新创建连接。这样就可以避免手动关闭连接，代码示例如下：

from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker

# 创建MySQL引擎
engine = create_engine('mysql+pymysql://username:password@localhost:3306/test', pool_recycle=3600, pool_pre_ping=True)

# 创建Session
Session = sessionmaker(bind=engine)
session = Session()

# 不需要显式关闭连接

示例程序

下面是一个完整的示例程序，用于演示如何在MySQL中关闭SQLAlchemy连接：

from sqlalchemy import create_engine, Column, Integer, String
from sqlalchemy.orm import sessionmaker
from sqlalchemy.ext.declarative import declarative_base

# 创建MySQL引擎
engine = create_engine('mysql+pymysql://username:password@localhost:3306/test', pool_recycle=3600)

# 创建ORM基类
Base = declarative_base()

# 定义模型
class User(Base):
    __tablename__ = 'users'
    id = Column(Integer, primary_key=True)
    name = Column(String(50))
    age = Column(Integer)

    def __repr__(self):
        return f"<User(id={self.id}, name='{self.name}', age={self.age})>"

# 创建Session
Session = sessionmaker(bind=engine)
session = Session()

# 查询所有用户
users = session.query(User).all()
print(f"查询结果：{users}")

# 关闭连接
engine.dispose()

总结

本文介绍了两种在MySQL中关闭SQLAlchemy连接的方式，即显式关闭连接和自动关闭连接。显式关闭连接需要在适当的时候手动调用engine.dispose()或session.close_all()，而自动关闭连接可以通过在创建引擎时指定pool_pre_ping=True来实现自动重连。选择哪种方式取决于具体的业务需求和应用场景。