Optimize default setting of PostgreSQL

This commit is contained in:
yangdx 2025-05-22 17:09:26 +08:00
parent 2ee809cf58
commit ada2443653

View file

@ -49,13 +49,13 @@ MAX_GRAPH_NODES = int(os.getenv("MAX_GRAPH_NODES", 1000))
class PostgreSQLDB: class PostgreSQLDB:
def __init__(self, config: dict[str, Any], **kwargs: Any): def __init__(self, config: dict[str, Any], **kwargs: Any):
self.host = config.get("host", "localhost") self.host = config["host"]
self.port = config.get("port", 5432) self.port = config["port"]
self.user = config.get("user", "postgres") self.user = config["user"]
self.password = config.get("password", None) self.password = config["password"]
self.database = config.get("database", "postgres") self.database = config["database"]
self.workspace = config.get("workspace", "default") self.workspace = config["workspace"]
self.max = int(config.get("max_connections", 20)) self.max = int(config["max_connections"])
self.increment = 1 self.increment = 1
self.pool: Pool | None = None self.pool: Pool | None = None
@ -298,7 +298,7 @@ class ClientManager:
"POSTGRES_PORT", config.get("postgres", "port", fallback=5432) "POSTGRES_PORT", config.get("postgres", "port", fallback=5432)
), ),
"user": os.environ.get( "user": os.environ.get(
"POSTGRES_USER", config.get("postgres", "user", fallback=None) "POSTGRES_USER", config.get("postgres", "user", fallback="postgres")
), ),
"password": os.environ.get( "password": os.environ.get(
"POSTGRES_PASSWORD", "POSTGRES_PASSWORD",
@ -306,7 +306,7 @@ class ClientManager:
), ),
"database": os.environ.get( "database": os.environ.get(
"POSTGRES_DATABASE", "POSTGRES_DATABASE",
config.get("postgres", "database", fallback=None), config.get("postgres", "database", fallback="postgres"),
), ),
"workspace": os.environ.get( "workspace": os.environ.get(
"POSTGRES_WORKSPACE", "POSTGRES_WORKSPACE",