diff --git a/api/db/services/database.py b/api/db/services/database.py index 0d365ac..29cdee0 100644 --- a/api/db/services/database.py +++ b/api/db/services/database.py @@ -1,9 +1,6 @@ -import mysql.connector import os -import redis from minio import Minio from dotenv import load_dotenv -from elasticsearch import Elasticsearch from pathlib import Path # 加载环境变量 @@ -25,36 +22,15 @@ def is_running_in_docker(): # 根据运行环境选择合适的主机地址和端口 if is_running_in_docker(): - MYSQL_HOST = "mysql" - MYSQL_PORT = 3306 MINIO_HOST = "minio" MINIO_VISIT_HOST = os.getenv("MINIO_VISIT_HOST", "localhost") MINIO_PORT = 9000 - ES_HOST = "es01" - ES_PORT = 9200 - REDIS_HOST = "redis" - REDIS_PORT = 6379 else: - MYSQL_HOST = "localhost" - MYSQL_PORT = int(os.getenv("MYSQL_PORT", "5455")) MINIO_HOST = "localhost" MINIO_VISIT_HOST = "localhost" MINIO_PORT = int(os.getenv("MINIO_PORT", "9000")) - ES_HOST = "localhost" - ES_PORT = int(os.getenv("ES_PORT", "9200")) - REDIS_HOST = "localhost" - REDIS_PORT = int(os.getenv("REDIS_PORT", "6379")) -# 数据库连接配置 -DB_CONFIG = { - "host": MYSQL_HOST, - "port": MYSQL_PORT, - "user": "root", - "password": os.getenv("MYSQL_PASSWORD", "infini_rag_flow"), - "database": "rag_flow", -} - # MinIO连接配置 MINIO_CONFIG = { "endpoint": f"{MINIO_HOST}:{MINIO_PORT}", @@ -64,32 +40,6 @@ MINIO_CONFIG = { "visit_point": f"{MINIO_VISIT_HOST}:{MINIO_PORT}", } -# Elasticsearch连接配置 -ES_CONFIG = { - "host": f"http://{ES_HOST}:{ES_PORT}", - "user": os.getenv("ELASTIC_USER", "elastic"), - "password": os.getenv("ELASTIC_PASSWORD", "infini_rag_flow"), - "use_ssl": os.getenv("ES_USE_SSL", "false").lower() == "true", -} - -# Redis连接配置 -REDIS_CONFIG = { - "host": REDIS_HOST, - "port": REDIS_PORT, - "password": os.getenv("REDIS_PASSWORD", "infini_rag_flow"), - "decode_responses": False, -} - - -def get_db_connection(): - """创建MySQL数据库连接""" - try: - conn = mysql.connector.connect(**DB_CONFIG) - return conn - except Exception as e: - print(f"MySQL连接失败: {str(e)}") - raise e - def get_minio_client(): """创建MinIO客户端连接""" @@ -99,38 +49,3 @@ def get_minio_client(): except Exception as e: print(f"MinIO连接失败: {str(e)}") raise e - - -def get_es_client(): - """创建Elasticsearch客户端连接""" - try: - # 构建连接参数 - es_params = {"hosts": [ES_CONFIG["host"]]} - - # 添加认证信息 - if ES_CONFIG["user"] and ES_CONFIG["password"]: - es_params["basic_auth"] = (ES_CONFIG["user"], ES_CONFIG["password"]) - - # 添加SSL配置 - if ES_CONFIG["use_ssl"]: - es_params["use_ssl"] = True - es_params["verify_certs"] = False # 在开发环境中可以设置为False,生产环境应该设置为True - - es_client = Elasticsearch(**es_params) - return es_client - except Exception as e: - print(f"Elasticsearch连接失败: {str(e)}") - raise e - - -def get_redis_connection(): - """创建Redis连接""" - try: - # 使用配置创建Redis连接 - r = redis.Redis(**REDIS_CONFIG) - # 测试连接 - r.ping() - return r - except Exception as e: - print(f"Redis连接失败: {str(e)}") - raise e