# 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))
|