iframe-proxy | Sunbelt Computer Software

History

77 lines (57 loc) · 2.44 KB

Raw

from .api_request import APIRequest

from .utils import timestamp

class ConfigLoader(object):

LOAD_DELAY = 2

LOAD_INTERVAL = 120

def __init__(self, agent):

self.agent = agent

self.load_timer = None

self.last_load_ts = 0

def start(self):

if self.agent.get_option('auto_profiling'):

self.load_timer = self.agent.schedule(self.LOAD_DELAY, self.LOAD_INTERVAL, self.load)

def stop(self):

if self.load_timer:

self.load_timer.cancel()

self.load_timer = None

def load(self, with_interval=False):

now = timestamp()

if with_interval and self.last_load_ts > now - self.LOAD_INTERVAL:

return

self.last_load_ts = now;

try:

api_request = APIRequest(self.agent)

config = api_request.post('config', {})

# agent_enabled yes|no

if 'agent_enabled' in config:

self.agent.config.set_agent_enabled(config['agent_enabled'] == 'yes')

else:

self.agent.config.set_agent_enabled(False)

# profiling_disabled yes|no

if 'profiling_disabled' in config:

self.agent.config.set_profiling_disabled(config['profiling_disabled'] == 'yes')

else:

self.agent.config.set_profiling_disabled(False)

if self.agent.config.is_agent_enabled() and not self.agent.config.is_profiling_disabled():

self.agent.cpu_reporter.start()

self.agent.allocation_reporter.start()

self.agent.block_reporter.start()

self.agent.tf_reporter.start()

else:

self.agent.cpu_reporter.stop()

self.agent.allocation_reporter.stop()

self.agent.block_reporter.stop()

self.agent.tf_reporter.stop()

if self.agent.config.is_agent_enabled():

self.agent.error_reporter.start()

self.agent.span_reporter.start()

self.agent.process_reporter.start()

self.agent.log('Agent activated')

else:

self.agent.error_reporter.stop()

self.agent.span_reporter.stop()

self.agent.process_reporter.stop()

self.agent.log('Agent deactivated')

except Exception:

self.agent.log('Error loading config')

self.agent.exception()

Sunbelt Computer Software

PL/B Language Development and Support

Navigation Menu

Uh oh!

Sunbelt Computer Software

PL/B Language Development and Support

Uh oh!

FilesExpand file tree

config_loader.py

Latest commit

History

config_loader.py

File metadata and controls