• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1"""Memory watchdog: periodically read the memory usage of the main test process
2and print it out, until terminated."""
3# stdin should refer to the process' /proc/<PID>/statm: we don't pass the
4# process' PID to avoid a race condition in case of - unlikely - PID recycling.
5# If the process crashes, reading from the /proc entry will fail with ESRCH.
6
7
8import os
9import sys
10import time
11
12
13try:
14    page_size = os.sysconf('SC_PAGESIZE')
15except (ValueError, AttributeError):
16    try:
17        page_size = os.sysconf('SC_PAGE_SIZE')
18    except (ValueError, AttributeError):
19        page_size = 4096
20
21while True:
22    sys.stdin.seek(0)
23    statm = sys.stdin.read()
24    data = int(statm.split()[5])
25    sys.stdout.write(" ... process data size: {data:.1f}G\n"
26                     .format(data=data * page_size / (1024 ** 3)))
27    sys.stdout.flush()
28    time.sleep(1)
29