summaryrefslogtreecommitdiffstats
path: root/policyhandler/onap/process_info.py
blob: 9fb633429137537e0d9dfc17a8b26eb2b16f309b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
# ================================================================================
# Copyright (c) 2018 AT&T Intellectual Property. All rights reserved.
# ================================================================================
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# ============LICENSE_END=========================================================
#
# ECOMP is a trademark and service mark of AT&T Intellectual Property.

"""generic class to keep get real time info about the current process"""

import gc
import sys
import threading
import traceback
from functools import wraps

import psutil


def safe_operation(func):
    """safequard the function against any exception"""
    if not func:
        return

    @wraps(func)
    def wrapper(*args, **kwargs):
        """wrapper around the function"""
        try:
            return func(*args, **kwargs)
        except Exception as ex:
            return "%s: %s" % (type(ex).__name__, str(ex))
    return wrapper


class ProcessInfo(object):
    """static class to calculate process info"""
    _BIBYTES_SYMBOLS = ('KiB', 'MiB', 'GiB', 'TiB', 'PiB', 'EiB', 'ZiB', 'YiB')
    _BIBYTES_VALS = {}
    _inited = False
    _lock = threading.Lock()

    @staticmethod
    def init():
        """init static constants"""
        if ProcessInfo._inited:
            return
        with ProcessInfo._lock:
            if ProcessInfo._inited:
                return

            for i, bibytes_symbol in enumerate(ProcessInfo._BIBYTES_SYMBOLS):
                ProcessInfo._BIBYTES_VALS[bibytes_symbol] = 1 << (i + 1) * 10
            ProcessInfo._BIBYTES_SYMBOLS = list(reversed(ProcessInfo._BIBYTES_SYMBOLS))
            ProcessInfo._inited = True

    @staticmethod
    def bytes_to_bibytes(byte_count):
        """converts byte count to human value in kibi-mebi-gibi-...-bytes"""
        if byte_count is None:
            return "unknown"
        if not byte_count or not isinstance(byte_count, int):
            return byte_count
        ProcessInfo.init()

        for bibytes_symbol in ProcessInfo._BIBYTES_SYMBOLS:
            bibytes_value = ProcessInfo._BIBYTES_VALS[bibytes_symbol]
            if byte_count >= bibytes_value:
                value = float(byte_count) / bibytes_value
                return '%.2f %s' % (value, bibytes_symbol)
        return "%s B" % byte_count

    @staticmethod
    @safe_operation
    def process_memory():
        """calculates the memory usage of the current process"""
        process = psutil.Process()
        with process.oneshot():
            return dict((k, ProcessInfo.bytes_to_bibytes(v))
                        for k, v in process.memory_full_info()._asdict().iteritems())


    @staticmethod
    @safe_operation
    def virtual_memory():
        """calculates the virtual memory usage of the whole vm"""
        return dict((k, ProcessInfo.bytes_to_bibytes(v))
                    for k, v in psutil.virtual_memory()._asdict().iteritems())


    @staticmethod
    @safe_operation
    def active_threads():
        """list of active threads"""
        return sorted([thr.name + "(" + str(thr.ident) + ")" for thr in threading.enumerate()])


    @staticmethod
    @safe_operation
    def thread_stacks():
        """returns the current threads with their stack"""
        thread_names = dict((thr.ident, thr.name) for thr in threading.enumerate())
        return [
            {
                "thread_id" : thread_id,
                "thread_name" : thread_names.get(thread_id),
                "thread_stack" : [
                    {
                        "filename" : filename,
                        "lineno" : lineno,
                        "function" : function_name,
                        "line" : line.strip() if line else None
                    }
                    for filename, lineno, function_name, line in traceback.extract_stack(stack)
                ]
            }
            for thread_id, stack in sys._current_frames().items()
        ]


    @staticmethod
    @safe_operation
    def gc_info(full=False):
        """gets info from garbage collector"""
        gc_info = {
            "gc_count" : str(gc.get_count()),
            "gc_threshold" : str(gc.get_threshold())
        }
        if gc.garbage:
            gc_info["gc_garbage"] = ([repr(stuck) for stuck in gc.garbage]
                                     if full else len(gc.garbage))
        return gc_info

    @staticmethod
    def get_all():
        """all info"""
        return {
            "active_threads" : ProcessInfo.active_threads(),
            "gc" : ProcessInfo.gc_info(full=True),
            "process_memory" : ProcessInfo.process_memory(),
            "virtual_memory" : ProcessInfo.virtual_memory(),
            "thread_stacks" : ProcessInfo.thread_stacks()
        }