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