Spaces:
Running
Running
Upload database_sql.py
Browse files- database_sql.py +4 -1
database_sql.py
CHANGED
|
@@ -35,7 +35,10 @@ else:
|
|
| 35 |
# PostgreSQL/MySQL:配置连接池参数
|
| 36 |
connect_args = {}
|
| 37 |
if "postgresql" in SQLALCHEMY_DATABASE_URL or "postgres" in SQLALCHEMY_DATABASE_URL:
|
| 38 |
-
|
|
|
|
|
|
|
|
|
|
| 39 |
engine = create_engine(
|
| 40 |
SQLALCHEMY_DATABASE_URL,
|
| 41 |
poolclass=QueuePool,
|
|
|
|
| 35 |
# PostgreSQL/MySQL:配置连接池参数
|
| 36 |
connect_args = {}
|
| 37 |
if "postgresql" in SQLALCHEMY_DATABASE_URL or "postgres" in SQLALCHEMY_DATABASE_URL:
|
| 38 |
+
sslmode = os.environ.get("DB_SSLMODE", "require")
|
| 39 |
+
connect_args["sslmode"] = sslmode
|
| 40 |
+
connect_args["connect_timeout"] = int(os.environ.get("DB_CONNECT_TIMEOUT", "10"))
|
| 41 |
+
logger.info(f"PostgreSQL SSL 配置: sslmode={sslmode}, connect_timeout={connect_args['connect_timeout']}")
|
| 42 |
engine = create_engine(
|
| 43 |
SQLALCHEMY_DATABASE_URL,
|
| 44 |
poolclass=QueuePool,
|