summaryrefslogtreecommitdiff
path: root/ctdb/config/events/legacy/05.system.script
blob: e2ffeac715a87d9fd01b1d597301232dfea354e3 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
#!/bin/sh
# ctdb event script for checking local file system utilization

[ -n "$CTDB_BASE" ] || \
    CTDB_BASE=$(d=$(dirname "$0") ; cd -P "$d" ; dirname "$PWD")

. "${CTDB_BASE}/functions"

load_script_options

ctdb_setup_state_dir "service" "system-monitoring"

validate_percentage ()
{
    case "$1" in
	"") return 1 ;;  # A failure that doesn't need a warning
	[0-9]|[0-9][0-9]|100) return 0 ;;
	*) echo "WARNING: ${1} is an invalid percentage${2:+ in \"}${2}${2:+\"} check"
	   return 1
    esac
}

check_thresholds ()
{
    _thing="$1"
    _thresholds="$2"
    _usage="$3"
    _unhealthy_callout="$4"

    case "$_thresholds" in
	*:*)
	    _warn_threshold="${_thresholds%:*}"
	    _unhealthy_threshold="${_thresholds#*:}"
	    ;;
	*)
	    _warn_threshold="$_thresholds"
	    _unhealthy_threshold=""
    esac

    _t=$(echo "$_thing" | sed -e 's@/@SLASH_@g' -e 's@ @_@g')
    # script_state_dir set by ctdb_setup_state_dir()
    # shellcheck disable=SC2154
    _cache="${script_state_dir}/cache_${_t}"
    if validate_percentage "$_unhealthy_threshold" "$_thing" ; then
        if [ "$_usage" -ge "$_unhealthy_threshold" ] ; then
	    echo "ERROR: ${_thing} utilization ${_usage}% >= threshold ${_unhealthy_threshold}%"
	    eval "$_unhealthy_callout"
	    echo "$_usage" >"$_cache"
	    exit 1
        fi
    fi

    if validate_percentage "$_warn_threshold" "$_thing" ; then
        if [ "$_usage" -ge "$_warn_threshold" ] ; then
	    if [ -r "$_cache" ] ; then
		read _prev <"$_cache"
	    else
		_prev=""
	    fi
	    if [ "$_usage" != "$_prev" ] ; then
		echo "WARNING: ${_thing} utilization ${_usage}% >= threshold ${_warn_threshold}%"
		echo "$_usage" >"$_cache"
	    fi
	else
	    if [ -r "$_cache" ] ; then
		echo "NOTICE: ${_thing} utilization ${_usage}% < threshold ${_warn_threshold}%"
	    fi
	    rm -f "$_cache"
        fi
    fi
}

set_monitor_filsystem_usage_defaults ()
{
    _fs_defaults_cache="${script_state_dir}/cache_filsystem_usage_defaults"

    if [ ! -r "$_fs_defaults_cache" ] ; then
	# Determine filesystem for each database directory, generate
	# an entry to warn at 90%, de-duplicate entries, put all items
	# on 1 line (so the read below gets everything)
	for _t in "${CTDB_DBDIR:-${CTDB_VARDIR}}" \
		      "${CTDB_DBDIR_PERSISTENT:-${CTDB_VARDIR}/persistent}" \
		      "${CTDB_DBDIR_STATE:-${CTDB_VARDIR}/state}" ; do
	    df -kP "$_t" | awk 'NR == 2 { printf "%s:90\n", $6 }'
	done | sort -u | xargs >"$_fs_defaults_cache"
    fi

    read CTDB_MONITOR_FILESYSTEM_USAGE <"$_fs_defaults_cache"
}

monitor_filesystem_usage ()
{
    if [ -z "$CTDB_MONITOR_FILESYSTEM_USAGE" ] ; then
	set_monitor_filsystem_usage_defaults
    fi

    # Check each specified filesystem, specified in format
    # <fs_mount>:<fs_warn_threshold>[:fs_unhealthy_threshold]
    for _fs in $CTDB_MONITOR_FILESYSTEM_USAGE ; do
	_fs_mount="${_fs%%:*}"
	_fs_thresholds="${_fs#*:}"

        if [ ! -d "$_fs_mount" ]; then
            echo "WARNING: Directory ${_fs_mount} does not exist"
	    continue
        fi

        # Get current utilization
        _fs_usage=$(df -kP "$_fs_mount" | \
			   sed -n -e 's@.*[[:space:]]\([[:digit:]]*\)%.*@\1@p')
        if [ -z "$_fs_usage" ] ; then
            echo "WARNING: Unable to get FS utilization for ${_fs_mount}"
	    continue
        fi

	check_thresholds "Filesystem ${_fs_mount}" \
			 "$_fs_thresholds" \
			 "$_fs_usage"
    done
}

dump_memory_info ()
{
    get_proc "meminfo"
    ps auxfww
    set_proc "sysrq-trigger" "m"
}

monitor_memory_usage ()
{
    # Defaults
    if [ -z "$CTDB_MONITOR_MEMORY_USAGE" ] ; then
	CTDB_MONITOR_MEMORY_USAGE=80
    fi
    if [ -z "$CTDB_MONITOR_SWAP_USAGE" ] ; then
	CTDB_MONITOR_SWAP_USAGE=25
    fi

    _meminfo=$(get_proc "meminfo")
    # Intentional word splitting here
    # shellcheck disable=SC2046
    set -- $(echo "$_meminfo" | awk '
$1 == "MemAvailable:" { memavail += $2 }
$1 == "MemFree:"      { memfree  += $2 }
$1 == "Cached:"       { memfree  += $2 }
$1 == "Buffers:"      { memfree  += $2 }
$1 == "MemTotal:"     { memtotal  = $2 }
$1 == "SwapFree:"     { swapfree  = $2 }
$1 == "SwapTotal:"    { swaptotal = $2 }
END {
    if (memavail != 0) { memfree = memavail ; }
    if (memtotal != 0) { print int((memtotal - memfree) / memtotal * 100) ; } else { print 0 ; }
    if (swaptotal != 0) { print int((swaptotal - swapfree) / swaptotal * 100) ; } else { print 0 ; }
}')
    _mem_usage="$1"
    _swap_usage="$2"

    check_thresholds "System memory" \
		     "$CTDB_MONITOR_MEMORY_USAGE" \
		     "$_mem_usage" \
		     dump_memory_info

    check_thresholds "System swap" \
		     "$CTDB_MONITOR_SWAP_USAGE" \
		     "$_swap_usage" \
		     dump_memory_info
}


case "$1" in
monitor)
	# Load/cache database options from configuration file
	ctdb_get_db_options

	monitor_filesystem_usage
	monitor_memory_usage
	;;
esac

exit 0