#!/bin/bash
# GENERE par Set-OPS (role serveur_postgresql). Ne pas editer a la main.
#
# SONDE « base » — la base repond a une requete et garde des connexions libres
#
# Contrat : docs/supervision-conception.md (API des greffons Nagios) — une ligne, 0/1/2.
# Mise en defaut PAR PARAMETRE : `serveur_postgresql_sonde_pct_avert` (seuil impossible).
set -uo pipefail

AVERT={{ serveur_postgresql_sonde_pct_avert }}
CRIT={{ serveur_postgresql_sonde_pct_crit }}

systemctl is-active --quiet postgresql || { echo "PostgreSQL n'est pas actif."; exit 2; }

# UNE REQUETE, PAS UN `pg_isready`. `pg_isready` ouvre une connexion et la ferme : il dit
# que le port repond, pas que la base sert. Une base en recuperation, en lecture seule ou
# a court de connexions passe `pg_isready` et refuse tout travail.
util=$(sudo -u postgres psql -Atc 'select 1' 2>&1)
[[ "${util}" == "1" ]] || { echo "La base n'execute plus de requete : ${util:0:70}"; exit 2; }

ouvertes=$(sudo -u postgres psql -Atc 'select count(*) from pg_stat_activity' 2>/dev/null)
max=$(sudo -u postgres psql -Atc 'show max_connections' 2>/dev/null)
[[ "${ouvertes}" =~ ^[0-9]+$ && "${max}" =~ ^[0-9]+$ ]] || { echo "Base vivante, comptes de connexions illisibles."; exit 1; }
pct=$(( ouvertes * 100 / max ))

if (( pct >= CRIT )); then
  echo "Connexions a ${pct}% (${ouvertes}/${max}) — les applications vont etre refusees.|conn=${ouvertes};;;0;${max}"
  exit 2
fi
if (( pct >= AVERT )); then
  echo "Connexions a ${pct}% (${ouvertes}/${max}), seuil ${AVERT}%.|conn=${ouvertes};;;0;${max}"
  exit 1
fi
echo "Base : requete servie, ${ouvertes}/${max} connexions (${pct}%).|conn=${ouvertes};;;0;${max}"