feat: optimzie polardb ThreadedConnectionPool (#1152)

* feat:optimzie polardb ThreadedConnectionPool

* feat:optimzie polardb ThreadedConnectionPool

* feat:optimzie polardb ThreadedConnectionPool

* feat:optimzie polardb ThreadedConnectionPool

* feat:optimzie polardb ThreadedConnectionPool

* feat:add _warm_up_on_startup

* feat:add _warm_up_on_startup

* feat:add _warm_up_on_startup

* feat:add _warm_up_on_startup

* feat:config format

---------

Co-authored-by: chunyu li <78344051+fridayL@users.noreply.github.com>

Author: wustzdy

src/memos/api/config.py

@@ -675,7 +675,17 @@ def get_polardb_config(user_id: str | None = None) -> dict[str, Any]:
             "user_name": user_name,
             "use_multi_db": use_multi_db,
             "auto_create": True,
-            "embedding_dimension": int(os.getenv("EMBEDDING_DIMENSION", 1024)),
+            "embedding_dimension": int(os.getenv("EMBEDDING_DIMENSION", "1024")),
+            # .env: CONNECTION_WAIT_TIMEOUT, SKIP_CONNECTION_HEALTH_CHECK, WARM_UP_ON_STARTUP_BY_FULL, WARM_UP_ON_STARTUP_BY_ALL
+            "connection_wait_timeout": int(os.getenv("CONNECTION_WAIT_TIMEOUT", "60")),
+            "skip_connection_health_check": os.getenv(
+                "SKIP_CONNECTION_HEALTH_CHECK", "false"
+            ).lower()
+            == "true",
+            "warm_up_on_startup_by_full": os.getenv("WARM_UP_ON_STARTUP_BY_FULL", "false").lower()
+            == "true",
+            "warm_up_on_startup_by_all": os.getenv("WARM_UP_ON_STARTUP_BY_ALL", "false").lower()
+            == "true",
         }
 
     @staticmethod

src/memos/configs/graph_db.py

@@ -202,6 +202,33 @@ class PolarDBGraphDBConfig(BaseConfig):
         default=100,
         description="Maximum number of connections in the connection pool",
     )
+    connection_wait_timeout: int = Field(
+        default=30,
+        ge=1,
+        le=3600,
+        description="Max seconds to wait for a connection slot before raising (0 = wait forever, not recommended)",
+    )
+    skip_connection_health_check: bool = Field(
+        default=False,
+        description=(
+            "If True, skip SELECT 1 health check when getting connections (~1-2ms saved per request). "
+            "Use only when pool/network is reliable."
+        ),
+    )
+    warm_up_on_startup_by_full: bool = Field(
+        default=True,
+        description=(
+            "If True, run search_by_fulltext warm-up on pool connections at init to reduce "
+            "first-query latency (~200ms planning). Requires user_name in config."
+        ),
+    )
+    warm_up_on_startup_by_all: bool = Field(
+        default=False,
+        description=(
+            "If True, run all connection warm-up on pool connections at init to reduce "
+            "first-query latency (~200ms planning). Requires user_name in config."
+        ),
+    )
 
     @model_validator(mode="after")
     def validate_config(self):