chore(database): 移除未使用的数据库连接代码
- 删除了 MySQL、Elasticsearch 和 Redis 的连接代码 - 保留了 MinIO 的连接代码 - 移除了未使用的导入语句
This commit is contained in:
parent
b30ddb43f7
commit
7ea1b7bbe8
|
@ -1,9 +1,6 @@
|
||||||
import mysql.connector
|
|
||||||
import os
|
import os
|
||||||
import redis
|
|
||||||
from minio import Minio
|
from minio import Minio
|
||||||
from dotenv import load_dotenv
|
from dotenv import load_dotenv
|
||||||
from elasticsearch import Elasticsearch
|
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
|
|
||||||
# 加载环境变量
|
# 加载环境变量
|
||||||
|
@ -25,36 +22,15 @@ def is_running_in_docker():
|
||||||
|
|
||||||
# 根据运行环境选择合适的主机地址和端口
|
# 根据运行环境选择合适的主机地址和端口
|
||||||
if is_running_in_docker():
|
if is_running_in_docker():
|
||||||
MYSQL_HOST = "mysql"
|
|
||||||
MYSQL_PORT = 3306
|
|
||||||
MINIO_HOST = "minio"
|
MINIO_HOST = "minio"
|
||||||
MINIO_VISIT_HOST = os.getenv("MINIO_VISIT_HOST", "localhost")
|
MINIO_VISIT_HOST = os.getenv("MINIO_VISIT_HOST", "localhost")
|
||||||
MINIO_PORT = 9000
|
MINIO_PORT = 9000
|
||||||
ES_HOST = "es01"
|
|
||||||
ES_PORT = 9200
|
|
||||||
REDIS_HOST = "redis"
|
|
||||||
REDIS_PORT = 6379
|
|
||||||
else:
|
else:
|
||||||
MYSQL_HOST = "localhost"
|
|
||||||
MYSQL_PORT = int(os.getenv("MYSQL_PORT", "5455"))
|
|
||||||
MINIO_HOST = "localhost"
|
MINIO_HOST = "localhost"
|
||||||
MINIO_VISIT_HOST = "localhost"
|
MINIO_VISIT_HOST = "localhost"
|
||||||
MINIO_PORT = int(os.getenv("MINIO_PORT", "9000"))
|
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连接配置
|
||||||
MINIO_CONFIG = {
|
MINIO_CONFIG = {
|
||||||
"endpoint": f"{MINIO_HOST}:{MINIO_PORT}",
|
"endpoint": f"{MINIO_HOST}:{MINIO_PORT}",
|
||||||
|
@ -64,32 +40,6 @@ MINIO_CONFIG = {
|
||||||
"visit_point": f"{MINIO_VISIT_HOST}:{MINIO_PORT}",
|
"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():
|
def get_minio_client():
|
||||||
"""创建MinIO客户端连接"""
|
"""创建MinIO客户端连接"""
|
||||||
|
@ -99,38 +49,3 @@ def get_minio_client():
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
print(f"MinIO连接失败: {str(e)}")
|
print(f"MinIO连接失败: {str(e)}")
|
||||||
raise 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
|
|
||||||
|
|
Loading…
Reference in New Issue