Don't register total_size metric if we don't expose it

Author: yuvipanda

prometheus_dirsize_exporter/exporter.py

@@ -2,7 +2,7 @@
 import time
 import argparse
 from typing import Callable, Never, Optional, Generator
-from prometheus_client import start_http_server
+from prometheus_client import REGISTRY, start_http_server
 from . import metrics
 from dataclasses import dataclass
 
@@ -192,6 +192,9 @@ def main() -> Never:
 
     args = argparser.parse_args()
 
+    if not args.disable_total_size:
+        REGISTRY.register(metrics.TOTAL_SIZE)
+
     start_http_server(args.port)
     while True:
         walker = BudgetedDirInfoWalker(args.iops_budget)

prometheus_dirsize_exporter/metrics.py

@@ -7,7 +7,8 @@
 
 TOTAL_SIZE = Gauge(
     "total_size_bytes", "Total Size of the Directory (in bytes)", namespace=NAMESPACE,
-    labelnames=("directory",)
+    labelnames=("directory",),
+    registry=None # Don't register this metric, as it's optional
 )
 
 LATEST_MTIME = Gauge(