Tweak pooling on the database.

Author: MikuAuahDark

npps4/db/achievement.py

@@ -1,6 +1,7 @@
 import sqlalchemy
 import sqlalchemy.ext.asyncio
 import sqlalchemy.orm
+import sqlalchemy.pool
 
 from . import common
 from ..download import download
@@ -183,6 +184,7 @@ class UnitTypeGroup(common.GameDBBase):
 
 engine = sqlalchemy.ext.asyncio.create_async_engine(
     f"sqlite+aiosqlite:///file:{download.get_db_path('achievement')}?mode=ro&uri=true",
+    poolclass=sqlalchemy.pool.NullPool,
     connect_args={"check_same_thread": False},
 )
 sessionmaker = sqlalchemy.ext.asyncio.async_sessionmaker(engine)

npps4/db/effort.py

@@ -1,6 +1,7 @@
 import sqlalchemy
 import sqlalchemy.ext.asyncio
 import sqlalchemy.orm
+import sqlalchemy.pool
 
 from . import common
 from ..download import download
@@ -51,6 +52,7 @@ class LiveEffortPointBoxSpec(common.GameDBBase, common.MaybeEncrypted):
 
 engine = sqlalchemy.ext.asyncio.create_async_engine(
     f"sqlite+aiosqlite:///file:{download.get_db_path('effort')}?mode=ro&uri=true",
+    poolclass=sqlalchemy.pool.NullPool,
     connect_args={"check_same_thread": False},
 )
 sessionmaker = sqlalchemy.ext.asyncio.async_sessionmaker(engine)

npps4/db/exchange.py

@@ -1,6 +1,7 @@
 import sqlalchemy
 import sqlalchemy.ext.asyncio
 import sqlalchemy.orm
+import sqlalchemy.pool
 
 from . import common
 from ..download import download
@@ -75,6 +76,7 @@ class ExchangePoint(common.GameDBBase, common.MaybeEncrypted):
 
 engine = sqlalchemy.ext.asyncio.create_async_engine(
     f"sqlite+aiosqlite:///file:{download.get_db_path('exchange')}?mode=ro&uri=true",
+    poolclass=sqlalchemy.pool.NullPool,
     connect_args={"check_same_thread": False},
 )
 sessionmaker = sqlalchemy.ext.asyncio.async_sessionmaker(engine)

npps4/db/game_mater.py

@@ -1,6 +1,7 @@
 import sqlalchemy
 import sqlalchemy.ext.asyncio
 import sqlalchemy.orm
+import sqlalchemy.pool
 
 from . import common
 from ..download import download
@@ -210,6 +211,7 @@ def load_client_setting():
 
 engine = sqlalchemy.ext.asyncio.create_async_engine(
     f"sqlite+aiosqlite:///file:{game_mater}?mode=ro&uri=true",
+    poolclass=sqlalchemy.pool.NullPool,
     connect_args={"check_same_thread": False},
 )
 sessionmaker = sqlalchemy.ext.asyncio.async_sessionmaker(engine)

npps4/db/item.py

@@ -1,6 +1,7 @@
 import sqlalchemy
 import sqlalchemy.ext.asyncio
 import sqlalchemy.orm
+import sqlalchemy.pool
 
 from . import common
 from ..download import download
@@ -412,6 +413,7 @@ class Memories(common.GameDBBase, common.MaybeEncrypted):
 
 engine = sqlalchemy.ext.asyncio.create_async_engine(
     f"sqlite+aiosqlite:///file:{download.get_db_path('item')}?mode=ro&uri=true",
+    poolclass=sqlalchemy.pool.NullPool,
     connect_args={"check_same_thread": False},
 )
 sessionmaker = sqlalchemy.ext.asyncio.async_sessionmaker(engine)

npps4/db/live.py

@@ -1,6 +1,7 @@
 import sqlalchemy
 import sqlalchemy.ext.asyncio
 import sqlalchemy.orm
