ACIL FM

Pilih semua

Nama	Ukuran	Aksi
server	-	Open
__pycache__	-	Open
health.py	3.38 MB	View DL Edit
inactivity.py	1.49 MB	View DL Edit
integration_conf.py	884 B	View DL Edit
jwt_issuer.py	3.26 MB	View DL Edit
newsfeed.py	4.3 MB	View DL Edit
pam_auth.py	779 B	View DL Edit
__init__.py	0 B	View DL Edit

Edit file: /opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/api/inactivity.py

"""This module implement inactivity tracker for ImunifyAV to automaticaly
shutdown the process when it is idle for certain time (no RPC calls and
long running tasks).
"""
import time
from contextlib import contextmanager, suppress
from logging import getLogger

logger = getLogger(__name__)

class InactivityTracker:
    def __init__(self):
        self._last_action_timestamp = time.monotonic()
        self._long_action_counter = 0
        self._long_actions_list = []
        self._timeout = 0

def __str__(self):
        return "Time from last action is {:.0f}, long actions {}".format(
            time.monotonic() - self._last_action_timestamp,
            self._long_actions_list,
        )

@contextmanager
    def task(self, name):
        self.start(name)
        try:
            yield
        finally:
            self.stop(name)

def reset_timer(self):
        self._last_action_timestamp = time.monotonic()

def start(self, name):
        self._long_action_counter += 1
        self._long_actions_list.append(name)
        self.reset_timer()

def stop(self, name):
        self._long_action_counter -= 1
        with suppress(ValueError):
            self._long_actions_list.remove(name)
        self.reset_timer()

def is_timeout(self):
        return (not self._long_action_counter) and (
            self._last_action_timestamp + self._timeout <= time.monotonic()
        )

def set_timeout(self, timeout: int) -> None:
        self._timeout = timeout

track = InactivityTracker()

Batal

Isi Zip:

Create