This commit is contained in:
2025-08-19 09:33:54 +08:00
parent ff39fdbf36
commit bfe7ac96a7

View File

@@ -258,28 +258,4 @@ class EsSearchUtil:
finally: finally:
# 确保释放连接回连接池 # 确保释放连接回连接池
if 'conn' in locals() and 'search_util' in locals(): if 'conn' in locals() and 'search_util' in locals():
search_util.es_pool.release_connection(conn) search_util.es_pool.release_connection(conn)
# 添加main函数进行测试
if __name__ == "__main__":
try:
# 创建EsSearchUtil实例
search_util = EsSearchUtil(ES_CONFIG)
# 查询"混凝土"
query = "混凝土"
logger.info(f"开始查询关键词: {query}")
results = search_util.text_search(query, size=5)
print(f"查询 '{query}' 完成,共找到 {len(results['hits']['hits'])} 条结果")
# 在打印结果数量后添加
for i, hit in enumerate(results['hits']['hits'], 1):
print(f"结果 {i}:")
print(f"得分: {hit['_score']}")
print(f"内容: {hit['_source'].get('user_input', '无内容')}")
print("-" * 50)
except Exception as e:
logger.error(f"测试失败: {str(e)}")
print(f"测试失败: {str(e)}")