+import sqlalchemy.pool
 
 from . import common
 from ..download import download
@@ -359,6 +360,7 @@ class SpecialLiveRotation(common.GameDBBase):
 
 engine = sqlalchemy.ext.asyncio.create_async_engine(
     f"sqlite+aiosqlite:///file:{download.get_db_path('live')}?mode=ro&uri=true",
+    poolclass=sqlalchemy.pool.NullPool,
     connect_args={"check_same_thread": False},
 )
 sessionmaker = sqlalchemy.ext.asyncio.async_sessionmaker(engine)

npps4/db/main.py

@@ -3,6 +3,7 @@
 import hmac
 
 import sqlalchemy
+import sqlalchemy.engine.url
 import sqlalchemy.ext.asyncio
 import sqlalchemy.orm
 
@@ -467,7 +468,14 @@ class MigrationFixes(common.Base, kw_only=True):
     revision: sqlalchemy.orm.Mapped[str] = sqlalchemy.orm.mapped_column(primary_key=True)
 
 
-engine = sqlalchemy.ext.asyncio.create_async_engine(config.get_database_url(), connect_args={"autocommit": False})
+# Have 0 connection pool with default max_overflow on SQLite3 only
+if sqlalchemy.engine.url.make_url(config.get_database_url()).get_backend_name() == "sqlite":
+    _poolsize = 0
+else:
+    _poolsize = 5
+engine = sqlalchemy.ext.asyncio.create_async_engine(
+    config.get_database_url(), pool_size=_poolsize, connect_args={"autocommit": False}
+)
 sessionmaker = sqlalchemy.ext.asyncio.async_sessionmaker(engine)
 
 

npps4/db/museum.py

@@ -1,6 +1,7 @@
 import sqlalchemy
 import sqlalchemy.ext.asyncio
 import sqlalchemy.orm
+import sqlalchemy.pool
 
 from . import common
 from ..download import download
@@ -49,6 +50,7 @@ class MuseumContents(common.GameDBBase, common.MaybeEncrypted):
 
 engine = sqlalchemy.ext.asyncio.create_async_engine(
     f"sqlite+aiosqlite:///file:{download.get_db_path('museum')}?mode=ro&uri=true",
+    poolclass=sqlalchemy.pool.NullPool,
     connect_args={"check_same_thread": False},
 )
 sessionmaker = sqlalchemy.ext.asyncio.async_sessionmaker(engine)

npps4/db/scenario.py

@@ -1,6 +1,7 @@
 import sqlalchemy
 import sqlalchemy.ext.asyncio
 import sqlalchemy.orm
+import sqlalchemy.pool
 
 from . import common
 from ..download import download
@@ -70,6 +71,8 @@ class Chapter(common.GameDBBase, common.MaybeEncrypted):
 
 engine = sqlalchemy.ext.asyncio.create_async_engine(
     f"sqlite+aiosqlite:///file:{download.get_db_path('scenario')}?mode=ro&uri=true",
+    poolclass=sqlalchemy.pool.NullPool,
+    connect_args={"check_same_thread": False},
 )
 sessionmaker = sqlalchemy.ext.asyncio.async_sessionmaker(engine)
 

npps4/db/subscenario.py

@@ -1,6 +1,7 @@
 import sqlalchemy
 import sqlalchemy.ext.asyncio
 import sqlalchemy.orm
+import sqlalchemy.pool
 
 from . import common
 from ..download import download
@@ -31,6 +32,7 @@ class SubScenario(common.GameDBBase, common.MaybeEncrypted):
 
 engine = sqlalchemy.ext.asyncio.create_async_engine(
     f"sqlite+aiosqlite:///file:{download.get_db_path('subscenario')}?mode=ro&uri=true",
+    poolclass=sqlalchemy.pool.NullPool,
     connect_args={"check_same_thread": False},
 )
 sessionmaker = sqlalchemy.ext.asyncio.async_sessionmaker(engine)