343 lines
11 KiB
Python
343 lines
11 KiB
Python
#!/usr/bin/env python3
|
|
import datetime
|
|
import hashlib
|
|
import json
|
|
import logging
|
|
import os
|
|
import time
|
|
import re
|
|
import subprocess
|
|
import sys
|
|
import traceback
|
|
|
|
from prometheus_client import start_http_server
|
|
from prometheus_client.core import GaugeMetricFamily, CounterMetricFamily, REGISTRY
|
|
|
|
|
|
class ResticCollector(object):
|
|
def __init__(
|
|
self, repository, password_file, exit_on_error, disable_check, disable_stats
|
|
):
|
|
self.repository = repository
|
|
self.password_file = password_file
|
|
self.exit_on_error = exit_on_error
|
|
self.disable_check = disable_check
|
|
self.disable_stats = disable_stats
|
|
# todo: the stats cache increases over time -> remove old ids
|
|
# todo: cold start -> the stats cache could be saved in a persistent
|
|
# volume
|
|
# todo: cold start -> the restic cache (/root/.cache/restic) could be
|
|
# saved in a persistent volume
|
|
self.stats_cache = {}
|
|
self.metrics = {}
|
|
self.refresh(exit_on_error)
|
|
|
|
def collect(self):
|
|
logging.debug("Incoming request")
|
|
|
|
common_label_names = [
|
|
"client_hostname",
|
|
"client_username",
|
|
"snapshot_hash",
|
|
"snapshot_tag",
|
|
]
|
|
|
|
check_success = GaugeMetricFamily(
|
|
"restic_check_success",
|
|
"Result of restic check operation in the repository",
|
|
labels=[],
|
|
)
|
|
|
|
snapshots_total = CounterMetricFamily(
|
|
"restic_snapshots_total",
|
|
"Total number of snapshots in the repository",
|
|
labels=[],
|
|
)
|
|
|
|
backup_timestamp = GaugeMetricFamily(
|
|
"restic_backup_timestamp",
|
|
"Timestamp of the last backup",
|
|
labels=common_label_names,
|
|
)
|
|
|
|
backup_files_total = CounterMetricFamily(
|
|
"restic_backup_files_total",
|
|
"Number of files in the backup",
|
|
labels=common_label_names,
|
|
)
|
|
|
|
backup_size_total = CounterMetricFamily(
|
|
"restic_backup_size_total",
|
|
"Total size of backup in bytes",
|
|
labels=common_label_names,
|
|
)
|
|
|
|
backup_snapshots_total = CounterMetricFamily(
|
|
"restic_backup_snapshots_total",
|
|
"Total number of snapshots",
|
|
labels=common_label_names,
|
|
)
|
|
|
|
scrape_duration_seconds = GaugeMetricFamily(
|
|
"restic_scrape_duration_seconds",
|
|
"Ammount of time each scrape takes",
|
|
labels=[],
|
|
)
|
|
|
|
check_success.add_metric([], self.metrics["check_success"])
|
|
snapshots_total.add_metric([], self.metrics["snapshots_total"])
|
|
|
|
for client in self.metrics["clients"]:
|
|
common_label_values = [
|
|
client["hostname"],
|
|
client["username"],
|
|
client["snapshot_hash"],
|
|
client["snapshot_tag"],
|
|
]
|
|
|
|
backup_timestamp.add_metric(common_label_values, client["timestamp"])
|
|
backup_files_total.add_metric(common_label_values, client["files_total"])
|
|
backup_size_total.add_metric(common_label_values, client["size_total"])
|
|
backup_snapshots_total.add_metric(
|
|
common_label_values, client["snapshots_total"]
|
|
)
|
|
|
|
scrape_duration_seconds.add_metric([], self.metrics["duration"])
|
|
|
|
yield check_success
|
|
yield snapshots_total
|
|
yield backup_timestamp
|
|
yield backup_files_total
|
|
yield backup_size_total
|
|
yield backup_snapshots_total
|
|
yield scrape_duration_seconds
|
|
|
|
def refresh(self, exit_on_error=False):
|
|
try:
|
|
self.metrics = self.get_metrics()
|
|
except Exception:
|
|
logging.error(
|
|
"Unable to collect metrics from Restic. %s",
|
|
traceback.format_exc(0).replace("\n", " "),
|
|
)
|
|
|
|
# Shutdown exporter for any error
|
|
if exit_on_error:
|
|
sys.exit(1)
|
|
|
|
def get_metrics(self):
|
|
duration = time.time()
|
|
all_snapshots = self.get_snapshots()
|
|
latest_snapshots = self.get_snapshots(True)
|
|
clients = []
|
|
for snap in latest_snapshots:
|
|
# Collect stats for each snap only if enabled
|
|
if self.disable_stats:
|
|
# return zero as "no-stats" value
|
|
stats = {
|
|
"total_size": -1,
|
|
"total_file_count": -1,
|
|
}
|
|
else:
|
|
stats = self.get_stats(snap["id"])
|
|
|
|
# use first element of tags if tags is present
|
|
if "tags" in snap:
|
|
tag = snap["tags"][0]
|
|
else:
|
|
tag = ""
|
|
|
|
time_parsed = re.sub(r"\.[^+-]+", "", snap["time"])
|
|
if len(time_parsed) > 19:
|
|
# restic 14: '2023-01-12T06:59:33.1576588+01:00' ->
|
|
# '2023-01-12T06:59:33+01:00'
|
|
time_format = "%Y-%m-%dT%H:%M:%S%z"
|
|
else:
|
|
# restic 12: '2023-02-01T14:14:19.30760523Z' ->
|
|
# '2023-02-01T14:14:19'
|
|
time_format = "%Y-%m-%dT%H:%M:%S"
|
|
timestamp = time.mktime(
|
|
datetime.datetime.strptime(time_parsed, time_format).timetuple()
|
|
)
|
|
|
|
snapshots_total = 0
|
|
for snap2 in all_snapshots:
|
|
if snap2["hash"] == snap["hash"]:
|
|
snapshots_total += 1
|
|
|
|
clients.append(
|
|
{
|
|
"hostname": snap["hostname"],
|
|
"username": snap["username"] if "username" in snap else "",
|
|
"snapshot_hash": snap["hash"],
|
|
"snapshot_tag": tag,
|
|
"timestamp": timestamp,
|
|
"size_total": stats["total_size"],
|
|
"files_total": stats["total_file_count"],
|
|
"snapshots_total": snapshots_total,
|
|
}
|
|
)
|
|
# todo: fix the commented code when the bug is fixed in restic
|
|
# https://github.com/restic/restic/issues/2126
|
|
# stats = self.get_stats()
|
|
|
|
if self.disable_check:
|
|
# return 2 as "no-check" value
|
|
check_success = 2
|
|
else:
|
|
check_success = self.get_check()
|
|
|
|
metrics = {
|
|
"check_success": check_success,
|
|
"clients": clients,
|
|
"snapshots_total": len(all_snapshots),
|
|
"duration": time.time() - duration
|
|
# 'size_total': stats['total_size'],
|
|
# 'files_total': stats['total_file_count'],
|
|
}
|
|
|
|
return metrics
|
|
|
|
def get_snapshots(self, only_latest=False):
|
|
cmd = [
|
|
"restic",
|
|
"-r",
|
|
self.repository,
|
|
"-p",
|
|
self.password_file,
|
|
"--no-lock",
|
|
"snapshots",
|
|
"--json",
|
|
]
|
|
|
|
if only_latest:
|
|
cmd.extend(["--latest", "1"])
|
|
|
|
result = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
|
if result.returncode != 0:
|
|
raise Exception(
|
|
"Error executing restic snapshot command: " + self.parse_stderr(result)
|
|
)
|
|
snapshots = json.loads(result.stdout.decode("utf-8"))
|
|
for snap in snapshots:
|
|
snap["hash"] = self.calc_snapshot_hash(snap)
|
|
return snapshots
|
|
|
|
def get_stats(self, snapshot_id=None):
|
|
# This command is expensive in CPU/Memory (1-5 seconds),
|
|
# and much more when snapshot_id=None (3 minutes) -> we avoid this call for now
|
|
# https://github.com/restic/restic/issues/2126
|
|
if snapshot_id is not None and snapshot_id in self.stats_cache:
|
|
return self.stats_cache[snapshot_id]
|
|
|
|
cmd = [
|
|
"restic",
|
|
"-r",
|
|
self.repository,
|
|
"-p",
|
|
self.password_file,
|
|
"--no-lock",
|
|
"stats",
|
|
"--json",
|
|
]
|
|
if snapshot_id is not None:
|
|
cmd.extend([snapshot_id])
|
|
|
|
result = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
|
if result.returncode != 0:
|
|
raise Exception(
|
|
"Error executing restic stats command: " + self.parse_stderr(result)
|
|
)
|
|
stats = json.loads(result.stdout.decode("utf-8"))
|
|
|
|
if snapshot_id is not None:
|
|
self.stats_cache[snapshot_id] = stats
|
|
|
|
return stats
|
|
|
|
def get_check(self):
|
|
# This command takes 20 seconds or more, but it's required
|
|
cmd = [
|
|
"restic",
|
|
"-r",
|
|
self.repository,
|
|
"-p",
|
|
self.password_file,
|
|
"--no-lock",
|
|
"check",
|
|
]
|
|
|
|
result = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
|
if result.returncode == 0:
|
|
return 1 # ok
|
|
else:
|
|
logging.warning(
|
|
"Error checking the repository health. " + self.parse_stderr(result)
|
|
)
|
|
return 0 # error
|
|
|
|
def calc_snapshot_hash(self, snapshot: dict) -> str:
|
|
text = snapshot["hostname"] + ",".join(snapshot["paths"])
|
|
return hashlib.sha256(text.encode("utf-8")).hexdigest()
|
|
|
|
def parse_stderr(self, result):
|
|
return (
|
|
result.stderr.decode("utf-8").replace("\n", " ")
|
|
+ " Exit code: "
|
|
+ str(result.returncode)
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
logging.basicConfig(
|
|
format="%(asctime)s %(levelname)-8s %(message)s",
|
|
level=logging.getLevelName(os.environ.get("LOG_LEVEL", "INFO")),
|
|
datefmt="%Y-%m-%d %H:%M:%S",
|
|
handlers=[logging.StreamHandler(sys.stdout)],
|
|
)
|
|
logging.info("Starting Restic Prometheus Exporter")
|
|
logging.info("It could take a while if the repository is remote")
|
|
|
|
try:
|
|
restic_repo_url = os.environ["RESTIC_REPO_URL"]
|
|
except Exception:
|
|
logging.error("The environment variable RESTIC_REPO_URL is mandatory")
|
|
sys.exit(1)
|
|
|
|
try:
|
|
restic_repo_password_file = os.environ["RESTIC_REPO_PASSWORD_FILE"]
|
|
except Exception:
|
|
logging.error("The environment variable RESTIC_REPO_PASSWORD_FILE is mandatory")
|
|
sys.exit(1)
|
|
|
|
exporter_address = os.environ.get("LISTEN_ADDRESS", "0.0.0.0")
|
|
exporter_port = int(os.environ.get("LISTEN_PORT", 8001))
|
|
exporter_refresh_interval = int(os.environ.get("REFRESH_INTERVAL", 60))
|
|
exporter_exit_on_error = bool(os.environ.get("EXIT_ON_ERROR", False))
|
|
exporter_disable_check = bool(os.environ.get("NO_CHECK", False))
|
|
exporter_disable_stats = bool(os.environ.get("NO_STATS", False))
|
|
|
|
try:
|
|
collector = ResticCollector(
|
|
restic_repo_url,
|
|
restic_repo_password_file,
|
|
exporter_exit_on_error,
|
|
exporter_disable_check,
|
|
exporter_disable_stats,
|
|
)
|
|
REGISTRY.register(collector)
|
|
start_http_server(exporter_port, exporter_address)
|
|
logging.info(
|
|
"Serving at http://{0}:{1}".format(exporter_address, exporter_port)
|
|
)
|
|
|
|
while True:
|
|
logging.info(
|
|
"Refreshing stats every {0} seconds".format(exporter_refresh_interval)
|
|
)
|
|
time.sleep(exporter_refresh_interval)
|
|
collector.refresh()
|
|
|
|
except KeyboardInterrupt:
|
|
logging.info("\nInterrupted")
|
|
exit(0)
|