| 12345678910111213141516171819202122232425262728 |
- """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)
|
