-
Notifications
You must be signed in to change notification settings - Fork 390
/
vnc_cpu_info.py
178 lines (144 loc) · 6.14 KB
/
vnc_cpu_info.py
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
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
#
# Copyright (c) 2013 Juniper Networks, Inc. All rights reserved.
#
"""
.. attention:: Fix the license string
"""
import os
import socket
import psutil
import gevent
from uve.cfgm_cpuinfo.ttypes import *
from uve.cfgm_cpuinfo.cpuinfo.ttypes import *
from buildinfo import build_info
from sandesh_common.vns.ttypes import Module
from sandesh_common.vns.constants import ModuleNames
import vnc_greenlets
# CpuInfo object for config-node
class CpuInfo(object):
def __init__(self, module_id, instance_id, sysinfo_req, sandesh,
time_interval, server_ip=None):
# store cpuinfo at init
self._module_id = module_id
self._instance_id = instance_id
self._sysinfo = sysinfo_req
self._sandesh = sandesh
self._time_interval = time_interval
self._rss = 0
self._vms = 0
self._pvms = 0
self._load_avg = (0, 0, 0)
self._virtmem_info = None
self._num_cpus = 0
self._cpu_share = 0
self._curr_build_info = None
self._new_build_info = None
self._curr_ip = server_ip
self._new_ip = None
# spawn a Greenlet object to do periodic collect and send.
vnc_greenlets.VncGreenlet("VNC CPU Info", self.cpu_stats)
# end __init__
def get_config_node_ip(self):
return self._curr_ip
# end __get_config_ndoe_ip
def set_config_node_ip(self, server_ip):
self._curr_ip = server_ip
# end __set_config_ndoe_ip
def cpu_stats(self):
cfg_process = psutil.Process(os.getpid())
while True:
# collect Vmsizes
self._ip_change = 0
self._build_change = 0
rss = cfg_process.get_memory_info().rss
if (self._rss != rss):
self._rss = rss
vms = cfg_process.get_memory_info().vms
if (self._vms != vms):
self._vms = vms
pvms = vms
if (pvms > self._pvms):
self._pvms = pvms
if self._sysinfo:
# collect CPU Load avg
load_avg = os.getloadavg()
if (load_avg != self._load_avg):
self._load_avg = load_avg
# collect system memory info
self._virtmem_info = psutil.virtual_memory()
if (self._new_ip != self._curr_ip):
self._new_ip = self.get_config_node_ip()
self._ip_change = 1
# Retrieve build_info from package/rpm and cache it
if self._curr_build_info is None:
command = "contrail-version contrail-config | grep 'contrail-config'"
version = os.popen(command).read()
_, rpm_version, build_num = version.split()
self._new_build_info = build_info + '"build-id" : "' + \
rpm_version + '", "build-number" : "' + \
build_num + '"}]}'
if (self._new_build_info != self._curr_build_info):
self._curr_build_info = self._new_build_info
self._build_change = 1
num_cpus = psutil.NUM_CPUS
if (num_cpus != self._num_cpus):
self._num_cpus = num_cpus
cpu_percent = cfg_process.get_cpu_percent(interval=0.1)
cpu_share = cpu_percent / num_cpus
self._cpu_share = cpu_share
self._send_cpustats()
gevent.sleep(self._time_interval)
# end cpu_stats
# Send Uve Object
def _send_cpustats(self):
mod_cpu = ModuleCpuInfo()
mod_cpu.module_id = self._module_id
mod_cpu.instance_id = self._instance_id
mod_cpu.cpu_info = CpuLoadInfo()
# populate number of available CPU
mod_cpu.cpu_info.num_cpu = self._num_cpus
if self._sysinfo:
# populate system memory details
mod_cpu.cpu_info.sys_mem_info = SysMemInfo()
mod_cpu.cpu_info.sys_mem_info.total = self._virtmem_info.total / 1024
mod_cpu.cpu_info.sys_mem_info.used = self._virtmem_info.used / 1024
mod_cpu.cpu_info.sys_mem_info.free = self._virtmem_info.free / 1024
mod_cpu.cpu_info.sys_mem_info.buffers = self._virtmem_info.buffers / 1024
mod_cpu.cpu_info.sys_mem_info.cached = self._virtmem_info.cached / 1024
# populate CPU Load avg
mod_cpu.cpu_info.cpuload = CpuLoadAvg()
mod_cpu.cpu_info.cpuload.one_min_avg = self._load_avg[0]
mod_cpu.cpu_info.cpuload.five_min_avg = self._load_avg[1]
mod_cpu.cpu_info.cpuload.fifteen_min_avg = self._load_avg[2]
# populate Virtual Memory details
mod_cpu.cpu_info.meminfo = MemInfo()
mod_cpu.cpu_info.meminfo.virt = self._vms / 1024
mod_cpu.cpu_info.meminfo.peakvirt = self._pvms / 1024
mod_cpu.cpu_info.meminfo.res = self._rss / 1024
# populate cpu_share, which is calibrated with num_cpu
mod_cpu.cpu_info.cpu_share = self._cpu_share
cpu_load_info_list = [mod_cpu]
cfgm_cpu_uve = ModuleCpuState(module_cpu_info=cpu_load_info_list)
cfgm_cpu_uve.name = socket.gethostname()
if self._sysinfo:
if self._ip_change:
cfgm_cpu_uve.config_node_ip = self._new_ip
if self._build_change:
cfgm_cpu_uve.build_info = self._curr_build_info
cpu_info_trace = ModuleCpuStateTrace(
data=cfgm_cpu_uve, sandesh=self._sandesh)
cpu_info_trace.send(sandesh=self._sandesh)
cnf_cpu_state = ConfigCpuState()
cnf_cpu_state.name = socket.gethostname()
cnf_cpu_info = ProcessCpuInfo()
cnf_cpu_info.module_id = self._module_id
cnf_cpu_info.inst_id = self._instance_id
cnf_cpu_info.cpu_share = self._cpu_share
cnf_cpu_info.mem_virt = mod_cpu.cpu_info.meminfo.virt
cnf_cpu_info.mem_res = mod_cpu.cpu_info.meminfo.res
cnf_cpu_state.cpu_info = [cnf_cpu_info]
cnf_cpu_state_trace = ConfigCpuStateTrace(
sandesh=self._sandesh, data=cnf_cpu_state)
cnf_cpu_state_trace.send(sandesh=self._sandesh)
# end _send_cpustats
# end class CpuInfo