39 lines
1.0 KiB
Python
39 lines
1.0 KiB
Python
"""Milvus 会话：全局单例 AsyncMilvusClient。
|
||
|
||
timeout 入参是数据操作默认超时（用 MILVUS_TIMEOUT）；
|
||
构造为急切连接、由注册表 init_db 做指数退避重试 + 严格模式兜底。
|
||
"""
|
||
from pymilvus import AsyncMilvusClient
|
||
|
||
from config.settings import settings
|
||
|
||
_client: AsyncMilvusClient | None = None
|
||
|
||
|
||
def client() -> AsyncMilvusClient:
|
||
global _client
|
||
if _client is None:
|
||
_client = AsyncMilvusClient(
|
||
uri=settings.milvus.uri,
|
||
token=settings.milvus.token,
|
||
user=settings.milvus.user,
|
||
password=settings.milvus.password,
|
||
db_name=settings.milvus.db_name,
|
||
timeout=settings.milvus.timeout,
|
||
)
|
||
return _client
|
||
|
||
|
||
async def init_db() -> None:
|
||
client() # 急切建连；失败由注册表重试（config/database/__init__.py）
|
||
|
||
|
||
async def dispose() -> None:
|
||
global _client
|
||
if _client is not None:
|
||
await _client.close()
|
||
_client = None
|
||
|
||
|
||
async def check_health() -> None:
|
||
await client().get_server_version() |