#!/opt/cloudlinux/venv/bin/python3 -bb
# coding=utf-8
#
# Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2019 All Rights Reserved
#
# Licensed under CLOUD LINUX LICENSE AGREEMENT
# http://cloudlinux.com/docs/LICENSE.TXT
import sys
from typing import List, Optional # NOQA
from lvestats.core.plugin import LveStatsPlugin # NOQA
from lvestats.core.plugin_loader import PluginLoader
from lvestats.eventloop.main_loop import MainLoop
from lvestats.plugins import default_plugins
DEFAULT_INTERVAL = '5' # seconds
DEFAULT_TIMEOUT = '5' # seconds
LOW_PRIORITY = sys.maxsize
class LveStatsServer(object):
def __init__(self, plugins, config, singleprocess, profiling_log, times):
self.singleprocess = singleprocess
self.plugins = plugins
self.config = config
self.profiling_log = profiling_log
self.times = times
def run(self):
loop = MainLoop(self.config,
interval=int(self.config.get('interval', DEFAULT_INTERVAL)),
plugins=self.plugins,
plugin_timeout=int(self.config.get('timeout', DEFAULT_TIMEOUT)),
multiproc=not self.singleprocess,
profiling_log=self.profiling_log)
loop.run(times=self.times)
def init_plugins(config, user_specified_list_of_plugins=None):
# type: (dict, Optional[List[str]]) -> List[LveStatsPlugin]
if user_specified_list_of_plugins is None:
user_specified_list_of_plugins = []
def_plugins_corteges = default_plugins()
user_plugins = []
plugins_folder = config.get('plugins', None)
if plugins_folder:
plugin_loader = PluginLoader(plugins_folder)
user_plugins.extend(plugin_loader.plugins)
for cls in user_plugins:
setattr(cls, '__is_user_plugin__', True)
plugins = []
plugins.extend(user_plugins)
for def_plugin_cortege in def_plugins_corteges:
cls, plugin_order = def_plugin_cortege
setattr(cls, '__is_user_plugin__', False)
setattr(cls, 'order', plugin_order)
plugins.append(cls)
if not user_specified_list_of_plugins:
plugins.sort(key=lambda pligin: pligin.order if hasattr(pligin, 'order') else LOW_PRIORITY)
else:
plugins = [p for p in plugins if p.__name__ in user_specified_list_of_plugins]
plugins.sort(key=lambda x: user_specified_list_of_plugins.index(x.__name__))
return plugins
def main(config, singleprocess, user_specified_list_of_plugins, profiling_log, times):
plugins = init_plugins(config, user_specified_list_of_plugins)
server = LveStatsServer(plugins, config, singleprocess, profiling_log, times)
try:
server.run()
except KeyboardInterrupt:
print('\nKeyboardInterrupt')
|