rawfile-localpv/metrics.py

75 lines
2.3 KiB
Python
Raw Normal View History

2020-04-25 19:39:16 +00:00
import json
import os
import subprocess
2020-04-25 19:39:16 +00:00
from prometheus_client.core import REGISTRY
2020-04-25 19:39:16 +00:00
from prometheus_client.exposition import start_http_server
from prometheus_client.metrics_core import GaugeMetricFamily
2020-04-25 19:39:16 +00:00
import rawfile_util
from rawfile_util import attached_loops
class VolumeStatsCollector(object):
def collect(self):
VOLUME_ID = "volume_id"
fs_size = GaugeMetricFamily(
"rawfile_filesystem_size_bytes",
"Filesystem size in bytes.",
labels=[VOLUME_ID],
)
fs_free = GaugeMetricFamily(
"rawfile_filesystem_avail_bytes",
"Filesystem free space in bytes",
labels=[VOLUME_ID],
)
dev_size = GaugeMetricFamily(
"rawfile_device_size_bytes", "Device size in bytes.", labels=[VOLUME_ID]
)
dev_free = GaugeMetricFamily(
"rawfile_device_free_bytes",
"Device free space in bytes.",
labels=[VOLUME_ID],
)
for volume_id in rawfile_util.list_all_volumes():
img_file = rawfile_util.img_file(volume_id)
labels = [volume_id]
dev_stat = img_file.stat()
dev_size.add_metric(labels, dev_stat.st_size)
dev_free.add_metric(labels, dev_stat.st_size - dev_stat.st_blocks * 512)
mountpoint = volume_to_mountpoint(img_file)
if mountpoint is not None:
fs_stat = os.statvfs(mountpoint)
fs_size.add_metric(labels, fs_stat.f_frsize * fs_stat.f_blocks)
fs_free.add_metric(labels, fs_stat.f_frsize * fs_stat.f_bfree)
return [fs_size, fs_free, dev_size, dev_free]
def volume_to_mountpoint(img_file):
for dev in attached_loops(img_file):
mountpoint = dev_to_mountpoint(dev)
if mountpoint is not None:
return mountpoint
return None
2020-04-25 19:39:16 +00:00
def dev_to_mountpoint(dev_name):
try:
output = subprocess.run(
f"findmnt --json --first-only {dev_name}",
shell=True,
check=True,
capture_output=True,
).stdout.decode()
data = json.loads(output)
return data["filesystems"][0]["target"]
except subprocess.CalledProcessError:
return None
def expose_metrics():
REGISTRY.register(VolumeStatsCollector())
2020-04-25 19:39:16 +00:00
start_http_server(9100)