chore(database): 移除未使用的数据库连接代码

- 删除了 MySQL、Elasticsearch 和 Redis 的连接代码
- 保留了 MinIO 的连接代码
- 移除了未使用的导入语句
This commit is contained in:
zstar 2025-06-09 15:43:59 +08:00
parent b30ddb43f7
commit 7ea1b7bbe8
1 changed files with 0 additions and 85 deletions

View File

@ -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