REDROOM

Edit File Size: 3.75 KB

/opt/cloudlinux/venv/lib64/python3.11/site-packages/prometheus_client/values.py

Text Base64

from __future__ import unicode_literals

import os
from threading import Lock

from .mmap_dict import mmap_key, MmapedDict

class MutexValue(object):
    """A float protected by a mutex."""

_multiprocess = False

def __init__(self, typ, metric_name, name, labelnames, labelvalues, **kwargs):
        self._value = 0.0
        self._lock = Lock()

def inc(self, amount):
        with self._lock:
            self._value += amount

def set(self, value):
        with self._lock:
            self._value = value

def get(self):
        with self._lock:
            return self._value

def MultiProcessValue(process_identifier=os.getpid):
    """Returns a MmapedValue class based on a process_identifier function.

The 'process_identifier' function MUST comply with this simple rule:
    when called in simultaneously running processes it MUST return distinct values.

Using a different function than the default 'os.getpid' is at your own risk.
    """
    files = {}
    values = []
    pid = {'value': process_identifier()}
    # Use a single global lock when in multi-processing mode
    # as we presume this means there is no threading going on.
    # This avoids the need to also have mutexes in __MmapDict.
    lock = Lock()

class MmapedValue(object):
        """A float protected by a mutex backed by a per-process mmaped file."""

_multiprocess = True

def __init__(self, typ, metric_name, name, labelnames, labelvalues, multiprocess_mode='', **kwargs):
            self._params = typ, metric_name, name, labelnames, labelvalues, multiprocess_mode
            with lock:
                self.__check_for_pid_change()
                self.__reset()
                values.append(self)

def __reset(self):
            typ, metric_name, name, labelnames, labelvalues, multiprocess_mode = self._params
            if typ == 'gauge':
                file_prefix = typ + '_' + multiprocess_mode
            else:
                file_prefix = typ
            if file_prefix not in files:
                filename = os.path.join(
                    os.environ['prometheus_multiproc_dir'],
                    '{0}_{1}.db'.format(file_prefix, pid['value']))

files[file_prefix] = MmapedDict(filename)
            self._file = files[file_prefix]
            self._key = mmap_key(metric_name, name, labelnames, labelvalues)
            self._value = self._file.read_value(self._key)

def __check_for_pid_change(self):
            actual_pid = process_identifier()
            if pid['value'] != actual_pid:
                pid['value'] = actual_pid
                # There has been a fork(), reset all the values.
                for f in files.values():
                    f.close()
                files.clear()
                for value in values:
                    value.__reset()

def inc(self, amount):
            with lock:
                self.__check_for_pid_change()
                self._value += amount
                self._file.write_value(self._key, self._value)

def set(self, value):
            with lock:
                self.__check_for_pid_change()
                self._value = value
                self._file.write_value(self._key, self._value)

def get(self):
            with lock:
                self.__check_for_pid_change()
                return self._value

return MmapedValue

def get_value_class():
    # Should we enable multi-process mode?
    # This needs to be chosen before the first metric is constructed,
    # and as that may be in some arbitrary library the user/admin has
    # no control over we use an environment variable.
    if 'prometheus_multiproc_dir' in os.environ:
        return MultiProcessValue()
    else:
        return MutexValue

ValueClass = get_value_class()

Exit & Reset Text mode: syntax highlighting auto-detects file type.

Name	Size	Perms	Modified	Actions
bridge DIR	-	drwxr-xr-x	2026-07-22 02:19:43	Edit Download recursive recursive
openmetrics DIR	-	drwxr-xr-x	2026-07-22 02:19:43	Edit Download recursive recursive
twisted DIR	-	drwxr-xr-x	2026-07-22 02:19:43	Edit Download recursive recursive
__pycache__ DIR	-	drwxr-xr-x	2026-07-22 02:19:43	Edit Download recursive recursive
asgi.py	1.16 KB	lrw-r--r--	2026-06-16 08:49:16	Edit Download recursive recursive
context_managers.py	1.65 KB	lrw-r--r--	2026-06-16 08:49:16	Edit Download recursive recursive
core.py	895 B	lrw-r--r--	2026-06-16 08:49:16	Edit Download recursive recursive
decorator.py	15.43 KB	lrw-r--r--	2026-06-16 08:49:16	Edit Download recursive recursive
exposition.py	14.92 KB	lrw-r--r--	2026-06-16 08:49:16	Edit Download recursive recursive
gc_collector.py	1.42 KB	lrw-r--r--	2026-06-16 08:49:16	Edit Download recursive recursive
metrics.py	21.66 KB	lrw-r--r--	2026-06-16 08:49:16	Edit Download recursive recursive
metrics_core.py	11.60 KB	lrw-r--r--	2026-06-16 08:49:16	Edit Download recursive recursive
mmap_dict.py	5.14 KB	lrw-r--r--	2026-06-16 08:49:16	Edit Download recursive recursive
multiprocess.py	6.32 KB	lrw-r--r--	2026-06-16 08:49:16	Edit Download recursive recursive
parser.py	7.09 KB	lrw-r--r--	2026-06-16 08:49:16	Edit Download recursive recursive
platform_collector.py	1.69 KB	lrw-r--r--	2026-06-16 08:49:16	Edit Download recursive recursive
process_collector.py	3.57 KB	lrw-r--r--	2026-06-16 08:49:16	Edit Download recursive recursive
registry.py	5.23 KB	lrw-r--r--	2026-06-16 08:49:16	Edit Download recursive recursive
samples.py	1.33 KB	lrw-r--r--	2026-06-16 08:49:16	Edit Download recursive recursive
utils.py	621 B	lrw-r--r--	2026-06-16 08:49:16	Edit Download recursive recursive
values.py	3.75 KB	lrw-r--r--	2026-06-16 08:49:16	Edit Download recursive recursive
__init__.py	1.77 KB	lrw-r--r--	2026-06-16 08:49:16	Edit Download recursive recursive

Directory Contents