268 lines
14 KiB
Bash
268 lines
14 KiB
Bash
#!/bin/bash
|
|
# ==============================================================================================
|
|
# ================================= Bandwidth Monitor ==========================================
|
|
# ==============================================================================================
|
|
# Logs rsync transfer history and generates weekly summary reports.
|
|
# Designed for minimal flash drive impact — one bounded write per rsync run.
|
|
#
|
|
# ── TWO MODES ─────────────────────────────────────────────────────────────────────────────────
|
|
#
|
|
# --log-transfer "profile" duration_seconds status
|
|
# Called automatically by rsync.sh after each sync completes.
|
|
# Appends one line to the log and trims entries older than BANDWIDTH_LOG_RETENTION.
|
|
# Flags syncs exceeding BANDWIDTH_WARN_GB in the log for weekly report highlighting.
|
|
#
|
|
# --report (or no args)
|
|
# Generates a summary from the accumulated log.
|
|
# Shows per-profile breakdown, last 7 days, and overall totals.
|
|
# This is a monitor script — SILENT_MODE=false — output is the point.
|
|
#
|
|
# ── LOG FORMAT ────────────────────────────────────────────────────────────────────────────────
|
|
# One line per transfer — version-proof, never needs rsync output parsing:
|
|
# YYYY-MM-DD|HH:MM|profile|duration_seconds|status|bytes_transferred
|
|
#
|
|
# Log file stays bounded to BANDWIDTH_LOG_RETENTION days — trimmed on every write.
|
|
# Minimal flash drive impact: one append + one trim per rsync run.
|
|
#
|
|
# ── SAFEGUARDS ────────────────────────────────────────────────────────────────────────────────
|
|
# acquire_lock "wait" — prevents log corruption from concurrent rsync completions
|
|
# validate_unraid_cmd — notify script validated before use
|
|
# Atomic log write — temp file + mv prevents partial writes on trim
|
|
# Log existence check — creates log directory if needed, exits cleanly if unwritable
|
|
#
|
|
# ── CONFIGURATION (master.conf) ───────────────────────────────────────────────────────────────
|
|
# BANDWIDTH_LOG — log file path
|
|
# BANDWIDTH_LOG_RETENTION — days before old entries are purged (default 90)
|
|
# BANDWIDTH_WARN_GB — flag syncs larger than this in report (default 50)
|
|
#
|
|
# ── USAGE ─────────────────────────────────────────────────────────────────────────────────────
|
|
# bandwidth_monitor.sh — generate report
|
|
# bandwidth_monitor.sh --report — generate report (explicit)
|
|
# bandwidth_monitor.sh --log-transfer profile secs ok — log a transfer (called by rsync.sh)
|
|
# bandwidth_monitor.sh --status — show config and exit
|
|
# bandwidth_monitor.sh --log — verbose output
|
|
# ==============================================================================================
|
|
|
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
|
|
|
source "$SCRIPT_DIR/../load_config.sh"
|
|
|
|
# Monitor script — output is the point
|
|
SILENT_MODE=false
|
|
|
|
parse_args "$@"
|
|
|
|
# ── Parse mode from PARSED_ARGS ───────────────────────────────────────────────────────────────
|
|
LOG_TRANSFER_MODE=false
|
|
TRANSFER_PROFILE=""
|
|
TRANSFER_DURATION=0
|
|
TRANSFER_STATUS="success"
|
|
TRANSFER_BYTES=0
|
|
|
|
for arg in "${PARSED_ARGS[@]}"; do
|
|
case "$arg" in
|
|
--log-transfer) LOG_TRANSFER_MODE=true ;;
|
|
--report) LOG_TRANSFER_MODE=false ;;
|
|
*)
|
|
if [[ "$LOG_TRANSFER_MODE" == true ]]; then
|
|
if [[ -z "$TRANSFER_PROFILE" ]]; then
|
|
TRANSFER_PROFILE="$arg"
|
|
elif [[ "$TRANSFER_DURATION" -eq 0 && "$arg" =~ ^[0-9]+$ ]]; then
|
|
TRANSFER_DURATION="$arg"
|
|
elif [[ "$arg" == "success" || "$arg" == "failed" ]]; then
|
|
TRANSFER_STATUS="$arg"
|
|
elif [[ "$arg" =~ ^[0-9]+$ ]]; then
|
|
TRANSFER_BYTES="$arg"
|
|
fi
|
|
fi
|
|
;;
|
|
esac
|
|
done
|
|
|
|
# ── Ensure log file exists and is writable ────────────────────────────────────────────────────
|
|
mkdir -p "$(dirname "$BANDWIDTH_LOG")"
|
|
touch "$BANDWIDTH_LOG" 2>/dev/null || {
|
|
error "Cannot write to bandwidth log: $BANDWIDTH_LOG"
|
|
exit 1
|
|
}
|
|
|
|
# ==============================================================================================
|
|
# ━━━ Setup ━━━
|
|
# ==============================================================================================
|
|
if [[ "$EUID" -ne 0 ]]; then
|
|
error "Must be run as root"
|
|
exit 1
|
|
fi
|
|
|
|
validate_unraid_cmd \
|
|
"/usr/local/emhttp/plugins/dynamix/scripts/notify" \
|
|
"" "" \
|
|
"unRAID notify script" || warn "unRAID notify script not found — native notifications disabled"
|
|
|
|
# detect_hosts() sets MY_ID for report header
|
|
detect_hosts
|
|
|
|
# ==============================================================================================
|
|
# ━━━ Status ━━━
|
|
# ==============================================================================================
|
|
if [[ "$SHOW_STATUS" == true ]]; then
|
|
echo ""
|
|
echo "━━━━━ $ICON_SUMMARY STATUS ━━━━━"
|
|
echo "$ICON_HOST Identity: $MY_ID ($LOCAL_SERVER_NAME)"
|
|
echo "$ICON_BANDWIDTH Log file: $BANDWIDTH_LOG"
|
|
echo "$ICON_BANDWIDTH Retention: ${BANDWIDTH_LOG_RETENTION} days"
|
|
echo "$ICON_BANDWIDTH Warn GB: ${BANDWIDTH_WARN_GB}GB"
|
|
local entry_count=0
|
|
[[ -f "$BANDWIDTH_LOG" ]] && entry_count=$(wc -l < "$BANDWIDTH_LOG")
|
|
echo "$ICON_BANDWIDTH Log entries: $entry_count"
|
|
echo "━━━━━━━━━━━━━━━━━━━━━━━"
|
|
exit 0
|
|
fi
|
|
|
|
# ==============================================================================================
|
|
# ━━━ Log Transfer Mode ━━━
|
|
# ==============================================================================================
|
|
# Called by rsync.sh after each sync — appends one line and trims old entries.
|
|
# Uses "wait" lock — if two rsync jobs finish simultaneously, wait and write in order.
|
|
# Usage: bandwidth_monitor.sh --log-transfer "profile" duration_seconds status [bytes]
|
|
|
|
if [[ "$LOG_TRANSFER_MODE" == true ]]; then
|
|
[[ -z "$TRANSFER_PROFILE" ]] && { error "No profile specified for --log-transfer"; exit 1; }
|
|
|
|
acquire_lock "wait"
|
|
|
|
TODAY=$(date '+%Y-%m-%d')
|
|
NOW=$(date '+%H:%M')
|
|
DURATION_FMT=$(format_duration "$TRANSFER_DURATION")
|
|
|
|
# Check if transfer exceeds warn threshold
|
|
WARN_FLAG=""
|
|
if [[ -n "$TRANSFER_BYTES" && "$TRANSFER_BYTES" -gt 0 ]]; then
|
|
WARN_BYTES=$(awk "BEGIN {printf \"%d\", $BANDWIDTH_WARN_GB * 1073741824}")
|
|
[[ "$TRANSFER_BYTES" -gt "$WARN_BYTES" ]] && WARN_FLAG="LARGE"
|
|
fi
|
|
|
|
# Append entry — format: date|time|profile|duration|status|bytes|warn_flag
|
|
echo "${TODAY}|${NOW}|${TRANSFER_PROFILE}|${TRANSFER_DURATION}|${TRANSFER_STATUS}|${TRANSFER_BYTES}|${WARN_FLAG}" \
|
|
>> "$BANDWIDTH_LOG"
|
|
log "$ICON_BANDWIDTH Logged: $TRANSFER_PROFILE — ${DURATION_FMT} — $TRANSFER_STATUS${WARN_FLAG:+ [$WARN_FLAG]}"
|
|
|
|
# Trim entries older than retention — atomic write via temp file
|
|
CUTOFF=$(date -d "${BANDWIDTH_LOG_RETENTION} days ago" '+%Y-%m-%d')
|
|
TEMP_FILE="${BANDWIDTH_LOG}.tmp"
|
|
awk -F'|' -v cutoff="$CUTOFF" '$1 >= cutoff' "$BANDWIDTH_LOG" > "$TEMP_FILE" && \
|
|
mv "$TEMP_FILE" "$BANDWIDTH_LOG"
|
|
log "$ICON_BANDWIDTH Log trimmed — retaining entries from $CUTOFF"
|
|
exit 0
|
|
fi
|
|
|
|
# ==============================================================================================
|
|
# ━━━ Report Mode ━━━
|
|
# ==============================================================================================
|
|
acquire_lock "wait"
|
|
|
|
echo ""
|
|
echo "━━━ $ICON_BANDWIDTH Bandwidth Report — $(date '+%Y-%m-%d %H:%M:%S') ━━━"
|
|
echo "$ICON_HOST $MY_ID ($LOCAL_SERVER_NAME)"
|
|
echo ""
|
|
|
|
if [[ ! -s "$BANDWIDTH_LOG" ]]; then
|
|
warn "No bandwidth data yet — log is empty"
|
|
warn "Data accumulates as rsync jobs complete via rsync.sh"
|
|
exit 0
|
|
fi
|
|
|
|
START=$(date +%s)
|
|
|
|
# ── Log overview ──────────────────────────────────────────────────────────────────────────────
|
|
OLDEST=$(awk -F'|' 'NR==1{print $1}' "$BANDWIDTH_LOG")
|
|
NEWEST=$(awk -F'|' 'END{print $1}' "$BANDWIDTH_LOG")
|
|
ENTRY_COUNT=$(wc -l < "$BANDWIDTH_LOG")
|
|
SUCCESS_COUNT=$(awk -F'|' '$5=="success"' "$BANDWIDTH_LOG" | wc -l)
|
|
FAILED_COUNT=$(awk -F'|' '$5=="failed"' "$BANDWIDTH_LOG" | wc -l)
|
|
LARGE_COUNT=$(awk -F'|' '$7=="LARGE"' "$BANDWIDTH_LOG" | wc -l)
|
|
|
|
info "Log covers: $OLDEST → $NEWEST ($ENTRY_COUNT runs)"
|
|
echo ""
|
|
|
|
# ── Per-profile breakdown ─────────────────────────────────────────────────────────────────────
|
|
echo "━━━ $ICON_BANDWIDTH Per-Profile Summary ━━━"
|
|
awk -F'|' '{
|
|
runs[$3]++
|
|
duration[$3] += $4
|
|
if ($5 == "failed") fails[$3]++
|
|
if ($7 == "LARGE") large[$3]++
|
|
}
|
|
END {
|
|
for (profile in runs) {
|
|
avg = (runs[profile] > 0) ? duration[profile] / runs[profile] : 0
|
|
mins = int(avg / 60)
|
|
secs = int(avg % 60)
|
|
fail_count = (profile in fails) ? fails[profile] : 0
|
|
large_count = (profile in large) ? large[profile] : 0
|
|
large_str = (large_count > 0) ? " ⚠️ " large_count " large" : ""
|
|
printf " %-22s %3d runs avg %dm%ds failed: %d%s\n", \
|
|
profile, runs[profile], mins, secs, fail_count, large_str
|
|
}
|
|
}' "$BANDWIDTH_LOG" | sort
|
|
echo ""
|
|
|
|
# ── Last 7 days ───────────────────────────────────────────────────────────────────────────────
|
|
echo "━━━ $ICON_BANDWIDTH Last 7 Days ━━━"
|
|
for i in 6 5 4 3 2 1 0; do
|
|
day=$(date -d "$i days ago" '+%Y-%m-%d')
|
|
day_name=$(date -d "$i days ago" '+%a')
|
|
day_runs=$(awk -F'|' -v d="$day" '$1==d' "$BANDWIDTH_LOG" | wc -l)
|
|
day_failed=$(awk -F'|' -v d="$day" '$1==d && $5=="failed"' "$BANDWIDTH_LOG" | wc -l)
|
|
day_large=$(awk -F'|' -v d="$day" '$1==d && $7=="LARGE"' "$BANDWIDTH_LOG" | wc -l)
|
|
day_duration=$(awk -F'|' -v d="$day" '$1==d{sum+=$4} END{print sum+0}' "$BANDWIDTH_LOG")
|
|
day_duration_fmt=$(format_duration "$day_duration")
|
|
|
|
if [[ "$day_runs" -eq 0 ]]; then
|
|
echo " $ICON_TIME $day ($day_name) — no syncs"
|
|
elif [[ "$day_failed" -gt 0 ]]; then
|
|
echo " $ICON_WARN $day ($day_name) — $day_runs runs / ${day_duration_fmt} total / $ICON_ERROR $day_failed failed"
|
|
elif [[ "$day_large" -gt 0 ]]; then
|
|
echo " $ICON_WARN $day ($day_name) — $day_runs runs / ${day_duration_fmt} total / ⚠️ $day_large large"
|
|
else
|
|
echo " $ICON_DONE $day ($day_name) — $day_runs runs / ${day_duration_fmt} total"
|
|
fi
|
|
done
|
|
echo ""
|
|
|
|
# ── Large transfers ───────────────────────────────────────────────────────────────────────────
|
|
if [[ "$LARGE_COUNT" -gt 0 ]]; then
|
|
echo "━━━ $ICON_WARN Large Transfers (>${BANDWIDTH_WARN_GB}GB) ━━━"
|
|
awk -F'|' '$7=="LARGE" {
|
|
bytes=$6+0
|
|
gb=bytes/1073741824
|
|
printf " %s %s %-20s %.1fGB\n", $1, $2, $3, gb
|
|
}' "$BANDWIDTH_LOG" | tail -10
|
|
echo ""
|
|
fi
|
|
|
|
# ── Totals ────────────────────────────────────────────────────────────────────────────────────
|
|
TOTAL_DURATION=$(awk -F'|' '{sum+=$4} END{print sum+0}' "$BANDWIDTH_LOG")
|
|
TOTAL_DURATION_FMT=$(format_duration "$TOTAL_DURATION")
|
|
|
|
END=$(date +%s)
|
|
|
|
echo "━━━━━ $ICON_SUMMARY BANDWIDTH SUMMARY ━━━━━"
|
|
echo "$ICON_HOST Identity: $MY_ID ($LOCAL_SERVER_NAME)"
|
|
echo "$ICON_BANDWIDTH Runs: $ENTRY_COUNT ($SUCCESS_COUNT success / $FAILED_COUNT failed)"
|
|
[[ "$LARGE_COUNT" -gt 0 ]] && \
|
|
warn "Large: $LARGE_COUNT transfer(s) exceeded ${BANDWIDTH_WARN_GB}GB"
|
|
echo "$ICON_TIME Total: $TOTAL_DURATION_FMT"
|
|
echo "$ICON_TIME Period: $OLDEST → $NEWEST"
|
|
echo "$ICON_GEAR Retention: ${BANDWIDTH_LOG_RETENTION} days"
|
|
echo "$ICON_TIME Generated: $(format_duration $(( END - START )))"
|
|
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
|
|
|
|
# Only notify if there are failures or large transfers worth flagging
|
|
if [[ "$FAILED_COUNT" -gt 0 ]]; then
|
|
notify "Bandwidth report on $(hostname) — $FAILED_COUNT failed sync(s) in ${BANDWIDTH_LOG_RETENTION} day window" \
|
|
"Bandwidth Monitor" "warning"
|
|
elif [[ "$LARGE_COUNT" -gt 0 ]]; then
|
|
notify "Bandwidth report on $(hostname) — $LARGE_COUNT large transfer(s) exceeded ${BANDWIDTH_WARN_GB}GB" \
|
|
"Bandwidth Monitor" "normal"
|
|
fi |