26 lines
903 B
Diff
26 lines
903 B
Diff
--- a/apps/blockchain-node/src/aitbc_chain/database.py
|
|
+++ b/apps/blockchain-node/src/aitbc_chain/database.py
|
|
@@ -3,11 +3,22 @@
|
|
from contextlib import contextmanager
|
|
|
|
from sqlmodel import Session, SQLModel, create_engine
|
|
+from sqlalchemy import event
|
|
|
|
from .config import settings
|
|
|
|
_engine = create_engine(f"sqlite:///{settings.db_path}", echo=False)
|
|
|
|
+@event.listens_for(_engine, "connect")
|
|
+def set_sqlite_pragma(dbapi_connection, connection_record):
|
|
+ cursor = dbapi_connection.cursor()
|
|
+ cursor.execute("PRAGMA journal_mode=WAL")
|
|
+ cursor.execute("PRAGMA synchronous=NORMAL")
|
|
+ cursor.execute("PRAGMA cache_size=-64000")
|
|
+ cursor.execute("PRAGMA temp_store=MEMORY")
|
|
+ cursor.execute("PRAGMA mmap_size=30000000000")
|
|
+ cursor.execute("PRAGMA busy_timeout=5000")
|
|
+ cursor.close()
|
|
|
|
def init_db() -> None:
|
|
settings.db_path.parent.mkdir(parents=True, exist_ok=True)
|