Charles-François Natali | 55bce63 | 2012-03-24 10:06:23 +0100 | [diff] [blame] | 1 | """Memory watchdog: periodically read the memory usage of the main test process |
| 2 | and 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 | |
| 8 | import os |
| 9 | import sys |
| 10 | import time |
| 11 | |
| 12 | |
| 13 | try: |
| 14 | page_size = os.sysconf('SC_PAGESIZE') |
| 15 | except (ValueError, AttributeError): |
| 16 | try: |
| 17 | page_size = os.sysconf('SC_PAGE_SIZE') |
| 18 | except (ValueError, AttributeError): |
| 19 | page_size = 4096 |
| 20 | |
| 21 | while 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) |