26 lines
592 B
Python
26 lines
592 B
Python
|
"""FastAPI Depends 工厂：把池化会话按请求注入，用后即还。"""
|
|||
|
|
from .database import milvus, mysql, neo4j, redis
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def get_db():
|
|||
|
|
"""仓储/服务层用 AsyncSession（UoW）：注入 BaseRepository。"""
|
|||
|
|
async with mysql.get_session_factory()() as session:
|
|||
|
|
yield session
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def get_mysql_conn():
|
|||
|
|
async with mysql.get_session_factory()() as conn:
|
|||
|
|
yield conn
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_redis():
|
|||
|
|
yield redis.client()
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def get_neo4j_session():
|
|||
|
|
async with neo4j.get_session() as s:
|
|||
|
|
yield s
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_milvus():
|
|||
|
|
yield milvus.client()
|