3
0
Files
confdroid_prometheus/templates/prune_blocks.erb
2025-12-07 18:03:52 +01:00

49 lines
1.3 KiB
Plaintext

#!/bin/bash
set -euo pipefail
# Load environment
source <%= @ps_env_file %>
LOG_FILE="<%= @ps_prune_log_file %>"
echo "$(date '+%Y-%m-%d %H:%M:%S') Starting prune run" >> "$LOG_FILE"
# -----------------------
# Iterate TSDB blocks
# -----------------------
for block in "$TSDB_DIR"/*/; do
[[ -d "$block" ]] || continue
meta="$block/meta.json"
[[ -f "$meta" ]] || continue
min_time=$(jq -r '.minTime' "$meta")
max_time=$(jq -r '.maxTime' "$meta")
# Skip last MIN_AGE_HOURS
block_age_hours=$(( ($(date +%s) - min_time/1000) / 3600 ))
if (( block_age_hours < MIN_AGE_HOURS + GRACE_HOURS )); then
log "Keeping block $block (age ${block_age_hours}h)"
continue
fi
# Check existence in PostgreSQL
check_sql="${CHECK_SQL_TEMPLATE//\{min\}/$min_time}"
check_sql="${check_sql//\{max\}/$max_time}"
exists=$(psql -h "$PGHOST" -p "$PGPORT" -U "$PGUSER" -d "$PGDATABASE" -t -c "$check_sql" | xargs || echo "0")
if [[ "$exists" != "1" ]]; then
log "Block $block not confirmed in PostgreSQL, skipping deletion"
continue
fi
# Delete block
if [[ "$DRY_RUN" == "true" ]]; then
log "[DRY-RUN] Would delete block $block"
else
log "Deleting block $block"
rm -rf "$block"
fi
done
echo "$(date '+%Y-%m-%d %H:%M:%S') Prune run finished" >> "$LOG_FILE"