fix: update monitoring

Author: Dmitriy Astapkovich

pyproject.toml

@@ -10,7 +10,7 @@ name = "temporal-boost"
 packages = [{include = "temporal_boost"}]
 readme = "README.md"
 repository = "https://github.com/northpowered/temporal-boost"
-version = "1.2.0"
+version = "2.0.0-rc.1"
 
 classifiers = [
   "Programming Language :: Python :: 3",

temporal_boost/cli/runner.py

@@ -1,6 +1,9 @@
 import importlib
 import logging
+import os
+import tempfile
 from multiprocessing import Process
+from pathlib import Path
 from typing import TYPE_CHECKING, Any
 
 import typer
@@ -14,6 +17,28 @@
 cli = typer.Typer(help="CLI runner for BoostApp services")
 
 
+def setup_prometheus_multiproc_dir() -> None:
+    prometheus_dir = os.getenv("PROMETHEUS_MULTIPROC_DIR")
+
+    if prometheus_dir:
+        path = Path(prometheus_dir)
+        if not path.exists():
+            try:
+                path.mkdir(parents=True, exist_ok=True)
+                logger.info(f"Created prometheus multiprocess dir at {prometheus_dir}")
+            except Exception as e:
+                logger.warning(f"Cannot create prometheus multiprocess dir '{prometheus_dir}': {e}")
+                del os.environ["PROMETHEUS_MULTIPROC_DIR"]
+                logger.warning("Unset PROMETHEUS_MULTIPROC_DIR due to errors")
+    else:
+        try:
+            temp_dir = tempfile.mkdtemp(prefix="prometheus_multiproc_")
+            os.environ["PROMETHEUS_MULTIPROC_DIR"] = temp_dir
+            logger.info(f"Set temporary PROMETHEUS_MULTIPROC_DIR at {temp_dir}")
+        except Exception as e:
+            logger.warning(f"Failed to set temporary PROMETHEUS_MULTIPROC_DIR: {e}")
+
+
 def _import_app_object(app_path: str) -> Any:
     if ":" not in app_path:
         raise typer.BadParameter("App path must be in format 'module.submodule:app'")
@@ -31,11 +56,14 @@ def _run_single(app_path: str, run_args: list[str]) -> None:
 
 
 def _run_multiprocess(app_path: str, processes: int, run_args: list[str]) -> None:
+    setup_prometheus_multiproc_dir()
+
     process_list: list[Process] = []
     for index in range(processes):
         process = Process(target=_run_single, args=(app_path, run_args), name=f"worker-{index}")
         process_list.append(process)
         process.start()
+
     for process in process_list:
         process.join()