jack před 1 měsícem
rodič
revize
cc304d3038
2 změnil soubory, kde provedl 3 přidání a 1 odebrání
  1. 2 0
      app/providers/database.py
  2. 1 1
      config/database.py

+ 2 - 0
app/providers/database.py

@@ -19,6 +19,7 @@ connect_args = {}
 database_url = db_settings.database_url
 engine = create_engine(
     database_url,
+    pool_pre_ping=True,  # 设置心跳
     connect_args=connect_args,
     poolclass=QueuePool,
     pool_size=db_settings.DB_POOL_SIZE,
@@ -32,6 +33,7 @@ async_database_url = db_settings.async_database_url
 async_engine = create_async_engine(
     async_database_url,
     connect_args=connect_args,
+    pool_pre_ping=True,  # 设置心跳
     poolclass=AsyncAdaptedQueuePool,
     pool_size=db_settings.DB_POOL_SIZE,
     pool_recycle=db_settings.DB_POOL_RECYCLE,

+ 1 - 1
config/database.py

@@ -11,7 +11,7 @@ class Settings(BaseSettings):
     DB_DATABASE: str = "open_assistant"
     DB_USER: str = "root"
     DB_PASSWORD: str = "123456"
-    DB_POOL_SIZE: int = 20
+    DB_POOL_SIZE: int = 20000
     DB_POOL_RECYCLE: int = 3600
     DB_OVERLOW: int = 10000