终于阻止发电机中总是运行吗？

Question

在sqlalchemy中，我以以下方式使用会话：

from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine
from sqlalchemy.orm import Session, sessionmaker

from app.core.config import get_settings

SQLALCHEMY_DATABASE_URL = get_settings().DB_URL
engine = create_async_engine(
    SQLALCHEMY_DATABASE_URL,
    future=True,
    echo=get_settings().DB_DEBUG,
    pool_size=get_settings().CONNECTION_POOL_SIZE,
    pool_pre_ping=True,
)

async_session = sessionmaker(
    engine,
    expire_on_commit=False,
    class_=AsyncSession,
)


async def get_session() -> Session:
    session: Session = async_session()
    try:
        yield session
    finally:
        await session.close()

在运行上述代码后的时间后，我会收到一个错误，这是从连接池中获得连接的超时。如果我不返回连接（将其关闭）到连接池，我用完了连接。我的问题是在get_session函数中的上述代码中，每个连接最终都会运行吗？（即每个连接最终返回池）

Answer 1

如果您的会话保持打开状态，则意味着创建它们的发电机将保持活力并运行。

发电机需要完成执行，关闭或GC，才能最终运行。如果您没有整体迭代以完成其执行，则应关闭发电机，以防止其保持会话的时间比必要的时间更长。

作为GC和关闭的示例，以下代码

def generator_func():
    try:
        yield
    finally:
        print("Ran finally cleanup")

generator = generator_func()
next(generator)
print("deleting generator")
del generator
print("deleted generator")
print()
generator = generator_func()
next(generator)
print("closing generator")
generator.close()
print("closed generator")

为您提供此输出，

deleting generator
Ran finally cleanup
deleted generator

closing generator
Ran finally cleanup
closed generator

以确保运行关闭， contextlib.closing 可用于通过上下文来克服生成器 经理。

这也导致您的 session 对象，似乎是上下文管理器本身，应在可能的情况下用作上下文管理器。