apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-06 17:17:14 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

metrics shared folder naming (#4007)

Browse Source 
* Fixed the issue of metrics file conflicts between multiple instances on a single machine

* Use uuid to name the metrics shared folder

* Use uuid to name the metrics shared folder
This commit is contained in:
guozhuangzhuang
2025-09-10 10:47:20 +08:00
committed by GitHub
parent 3b1da6e4dd
commit f078a959b6
1 changed files with 12 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
fastdeploy/metrics/metrics.py
View File

@@ -19,6 +19,7 @@ metrics
""" """
import os import os
import shutil import shutil
import uuid
from typing import Set from typing import Set
from prometheus_client import ( from prometheus_client import (
@@ -35,24 +36,20 @@ from fastdeploy.metrics import build_1_2_5_buckets
from fastdeploy.metrics.work_metrics import work_process_metrics from fastdeploy.metrics.work_metrics import work_process_metrics
def cleanup_prometheus_files(is_main): def cleanup_prometheus_files(is_main: bool, instance_id: str = None):
""" """
Cleans and recreates the Prometheus multiprocess directory. Cleans and recreates the Prometheus multiprocess directory.
Depending on whether it's the main process or a worker, this function removes the corresponding
Prometheus multiprocess directory (/tmp/prom_main or /tmp/prom_worker) and recreates it as an empty directory.
Args:
is_main (bool): Indicates whether the current process is the main process.
Returns:
str: The path to the newly created Prometheus multiprocess directory.
""" """
PROM_DIR = "/tmp/prom_main" if is_main else "/tmp/prom_worker" base_dir = "/tmp/prom_main" if is_main else "/tmp/prom_worker"
if os.path.exists(PROM_DIR): if instance_id is None:
shutil.rmtree(PROM_DIR) instance_id = str(uuid.uuid4())
os.makedirs(PROM_DIR, exist_ok=True) prom_dir = f"{base_dir}_{instance_id}"
return PROM_DIR
if os.path.exists(prom_dir):
shutil.rmtree(prom_dir, ignore_errors=True)
os.makedirs(prom_dir, exist_ok=True)
return prom_dir
class SimpleCollector(Collector): class SimpleCollector(Collector):