stop even using pooled DBs in peewee

This commit is contained in:
perf3ct 2024-06-17 09:56:31 -07:00
parent 981866eb93
commit 5c655f298b
No known key found for this signature in database
GPG Key ID: 569C4EEC436F5232
1 changed files with 4 additions and 16 deletions

View File

@ -22,23 +22,11 @@ class PeeweeConnectionState(object):
def register_connection(db_url):
db = connect(db_url)
if isinstance(db, PostgresqlDatabase):
db = PooledPostgresqlExtDatabase(
db.database,
max_connections=8,
stale_timeout=300,
timeout=None,
autoconnect=True,
**db.connect_params
)
# Directly use PostgresqlDatabase without pooling
db.autoconnect = True
elif isinstance(db, SqliteDatabase):
db = PooledSqliteDatabase(
db.database,
max_connections=8,
stale_timeout=300,
timeout=None,
autoconnect=True,
**db.connect_params
)
# Directly use SqliteDatabase without pooling
db.autoconnect = True
else:
raise ValueError('Unsupported database connection')
return db