Viewing file:  lvestats_memory_usage.py (1.59 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
from __future__ import absolute_import
from __future__ import division
import logging
import os
import psutil
from guppy import hpy
from lvestats.core.plugin import LveStatsPlugin
class LVEStatsMemoryUsage(LveStatsPlugin):
def __init__(self):
self.log = logging.getLogger('plugin.LVEStatsMemoryUsage')
self.now = 0 # This changes in MainLoop
self.period = 60 * 60
self.threshold = 512
try:
self._process = psutil.Process(os.getpid())
except AttributeError:
self._process = None
self.log.error("Process attribute in psutil is not available. psutil.__version__=%s", psutil.__version__)
self._hpy = hpy()
self._hpy.setrelheap()
@staticmethod
def split(value):
return str(value).split("\n")
def execute(self, lve_data):
if self._process is None:
return
rss = self._process.memory_info()[0] / float(1024 * 1024)
self.log.debug(str(rss))
if rss >= self.threshold:
heap = self._hpy.heap()[:10]
result = {
'heap': self.split(heap),
'byid': self.split(heap.byid),
'byvia': self.split(heap.byvia),
'byrcs': self.split(heap.byrcs)}
self.log.error("LVEStats uses too much memory", extra=result)
elif rss < self.threshold/2:
self._hpy.setrelheap()
|