add polardb pool config (#702)

Author: wustzdy

src/memos/api/config.py

@@ -590,6 +590,7 @@ def get_polardb_config(user_id: str | None = None) -> dict[str, Any]:
             "user": os.getenv("POLAR_DB_USER", "root"),
             "password": os.getenv("POLAR_DB_PASSWORD", "123456"),
             "db_name": db_name,
+            "maxconn": int(os.getenv("POLARDB_POOL_MAX_CONN", "100")),
             "user_name": user_name,
             "use_multi_db": use_multi_db,
             "auto_create": True,

src/memos/configs/graph_db.py

@@ -198,6 +198,10 @@ class PolarDBGraphDBConfig(BaseConfig):
         ),
     )
     embedding_dimension: int = Field(default=1024, description="Dimension of vector embedding")
+    maxconn: int = Field(
+        default=100,
+        description="Maximum number of connections in the connection pool",
+    )
 
     @model_validator(mode="after")
     def validate_config(self):

src/memos/graph_dbs/polardb.py

@@ -136,24 +136,27 @@ def __init__(self, config: PolarDBGraphDBConfig):
             port = config.get("port")
             user = config.get("user")
             password = config.get("password")
+            maxconn = config.get("maxconn", 100)  # De
         else:
             self.db_name = config.db_name
             self.user_name = config.user_name
             host = config.host
             port = config.port
             user = config.user
             password = config.password
+            maxconn = config.maxconn if hasattr(config, "maxconn") else 100
         """
         # Create connection
         self.connection = psycopg2.connect(
             host=host, port=port, user=user, password=password, dbname=self.db_name,minconn=10, maxconn=2000
         )
         """
+        logger.info(f" db_name: {self.db_name} current maxconn is:'{maxconn}'")
 
         # Create connection pool
         self.connection_pool = psycopg2.pool.ThreadedConnectionPool(
             minconn=5,
-            maxconn=100,
+            maxconn=maxconn,
             host=host,
             port=port,
             user=user,
@@ -216,6 +219,7 @@ def _get_connection(self):
         Raises:
             RuntimeError: If connection pool is closed or exhausted after retries
         """
+        logger.info(f" db_name: {self.db_name} pool maxconn is:'{self.connection_pool.maxconn}'")
         if self._pool_closed:
             raise RuntimeError("Connection pool has been closed")