Viewing file: json_stats_saver.py (1.93 KB) -rw-r--r-- Select action/file-type: (+) | (+) | (+) | Code (+) | Session (+) | (+) | SDB (+) | (+) | (+) | (+) | (+) | (+) |
# 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))
|