name : res_mem_collector.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 from clcommon.utils import run_command from lvestats.core.plugin import LveStatsPlugin LVEPS_COMMAND = '/usr/sbin/lveps' class ResMEMCollector(LveStatsPlugin): order = 1500 # should run after other collectors, but before analyzers timeout = 15 """ The goal of this plugin is to substitute physical memory readings from /proc/lve/list with RES data from /proc/* for each process within the LVE The data is extracted from lveps -p command """ @staticmethod def parse_mem_per_lve(lines): """ parse the output of lveps -p -n -o id:10,mem:15 lveps returns memory in megabytes. The data for LVE is taken from /proc/lve/list the data for processes is taken from /proc/PID/... and is what we need. :param: lines list of lines, as outputed by lveps :return: dictionary with LVE id as keys, and memory in bytes as values """ result = {} lve_id = None mem = 0 for i in range(1, len(lines)): val = lines[i].split() if len(val) == 2: if lve_id is not None: result[lve_id] = mem * 256 # the mem comes in MB, we need to convert it into num of 4k pages mem = 0 lve_id = int(val[0]) elif len(val) == 1: mem += float(val[0]) if lve_id is not None: result[lve_id] = mem * 256 # the mem comes in MB, we need to convert it into num of 4k pages return result @staticmethod def get_mem_per_lve(): """ Get amount of memory processes inside LVE use :return: dictionary of LVE_ID <-> MEM """ lveps_output = run_command([LVEPS_COMMAND, '-p', '-n', '-o', 'id:10,mem:15']) return ResMEMCollector.parse_mem_per_lve(lveps_output.split('\n')) def execute(self, lve_data): self.update_lve_data(lve_data, self.get_mem_per_lve()) @staticmethod def update_lve_data(lve_data, pmem_dict): """ updates lve_data['stats'] pmem values for all LVEs Only update LVEs already in lve_data['stats'], skip others :param lve_data: plugin's lve_data :param pmem_dict: dictionary of LVE id <-> memory used in bytes :return: updated stats structure in lve_data """ stats = lve_data['stats'] for lve_id in stats: stat = stats[lve_id] if lve_id in pmem_dict: limit = stat.lmemphy pmem = pmem_dict[lve_id] if limit and pmem > limit: stat.memphy = limit else: stat.memphy = pmem_dict[lve_id] else: stat.memphy = 0