main
HuangHai 4 weeks ago
parent a16b86559e
commit 8a42084208

@ -1,16 +1,12 @@
import logging import logging
from Config.Config import ES_CONFIG from Config.Config import ES_CONFIG
from Util.EsSearchUtil import EsSearchUtil
# 初始化日志 # 初始化日志
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
logger.setLevel(logging.INFO) logger.setLevel(logging.INFO)
# 导入EsSearchUtil
from Util.EsSearchUtil import EsSearchUtil
# 初始化EsSearchUtil # 初始化EsSearchUtil
es_search_util = EsSearchUtil(ES_CONFIG) es_search_util = EsSearchUtil(ES_CONFIG)

@ -3,6 +3,8 @@ import os
from logging.handlers import RotatingFileHandler from logging.handlers import RotatingFileHandler
import jieba import jieba
from elasticsearch import Elasticsearch from elasticsearch import Elasticsearch
from gensim.models import KeyedVectors
from Config.Config import MS_MODEL_PATH, MS_MODEL_LIMIT
from ElasticSearch.Utils.ElasticsearchConnectionPool import ElasticsearchConnectionPool from ElasticSearch.Utils.ElasticsearchConnectionPool import ElasticsearchConnectionPool
@ -16,12 +18,11 @@ handler.setFormatter(logging.Formatter('%(asctime)s - %(levelname)s - %(message)
logger.addHandler(handler) logger.addHandler(handler)
class EsSearchUtil: class EsSearchUtil:
def __init__(self, es_config): def __init__(self, es_config):
""" """
初始化Elasticsearch搜索工具 初始化Elasticsearch搜索工具
:param es_config: Elasticsearch配置字典包含hosts, username, password, index_name等 :param es_config: Elasticsearch配置字典包含hosts, username, password, index_name等
""" """
self.es_config = es_config self.es_config = es_config
# 初始化连接池 # 初始化连接池
@ -43,9 +44,9 @@ class EsSearchUtil:
# 确保es_conn属性存在以兼容旧代码 # 确保es_conn属性存在以兼容旧代码
self.es_conn = self.es self.es_conn = self.es
def __init__(self, es_config): # 确保es_conn属性存在以兼容旧代码
from gensim.models import KeyedVectors self.es_conn = self.es
from Config.Config import MS_MODEL_PATH, MS_MODEL_LIMIT
# 加载预训练模型 # 加载预训练模型
self.model = KeyedVectors.load_word2vec_format(MS_MODEL_PATH, binary=False, limit=MS_MODEL_LIMIT) self.model = KeyedVectors.load_word2vec_format(MS_MODEL_PATH, binary=False, limit=MS_MODEL_LIMIT)

Loading…
Cancel
Save