Current File : //opt/imunify360/venv/lib64/python3.11/site-packages/defence360agent/plugins/checkpoint.py
from defence360agent.contracts.plugins import MessageSink
from defence360agent.model.instance import db
from defence360agent.utils import recurring_check
class Checkpoint(MessageSink):
"""
Checkpoint imunify360.db periodically to limit unexpected WAL file growing.
"""
ONE_DAY = 24 * 60 * 60
def __init__(self, *, checkpoint_period=ONE_DAY, db=db):
self._checkpoint_period = checkpoint_period
self._db = db
self._task = None
async def create_sink(self, loop):
self._loop = loop
self._task = self._loop.create_task(
recurring_check(self._checkpoint_period)(self._checkpoint)()
)
async def shutdown(self):
task, self._task = self._task, None # avoid cancelling twice
if task is None or task.cancelled():
return
task.cancel()
# CancelledError is handled by @recurring_check():
await task
async def _checkpoint(self):
# 1. may not shrink database wal file in case of this command will be
# during external read process took place
# 2. returning immediately without result if database
# has concurrent transaction
self._db.execute_sql("PRAGMA wal_checkpoint(TRUNCATE)")
Mini Shell