files relocation
This commit is contained in:
Alexey Avramov
478
src/psi2log
Executable file
478
src/psi2log
Executable file
@@ -0,0 +1,478 @@
|
||||
#!/usr/bin/env python3
|
||||
"""psi2log"""
|
||||
|
||||
import os
|
||||
from time import sleep, monotonic
|
||||
from ctypes import CDLL
|
||||
from sys import stdout, exit
|
||||
from argparse import ArgumentParser
|
||||
from signal import signal, SIGTERM, SIGINT, SIGQUIT, SIGHUP
|
||||
|
||||
|
||||
def form(num):
|
||||
"""
|
||||
"""
|
||||
s = str(num).split('.')
|
||||
return '{}.{:0<2}'.format(s[0], s[1])
|
||||
|
||||
|
||||
def signal_handler(signum, frame):
|
||||
"""
|
||||
"""
|
||||
def signal_handler_inner(signum, frame):
|
||||
pass
|
||||
for i in sig_list:
|
||||
signal(i, signal_handler_inner)
|
||||
|
||||
log('')
|
||||
|
||||
lpd = len(peaks_dict)
|
||||
if lpd != 15:
|
||||
exit()
|
||||
|
||||
log('=================================')
|
||||
log('Peak values: avg10 avg60 avg300')
|
||||
|
||||
log('----------- ------ ------ ------')
|
||||
|
||||
log('some cpu {:>6} {:>6} {:>6}'.format(
|
||||
form(peaks_dict['c_some_avg10']),
|
||||
form(peaks_dict['c_some_avg60']),
|
||||
form(peaks_dict['c_some_avg300']),
|
||||
))
|
||||
|
||||
log('----------- ------ ------ ------')
|
||||
|
||||
log('some memory {:>6} {:>6} {:>6}'.format(
|
||||
form(peaks_dict['m_some_avg10']),
|
||||
form(peaks_dict['m_some_avg60']),
|
||||
form(peaks_dict['m_some_avg300']),
|
||||
))
|
||||
|
||||
log('full memory {:>6} {:>6} {:>6}'.format(
|
||||
form(peaks_dict['m_full_avg10']),
|
||||
form(peaks_dict['m_full_avg60']),
|
||||
form(peaks_dict['m_full_avg300']),
|
||||
))
|
||||
|
||||
log('----------- ------ ------ ------')
|
||||
|
||||
log('some io {:>6} {:>6} {:>6}'.format(
|
||||
form(peaks_dict['i_some_avg10']),
|
||||
form(peaks_dict['i_some_avg60']),
|
||||
form(peaks_dict['i_some_avg300']),
|
||||
))
|
||||
|
||||
log('full io {:>6} {:>6} {:>6}'.format(
|
||||
form(peaks_dict['i_full_avg10']),
|
||||
form(peaks_dict['i_full_avg60']),
|
||||
form(peaks_dict['i_full_avg300']),
|
||||
))
|
||||
|
||||
exit()
|
||||
|
||||
|
||||
def cgroup2_root():
|
||||
"""
|
||||
"""
|
||||
with open(mounts) as f:
|
||||
for line in f:
|
||||
if cgroup2_separator in line:
|
||||
return line.partition(cgroup2_separator)[0].partition(' ')[2]
|
||||
|
||||
|
||||
def mlockall():
|
||||
"""
|
||||
"""
|
||||
|
||||
MCL_CURRENT = 1
|
||||
MCL_FUTURE = 2
|
||||
MCL_ONFAULT = 4
|
||||
|
||||
CDLL('libc.so.6').mlockall(MCL_CURRENT | MCL_FUTURE | MCL_ONFAULT)
|
||||
|
||||
|
||||
def psi_file_mem_to_metrics(psi_path):
|
||||
"""
|
||||
"""
|
||||
|
||||
with open(psi_path) as f:
|
||||
psi_list = f.readlines()
|
||||
some_list, full_list = psi_list[0].split(' '), psi_list[1].split(' ')
|
||||
some_avg10 = some_list[1].split('=')[1]
|
||||
some_avg60 = some_list[2].split('=')[1]
|
||||
some_avg300 = some_list[3].split('=')[1]
|
||||
full_avg10 = full_list[1].split('=')[1]
|
||||
full_avg60 = full_list[2].split('=')[1]
|
||||
full_avg300 = full_list[3].split('=')[1]
|
||||
return (some_avg10, some_avg60, some_avg300,
|
||||
full_avg10, full_avg60, full_avg300)
|
||||
|
||||
|
||||
def psi_file_cpu_to_metrics(psi_path):
|
||||
"""
|
||||
"""
|
||||
|
||||
with open(psi_path) as f:
|
||||
psi_list = f.readlines()
|
||||
some_list = psi_list[0].split(' ')
|
||||
some_avg10 = some_list[1].split('=')[1]
|
||||
some_avg60 = some_list[2].split('=')[1]
|
||||
some_avg300 = some_list[3].split('=')[1]
|
||||
return (some_avg10, some_avg60, some_avg300)
|
||||
|
||||
|
||||
def psi_file_mem_to_total(psi_path):
|
||||
"""
|
||||
"""
|
||||
|
||||
with open(psi_path) as f:
|
||||
psi_list = f.readlines()
|
||||
some_list, full_list = psi_list[0].split(' '), psi_list[1].split(' ')
|
||||
some_total = some_list[4].split('=')[1]
|
||||
full_total = full_list[4].split('=')[1]
|
||||
|
||||
return int(some_total), int(full_total)
|
||||
|
||||
|
||||
def psi_file_cpu_to_total(psi_path):
|
||||
"""
|
||||
"""
|
||||
|
||||
with open(psi_path) as f:
|
||||
psi_list = f.readlines()
|
||||
some_list = psi_list[0].split(' ')
|
||||
some_total = some_list[4].split('=')[1]
|
||||
|
||||
return int(some_total)
|
||||
|
||||
|
||||
def print_head_1():
|
||||
log('=================================================================='
|
||||
'================================================')
|
||||
log(' some cpu pressure || some memory pressure | full memory pressur'
|
||||
'e || some io pressure | full io pressure')
|
||||
log('-------------------- || -------------------- | -------------------'
|
||||
'- || -------------------- | --------------------')
|
||||
log(' avg10 avg60 avg300 || avg10 avg60 avg300 | avg10 avg60 avg30'
|
||||
'0 || avg10 avg60 avg300 | avg10 avg60 avg300')
|
||||
log('------ ------ ------ || ------ ------ ------ | ------ ------ -----'
|
||||
'- || ------ ------ ------ | ------ ------ ------')
|
||||
|
||||
|
||||
def print_head_2():
|
||||
log('============================================')
|
||||
log(' cpu | memory | io |')
|
||||
log('----- | ----------- | ----------- |')
|
||||
log(' some | some full | some full | interval')
|
||||
log('----- | ----- ----- | ----- ----- | --------')
|
||||
|
||||
|
||||
def log(*msg):
|
||||
"""
|
||||
"""
|
||||
print(*msg)
|
||||
if separate_log:
|
||||
logging.info(*msg)
|
||||
|
||||
|
||||
parser = ArgumentParser()
|
||||
|
||||
parser.add_argument(
|
||||
'-t',
|
||||
'--target',
|
||||
help="""target (cgroup_v2 or SYTSTEM_WIDE)""",
|
||||
default='SYSTEM_WIDE',
|
||||
type=str
|
||||
)
|
||||
|
||||
|
||||
parser.add_argument(
|
||||
'-i',
|
||||
'--interval',
|
||||
help="""interval in sec""",
|
||||
default=2,
|
||||
type=float
|
||||
)
|
||||
|
||||
|
||||
parser.add_argument(
|
||||
'-l',
|
||||
'--log',
|
||||
help="""path to log file""",
|
||||
default=None,
|
||||
type=str
|
||||
)
|
||||
|
||||
|
||||
parser.add_argument(
|
||||
'-m',
|
||||
'--mode',
|
||||
help="""mode (1 or 2)""",
|
||||
default='1',
|
||||
type=str
|
||||
)
|
||||
|
||||
|
||||
args = parser.parse_args()
|
||||
target = args.target
|
||||
interval = args.interval
|
||||
log_file = args.log
|
||||
mode = args.mode
|
||||
|
||||
|
||||
if log_file is None:
|
||||
separate_log = False
|
||||
else:
|
||||
separate_log = True
|
||||
import logging
|
||||
|
||||
sig_list = [SIGTERM, SIGINT, SIGQUIT, SIGHUP]
|
||||
|
||||
|
||||
for i in sig_list:
|
||||
signal(i, signal_handler)
|
||||
|
||||
|
||||
if separate_log:
|
||||
logging.basicConfig(
|
||||
filename=log_file,
|
||||
level=logging.INFO,
|
||||
format="%(asctime)s: %(message)s")
|
||||
|
||||
log('Starting psi2log')
|
||||
log('target: {}'.format(target))
|
||||
log('interval: {} sec'.format(interval))
|
||||
if log_file is not None:
|
||||
log('log file: {}'.format(log_file))
|
||||
log('mode: {}'.format(mode))
|
||||
|
||||
|
||||
try:
|
||||
psi_file_mem_to_metrics('/proc/pressure/memory')
|
||||
except Exception as e:
|
||||
print('ERROR: {}'.format(e))
|
||||
print('PSI metrics are not provided by the kernel. Exit.')
|
||||
exit(1)
|
||||
|
||||
|
||||
if target == 'SYSTEM_WIDE':
|
||||
|
||||
cpu_file = "/proc/pressure/cpu"
|
||||
memory_file = "/proc/pressure/memory"
|
||||
io_file = "/proc/pressure/io"
|
||||
|
||||
else:
|
||||
|
||||
mounts = '/proc/mounts'
|
||||
cgroup2_separator = ' cgroup2 rw,'
|
||||
cgroup2_mountpoint = cgroup2_root()
|
||||
|
||||
if cgroup2_mountpoint is None:
|
||||
|
||||
log('ERROR: unified cgroup hierarchy is not mounted, exit')
|
||||
exit(1)
|
||||
|
||||
else:
|
||||
|
||||
log('cgroup_v2 mountpoint: {}'.format(cgroup2_mountpoint))
|
||||
|
||||
cpu_file = cgroup2_mountpoint + target + "/cpu.pressure"
|
||||
memory_file = cgroup2_mountpoint + target + "/memory.pressure"
|
||||
io_file = cgroup2_mountpoint + target + "/io.pressure"
|
||||
|
||||
|
||||
peaks_dict = dict()
|
||||
|
||||
|
||||
mlockall()
|
||||
|
||||
|
||||
if mode == '2':
|
||||
|
||||
print_head_2()
|
||||
|
||||
total_cs0 = psi_file_cpu_to_total(cpu_file)
|
||||
total_ms0, total_mf0 = psi_file_mem_to_total(memory_file)
|
||||
total_is0, total_if0 = psi_file_mem_to_total(io_file)
|
||||
monotonic0 = monotonic()
|
||||
sleep(interval)
|
||||
|
||||
while True:
|
||||
|
||||
total_cs1 = psi_file_cpu_to_total(cpu_file)
|
||||
total_ms1, total_mf1 = psi_file_mem_to_total(memory_file)
|
||||
total_is1, total_if1 = psi_file_mem_to_total(io_file)
|
||||
monotonic1 = monotonic()
|
||||
dm = monotonic1 - monotonic0
|
||||
monotonic0 = monotonic1
|
||||
|
||||
dtotal_cs = total_cs1 - total_cs0
|
||||
avg_cs = dtotal_cs / dm / 10000
|
||||
total_cs0 = total_cs1
|
||||
|
||||
dtotal_ms = total_ms1 - total_ms0
|
||||
avg_ms = dtotal_ms / dm / 10000
|
||||
total_ms0 = total_ms1
|
||||
|
||||
dtotal_mf = total_mf1 - total_mf0
|
||||
avg_mf = dtotal_mf / dm / 10000
|
||||
total_mf0 = total_mf1
|
||||
|
||||
dtotal_is = total_is1 - total_is0
|
||||
avg_is = dtotal_is / dm / 10000
|
||||
total_is0 = total_is1
|
||||
|
||||
dtotal_if = total_if1 - total_if0
|
||||
avg_if = dtotal_if / dm / 10000
|
||||
total_if0 = total_if1
|
||||
|
||||
log('{:>5} | {:>5} {:>5} | {:>5} {:>5} | {}'.format(
|
||||
|
||||
round(avg_cs, 1),
|
||||
|
||||
round(avg_ms, 1),
|
||||
round(avg_mf, 1),
|
||||
|
||||
round(avg_is, 1),
|
||||
round(avg_if, 1),
|
||||
|
||||
round(dm, 3)
|
||||
|
||||
))
|
||||
|
||||
stdout.flush()
|
||||
sleep(interval)
|
||||
|
||||
|
||||
if mode != '1':
|
||||
log('ERROR: invalid mode. Exit.')
|
||||
exit(1)
|
||||
|
||||
|
||||
print_head_1()
|
||||
|
||||
|
||||
while True:
|
||||
|
||||
if not os.path.exists(cpu_file):
|
||||
log('ERROR: cpu pressure file does not exist: {}'.format(cpu_file))
|
||||
sleep(interval)
|
||||
continue
|
||||
|
||||
if not os.path.exists(memory_file):
|
||||
log('ERROR: memory pressure file does not exist: {}'.format(
|
||||
memory_file))
|
||||
sleep(interval)
|
||||
continue
|
||||
|
||||
if not os.path.exists(io_file):
|
||||
log('ERROR: io pressure file does not exist: {}'.format(cpu_file))
|
||||
sleep(interval)
|
||||
continue
|
||||
|
||||
(c_some_avg10, c_some_avg60, c_some_avg300
|
||||
) = psi_file_cpu_to_metrics(cpu_file)
|
||||
|
||||
(m_some_avg10, m_some_avg60, m_some_avg300,
|
||||
m_full_avg10, m_full_avg60, m_full_avg300
|
||||
) = psi_file_mem_to_metrics(memory_file)
|
||||
|
||||
(i_some_avg10, i_some_avg60, i_some_avg300,
|
||||
i_full_avg10, i_full_avg60, i_full_avg300
|
||||
) = psi_file_mem_to_metrics(io_file)
|
||||
|
||||
log('{:>6} {:>6} {:>6} || {:>6} {:>6} {:>6} | {:>6} {:>6} {:>6} || {:>6}'
|
||||
' {:>6} {:>6} | {:>6} {:>6} {:>6}'.format(
|
||||
|
||||
c_some_avg10, c_some_avg60, c_some_avg300,
|
||||
|
||||
m_some_avg10, m_some_avg60, m_some_avg300,
|
||||
m_full_avg10, m_full_avg60, m_full_avg300,
|
||||
|
||||
i_some_avg10, i_some_avg60, i_some_avg300,
|
||||
i_full_avg10, i_full_avg60, i_full_avg300
|
||||
|
||||
))
|
||||
|
||||
c_some_avg10 = float(c_some_avg10)
|
||||
if ('c_some_avg10' not in peaks_dict or
|
||||
peaks_dict['c_some_avg10'] < c_some_avg10):
|
||||
peaks_dict['c_some_avg10'] = c_some_avg10
|
||||
|
||||
c_some_avg60 = float(c_some_avg60)
|
||||
if ('c_some_avg60' not in peaks_dict or
|
||||
peaks_dict['c_some_avg60'] < c_some_avg60):
|
||||
peaks_dict['c_some_avg60'] = c_some_avg60
|
||||
|
||||
c_some_avg300 = float(c_some_avg300)
|
||||
if ('c_some_avg300' not in peaks_dict or
|
||||
peaks_dict['c_some_avg300'] < c_some_avg300):
|
||||
peaks_dict['c_some_avg300'] = c_some_avg300
|
||||
|
||||
#######################################################################
|
||||
|
||||
m_some_avg10 = float(m_some_avg10)
|
||||
if ('m_some_avg10' not in peaks_dict or
|
||||
peaks_dict['m_some_avg10'] < m_some_avg10):
|
||||
peaks_dict['m_some_avg10'] = m_some_avg10
|
||||
|
||||
m_some_avg60 = float(m_some_avg60)
|
||||
if ('m_some_avg60' not in peaks_dict or
|
||||
peaks_dict['m_some_avg60'] < m_some_avg60):
|
||||
peaks_dict['m_some_avg60'] = m_some_avg60
|
||||
|
||||
m_some_avg300 = float(m_some_avg300)
|
||||
if ('m_some_avg300' not in peaks_dict or
|
||||
peaks_dict['m_some_avg300'] < m_some_avg300):
|
||||
peaks_dict['m_some_avg300'] = m_some_avg300
|
||||
|
||||
m_full_avg10 = float(m_full_avg10)
|
||||
if ('m_full_avg10' not in peaks_dict or
|
||||
peaks_dict['m_full_avg10'] < m_full_avg10):
|
||||
peaks_dict['m_full_avg10'] = m_full_avg10
|
||||
|
||||
m_full_avg60 = float(m_full_avg60)
|
||||
if ('m_full_avg60' not in peaks_dict or
|
||||
peaks_dict['m_full_avg60'] < m_full_avg60):
|
||||
peaks_dict['m_full_avg60'] = m_full_avg60
|
||||
|
||||
m_full_avg300 = float(m_full_avg300)
|
||||
if ('m_full_avg300' not in peaks_dict or
|
||||
peaks_dict['m_full_avg300'] < m_full_avg300):
|
||||
peaks_dict['m_full_avg300'] = m_full_avg300
|
||||
|
||||
#######################################################################
|
||||
|
||||
i_some_avg10 = float(i_some_avg10)
|
||||
if ('i_some_avg10' not in peaks_dict or
|
||||
peaks_dict['i_some_avg10'] < i_some_avg10):
|
||||
peaks_dict['i_some_avg10'] = i_some_avg10
|
||||
|
||||
i_some_avg60 = float(i_some_avg60)
|
||||
if ('i_some_avg60' not in peaks_dict or
|
||||
peaks_dict['i_some_avg60'] < i_some_avg60):
|
||||
peaks_dict['i_some_avg60'] = i_some_avg60
|
||||
|
||||
i_some_avg300 = float(i_some_avg300)
|
||||
if ('i_some_avg300' not in peaks_dict or
|
||||
peaks_dict['i_some_avg300'] < i_some_avg300):
|
||||
peaks_dict['i_some_avg300'] = i_some_avg300
|
||||
|
||||
i_full_avg10 = float(i_full_avg10)
|
||||
if ('i_full_avg10' not in peaks_dict or
|
||||
peaks_dict['i_full_avg10'] < i_full_avg10):
|
||||
peaks_dict['i_full_avg10'] = i_full_avg10
|
||||
|
||||
i_full_avg60 = float(i_full_avg60)
|
||||
if ('i_full_avg60' not in peaks_dict or
|
||||
peaks_dict['i_full_avg60'] < i_full_avg60):
|
||||
peaks_dict['i_full_avg60'] = i_full_avg60
|
||||
|
||||
i_full_avg300 = float(i_full_avg300)
|
||||
if ('i_full_avg300' not in peaks_dict or
|
||||
peaks_dict['i_full_avg300'] < i_full_avg300):
|
||||
peaks_dict['i_full_avg300'] = i_full_avg300
|
||||
|
||||
stdout.flush()
|
||||
sleep(interval)
|
||||
Reference in New Issue
Block a user