psi killer

mempressure.py

import os
import subprocess
import time
import re

THRESHOLD = 10

def main():
  sysrq_mask = int(open("/proc/sys/kernel/sysrq").read())
  sysrq_mask = sysrq_mask | 0x40
  open("/proc/sys/kernel/sysrq", "w").write(str(sysrq_mask))

  while True:
    time.sleep(5)
    pressure_info = open("/proc/pressure/memory").readlines()[1]
    match = re.match(r'full avg10=([\d\.]+).*', pressure_info)
    full_avg10 = float(match.groups()[0])
    print (full_avg10)

    if full_avg10 > THRESHOLD:
      print("Above threshold limit, killing task and pausing for recovery")
      open("/proc/sysrq-trigger", "w").write("f")
      time.sleep(10)

main()