From 045d0960c2369d78227a36199dd14b4ea00f5e89 Mon Sep 17 00:00:00 2001
From: fewensa <37804932+fewensa@users.noreply.github.com>
Date: Tue, 22 Oct 2024 11:08:06 +0800
Subject: [PATCH] Update disk check (#10)

---
 ansible/roles/monitor/defaults/main.yml  |  5 ++---
 ansible/roles/monitor/templates/crawl.sh | 22 +++++++++++-----------
 2 files changed, 13 insertions(+), 14 deletions(-)

diff --git a/ansible/roles/monitor/defaults/main.yml b/ansible/roles/monitor/defaults/main.yml
index 366bb00..8e412d5 100644
--- a/ansible/roles/monitor/defaults/main.yml
+++ b/ansible/roles/monitor/defaults/main.yml
@@ -5,12 +5,11 @@ monitor:
   notify_slack_channel: 'darwinia-alert-notification'
   server_name: ''
   check_disks:
-    - /dev/sda
     - /dev/sdb
   alert_thread_cpu_p2: 90
   alert_thread_cpu_p1: 98
   alert_thread_ram_p2: 90
   alert_thread_ram_p1: 98
-  alert_thread_disk_p2: 98
-  alert_thread_disk_p1: 99
+  alert_thread_disk_p2: 6
+  alert_thread_disk_p1: 3
 
diff --git a/ansible/roles/monitor/templates/crawl.sh b/ansible/roles/monitor/templates/crawl.sh
index b149b52..acaf680 100755
--- a/ansible/roles/monitor/templates/crawl.sh
+++ b/ansible/roles/monitor/templates/crawl.sh
@@ -25,10 +25,10 @@ memory_usage() {
   free | grep Mem | awk '{print $3/$2 * 100.0}'
 }
 
-disk_usage() {
+disk_remain() {
   for disk in "${DISKS_TO_MONITOR[@]}"; do
-    usage=$(df -h | grep "^$disk" | awk '{print $5}' | sed 's/%//')
-    echo "$disk $usage"
+    remain=$(df -h | grep "^$disk" | awk '{print $4}' | sed 's/G//')
+    echo "$disk $remain"
   done
 }
 
@@ -79,19 +79,19 @@ generate_disk_alert_message() {
 
   while IFS= read -r line; do
     local disk=$(echo $line | awk '{print $1}')
-    local usage=$(echo $line | awk '{print $2}')
-    if [[ -z "$usage" ]]; then
+    local remain=$(echo $line | awk '{print $2}')
+    if [[ -z "$remain" ]]; then
       continue
     fi
 
-    if (( $(echo "$usage > $ALERT_THREAD_DISK_P1" | bc -l) )); then
+    if (( $(echo "$remain < $ALERT_THREAD_DISK_P1" | bc -l) )); then
       priority='P1'
     fi
-    if (( $(echo "$usage > $ALERT_THREAD_DISK_P2" | bc -l) )); then
-      disk_alert=$(jq -n --arg disk "*DISK* ($disk)" --arg usage "${usage}%" '[{"type":"mrkdwn","text":$disk},{"type":"plain_text","text":$usage}]')
+    if (( $(echo "$remain < $ALERT_THREAD_DISK_P2" | bc -l) )); then
+      disk_alert=$(jq -n --arg disk "*DISK* ($disk)" --arg remain "${remain}G" '[{"type":"mrkdwn","text":$disk},{"type":"plain_text","text":$remain}]')
       alert_message=$(echo "$alert_message" | jq --argjson disk_alert "$disk_alert" '. += $disk_alert')
     fi
-  done < <(disk_usage)
+  done < <(disk_remain)
 
   if [[ "P1" == "$priority" ]]; then
     priority_alert=$(jq -n --arg priority "${priority}" '[{"type":"mrkdwn","text":"*Priority*"},{"type":"plain_text","text":$priority}]')
@@ -158,9 +158,9 @@ send_alert() {
 main() {
   local cpu=$(cpu_usage)
   local ram=$(memory_usage)
-  local disk=$(disk_usage)
+  local disk=$(disk_remain)
   local requests=$(request_count)
-  echo "$(timestamp) CPU: ${cpu}% RAM: ${ram}% Disk: ${disk}% Requests: ${requests}"
+  echo "$(timestamp) CPU: ${cpu}% RAM: ${ram}% Remain Disk: ${disk}G Requests: ${requests}"
 
   check_and_send_alert
 }