fix(recovery): detect all-host reboot recovery gaps
Some checks failed
CD Pipeline / workflow-shape (push) Successful in 0s
CD Pipeline / cancel-stale-cd (push) Has been skipped
CD Pipeline / build-and-deploy (push) Has been cancelled
CD Pipeline / post-deploy-checks (push) Has been cancelled
CD Pipeline / tests (push) Has been cancelled
Some checks failed
CD Pipeline / workflow-shape (push) Successful in 0s
CD Pipeline / cancel-stale-cd (push) Has been skipped
CD Pipeline / build-and-deploy (push) Has been cancelled
CD Pipeline / post-deploy-checks (push) Has been cancelled
CD Pipeline / tests (push) Has been cancelled
This commit is contained in:
@@ -659,6 +659,10 @@ def _base_lines(host: str) -> list[str]:
|
||||
"# TYPE awoooi_backup_health_last_run_timestamp gauge",
|
||||
"# HELP awoooi_backup_expected_job_info Expected backup job inventory.",
|
||||
"# TYPE awoooi_backup_expected_job_info gauge",
|
||||
"# HELP awoooi_backup_coverage_domain_expected_info Expected backup coverage domains for host, DB, website, service, package, tool, and log recovery.",
|
||||
"# TYPE awoooi_backup_coverage_domain_expected_info gauge",
|
||||
"# HELP awoooi_backup_coverage_domain_fresh Whether every required evidence job for a backup coverage domain is fresh.",
|
||||
"# TYPE awoooi_backup_coverage_domain_fresh gauge",
|
||||
"# HELP awoooi_backup_job_configured Whether the expected backup cron/config is present.",
|
||||
"# TYPE awoooi_backup_job_configured gauge",
|
||||
"# HELP awoooi_backup_script_present Whether the backup script exists on this host.",
|
||||
@@ -808,6 +812,7 @@ def _collect_110(host: str) -> list[str]:
|
||||
labels = f'host="{_escape_label(host)}",script="{_escape_label(script)}"'
|
||||
lines.append(f"awoooi_backup_script_present{{{labels}}} {int(Path('/backup/scripts', script).exists())}")
|
||||
|
||||
job_freshness: dict[str, int] = {}
|
||||
for job, repo, max_age in [
|
||||
("awoooi_db", "/backup/awoooi", 7),
|
||||
("configs", "/backup/configs", 48),
|
||||
@@ -824,6 +829,8 @@ def _collect_110(host: str) -> list[str]:
|
||||
("public_routes", "/backup/public-routes", 168),
|
||||
]:
|
||||
timestamp, count = _latest_restic_snapshot(repo)
|
||||
age = int(time.time()) - timestamp if timestamp else 0
|
||||
job_freshness[job] = 1 if timestamp and age <= max_age * 3600 else 0
|
||||
lines.extend(
|
||||
_metric_lines_for_job(
|
||||
host=host,
|
||||
@@ -837,6 +844,26 @@ def _collect_110(host: str) -> list[str]:
|
||||
)
|
||||
)
|
||||
|
||||
coverage_domains = {
|
||||
"host": ["configs"],
|
||||
"database": ["awoooi_db"],
|
||||
"website": ["public_routes", "momo"],
|
||||
"service": ["gitea", "harbor", "sentry", "monitoring", "signoz"],
|
||||
"package": ["configs", "ai_artifacts"],
|
||||
"tool": ["ai_artifacts", "open_webui", "clawbot", "langfuse"],
|
||||
"log": ["monitoring", "signoz", "sentry"],
|
||||
}
|
||||
for domain, jobs in coverage_domains.items():
|
||||
required = ",".join(jobs)
|
||||
fresh = 1 if all(job_freshness.get(job, 0) == 1 for job in jobs) else 0
|
||||
labels = (
|
||||
f'host="{_escape_label(host)}",'
|
||||
f'domain="{_escape_label(domain)}",'
|
||||
f'required_jobs="{_escape_label(required)}"'
|
||||
)
|
||||
lines.append(f"awoooi_backup_coverage_domain_expected_info{{{labels}}} 1")
|
||||
lines.append(f"awoooi_backup_coverage_domain_fresh{{{labels}}} {fresh}")
|
||||
|
||||
backup_all_ts, failed_count = _latest_backup_all_failed_count()
|
||||
labels = f'host="{_escape_label(host)}",job="backup_all"'
|
||||
lines.append(f"awoooi_backup_last_run_failed_count{{{labels}}} {failed_count}")
|
||||
|
||||
@@ -7,11 +7,21 @@
|
||||
|
||||
set -uo pipefail
|
||||
|
||||
SSH_OPTS=(-o BatchMode=yes -o ConnectTimeout="${SSH_CONNECT_TIMEOUT_SECONDS:-6}")
|
||||
SSH_OPTS=(
|
||||
-o BatchMode=yes
|
||||
-o ConnectTimeout="${SSH_CONNECT_TIMEOUT_SECONDS:-6}"
|
||||
-o ConnectionAttempts=1
|
||||
-o ServerAliveInterval="${SSH_SERVER_ALIVE_INTERVAL_SECONDS:-5}"
|
||||
-o ServerAliveCountMax="${SSH_SERVER_ALIVE_COUNT_MAX:-1}"
|
||||
)
|
||||
SSH_COMMAND_TIMEOUT_SECONDS="${SSH_COMMAND_TIMEOUT_SECONDS:-15}"
|
||||
NODE_EXPORTER_PORT="${NODE_EXPORTER_PORT:-9100}"
|
||||
NODE_EXPORTER_TIMEOUT_SECONDS="${NODE_EXPORTER_TIMEOUT_SECONDS:-4}"
|
||||
HOST_SPECS=(
|
||||
"99=192.168.0.99:vmware-host-autostart"
|
||||
"110=wooo@192.168.0.110:awoooi-startup-110.service"
|
||||
"111=192.168.0.111:vm-host-boot"
|
||||
"112=192.168.0.112:vm-host-boot"
|
||||
"120=wooo@192.168.0.120:k3s.service"
|
||||
"121=wooo@192.168.0.121:k3s.service"
|
||||
"188=ollama@192.168.0.188:awoooi-startup.service"
|
||||
@@ -37,6 +47,42 @@ is_local_target() {
|
||||
grep -Fxq "$target_host" <<<"$ips"
|
||||
}
|
||||
|
||||
run_with_timeout() {
|
||||
local timeout_seconds="$1"
|
||||
shift
|
||||
if command -v timeout >/dev/null 2>&1; then
|
||||
timeout "$timeout_seconds" "$@"
|
||||
return $?
|
||||
fi
|
||||
if command -v python3 >/dev/null 2>&1; then
|
||||
python3 - "$timeout_seconds" "$@" <<'PY'
|
||||
import subprocess
|
||||
import sys
|
||||
|
||||
timeout_seconds = float(sys.argv[1])
|
||||
command = sys.argv[2:]
|
||||
try:
|
||||
result = subprocess.run(
|
||||
command,
|
||||
capture_output=True,
|
||||
check=False,
|
||||
text=True,
|
||||
timeout=timeout_seconds,
|
||||
)
|
||||
except subprocess.TimeoutExpired as exc:
|
||||
if isinstance(exc.stdout, str):
|
||||
print(exc.stdout, end="")
|
||||
sys.exit(124)
|
||||
print(result.stdout, end="")
|
||||
if result.stderr:
|
||||
print(result.stderr, end="", file=sys.stderr)
|
||||
sys.exit(result.returncode)
|
||||
PY
|
||||
return $?
|
||||
fi
|
||||
"$@"
|
||||
}
|
||||
|
||||
emit_boot_row() {
|
||||
local alias="$1"
|
||||
local target="$2"
|
||||
@@ -94,19 +140,54 @@ probe_node_exporter() {
|
||||
emit_boot_row "$alias" "$target" "$unit" 1 "node_exporter_${boot_time}" "$uptime_seconds" "node_exporter" "unknown" "unknown"
|
||||
}
|
||||
|
||||
probe_reachable_only() {
|
||||
local alias="$1"
|
||||
local target="$2"
|
||||
local unit="$3"
|
||||
local target_host="${target##*@}"
|
||||
|
||||
if run_with_timeout "${PING_TIMEOUT_SECONDS:-2}" ping -c 1 "$target_host" >/dev/null 2>&1; then
|
||||
emit_boot_row "$alias" "$target" "$unit" 1 "reachable_unknown_boot" "unknown" "ping_reachable" "unknown" "unknown"
|
||||
return 0
|
||||
fi
|
||||
|
||||
if command -v nc >/dev/null 2>&1; then
|
||||
for port in ${BOOT_PROBE_TCP_PORTS:-22 80 443 3389 5985 9100}; do
|
||||
if nc -z -w "${TCP_CONNECT_TIMEOUT_SECONDS:-2}" "$target_host" "$port" >/dev/null 2>&1; then
|
||||
emit_boot_row "$alias" "$target" "$unit" 1 "reachable_unknown_boot" "unknown" "tcp_${port}_reachable" "unknown" "unknown"
|
||||
return 0
|
||||
fi
|
||||
done
|
||||
fi
|
||||
|
||||
return 1
|
||||
}
|
||||
|
||||
probe_host() {
|
||||
local alias="$1"
|
||||
local target="$2"
|
||||
local unit="$3"
|
||||
local output boot_id uptime_seconds systemd_state enabled active
|
||||
local target_host="${target##*@}"
|
||||
local remote_script
|
||||
|
||||
if is_local_target "$target_host"; then
|
||||
probe_local_host "$alias" "$target" "$unit"
|
||||
return 0
|
||||
fi
|
||||
|
||||
output="$(ssh "${SSH_OPTS[@]}" "$target" "unit='$unit'; \
|
||||
if [[ "$target" != *@* ]]; then
|
||||
if probe_node_exporter "$alias" "$target" "$unit"; then
|
||||
return 0
|
||||
fi
|
||||
if probe_reachable_only "$alias" "$target" "$unit"; then
|
||||
return 0
|
||||
fi
|
||||
emit_boot_row "$alias" "$target" "$unit" 0 "unknown" "unknown" "unknown" "unknown" "unknown"
|
||||
return 0
|
||||
fi
|
||||
|
||||
remote_script="unit='$unit'; \
|
||||
boot_id=\$(cat /proc/sys/kernel/random/boot_id 2>/dev/null || echo unknown); \
|
||||
uptime_seconds=\$(awk '{print int(\$1)}' /proc/uptime 2>/dev/null || echo unknown); \
|
||||
systemd_state=\$(systemctl is-system-running 2>/dev/null || true); \
|
||||
@@ -115,11 +196,15 @@ probe_host() {
|
||||
enabled=\${enabled:-unknown}; \
|
||||
active=\${active:-unknown}; \
|
||||
printf 'boot_id=%s uptime_seconds=%s systemd_state=%s startup_enabled=%s startup_active=%s\n' \"\$boot_id\" \"\$uptime_seconds\" \"\$systemd_state\" \"\$enabled\" \"\$active\" \
|
||||
" 2>/dev/null)"
|
||||
"
|
||||
output="$(run_with_timeout "$SSH_COMMAND_TIMEOUT_SECONDS" ssh "${SSH_OPTS[@]}" "$target" "$remote_script" 2>/dev/null)"
|
||||
if [[ $? -ne 0 || -z "$output" ]]; then
|
||||
if probe_node_exporter "$alias" "$target" "$unit"; then
|
||||
return 0
|
||||
fi
|
||||
if probe_reachable_only "$alias" "$target" "$unit"; then
|
||||
return 0
|
||||
fi
|
||||
emit_boot_row "$alias" "$target" "$unit" 0 "unknown" "unknown" "unknown" "unknown" "unknown"
|
||||
return 0
|
||||
fi
|
||||
@@ -134,7 +219,7 @@ probe_host() {
|
||||
}
|
||||
|
||||
echo "AWOOOI_REBOOT_AUTO_RECOVERY_HOST_PROBE=1"
|
||||
echo "TARGET_HOSTS=110,120,121,188"
|
||||
echo "TARGET_HOSTS=99,110,111,112,120,121,188"
|
||||
echo "GENERATED_AT=$(date '+%Y-%m-%dT%H:%M:%S%z')"
|
||||
|
||||
for spec in "${HOST_SPECS[@]}"; do
|
||||
|
||||
@@ -29,12 +29,21 @@ artifact_dir="$LOG_DIR/reboot-auto-recovery-slo-$run_id"
|
||||
mkdir -p "$artifact_dir"
|
||||
|
||||
host_probe="$artifact_dir/host-probe.txt"
|
||||
reboot_event_file="$artifact_dir/reboot-event.json"
|
||||
reboot_event_prom="$artifact_dir/reboot-event.prom"
|
||||
summary_file="$artifact_dir/summary.txt"
|
||||
scorecard_file="$artifact_dir/scorecard.json"
|
||||
stock_freshness_file="$artifact_dir/stock-freshness.json"
|
||||
stock_ingestion_file="$artifact_dir/stock-ingestion.json"
|
||||
reboot_event_state_file="${REBOOT_EVENT_STATE_FILE:-${LOG_DIR}/reboot-event-state.json}"
|
||||
|
||||
bash "$ROOT_DIR/scripts/reboot-recovery/reboot-auto-recovery-host-probe.sh" >"$host_probe" 2>&1 || true
|
||||
python3 "$ROOT_DIR/scripts/reboot-recovery/reboot-event-detector.py" \
|
||||
--host-probe-file "$host_probe" \
|
||||
--state-file "$reboot_event_state_file" \
|
||||
--target-minutes "$TARGET_MINUTES" \
|
||||
--output "$reboot_event_file" \
|
||||
--prometheus-output "$reboot_event_prom" || true
|
||||
ARTIFACT_DIR="$artifact_dir/post-reboot-readiness" \
|
||||
bash "$ROOT_DIR/scripts/reboot-recovery/post-reboot-readiness-summary.sh" --no-color >"$summary_file" 2>&1 || true
|
||||
|
||||
@@ -51,6 +60,7 @@ scorecard_args=(
|
||||
"$ROOT_DIR/scripts/reboot-recovery/reboot-auto-recovery-slo-scorecard.py"
|
||||
--summary-file "$summary_file" \
|
||||
--host-probe-file "$host_probe" \
|
||||
--reboot-event-file "$reboot_event_file" \
|
||||
--target-minutes "$TARGET_MINUTES" \
|
||||
--min-free-gib "$MIN_FREE_GIB" \
|
||||
--disk-path / \
|
||||
@@ -84,21 +94,33 @@ payload=json.load(open(sys.argv[1], encoding="utf-8"))
|
||||
print(payload.get("host_boot_detection", {}).get("max_observed_uptime_seconds", 0))
|
||||
PY
|
||||
)"
|
||||
target_seconds_remaining="$(python3 - "$scorecard_file" <<'PY'
|
||||
import json, sys
|
||||
payload=json.load(open(sys.argv[1], encoding="utf-8"))
|
||||
print(payload.get("sla_recovery_eta", {}).get("target_seconds_remaining", 0) or 0)
|
||||
PY
|
||||
)"
|
||||
|
||||
tmp_metric="$(mktemp "$TEXTFILE_DIR/.reboot_auto_recovery_slo.XXXXXX")"
|
||||
cat >"$tmp_metric" <<METRICS
|
||||
# HELP awoooi_reboot_auto_recovery_slo_ready Whether all P0 hosts auto-recovered within the target window.
|
||||
# TYPE awoooi_reboot_auto_recovery_slo_ready gauge
|
||||
awoooi_reboot_auto_recovery_slo_ready{scope="110_120_121_188",target_minutes="$TARGET_MINUTES"} $ready
|
||||
awoooi_reboot_auto_recovery_slo_ready{scope="99_110_111_112_120_121_188",target_minutes="$TARGET_MINUTES"} $ready
|
||||
# HELP awoooi_reboot_auto_recovery_slo_blocker_count Number of active fail-closed blockers for the reboot recovery SLO.
|
||||
# TYPE awoooi_reboot_auto_recovery_slo_blocker_count gauge
|
||||
awoooi_reboot_auto_recovery_slo_blocker_count{scope="110_120_121_188",target_minutes="$TARGET_MINUTES"} $blocker_count
|
||||
awoooi_reboot_auto_recovery_slo_blocker_count{scope="99_110_111_112_120_121_188",target_minutes="$TARGET_MINUTES"} $blocker_count
|
||||
# HELP awoooi_reboot_auto_recovery_slo_max_host_uptime_seconds Max observed host uptime in the boot probe.
|
||||
# TYPE awoooi_reboot_auto_recovery_slo_max_host_uptime_seconds gauge
|
||||
awoooi_reboot_auto_recovery_slo_max_host_uptime_seconds{scope="110_120_121_188"} $max_uptime
|
||||
awoooi_reboot_auto_recovery_slo_max_host_uptime_seconds{scope="99_110_111_112_120_121_188"} $max_uptime
|
||||
# HELP awoooi_reboot_auto_recovery_slo_target_seconds_remaining Seconds remaining before the 10-minute target expires.
|
||||
# TYPE awoooi_reboot_auto_recovery_slo_target_seconds_remaining gauge
|
||||
awoooi_reboot_auto_recovery_slo_target_seconds_remaining{scope="99_110_111_112_120_121_188",target_minutes="$TARGET_MINUTES"} $target_seconds_remaining
|
||||
# HELP awoooi_reboot_auto_recovery_slo_last_run_timestamp Last SLO exporter run timestamp.
|
||||
# TYPE awoooi_reboot_auto_recovery_slo_last_run_timestamp gauge
|
||||
awoooi_reboot_auto_recovery_slo_last_run_timestamp{scope="110_120_121_188"} $now
|
||||
awoooi_reboot_auto_recovery_slo_last_run_timestamp{scope="99_110_111_112_120_121_188"} $now
|
||||
METRICS
|
||||
if [ -s "$reboot_event_prom" ]; then
|
||||
cat "$reboot_event_prom" >>"$tmp_metric"
|
||||
fi
|
||||
chmod 0644 "$tmp_metric"
|
||||
mv "$tmp_metric" "$TEXTFILE_DIR/$OUTPUT_NAME"
|
||||
|
||||
@@ -15,7 +15,7 @@ from typing import Any
|
||||
|
||||
ROOT = Path(__file__).resolve().parents[2]
|
||||
SCHEMA_VERSION = "awoooi_reboot_auto_recovery_slo_scorecard_v1"
|
||||
REQUIRED_HOSTS = {"110", "120", "121", "188"}
|
||||
REQUIRED_HOSTS = {"99", "110", "111", "112", "120", "121", "188"}
|
||||
|
||||
|
||||
def parse_args() -> argparse.Namespace:
|
||||
@@ -24,6 +24,7 @@ def parse_args() -> argparse.Namespace:
|
||||
)
|
||||
parser.add_argument("--summary-file", type=Path, help="post-reboot-readiness-summary output.")
|
||||
parser.add_argument("--host-probe-file", type=Path, help="reboot-auto-recovery-host-probe output.")
|
||||
parser.add_argument("--reboot-event-file", type=Path, help="reboot-event-detector JSON output.")
|
||||
parser.add_argument("--target-minutes", type=int, default=10)
|
||||
parser.add_argument("--min-free-gib", type=float, default=2.0)
|
||||
parser.add_argument("--disk-path", type=Path, help="Optionally check local free space.")
|
||||
@@ -38,6 +39,12 @@ def parse_args() -> argparse.Namespace:
|
||||
help="Optional StockPlatform /api/v1/system/ingestion JSON readback.",
|
||||
)
|
||||
parser.add_argument("--generated-at", help="Override generated_at for stable snapshots.")
|
||||
parser.add_argument(
|
||||
"--required-host",
|
||||
action="append",
|
||||
dest="required_hosts",
|
||||
help="Required host alias. May be passed more than once.",
|
||||
)
|
||||
parser.add_argument("--output", type=Path, help="Write JSON to this path.")
|
||||
return parser.parse_args()
|
||||
|
||||
@@ -141,6 +148,9 @@ def source_controls() -> dict[str, bool]:
|
||||
"host_boot_probe_source_present": source_file(
|
||||
"scripts/reboot-recovery/reboot-auto-recovery-host-probe.sh"
|
||||
).exists(),
|
||||
"reboot_event_detector_source_present": source_file(
|
||||
"scripts/reboot-recovery/reboot-event-detector.py"
|
||||
).exists(),
|
||||
"slo_exporter_source_present": source_file(
|
||||
"scripts/reboot-recovery/reboot-auto-recovery-slo-exporter.sh"
|
||||
).exists(),
|
||||
@@ -166,6 +176,24 @@ def source_controls() -> dict[str, bool]:
|
||||
"cold_start_textfile_exporter_source_present": source_file(
|
||||
"scripts/reboot-recovery/cold-start-textfile-exporter.sh"
|
||||
).exists(),
|
||||
"windows_99_vmware_autostart_source_present": source_file(
|
||||
"scripts/reboot-recovery/windows99-vmware-autostart.ps1"
|
||||
).exists(),
|
||||
"public_maintenance_fallback_source_present": source_file(
|
||||
"ops/maintenance/maintenance.html"
|
||||
).exists()
|
||||
and file_contains(
|
||||
source_file("docs/runbooks/PUBLIC-MAINTENANCE-FALLBACK-RUNBOOK.md"),
|
||||
"L0",
|
||||
"L1",
|
||||
"502",
|
||||
),
|
||||
"telegram_reboot_backup_alert_rules_source_present": file_contains(
|
||||
source_file("ops/monitoring/alerts-unified.yml"),
|
||||
"HostRebootEventDetected",
|
||||
"RebootAutoRecoverySLOMissed",
|
||||
"BackupCoverageDomainStale",
|
||||
),
|
||||
}
|
||||
|
||||
|
||||
@@ -402,8 +430,10 @@ def choose_safe_next_step(
|
||||
def build_scorecard(args: argparse.Namespace) -> dict[str, Any]:
|
||||
target_seconds = args.target_minutes * 60
|
||||
generated_at = args.generated_at or datetime.now().astimezone().isoformat(timespec="seconds")
|
||||
required_hosts = set(args.required_hosts or REQUIRED_HOSTS)
|
||||
summary = parse_kv(read_text(args.summary_file))
|
||||
host_rows = parse_host_probe(read_text(args.host_probe_file))
|
||||
reboot_event = read_json_object(args.reboot_event_file)
|
||||
stockplatform = build_stockplatform_readback(
|
||||
summary=summary,
|
||||
freshness=read_json_object(args.stock_freshness_file),
|
||||
@@ -419,7 +449,7 @@ def build_scorecard(args: argparse.Namespace) -> dict[str, Any]:
|
||||
blockers.append(key.replace("_present", "_missing"))
|
||||
|
||||
host_aliases = {str(row.get("alias", "")) for row in host_rows}
|
||||
missing_hosts = sorted(REQUIRED_HOSTS - host_aliases)
|
||||
missing_hosts = sorted(required_hosts - host_aliases)
|
||||
unreachable_hosts = sorted(str(row.get("alias")) for row in host_rows if not row.get("reachable"))
|
||||
stale_hosts = sorted(
|
||||
str(row.get("alias"))
|
||||
@@ -442,6 +472,27 @@ def build_scorecard(args: argparse.Namespace) -> dict[str, Any]:
|
||||
if unknown_uptime_hosts:
|
||||
blockers.append("host_uptime_unknown")
|
||||
|
||||
reboot_event_missing_hosts = strings(reboot_event.get("missing_hosts"))
|
||||
reboot_event_unreachable_hosts = strings(reboot_event.get("unreachable_hosts"))
|
||||
reboot_event_fresh_hosts = strings(reboot_event.get("fresh_boot_hosts"))
|
||||
reboot_event_detected = reboot_event.get("reboot_detected") is True
|
||||
all_required_in_reboot_window = reboot_event.get("all_required_hosts_in_reboot_window") is True
|
||||
target_seconds_remaining = int_value(reboot_event.get("target_seconds_remaining"), 0)
|
||||
reboot_deadline_status = str(
|
||||
reboot_event.get("recovery_deadline_status") or "event_readback_missing"
|
||||
)
|
||||
if not reboot_event:
|
||||
blockers.append("stateful_reboot_event_detection_missing")
|
||||
else:
|
||||
if reboot_event_missing_hosts:
|
||||
blockers.append("reboot_event_missing_required_hosts")
|
||||
if reboot_event_unreachable_hosts:
|
||||
blockers.append("reboot_event_required_host_unreachable")
|
||||
if not reboot_event_detected:
|
||||
blockers.append("fresh_all_host_reboot_event_missing")
|
||||
if not all_required_in_reboot_window:
|
||||
blockers.append("all_required_hosts_not_in_10_minute_reboot_window")
|
||||
|
||||
service_green = truthy(summary.get("SERVICE_GREEN"))
|
||||
product_data_green = truthy(summary.get("PRODUCT_DATA_GREEN"))
|
||||
backup_core_green = truthy(summary.get("BACKUP_CORE_GREEN"))
|
||||
@@ -488,7 +539,7 @@ def build_scorecard(args: argparse.Namespace) -> dict[str, Any]:
|
||||
"can_claim_all_services_recovered_within_target": can_claim,
|
||||
"source_controls": controls,
|
||||
"host_boot_detection": {
|
||||
"required_hosts": sorted(REQUIRED_HOSTS),
|
||||
"required_hosts": sorted(required_hosts),
|
||||
"observed_hosts": sorted(host_aliases),
|
||||
"missing_hosts": missing_hosts,
|
||||
"unreachable_hosts": unreachable_hosts,
|
||||
@@ -497,6 +548,43 @@ def build_scorecard(args: argparse.Namespace) -> dict[str, Any]:
|
||||
"max_observed_uptime_seconds": max_uptime,
|
||||
"host_rows": host_rows,
|
||||
},
|
||||
"reboot_event_detection": {
|
||||
"readback_present": bool(reboot_event),
|
||||
"reboot_detected": reboot_event_detected,
|
||||
"rebooted_hosts": strings(reboot_event.get("rebooted_hosts")),
|
||||
"fresh_boot_hosts": reboot_event_fresh_hosts,
|
||||
"changed_boot_id_hosts": strings(reboot_event.get("changed_boot_id_hosts")),
|
||||
"missing_hosts": reboot_event_missing_hosts,
|
||||
"unreachable_hosts": reboot_event_unreachable_hosts,
|
||||
"all_required_hosts_observed": reboot_event.get("all_required_hosts_observed") is True,
|
||||
"all_required_hosts_in_reboot_window": all_required_in_reboot_window,
|
||||
"state_file": reboot_event.get("state_file") if reboot_event else None,
|
||||
"state_written": reboot_event.get("state_written") if reboot_event else False,
|
||||
},
|
||||
"sla_recovery_eta": {
|
||||
"target_minutes": args.target_minutes,
|
||||
"target_seconds_remaining": target_seconds_remaining,
|
||||
"deadline_status": reboot_deadline_status,
|
||||
"estimated_recovery_status": (
|
||||
"inside_10_minute_window"
|
||||
if target_seconds_remaining > 0 and not unique_blockers
|
||||
else (
|
||||
"inside_10_minute_window_with_blockers"
|
||||
if target_seconds_remaining > 0
|
||||
else "target_window_elapsed_or_not_detected"
|
||||
)
|
||||
),
|
||||
"next_update_due_seconds": 60,
|
||||
"fixed_triage_order": [
|
||||
"99_vmware_autostart_and_vm_power",
|
||||
"host_boot_event_and_node_exporter",
|
||||
"public_routes_and_maintenance_fallback",
|
||||
"awoooi_k3s_workloads_and_registry",
|
||||
"stockplatform_public_api_and_freshness",
|
||||
"backup_health_and_offsite_evidence",
|
||||
"telegram_alert_delivery_readback",
|
||||
],
|
||||
},
|
||||
"post_reboot_readiness": {
|
||||
"summary_present": bool(summary),
|
||||
"post_start_result": summary.get("POST_START_RESULT", "unknown"),
|
||||
|
||||
236
scripts/reboot-recovery/reboot-event-detector.py
Normal file
236
scripts/reboot-recovery/reboot-event-detector.py
Normal file
@@ -0,0 +1,236 @@
|
||||
#!/usr/bin/env python3
|
||||
"""Stateful all-host reboot event detector for AWOOOI recovery automation."""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import argparse
|
||||
import json
|
||||
from datetime import datetime, timedelta
|
||||
from pathlib import Path
|
||||
from typing import Any
|
||||
|
||||
|
||||
SCHEMA_VERSION = "awoooi_reboot_event_detector_v1"
|
||||
REQUIRED_HOSTS = ("99", "110", "111", "112", "120", "121", "188")
|
||||
|
||||
|
||||
def parse_args() -> argparse.Namespace:
|
||||
parser = argparse.ArgumentParser(
|
||||
description="Detect host reboot events from reboot-auto-recovery-host-probe output.",
|
||||
)
|
||||
parser.add_argument("--host-probe-file", type=Path, required=True)
|
||||
parser.add_argument("--state-file", type=Path, required=True)
|
||||
parser.add_argument("--target-minutes", type=int, default=10)
|
||||
parser.add_argument("--generated-at")
|
||||
parser.add_argument("--output", type=Path)
|
||||
parser.add_argument("--prometheus-output", type=Path)
|
||||
parser.add_argument(
|
||||
"--required-host",
|
||||
action="append",
|
||||
dest="required_hosts",
|
||||
help="Required host alias. May be passed more than once.",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--no-write-state",
|
||||
action="store_true",
|
||||
help="Evaluate without updating the state file.",
|
||||
)
|
||||
return parser.parse_args()
|
||||
|
||||
|
||||
def int_value(value: Any, default: int = -1) -> int:
|
||||
try:
|
||||
return int(str(value))
|
||||
except (TypeError, ValueError):
|
||||
return default
|
||||
|
||||
|
||||
def parse_host_probe(text: str) -> list[dict[str, Any]]:
|
||||
rows: list[dict[str, Any]] = []
|
||||
for raw_line in text.splitlines():
|
||||
line = raw_line.strip()
|
||||
if not line.startswith("HOST_BOOT "):
|
||||
continue
|
||||
row: dict[str, Any] = {}
|
||||
for token in line.split()[1:]:
|
||||
if "=" not in token:
|
||||
continue
|
||||
key, value = token.split("=", 1)
|
||||
row[key] = value
|
||||
row["alias"] = str(row.get("alias", ""))
|
||||
row["reachable"] = row.get("reachable") == "1"
|
||||
row["uptime_seconds"] = int_value(row.get("uptime_seconds"))
|
||||
rows.append(row)
|
||||
return rows
|
||||
|
||||
|
||||
def load_state(path: Path) -> dict[str, Any]:
|
||||
try:
|
||||
payload = json.loads(path.read_text(encoding="utf-8"))
|
||||
except (FileNotFoundError, json.JSONDecodeError):
|
||||
return {"hosts": {}}
|
||||
return payload if isinstance(payload, dict) else {"hosts": {}}
|
||||
|
||||
|
||||
def host_state(rows: list[dict[str, Any]]) -> dict[str, dict[str, Any]]:
|
||||
state: dict[str, dict[str, Any]] = {}
|
||||
for row in rows:
|
||||
alias = str(row.get("alias") or "")
|
||||
if not alias:
|
||||
continue
|
||||
state[alias] = {
|
||||
"boot_id": str(row.get("boot_id") or "unknown"),
|
||||
"uptime_seconds": int_value(row.get("uptime_seconds")),
|
||||
"reachable": bool(row.get("reachable")),
|
||||
"systemd_state": str(row.get("systemd_state") or "unknown"),
|
||||
"startup_unit": str(row.get("startup_unit") or "unknown"),
|
||||
"startup_active": str(row.get("startup_active") or "unknown"),
|
||||
}
|
||||
return state
|
||||
|
||||
|
||||
def build_payload(args: argparse.Namespace) -> dict[str, Any]:
|
||||
generated_at = args.generated_at or datetime.now().astimezone().isoformat(timespec="seconds")
|
||||
observed_at = datetime.fromisoformat(generated_at)
|
||||
target_seconds = args.target_minutes * 60
|
||||
required_hosts = tuple(args.required_hosts or REQUIRED_HOSTS)
|
||||
rows = parse_host_probe(args.host_probe_file.read_text(encoding="utf-8"))
|
||||
current_hosts = host_state(rows)
|
||||
previous = load_state(args.state_file)
|
||||
previous_hosts = previous.get("hosts") if isinstance(previous.get("hosts"), dict) else {}
|
||||
|
||||
rebooted_hosts: list[str] = []
|
||||
fresh_boot_hosts: list[str] = []
|
||||
changed_boot_id_hosts: list[str] = []
|
||||
unreachable_hosts: list[str] = []
|
||||
missing_hosts = sorted(set(required_hosts) - set(current_hosts))
|
||||
|
||||
events: list[dict[str, Any]] = []
|
||||
for alias in required_hosts:
|
||||
current = current_hosts.get(alias)
|
||||
previous_host = previous_hosts.get(alias) if isinstance(previous_hosts, dict) else None
|
||||
if not current:
|
||||
continue
|
||||
if not current["reachable"]:
|
||||
unreachable_hosts.append(alias)
|
||||
previous_boot_id = (
|
||||
str(previous_host.get("boot_id"))
|
||||
if isinstance(previous_host, dict) and previous_host.get("boot_id")
|
||||
else ""
|
||||
)
|
||||
current_boot_id = str(current.get("boot_id") or "")
|
||||
boot_id_changed = bool(
|
||||
previous_boot_id
|
||||
and previous_boot_id != "unknown"
|
||||
and current_boot_id
|
||||
and current_boot_id != "unknown"
|
||||
and previous_boot_id != current_boot_id
|
||||
)
|
||||
fresh_boot = bool(current.get("reachable") and int_value(current.get("uptime_seconds")) <= target_seconds)
|
||||
if boot_id_changed:
|
||||
changed_boot_id_hosts.append(alias)
|
||||
if fresh_boot:
|
||||
fresh_boot_hosts.append(alias)
|
||||
if boot_id_changed or fresh_boot:
|
||||
rebooted_hosts.append(alias)
|
||||
events.append(
|
||||
{
|
||||
"host": alias,
|
||||
"event": "boot_id_changed" if boot_id_changed else "fresh_boot_window",
|
||||
"previous_boot_id": previous_boot_id or "unknown",
|
||||
"current_boot_id": current_boot_id or "unknown",
|
||||
"uptime_seconds": current.get("uptime_seconds"),
|
||||
"deadline_at": (
|
||||
observed_at
|
||||
+ timedelta(seconds=max(0, target_seconds - int_value(current.get("uptime_seconds"), 0)))
|
||||
).isoformat(timespec="seconds"),
|
||||
}
|
||||
)
|
||||
|
||||
max_uptime = max(
|
||||
[int_value(row.get("uptime_seconds"), 0) for row in rows if row.get("reachable")] or [0]
|
||||
)
|
||||
remaining_seconds = max(0, target_seconds - max_uptime)
|
||||
reboot_detected = bool(rebooted_hosts)
|
||||
all_required_observed = not missing_hosts and not unreachable_hosts
|
||||
all_required_in_reboot_window = set(required_hosts).issubset(set(fresh_boot_hosts))
|
||||
|
||||
next_state = {
|
||||
"schema_version": SCHEMA_VERSION,
|
||||
"updated_at": generated_at,
|
||||
"target_seconds": target_seconds,
|
||||
"hosts": current_hosts,
|
||||
}
|
||||
if not args.no_write_state:
|
||||
args.state_file.parent.mkdir(parents=True, exist_ok=True)
|
||||
tmp = args.state_file.with_suffix(args.state_file.suffix + ".tmp")
|
||||
tmp.write_text(json.dumps(next_state, ensure_ascii=False, indent=2, sort_keys=True) + "\n", encoding="utf-8")
|
||||
tmp.replace(args.state_file)
|
||||
|
||||
return {
|
||||
"schema_version": SCHEMA_VERSION,
|
||||
"generated_at": generated_at,
|
||||
"target_minutes": args.target_minutes,
|
||||
"target_seconds": target_seconds,
|
||||
"required_hosts": list(required_hosts),
|
||||
"observed_hosts": sorted(current_hosts),
|
||||
"missing_hosts": missing_hosts,
|
||||
"unreachable_hosts": sorted(unreachable_hosts),
|
||||
"reboot_detected": reboot_detected,
|
||||
"rebooted_hosts": sorted(set(rebooted_hosts)),
|
||||
"fresh_boot_hosts": sorted(set(fresh_boot_hosts)),
|
||||
"changed_boot_id_hosts": sorted(set(changed_boot_id_hosts)),
|
||||
"all_required_hosts_observed": all_required_observed,
|
||||
"all_required_hosts_in_reboot_window": all_required_in_reboot_window,
|
||||
"max_observed_uptime_seconds": max_uptime,
|
||||
"target_seconds_remaining": remaining_seconds,
|
||||
"recovery_deadline_status": "within_target_window" if remaining_seconds > 0 else "target_window_elapsed",
|
||||
"events": events,
|
||||
"state_file": str(args.state_file),
|
||||
"state_written": not args.no_write_state,
|
||||
}
|
||||
|
||||
|
||||
def write_prometheus(path: Path, payload: dict[str, Any]) -> None:
|
||||
path.parent.mkdir(parents=True, exist_ok=True)
|
||||
lines = [
|
||||
"# HELP awoooi_reboot_event_detected Whether a reboot event was detected by boot_id or fresh uptime.",
|
||||
"# TYPE awoooi_reboot_event_detected gauge",
|
||||
f'awoooi_reboot_event_detected{{scope="99_110_111_112_120_121_188"}} {1 if payload["reboot_detected"] else 0}',
|
||||
"# HELP awoooi_reboot_event_required_host_observed Whether each required host was observed.",
|
||||
"# TYPE awoooi_reboot_event_required_host_observed gauge",
|
||||
]
|
||||
observed = set(payload.get("observed_hosts") or [])
|
||||
rebooted = set(payload.get("rebooted_hosts") or [])
|
||||
for host in payload.get("required_hosts") or []:
|
||||
lines.append(f'awoooi_reboot_event_required_host_observed{{host="{host}"}} {1 if host in observed else 0}')
|
||||
lines.append(f'awoooi_reboot_event_host_rebooted{{host="{host}"}} {1 if host in rebooted else 0}')
|
||||
lines.extend(
|
||||
[
|
||||
"# HELP awoooi_reboot_event_target_seconds_remaining Seconds remaining in the reboot recovery target window.",
|
||||
"# TYPE awoooi_reboot_event_target_seconds_remaining gauge",
|
||||
f'awoooi_reboot_event_target_seconds_remaining{{scope="99_110_111_112_120_121_188"}} {payload["target_seconds_remaining"]}',
|
||||
"# HELP awoooi_reboot_event_max_observed_uptime_seconds Maximum uptime observed across reachable hosts.",
|
||||
"# TYPE awoooi_reboot_event_max_observed_uptime_seconds gauge",
|
||||
f'awoooi_reboot_event_max_observed_uptime_seconds{{scope="99_110_111_112_120_121_188"}} {payload["max_observed_uptime_seconds"]}',
|
||||
]
|
||||
)
|
||||
path.write_text("\n".join(lines) + "\n", encoding="utf-8")
|
||||
|
||||
|
||||
def main() -> int:
|
||||
args = parse_args()
|
||||
payload = build_payload(args)
|
||||
text = json.dumps(payload, ensure_ascii=False, indent=2, sort_keys=True) + "\n"
|
||||
if args.output:
|
||||
args.output.parent.mkdir(parents=True, exist_ok=True)
|
||||
args.output.write_text(text, encoding="utf-8")
|
||||
else:
|
||||
print(text, end="")
|
||||
if args.prometheus_output:
|
||||
write_prometheus(args.prometheus_output, payload)
|
||||
return 0
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
raise SystemExit(main())
|
||||
@@ -26,19 +26,47 @@ NEXT_REQUIRED_GATES=none
|
||||
|
||||
HOST_PROBE_GREEN = """\
|
||||
AWOOOI_REBOOT_AUTO_RECOVERY_HOST_PROBE=1
|
||||
TARGET_HOSTS=110,120,121,188
|
||||
TARGET_HOSTS=99,110,111,112,120,121,188
|
||||
HOST_BOOT alias=99 target=192.168.0.99 startup_unit=vmware-host-autostart reachable=1 boot_id=aa uptime_seconds=100 systemd_state=windows_exporter startup_enabled=enabled startup_active=active
|
||||
HOST_BOOT alias=110 target=wooo@192.168.0.110 startup_unit=awoooi-startup-110.service reachable=1 boot_id=a uptime_seconds=120 systemd_state=running startup_enabled=enabled startup_active=active
|
||||
HOST_BOOT alias=111 target=192.168.0.111 startup_unit=vm-host-boot reachable=1 boot_id=bb uptime_seconds=125 systemd_state=node_exporter startup_enabled=unknown startup_active=unknown
|
||||
HOST_BOOT alias=112 target=192.168.0.112 startup_unit=vm-host-boot reachable=1 boot_id=cc uptime_seconds=128 systemd_state=node_exporter startup_enabled=unknown startup_active=unknown
|
||||
HOST_BOOT alias=120 target=wooo@192.168.0.120 startup_unit=k3s.service reachable=1 boot_id=b uptime_seconds=130 systemd_state=running startup_enabled=enabled startup_active=active
|
||||
HOST_BOOT alias=121 target=wooo@192.168.0.121 startup_unit=k3s.service reachable=1 boot_id=c uptime_seconds=140 systemd_state=running startup_enabled=enabled startup_active=active
|
||||
HOST_BOOT alias=188 target=ollama@192.168.0.188 startup_unit=awoooi-startup.service reachable=1 boot_id=d uptime_seconds=150 systemd_state=running startup_enabled=enabled startup_active=active
|
||||
"""
|
||||
|
||||
|
||||
REBOOT_EVENT_GREEN = {
|
||||
"schema_version": "awoooi_reboot_event_detector_v1",
|
||||
"generated_at": "2026-06-29T14:30:00+08:00",
|
||||
"target_minutes": 10,
|
||||
"target_seconds": 600,
|
||||
"required_hosts": ["99", "110", "111", "112", "120", "121", "188"],
|
||||
"observed_hosts": ["99", "110", "111", "112", "120", "121", "188"],
|
||||
"missing_hosts": [],
|
||||
"unreachable_hosts": [],
|
||||
"reboot_detected": True,
|
||||
"rebooted_hosts": ["99", "110", "111", "112", "120", "121", "188"],
|
||||
"fresh_boot_hosts": ["99", "110", "111", "112", "120", "121", "188"],
|
||||
"changed_boot_id_hosts": [],
|
||||
"all_required_hosts_observed": True,
|
||||
"all_required_hosts_in_reboot_window": True,
|
||||
"max_observed_uptime_seconds": 150,
|
||||
"target_seconds_remaining": 450,
|
||||
"recovery_deadline_status": "within_target_window",
|
||||
"state_file": "/tmp/reboot-event-state.json",
|
||||
"state_written": True,
|
||||
}
|
||||
|
||||
|
||||
def run_scorecard(tmp_path: Path, summary: str, probe: str = HOST_PROBE_GREEN) -> dict:
|
||||
summary_path = tmp_path / "summary.txt"
|
||||
probe_path = tmp_path / "probe.txt"
|
||||
reboot_event_path = tmp_path / "reboot-event.json"
|
||||
summary_path.write_text(summary, encoding="utf-8")
|
||||
probe_path.write_text(probe, encoding="utf-8")
|
||||
reboot_event_path.write_text(json.dumps(REBOOT_EVENT_GREEN), encoding="utf-8")
|
||||
result = subprocess.run(
|
||||
[
|
||||
sys.executable,
|
||||
@@ -47,6 +75,8 @@ def run_scorecard(tmp_path: Path, summary: str, probe: str = HOST_PROBE_GREEN) -
|
||||
str(summary_path),
|
||||
"--host-probe-file",
|
||||
str(probe_path),
|
||||
"--reboot-event-file",
|
||||
str(reboot_event_path),
|
||||
"--generated-at",
|
||||
"2026-06-29T14:30:00+08:00",
|
||||
],
|
||||
@@ -66,10 +96,12 @@ def run_scorecard_with_stock(
|
||||
) -> dict:
|
||||
summary_path = tmp_path / "summary.txt"
|
||||
probe_path = tmp_path / "probe.txt"
|
||||
reboot_event_path = tmp_path / "reboot-event.json"
|
||||
freshness_path = tmp_path / "freshness.json"
|
||||
ingestion_path = tmp_path / "ingestion.json"
|
||||
summary_path.write_text(summary, encoding="utf-8")
|
||||
probe_path.write_text(HOST_PROBE_GREEN, encoding="utf-8")
|
||||
reboot_event_path.write_text(json.dumps(REBOOT_EVENT_GREEN), encoding="utf-8")
|
||||
freshness_path.write_text(json.dumps(freshness), encoding="utf-8")
|
||||
ingestion_path.write_text(json.dumps(ingestion), encoding="utf-8")
|
||||
result = subprocess.run(
|
||||
@@ -80,6 +112,8 @@ def run_scorecard_with_stock(
|
||||
str(summary_path),
|
||||
"--host-probe-file",
|
||||
str(probe_path),
|
||||
"--reboot-event-file",
|
||||
str(reboot_event_path),
|
||||
"--stock-freshness-file",
|
||||
str(freshness_path),
|
||||
"--stock-ingestion-file",
|
||||
|
||||
112
scripts/reboot-recovery/tests/test_reboot_event_detector.py
Normal file
112
scripts/reboot-recovery/tests/test_reboot_event_detector.py
Normal file
@@ -0,0 +1,112 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
import subprocess
|
||||
import sys
|
||||
from pathlib import Path
|
||||
|
||||
|
||||
ROOT = Path(__file__).resolve().parents[3]
|
||||
SCRIPT = ROOT / "scripts" / "reboot-recovery" / "reboot-event-detector.py"
|
||||
|
||||
|
||||
HOST_PROBE = """\
|
||||
AWOOOI_REBOOT_AUTO_RECOVERY_HOST_PROBE=1
|
||||
TARGET_HOSTS=99,110,111,112,120,121,188
|
||||
HOST_BOOT alias=99 target=192.168.0.99 startup_unit=vmware-host-autostart reachable=1 boot_id=win-boot-2 uptime_seconds=80 systemd_state=windows_exporter startup_enabled=enabled startup_active=active
|
||||
HOST_BOOT alias=110 target=wooo@192.168.0.110 startup_unit=awoooi-startup-110.service reachable=1 boot_id=linux-boot-2 uptime_seconds=100 systemd_state=running startup_enabled=enabled startup_active=active
|
||||
HOST_BOOT alias=111 target=192.168.0.111 startup_unit=vm-host-boot reachable=1 boot_id=vm111-boot-2 uptime_seconds=110 systemd_state=node_exporter startup_enabled=unknown startup_active=unknown
|
||||
HOST_BOOT alias=112 target=192.168.0.112 startup_unit=vm-host-boot reachable=1 boot_id=vm112-boot-2 uptime_seconds=115 systemd_state=node_exporter startup_enabled=unknown startup_active=unknown
|
||||
HOST_BOOT alias=120 target=wooo@192.168.0.120 startup_unit=k3s.service reachable=1 boot_id=mon-boot-2 uptime_seconds=120 systemd_state=running startup_enabled=enabled startup_active=active
|
||||
HOST_BOOT alias=121 target=wooo@192.168.0.121 startup_unit=k3s.service reachable=1 boot_id=worker-boot-2 uptime_seconds=130 systemd_state=running startup_enabled=enabled startup_active=active
|
||||
HOST_BOOT alias=188 target=ollama@192.168.0.188 startup_unit=awoooi-startup.service reachable=1 boot_id=ai-boot-2 uptime_seconds=140 systemd_state=running startup_enabled=enabled startup_active=active
|
||||
"""
|
||||
|
||||
|
||||
def test_reboot_detector_detects_all_required_hosts_and_writes_metrics(tmp_path: Path) -> None:
|
||||
probe_path = tmp_path / "host-probe.txt"
|
||||
state_path = tmp_path / "state.json"
|
||||
output_path = tmp_path / "event.json"
|
||||
prom_path = tmp_path / "event.prom"
|
||||
probe_path.write_text(HOST_PROBE, encoding="utf-8")
|
||||
state_path.write_text(
|
||||
json.dumps(
|
||||
{
|
||||
"hosts": {
|
||||
"99": {"boot_id": "win-boot-1"},
|
||||
"110": {"boot_id": "linux-boot-1"},
|
||||
"111": {"boot_id": "vm111-boot-1"},
|
||||
"112": {"boot_id": "vm112-boot-1"},
|
||||
"120": {"boot_id": "mon-boot-1"},
|
||||
"121": {"boot_id": "worker-boot-1"},
|
||||
"188": {"boot_id": "ai-boot-1"},
|
||||
}
|
||||
}
|
||||
),
|
||||
encoding="utf-8",
|
||||
)
|
||||
|
||||
subprocess.run(
|
||||
[
|
||||
sys.executable,
|
||||
str(SCRIPT),
|
||||
"--host-probe-file",
|
||||
str(probe_path),
|
||||
"--state-file",
|
||||
str(state_path),
|
||||
"--target-minutes",
|
||||
"10",
|
||||
"--generated-at",
|
||||
"2026-06-30T18:00:00+08:00",
|
||||
"--output",
|
||||
str(output_path),
|
||||
"--prometheus-output",
|
||||
str(prom_path),
|
||||
],
|
||||
check=True,
|
||||
)
|
||||
|
||||
payload = json.loads(output_path.read_text(encoding="utf-8"))
|
||||
metrics = prom_path.read_text(encoding="utf-8")
|
||||
|
||||
assert payload["required_hosts"] == ["99", "110", "111", "112", "120", "121", "188"]
|
||||
assert payload["reboot_detected"] is True
|
||||
assert payload["all_required_hosts_in_reboot_window"] is True
|
||||
assert payload["target_seconds_remaining"] == 460
|
||||
assert 'awoooi_reboot_event_host_rebooted{host="99"} 1' in metrics
|
||||
assert 'awoooi_reboot_event_required_host_observed{host="188"} 1' in metrics
|
||||
|
||||
|
||||
def test_reboot_detector_fails_visible_when_windows_or_vm_host_missing(tmp_path: Path) -> None:
|
||||
probe_path = tmp_path / "host-probe.txt"
|
||||
state_path = tmp_path / "state.json"
|
||||
output_path = tmp_path / "event.json"
|
||||
probe_path.write_text(
|
||||
"\n".join(line for line in HOST_PROBE.splitlines() if "alias=99 " not in line) + "\n",
|
||||
encoding="utf-8",
|
||||
)
|
||||
|
||||
subprocess.run(
|
||||
[
|
||||
sys.executable,
|
||||
str(SCRIPT),
|
||||
"--host-probe-file",
|
||||
str(probe_path),
|
||||
"--state-file",
|
||||
str(state_path),
|
||||
"--target-minutes",
|
||||
"10",
|
||||
"--generated-at",
|
||||
"2026-06-30T18:00:00+08:00",
|
||||
"--output",
|
||||
str(output_path),
|
||||
"--no-write-state",
|
||||
],
|
||||
check=True,
|
||||
)
|
||||
|
||||
payload = json.loads(output_path.read_text(encoding="utf-8"))
|
||||
|
||||
assert "99" in payload["missing_hosts"]
|
||||
assert payload["all_required_hosts_observed"] is False
|
||||
assert payload["all_required_hosts_in_reboot_window"] is False
|
||||
@@ -0,0 +1,53 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from pathlib import Path
|
||||
|
||||
|
||||
ROOT = Path(__file__).resolve().parents[3]
|
||||
|
||||
|
||||
def read(path: str) -> str:
|
||||
return (ROOT / path).read_text(encoding="utf-8")
|
||||
|
||||
|
||||
def test_reboot_p0_contract_covers_all_required_hosts_and_vmware_autostart() -> None:
|
||||
host_probe = read("scripts/reboot-recovery/reboot-auto-recovery-host-probe.sh")
|
||||
windows99 = read("scripts/reboot-recovery/windows99-vmware-autostart.ps1")
|
||||
|
||||
for host in ["99", "110", "111", "112", "120", "121", "188"]:
|
||||
assert host in host_probe
|
||||
assert "AWOOOI-Start-VMware-VMs" in windows99
|
||||
assert "NoAutoRebootWithLoggedOnUsers" in windows99
|
||||
assert "Host111Vmx" in windows99
|
||||
assert "Host188Vmx" in windows99
|
||||
assert "Host120Vmx" in windows99
|
||||
assert "Host121Vmx" in windows99
|
||||
assert "Host112Vmx" in windows99
|
||||
|
||||
|
||||
def test_reboot_p0_contract_has_maintenance_and_telegram_alerts() -> None:
|
||||
alerts = read("ops/monitoring/alerts-unified.yml")
|
||||
runbook = read("docs/runbooks/PUBLIC-MAINTENANCE-FALLBACK-RUNBOOK.md")
|
||||
snippet = read("ops/maintenance/nginx-502-maintenance-snippet.conf")
|
||||
|
||||
for alert_name in [
|
||||
"HostRebootEventDetected",
|
||||
"RebootAutoRecoverySLOMissed",
|
||||
"PublicRouteServing5xxWithoutMaintenanceFallback",
|
||||
"BackupCoverageDomainStale",
|
||||
]:
|
||||
assert alert_name in alerts
|
||||
assert "notification_type: TYPE-3" in alerts
|
||||
assert "proxy_intercept_errors on" in snippet
|
||||
assert "502" in runbook
|
||||
assert "L0" in runbook
|
||||
assert "L1" in runbook
|
||||
|
||||
|
||||
def test_backup_exporter_emits_domain_level_backup_coverage() -> None:
|
||||
exporter = read("scripts/ops/backup-health-textfile-exporter.py")
|
||||
|
||||
assert "awoooi_backup_coverage_domain_expected_info" in exporter
|
||||
assert "awoooi_backup_coverage_domain_fresh" in exporter
|
||||
for domain in ["host", "database", "website", "service", "package", "tool", "log"]:
|
||||
assert f'"{domain}"' in exporter
|
||||
162
scripts/reboot-recovery/windows99-vmware-autostart.ps1
Normal file
162
scripts/reboot-recovery/windows99-vmware-autostart.ps1
Normal file
@@ -0,0 +1,162 @@
|
||||
param(
|
||||
[ValidateSet("Verify", "DryRun", "Apply")]
|
||||
[string]$Mode = "Verify",
|
||||
[string]$VmrunPath = "C:\Program Files (x86)\VMware\VMware Workstation\vmrun.exe",
|
||||
[string]$Host111Vmx = "",
|
||||
[string]$Host188Vmx = "",
|
||||
[string]$Host120Vmx = "",
|
||||
[string]$Host121Vmx = "",
|
||||
[string]$Host112Vmx = "",
|
||||
[string[]]$DiscoveryRoot = @("D:\VMs", "E:\VMs", "C:\VMs", "C:\Users\Public\Documents\Virtual Machines")
|
||||
)
|
||||
|
||||
$ErrorActionPreference = "Stop"
|
||||
|
||||
$TaskName = "AWOOOI-Start-VMware-VMs"
|
||||
$ProgramDataDir = "C:\ProgramData\AWOOOI"
|
||||
$StartScript = Join-Path $ProgramDataDir "Start-AWOOOI-VMs.ps1"
|
||||
$VmOrder = @("111", "188", "120", "121", "112")
|
||||
$SuppliedVmx = @{
|
||||
"111" = $Host111Vmx
|
||||
"188" = $Host188Vmx
|
||||
"120" = $Host120Vmx
|
||||
"121" = $Host121Vmx
|
||||
"112" = $Host112Vmx
|
||||
}
|
||||
|
||||
function Resolve-VmxPath {
|
||||
param([string]$HostAlias)
|
||||
|
||||
$supplied = $SuppliedVmx[$HostAlias]
|
||||
if ($supplied -and (Test-Path -LiteralPath $supplied)) {
|
||||
return (Resolve-Path -LiteralPath $supplied).Path
|
||||
}
|
||||
|
||||
foreach ($root in $DiscoveryRoot) {
|
||||
if (-not (Test-Path -LiteralPath $root)) {
|
||||
continue
|
||||
}
|
||||
$match = Get-ChildItem -LiteralPath $root -Recurse -Filter "*.vmx" -ErrorAction SilentlyContinue |
|
||||
Where-Object { $_.FullName -match "(^|[\\_\-\s])$HostAlias([\\_\-\s]|$)" } |
|
||||
Select-Object -First 1
|
||||
if ($match) {
|
||||
return $match.FullName
|
||||
}
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
function Get-VmMap {
|
||||
$map = [ordered]@{}
|
||||
foreach ($alias in $VmOrder) {
|
||||
$map[$alias] = Resolve-VmxPath -HostAlias $alias
|
||||
}
|
||||
return $map
|
||||
}
|
||||
|
||||
function Write-StartupScript {
|
||||
param([hashtable]$VmMap)
|
||||
|
||||
New-Item -ItemType Directory -Path $ProgramDataDir -Force | Out-Null
|
||||
$vmRows = foreach ($alias in $VmOrder) {
|
||||
" @{ Alias = `"$alias`"; Path = `"$($VmMap[$alias])`" }"
|
||||
}
|
||||
$body = @"
|
||||
`$ErrorActionPreference = "Continue"
|
||||
`$VmrunPath = "$VmrunPath"
|
||||
`$VMs = @(
|
||||
$($vmRows -join ",`n")
|
||||
)
|
||||
|
||||
Start-Service -Name "VMAuthdService" -ErrorAction SilentlyContinue
|
||||
Start-Service -Name "VMnetDHCP" -ErrorAction SilentlyContinue
|
||||
Start-Service -Name "VMware NAT Service" -ErrorAction SilentlyContinue
|
||||
Start-Sleep -Seconds 30
|
||||
|
||||
foreach (`$vm in `$VMs) {
|
||||
if (-not (Test-Path -LiteralPath `$vm.Path)) {
|
||||
Write-Output "VMX_MISSING alias=`$(`$vm.Alias) path=`$(`$vm.Path)"
|
||||
continue
|
||||
}
|
||||
& `$VmrunPath -T ws start `$vm.Path nogui
|
||||
Write-Output "VM_START_REQUESTED alias=`$(`$vm.Alias) path=`$(`$vm.Path) exit=`$LASTEXITCODE"
|
||||
Start-Sleep -Seconds 20
|
||||
}
|
||||
"@
|
||||
Set-Content -LiteralPath $StartScript -Value $body -Encoding UTF8
|
||||
}
|
||||
|
||||
function Apply-WindowsUpdatePolicy {
|
||||
$auPath = "HKLM:\SOFTWARE\Policies\Microsoft\Windows\WindowsUpdate\AU"
|
||||
New-Item -Path $auPath -Force | Out-Null
|
||||
New-ItemProperty -Path $auPath -Name "NoAutoRebootWithLoggedOnUsers" -Value 1 -PropertyType DWord -Force | Out-Null
|
||||
New-ItemProperty -Path $auPath -Name "AlwaysAutoRebootAtScheduledTime" -Value 0 -PropertyType DWord -Force | Out-Null
|
||||
New-ItemProperty -Path $auPath -Name "AUOptions" -Value 3 -PropertyType DWord -Force | Out-Null
|
||||
New-ItemProperty -Path $auPath -Name "ScheduledInstallDay" -Value 0 -PropertyType DWord -Force | Out-Null
|
||||
New-ItemProperty -Path $auPath -Name "ScheduledInstallTime" -Value 3 -PropertyType DWord -Force | Out-Null
|
||||
New-ItemProperty -Path $auPath -Name "AUPowerManagement" -Value 0 -PropertyType DWord -Force | Out-Null
|
||||
}
|
||||
|
||||
function Apply-ScheduledTask {
|
||||
$action = New-ScheduledTaskAction `
|
||||
-Execute "powershell.exe" `
|
||||
-Argument "-NoProfile -ExecutionPolicy Bypass -File `"$StartScript`""
|
||||
$trigger = New-ScheduledTaskTrigger -AtStartup
|
||||
$principal = New-ScheduledTaskPrincipal -UserId "SYSTEM" -RunLevel Highest
|
||||
$settings = New-ScheduledTaskSettingsSet `
|
||||
-AllowStartIfOnBatteries `
|
||||
-DontStopIfGoingOnBatteries `
|
||||
-MultipleInstances IgnoreNew `
|
||||
-RestartCount 3 `
|
||||
-RestartInterval (New-TimeSpan -Minutes 2) `
|
||||
-StartWhenAvailable
|
||||
Register-ScheduledTask `
|
||||
-TaskName $TaskName `
|
||||
-Action $action `
|
||||
-Trigger $trigger `
|
||||
-Principal $principal `
|
||||
-Settings $settings `
|
||||
-Force | Out-Null
|
||||
}
|
||||
|
||||
$vmMap = Get-VmMap
|
||||
$missing = @($VmOrder | Where-Object { -not $vmMap[$_] })
|
||||
$vmrunPresent = Test-Path -LiteralPath $VmrunPath
|
||||
|
||||
Write-Output "AWOOOI_WINDOWS99_VMWARE_AUTOSTART=1"
|
||||
Write-Output "MODE=$Mode"
|
||||
Write-Output "VMRUN_PRESENT=$([int]$vmrunPresent)"
|
||||
foreach ($alias in $VmOrder) {
|
||||
Write-Output "VMX alias=$alias path=$($vmMap[$alias]) present=$([int][bool]$vmMap[$alias])"
|
||||
}
|
||||
|
||||
if (-not $vmrunPresent) {
|
||||
Write-Error "vmrun.exe not found at $VmrunPath"
|
||||
}
|
||||
|
||||
if ($missing.Count -gt 0) {
|
||||
Write-Output "MISSING_VMX_ALIASES=$($missing -join ',')"
|
||||
if ($Mode -eq "Apply") {
|
||||
throw "Apply requires explicit or discoverable VMX paths for every required VM."
|
||||
}
|
||||
}
|
||||
|
||||
if ($Mode -eq "Verify") {
|
||||
exit 0
|
||||
}
|
||||
|
||||
if ($Mode -eq "DryRun") {
|
||||
Write-Output "DRY_RUN would_write_start_script=$StartScript"
|
||||
Write-Output "DRY_RUN would_register_task=$TaskName"
|
||||
Write-Output "DRY_RUN would_apply_windows_update_no_auto_reboot_policy=1"
|
||||
exit 0
|
||||
}
|
||||
|
||||
Write-StartupScript -VmMap $vmMap
|
||||
Set-Service -Name "VMAuthdService" -StartupType Automatic -ErrorAction SilentlyContinue
|
||||
Set-Service -Name "VMnetDHCP" -StartupType Automatic -ErrorAction SilentlyContinue
|
||||
Set-Service -Name "VMware NAT Service" -StartupType Automatic -ErrorAction SilentlyContinue
|
||||
Apply-ScheduledTask
|
||||
Apply-WindowsUpdatePolicy
|
||||
|
||||
Write-Output "APPLIED scheduled_task=$TaskName start_script=$StartScript windows_update_no_auto_reboot_policy=1"
|
||||
Reference in New Issue
Block a user