40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
"""Neo4j 会话：全局单例 AsyncDriver（内部自带连接池与事务重试）。"""
|
||
from neo4j import AsyncGraphDatabase
|
||
from config.settings import settings
|
||
|
||
_driver = None
|
||
|
||
|
||
def client():
|
||
global _driver
|
||
if _driver is None:
|
||
_driver = AsyncGraphDatabase.driver(
|
||
settings.neo4j.uri,
|
||
auth=(settings.neo4j.user, settings.neo4j.password),
|
||
max_connection_pool_size=settings.neo4j.max_connection_pool_size,
|
||
connection_acquisition_timeout=settings.neo4j.connection_acquisition_timeout,
|
||
connection_timeout=settings.neo4j.connection_timeout,
|
||
max_transaction_retry_time=settings.neo4j.max_transaction_retry_time,
|
||
)
|
||
return _driver
|
||
|
||
|
||
async def init_db() -> None:
|
||
client() # 构造为懒连接，仅建出 driver 单例
|
||
|
||
|
||
async def get_session():
|
||
"""每次操作取一个轻量 pool 内 session；execute_read/write 自带事务重试。"""
|
||
async with client().session() as s:
|
||
yield s
|
||
|
||
|
||
async def dispose() -> None:
|
||
global _driver
|
||
if _driver is not None:
|
||
await _driver.close()
|
||
_driver = None
|
||
|
||
|
||
async def check_health() -> None:
|
||
await client().verify_connectivity() |