29 lines
859 B
Python
29 lines
859 B
Python
"""Memory watchdog: periodically read the memory usage of the main test process
|
|
and print it out, until terminated."""
|
|
# stdin should refer to the process' /proc/<PID>/statm: we don't pass the
|
|
# process' PID to avoid a race condition in case of - unlikely - PID recycling.
|
|
# If the process crashes, reading from the /proc entry will fail with ESRCH.
|
|
|
|
|
|
import os
|
|
import sys
|
|
import time
|
|
|
|
|
|
try:
|
|
page_size = os.sysconf('SC_PAGESIZE')
|
|
except (ValueError, AttributeError):
|
|
try:
|
|
page_size = os.sysconf('SC_PAGE_SIZE')
|
|
except (ValueError, AttributeError):
|
|
page_size = 4096
|
|
|
|
while True:
|
|
sys.stdin.seek(0)
|
|
statm = sys.stdin.read()
|
|
data = int(statm.split()[5])
|
|
sys.stdout.write(" ... process data size: {data:.1f}G\n"
|
|
.format(data=data * page_size / (1024 ** 3)))
|
|
sys.stdout.flush()
|
|
time.sleep(1)
|