prehistoric-systems/web
1
0
Fork 0
mirror of https://codeberg.org/Freedium-cfd/web.git synced 2026-03-11 09:04:37 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

More safe check

Browse source
This commit is contained in:
ZhymabekRoman 2024-02-24 08:10:47 +06:00
parent ecbe4b46ca
commit b67f487be9
1 changed files with 12 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
database-lib/database_lib/cache_db.py
View file

@ -26,15 +26,16 @@ class CacheResponse:
return self.data return self.data
class SQLiteCacheBackend: class SQLiteCacheBackend:
__slots__ = ('connection', 'cursor', 'database') __slots__ = ('connection', 'cursor', 'database', 'lock')
def __init__(self, database: str): def __init__(self, database: str):
self.database = database self.database = database
self.connection = sqlite3.connect(database) self.connection = sqlite3.connect(database, timeout=10.0)
self.connection.enable_load_extension(True) # Enable loading of extensions self.connection.enable_load_extension(True) # Enable loading of extensions
self.connection.execute("PRAGMA foreign_keys = ON;") # Need for working with foreign keys in db self.connection.execute("PRAGMA foreign_keys = ON;") # Need for working with foreign keys in db
self.connection.execute("PRAGMA journal_mode=WAL;") # Need to properly work with ZSTD compression self.connection.execute("PRAGMA journal_mode=WAL;") # Need to properly work with ZSTD compression
self.connection.execute("PRAGMA auto_vacuum=full;") # Same as above thing self.connection.execute("PRAGMA auto_vacuum=full;") # Same as above thing
self.cursor = self.connection.cursor() self.cursor = self.connection.cursor()
self.lock = threading.Lock()
if sqlite_zstd is not None: if sqlite_zstd is not None:
sqlite_zstd.load(self.connection) sqlite_zstd.load(self.connection)
@ -86,8 +87,10 @@ class SQLiteCacheBackend:
raise ValueError(f"Unable to serialize value to JSON: {e}") raise ValueError(f"Unable to serialize value to JSON: {e}")
elif not isinstance(value, str): elif not isinstance(value, str):
raise ValueError(f"value argument should be a string or dict, not {type(value).__name__}") raise ValueError(f"value argument should be a string or dict, not {type(value).__name__}")
with self.connection:
self.cursor.execute("INSERT OR REPLACE INTO cache VALUES (:0, :1)", {'0': key, '1': value}) with self.lock:
with self.connection:
self.cursor.execute("INSERT OR REPLACE INTO cache VALUES (:0, :1)", {'0': key, '1': value})
def delete(self, key: str) -> None: def delete(self, key: str) -> None:
with self.connection: with self.connection:
@ -105,8 +108,10 @@ class SQLiteCacheBackend:
connection.execute("PRAGMA foreign_keys = ON;") # Need for working with foreign keys in db connection.execute("PRAGMA foreign_keys = ON;") # Need for working with foreign keys in db
connection.execute("PRAGMA journal_mode=WAL;") # Need to properly work with ZSTD compression connection.execute("PRAGMA journal_mode=WAL;") # Need to properly work with ZSTD compression
connection.execute("PRAGMA auto_vacuum=full;") # Same as above thing connection.execute("PRAGMA auto_vacuum=full;") # Same as above thing
if sqlite_zstd is not None: if sqlite_zstd is not None:
sqlite_zstd.load(connection) sqlite_zstd.load(connection)
with connection: with connection:
if time is not None: if time is not None:
cursor.execute("SELECT zstd_incremental_maintenance(?, ?);", (time, blocking_time)) cursor.execute("SELECT zstd_incremental_maintenance(?, ?);", (time, blocking_time))
@ -115,6 +120,9 @@ class SQLiteCacheBackend:
cursor.execute("VACUUM") cursor.execute("VACUUM")
cursor.execute("ANALYZE") cursor.execute("ANALYZE")
cursor.close()
connection.close()
def maintenance_thread(self): def maintenance_thread(self):
maintenance_thread = threading.Thread(target=self.maintenance, daemon=True) maintenance_thread = threading.Thread(target=self.maintenance, daemon=True)
maintenance_thread.start() maintenance_thread.start()