mirror of
https://github.com/simple-login/app.git
synced 2024-11-10 17:35:27 +08:00
2660c96fa7
* Fix: Track processes that start with the same chars independently (smtp vs stmpd) * Format * Added more test Co-authored-by: Adrià Casajús <adria.casajus@proton.ch>
89 lines
2.8 KiB
Python
89 lines
2.8 KiB
Python
import os
|
|
import subprocess
|
|
from time import sleep
|
|
from typing import List, Dict
|
|
|
|
import newrelic.agent
|
|
|
|
from app.db import Session
|
|
from app.log import LOG
|
|
|
|
# the number of consecutive fails
|
|
# if more than _max_nb_fails, alert
|
|
# reset whenever the system comes back to normal
|
|
# a system is considered fail if incoming_queue + active_queue > 50
|
|
_nb_failed = 0
|
|
|
|
_max_nb_fails = 10
|
|
|
|
# the maximum number of emails in incoming & active queue
|
|
_max_incoming = 50
|
|
|
|
|
|
@newrelic.agent.background_task()
|
|
def log_postfix_metrics():
|
|
"""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)
|
|
|
|
newrelic.agent.record_custom_metric("Custom/postfix_incoming_queue", incoming_queue)
|
|
newrelic.agent.record_custom_metric("Custom/postfix_active_queue", active_queue)
|
|
newrelic.agent.record_custom_metric("Custom/postfix_deferred_queue", deferred_queue)
|
|
|
|
proc_counts = get_num_procs(["smtp", "smtpd", "bounce", "cleanup"])
|
|
for proc_name in proc_counts:
|
|
LOG.d(f"Process count {proc_counts}")
|
|
newrelic.agent.record_custom_metric(
|
|
f"Custom/process_{proc_name}_count", proc_counts[proc_name]
|
|
)
|
|
|
|
|
|
def nb_files(directory) -> int:
|
|
"""return the number of files in directory and its subdirectories"""
|
|
return sum(len(files) for _, _, files in os.walk(directory))
|
|
|
|
|
|
def get_num_procs(proc_names: List[str]) -> Dict[str, int]:
|
|
data = (
|
|
subprocess.Popen(["ps", "ax"], stdout=subprocess.PIPE)
|
|
.communicate()[0]
|
|
.decode("utf-8")
|
|
)
|
|
return _process_ps_output(proc_names, data)
|
|
|
|
|
|
def _process_ps_output(proc_names: List[str], data: str) -> Dict[str, int]:
|
|
proc_counts = {proc_name: 0 for proc_name in proc_names}
|
|
lines = data.split("\n")
|
|
for line in lines:
|
|
entry = [field for field in line.strip().split() if field.strip()]
|
|
if len(entry) < 5:
|
|
continue
|
|
if entry[4][0] == "[":
|
|
continue
|
|
for proc_name in proc_names:
|
|
if entry[4] == proc_name:
|
|
proc_counts[proc_name] += 1
|
|
return proc_counts
|
|
|
|
|
|
@newrelic.agent.background_task()
|
|
def log_nb_db_connection():
|
|
# get the number of connections to the DB
|
|
r = Session.execute("select count(*) from pg_stat_activity;")
|
|
nb_connection = list(r)[0][0]
|
|
|
|
LOG.d("number of db connections %s", nb_connection)
|
|
newrelic.agent.record_custom_metric("Custom/nb_db_connections", nb_connection)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
while True:
|
|
log_postfix_metrics()
|
|
log_nb_db_connection()
|
|
Session.close()
|
|
|
|
# 1 min
|
|
sleep(60)
|