You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
24 lines
671 B
24 lines
671 B
from neo4j import GraphDatabase, AsyncGraphDatabase
|
|
|
|
from app.config import settings
|
|
|
|
class Neo4jConnection:
|
|
def __init__(self, uri, user, password):
|
|
# 初始化连接，只负责连接
|
|
self.driver = AsyncGraphDatabase.driver(uri, auth=(user, password), max_connection_pool_size=100)
|
|
|
|
async def close(self):
|
|
# 关闭连接
|
|
await self.driver.close()
|
|
|
|
def get_session(self):
|
|
# 提供 session 对象，供外部查询使用
|
|
return self.driver.session()
|
|
|
|
# 初始化全局 Neo4j 连接
|
|
neo4j_conn = Neo4jConnection(
|
|
uri=settings.neo4j_uri,
|
|
user=settings.neo4j_user,
|
|
password=settings.neo4j_password,
|
|
)
|