observability.svc.plus/bin/pgmon-add
2026-02-01 20:53:55 +08:00

75 lines
2.7 KiB
Bash
Executable File

#!/bin/bash
set -uo pipefail
#==============================================================#
# File : pgmon-add
# Desc : Add remote database to surveillance
# Ctime : 2023-01-11
# Mtime : 2025-12-31
# Path : bin/pgmon-add
# Deps : ansible-playbook, pgsql-monitor.yml
# License : Apache-2.0 @ https://pigsty.io/docs/about/license/
# Copyright : 2018-2026 Ruohang Feng / Vonng (rh@vonng.com)
#==============================================================#
APP_NAME="$(basename $0)"
APP_DIR="$(cd $(dirname $0) && pwd)"
PIGSTY_HOME=$(cd $(dirname ${APP_DIR}) && pwd)
#--------------------------------------------------------------#
# Usage
#--------------------------------------------------------------#
# bin/pgmon-add <cls> # monitor one remote pgsql cluster 'cls'
# bin/pgmon-add <cls>... # monitor multiple pgsql clusters
#--------------------------------------------------------------#
# Utils
#--------------------------------------------------------------#
__CN='\033[0m';__CK='\033[0;30m';__CR='\033[0;31m';__CG='\033[0;32m';
__CY='\033[0;33m';__CB='\033[0;34m';__CM='\033[0;35m';__CC='\033[0;36m';__CW='\033[0;37m';
function log_info() { printf "[${__CG} OK ${__CN}] ${__CG}$*${__CN}\n"; }
function log_warn() { printf "[${__CY}WARN${__CN}] ${__CY}$*${__CN}\n"; }
function log_error() { printf "[${__CR}FAIL${__CN}] ${__CR}$*${__CN}\n"; }
function log_debug() { printf "[${__CB}HINT${__CN}] ${__CB}$*${__CN}\n"; }
function log_input() { printf "[${__CM} IN ${__CN}] ${__CM}$*\n=> ${__CN}"; }
function log_hint() { printf "${__CB}$*${__CN}\n"; }
function log_line() { printf "${__CM}[$*] ===========================================${__CN}\n"; }
function is_valid_ip(){
if [[ "$1" =~ (([0-9]|[0-9]{2}|1[0-9]{2}|2[0-4][0-9]|25[0-5])\.){3}([0-9]|[0-9]{2}|1[0-9]{2}|2[0-4][0-9]|25[0-5]) ]]; then
return 0
else
return 1
fi
}
if (($# < 1)); then
log_error "missing arguments: cluster name of remote postgres cluster"
log_hint "bin/pgmon-add <cls>..."
exit 1
fi
#--------------------------------------------------------------#
# Execute
#--------------------------------------------------------------#
log_line "PLANNING"
for ((i=1; i<=$#; i++))
do
log_info "monitor remote postgres cluster: ${!i}"
done
log_line "EXECUTE"
for ((i=1; i<=$#; i++))
do
log_info "monitor remote postgres cluster: ${!i}"
log_hint "$ ./pgsql-monitor.yml -e 'clsname=${!i}'"
"${PIGSTY_HOME}/pgsql-monitor.yml" -e "clsname=${!i}"
if [[ $? -ne 0 ]]; then
log_line "FAILURE"
log_error "fail to monitor pgsql cluster for '${!i}'"
exit 2
fi
log_line "SUMMARY"
log_info "monitor remote pgsql cluster ${!i} complete!"
done
exit 0