From 4bbb07c3ce65c0e1318b5a0479d109ea60c5007f Mon Sep 17 00:00:00 2001
From: Son NK <>
Date: Sat, 15 Aug 2020 13:16:51 +0200
Subject: [PATCH] add monitoring cronjob that monitors how many emails in
 Postfix queues

---
 cron.py     | 38 +++++++++++++++++++++++++++++++++++++-
 crontab.yml |  6 ++++++
 2 files changed, 43 insertions(+), 1 deletion(-)

diff --git a/cron.py b/cron.py
index 256e1670..ed181053 100644
--- a/cron.py
+++ b/cron.py
@@ -1,4 +1,5 @@
 import argparse
+import os
 from dataclasses import dataclass
 from time import sleep
 
@@ -12,6 +13,7 @@ from app.config import (
     ADMIN_EMAIL,
     MACAPP_APPLE_API_SECRET,
     APPLE_API_SECRET,
+    HOST,
 )
 from app.email_utils import (
     send_email,
@@ -26,13 +28,13 @@ from app.models import (
     User,
     Alias,
     EmailLog,
-    Contact,
     CustomDomain,
     Client,
     ManualSubscription,
     RefusedEmail,
     AppleSubscription,
     Mailbox,
+    Monitoring,
 )
 from server import create_app
 
@@ -313,6 +315,36 @@ def sanity_check():
     LOG.d("Finish sanity check")
 
 
+def monitoring():
+    """Look at different metrics and alert appropriately"""
+    incoming_queue = nb_files("/var/spool/postfix/incoming")
+    active_queue = nb_files("/var/spool/postfix/active")
+    deferred_queue = nb_files("/var/spool/postfix/deferred")
+    LOG.d("postfix queue sizes %s %s %s", incoming_queue, active_queue, deferred_queue)
+
+    Monitoring.create(
+        host=HOST,
+        incoming_queue=incoming_queue,
+        active_queue=active_queue,
+        deferred_queue=deferred_queue,
+    )
+    db.session.commit()
+
+    # alert when too many emails in incoming + active queue
+    # 20 is an arbitrary number here
+    if incoming_queue + active_queue > 20:
+        LOG.exception(
+            "Too many emails in incoming & active queue %s %s",
+            incoming_queue,
+            active_queue,
+        )
+
+
+def nb_files(directory) -> int:
+    """return the number of files in directory and its sub-directories"""
+    return sum(len(files) for _, _, files in os.walk(directory))
+
+
 if __name__ == "__main__":
     LOG.d("Start running cronjob")
     parser = argparse.ArgumentParser()
@@ -329,6 +361,7 @@ if __name__ == "__main__":
             "delete_refused_emails",
             "poll_apple_subscription",
             "sanity_check",
+            "monitoring",
         ],
     )
     args = parser.parse_args()
@@ -357,3 +390,6 @@ if __name__ == "__main__":
         elif args.job == "sanity_check":
             LOG.d("Check data consistency")
             sanity_check()
+        elif args.job == "monitoring":
+            LOG.d("Collect and monitor system heath")
+            monitoring()
diff --git a/crontab.yml b/crontab.yml
index 4f5848af..033c1aa1 100644
--- a/crontab.yml
+++ b/crontab.yml
@@ -40,3 +40,9 @@ jobs:
     shell: /bin/bash
     schedule: "0 2 * * *"
     captureStderr: true
+
+  - name: SimpleLogin System Monitoring
+    command: python /code/cron.py -j monitor
+    shell: /bin/bash
+    schedule: "*/5 * * * *"
+    captureStderr: true