name : json_stats_saver.py # 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 datetime import os import platform from lvestat import LVEStat from lvestats.core.plugin import LveStatsPlugin class LVEStatsData(LveStatsPlugin): """lve_stats_data encapsulates LVE data gathered from /proc/lve/list.""" order = 1750 def __init__(self, dump_dir="/var/lve/json-stats"): self.period = 60 self.attributes = set(slot for slot in LVEStat.__slots__ if not slot.startswith('_')) - {'id'} self.timestamp = '' self.dump_dir = dump_dir if not os.path.exists(self.dump_dir): os.makedirs(self.dump_dir) def _set_timestamp(self, tstmp=None): if tstmp: self.timestamp = tstmp else: self.timestamp = datetime.datetime.utcnow().isoformat() def format_json(self, lve_usages): """Returns LVE usage data contained in this object in JSON format.""" lve_json_set = [] for lve_id, lve in lve_usages.items(): lve_json_element = f'{{ "id": "{lve_id}"' for lve_attr in dir(lve): if lve_attr in self.attributes: lve_json_element += f', "{lve_attr}": "{getattr(lve, lve_attr)}"' lve_json_element += ' }' lve_json_set.append(lve_json_element) result = '{ ' result += f'"hostname": "{platform.node()}", ' result += f'"timestamp": "{self.timestamp}", ' result += '"data": [ ' + ', '.join(lve_json_set) + ' ]' result += ' }' return result def execute(self, lve_data): usages = lve_data.get('stats', {}) self._set_timestamp() name = self.dump_dir + "/stat." + self.timestamp with open(name, 'w', encoding='utf-8') as f: f.write(self.format_json(usages))