From f34b4fdf32c6d97f69a6e4bf9e851023b04c8748 Mon Sep 17 00:00:00 2001 From: Michael Kroes Date: Sun, 1 Nov 2015 03:29:45 -0500 Subject: [PATCH] Add checks to the management interface to report memory usage --- management/status_checks.py | 23 +++++++++++++++++++++++ setup/management.sh | 2 +- 2 files changed, 24 insertions(+), 1 deletion(-) diff --git a/management/status_checks.py b/management/status_checks.py index 9942f34c..bace461c 100755 --- a/management/status_checks.py +++ b/management/status_checks.py @@ -11,6 +11,7 @@ import sys, os, os.path, re, subprocess, datetime, multiprocessing.pool import dns.reversename, dns.resolver import dateutil.parser, dateutil.tz import idna +import psutil from dns_update import get_dns_zones, build_tlsa_record, get_custom_dns_config, get_secondary_dns from web_update import get_web_domains, get_default_www_redirects, get_ssl_certificates, get_domain_ssl_files, get_domains_with_a_records @@ -152,6 +153,7 @@ def run_system_checks(rounded_values, env, output): check_miab_version(env, output) check_system_aliases(env, output) check_free_disk_space(rounded_values, env, output) + check_free_memory(rounded_values, env, output) def check_ssh_password(env, output): # Check that SSH login with password is disabled. The openssh-server @@ -202,6 +204,27 @@ def check_free_disk_space(rounded_values, env, output): else: output.print_error(disk_msg) +def check_free_memory(rounded_values, env, output): + # Check free memory. + percent_used = psutil.virtual_memory().percent + percent_left = 100 - percent_used + if not rounded_values: + memory_msg = "The system has allocated %s%% of the memory." % str(round(percent_used)) + if percent_left > 20: + output.print_ok(memory_msg) + elif percent_left > 15: + output.print_warning(memory_msg) + else: + output.print_error(memory_msg) + else: + memory_msg = "The system has less than %s%% memory left." % str(round(percent_left)) + if percent_left > 20: + output.print_ok("The system has more than 20% memory left") + elif percent_left > 15: + output.print_warning("The system has less than 20% memory left but more than 15%") + else: + output.print_error("The system has less than 15% memory left") + def run_network_checks(env, output): # Also see setup/network-checks.sh. diff --git a/setup/management.sh b/setup/management.sh index 468ba91a..2cd6579a 100755 --- a/setup/management.sh +++ b/setup/management.sh @@ -7,7 +7,7 @@ echo "Installing Mail-in-a-Box system management daemon..." # build-essential libssl-dev libffi-dev python3-dev: Required to pip install cryptography. apt_install python3-flask links duplicity libyaml-dev python3-dnspython python3-dateutil \ build-essential libssl-dev libffi-dev python3-dev python-pip -hide_output pip3 install --upgrade rtyaml email_validator>=1.0.0 idna>=2.0.0 cryptography>=1.0.2 boto +hide_output pip3 install --upgrade rtyaml email_validator>=1.0.0 idna>=2.0.0 cryptography>=1.0.2 boto psutil # duplicity uses python 2 so we need to use the python 2 package of boto hide_output pip install --upgrade boto