mirror of
https://github.com/recklessop/zroc.git
synced 2026-07-03 13:23:15 -04:00
Justin
80 lines
3.4 KiB
Python
80 lines
3.4 KiB
Python
"""
|
|
Configuration for the zROC Planner vCenter metrics collector.
|
|
All settings are driven by environment variables with sensible defaults.
|
|
"""
|
|
|
|
import os
|
|
import logging
|
|
|
|
|
|
def _get_int(key: str, default: int) -> int:
|
|
try:
|
|
return int(os.environ.get(key, default))
|
|
except (ValueError, TypeError):
|
|
return default
|
|
|
|
|
|
def _get_float(key: str, default: float) -> float:
|
|
try:
|
|
return float(os.environ.get(key, default))
|
|
except (ValueError, TypeError):
|
|
return default
|
|
|
|
|
|
def _get_bool(key: str, default: bool) -> bool:
|
|
val = os.environ.get(key, "").strip().lower()
|
|
if val in ("1", "true", "yes"):
|
|
return True
|
|
if val in ("0", "false", "no"):
|
|
return False
|
|
return default
|
|
|
|
|
|
# ── vCenter connection ────────────────────────────────────────────────────────
|
|
VCENTER_HOST: str = os.environ.get("VCENTER_HOST", "vcenter.local")
|
|
VCENTER_USER: str = os.environ.get("VCENTER_USER", "administrator@vsphere.local")
|
|
VCENTER_PASSWORD: str = os.environ.get("VCENTER_PASSWORD", "")
|
|
VCENTER_PORT: int = _get_int("VCENTER_PORT", 443)
|
|
VCENTER_SSL_VERIFY: bool = _get_bool("VCENTER_SSL_VERIFY", False)
|
|
|
|
# ── collection tuning ─────────────────────────────────────────────────────────
|
|
# How often (seconds) the collector runs a full poll cycle
|
|
POLL_INTERVAL: int = _get_int("POLL_INTERVAL", 300)
|
|
|
|
# Size of each QueryPerf batch (VMware recommends ≤200)
|
|
BATCH_SIZE: int = _get_int("BATCH_SIZE", 100)
|
|
|
|
# Seconds to sleep between batches to avoid hammering vCenter
|
|
BATCH_DELAY: float = _get_float("BATCH_DELAY", 0.5)
|
|
|
|
# How often (seconds) to refresh the VM inventory list
|
|
VM_INVENTORY_TTL: int = _get_int("VM_INVENTORY_TTL", 600)
|
|
|
|
# Rollup interval for performance counters (300 = 5-minute stats)
|
|
PERF_INTERVAL_ID: int = _get_int("PERF_INTERVAL_ID", 300)
|
|
|
|
# ── HTTP server ───────────────────────────────────────────────────────────────
|
|
HTTP_HOST: str = os.environ.get("HTTP_HOST", "0.0.0.0")
|
|
HTTP_PORT: int = _get_int("HTTP_PORT", 9272)
|
|
|
|
# ── logging ───────────────────────────────────────────────────────────────────
|
|
LOG_LEVEL: str = os.environ.get("LOG_LEVEL", "INFO").upper()
|
|
|
|
# ── counter names we care about ───────────────────────────────────────────────
|
|
# These are the vSphere performance counter names for disk metrics.
|
|
# Mapped as: human_label -> (group, name, rollup_type)
|
|
COUNTERS_WANTED: dict[str, tuple[str, str, str]] = {
|
|
"disk_write_iops": ("virtualDisk", "numberWriteAveraged", "average"),
|
|
"disk_write_throughput": ("virtualDisk", "write", "average"),
|
|
"disk_write_latency": ("virtualDisk", "totalWriteLatency", "average"),
|
|
}
|
|
|
|
|
|
def configure_logging() -> None:
|
|
level = getattr(logging, LOG_LEVEL, logging.INFO)
|
|
logging.basicConfig(
|
|
format="%(asctime)s %(levelname)-8s %(name)s: %(message)s",
|
|
datefmt="%Y-%m-%dT%H:%M:%S",
|
|
level=level,
|
|
)
|