ZHIWEI666 commited on
Commit
57fd107
·
verified ·
1 Parent(s): 820f705

Upload database_sql.py

Browse files
Files changed (1) hide show
  1. 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
- connect_args["sslmode"] = os.environ.get("DB_SSLMODE", "prefer")
 
 
 
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,