OXIESEC PANEL

Name	Size	Modified	Perms
📁 ..	-	09/05/2025 09:34:06 AM	rwxr-xr-x
📄 __init__.py	1.77 KB	12/18/2024 10:23:15 AM	rw-r--r--
📁 __pycache__	-	02/07/2025 10:01:34 PM	rwxr-xr-x
📄 asgi.py	1.16 KB	12/18/2024 10:23:15 AM	rw-r--r--
📁 bridge	-	02/07/2025 10:01:34 PM	rwxr-xr-x
📄 context_managers.py	1.65 KB	12/18/2024 10:23:15 AM	rw-r--r--
📄 core.py	895 bytes	12/18/2024 10:23:15 AM	rw-r--r--
📄 decorator.py	15.43 KB	12/18/2024 10:23:15 AM	rw-r--r--
📄 exposition.py	14.92 KB	12/18/2024 10:23:15 AM	rw-r--r--
📄 gc_collector.py	1.42 KB	12/18/2024 10:23:15 AM	rw-r--r--
📄 metrics.py	21.66 KB	12/18/2024 10:23:15 AM	rw-r--r--
📄 metrics_core.py	11.6 KB	12/18/2024 10:23:15 AM	rw-r--r--
📄 mmap_dict.py	5.14 KB	12/18/2024 10:23:15 AM	rw-r--r--
📄 multiprocess.py	6.32 KB	12/18/2024 10:23:15 AM	rw-r--r--
📁 openmetrics	-	02/07/2025 10:01:34 PM	rwxr-xr-x
📄 parser.py	7.09 KB	12/18/2024 10:23:15 AM	rw-r--r--
📄 platform_collector.py	1.69 KB	12/18/2024 10:23:15 AM	rw-r--r--
📄 process_collector.py	3.57 KB	12/18/2024 10:23:15 AM	rw-r--r--
📄 registry.py	5.23 KB	12/18/2024 10:23:15 AM	rw-r--r--
📄 samples.py	1.33 KB	12/18/2024 10:23:15 AM	rw-r--r--
📁 twisted	-	02/07/2025 10:01:34 PM	rwxr-xr-x
📄 utils.py	621 bytes	12/18/2024 10:23:15 AM	rw-r--r--
📄 values.py	3.75 KB	12/18/2024 10:23:15 AM	rw-r--r--

Editing: values.py

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()