Merge branch 'pg-max-connection'

This commit is contained in:
yangdx
2025-05-09 14:20:53 +08:00
3 changed files with 7 additions and 1 deletions

View File

@@ -20,3 +20,4 @@ user = your_username
password = your_password
database = your_database
workspace = default # 可选,默认为default
max_connections = 12

View File

@@ -131,6 +131,7 @@ POSTGRES_PORT=5432
POSTGRES_USER=your_username
POSTGRES_PASSWORD='your_password'
POSTGRES_DATABASE=your_database
POSTGRES_MAX_CONNECTIONS=12
### separating all data from difference Lightrag instances(deprecating)
# POSTGRES_WORKSPACE=default

View File

@@ -55,7 +55,7 @@ class PostgreSQLDB:
self.password = config.get("password", None)
self.database = config.get("database", "postgres")
self.workspace = config.get("workspace", "default")
self.max = 12
self.max = int(config.get("max_connections", 12))
self.increment = 1
self.pool: Pool | None = None
@@ -312,6 +312,10 @@ class ClientManager:
"POSTGRES_WORKSPACE",
config.get("postgres", "workspace", fallback="default"),
),
"max_connections": os.environ.get(
"POSTGRES_MAX_CONNECTIONS",
config.get("postgres", "max_connections", fallback=12),
),
}
@classmethod