45 lines
1.4 KiB
Python
45 lines
1.4 KiB
Python
# _*_ coding : UTF-8 _*_
|
||
# @Time : 2025/02/13 17:28
|
||
# @UpdateTime : 2025/02/13 17:28
|
||
# @Author : sonder
|
||
# @File : get_ElasticSearch.py
|
||
# @Software : PyCharm
|
||
# @Comment : 本程序
|
||
|
||
from elasticsearch import AsyncElasticsearch
|
||
from config.env import ElasticSearchConfig
|
||
from utils.log import logger
|
||
|
||
|
||
class ElasticSearch:
|
||
"""
|
||
ElasticSearch工具类
|
||
"""
|
||
|
||
@classmethod
|
||
async def init_elasticsearch(cls):
|
||
"""
|
||
初始化elasticsearch
|
||
"""
|
||
try:
|
||
# 创建异步Elasticsearch客户端
|
||
es = AsyncElasticsearch(
|
||
hosts=[{
|
||
"scheme": "http", # 传递scheme
|
||
"host": ElasticSearchConfig.ES_HOST, # 传递主机名
|
||
"port": ElasticSearchConfig.ES_PORT, # 传递端口
|
||
}],
|
||
http_auth=(ElasticSearchConfig.ES_USER, ElasticSearchConfig.ES_PASSWORD), # 传递http_auth
|
||
)
|
||
|
||
# 检查连接是否成功
|
||
if await es.ping():
|
||
logger.success(f"ElasticSearch初始化成功: {ElasticSearchConfig.ES_HOST}:{ElasticSearchConfig.ES_PORT}")
|
||
else:
|
||
logger.warning("ElasticSearch初始化失败：无法ping通服务器")
|
||
es = None
|
||
return es
|
||
except Exception as e:
|
||
logger.error(f"ElasticSearch初始化失败: {e}")
|
||
return None
|