| Server IP : 188.40.95.74 / Your IP : 216.73.216.142 Web Server : Apache System : Linux cp01.striminghost.net 3.10.0-1160.119.1.el7.tuxcare.els13.x86_64 #1 SMP Fri Nov 22 06:29:45 UTC 2024 x86_64 User : vlasotin ( 1054) PHP Version : 5.6.40 Disable Function : NONE MySQL : ON | cURL : ON | WGET : ON | Perl : ON | Python : ON | Sudo : ON | Pkexec : ON Directory : /opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/plugins/ |
Upload File : |
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)")