Skip to content

Refactoring and metric wrapper unification #313

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
akx wants to merge 10 commits into from
Closed

Refactoring and metric wrapper unification #313

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
53edebd
Add coverage's htmlcov directory and pytest's caches to .gitignore
akx Sep 28, 2018
7d794ec
Move mmap stuff out of core.py
akx Sep 28, 2018
f5aef48
Move context managers to a file of their own
akx Sep 28, 2018
3314818
Move low-level Metric/Family objects to metrics.py (keep re-exports i…
akx Sep 28, 2018
38bb999
Move CollectorRegistry to its own module
akx Sep 28, 2018
bdbb1be
Move _ValueClass selection out of core.py
akx Sep 28, 2018
cd3d0f4
Move _floatToGoString to utils.py
akx Sep 28, 2018
fc2af36
Move metric wrappers to a module of their own (with some other change…
akx Sep 28, 2018
50b6d3c
Refactor MetricWrapper magic to class hierarchy
akx Sep 28, 2018
0194bdc
Fix flake8 kvetches with autopep8
akx Sep 29, 2018
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 5 additions & 3 deletions .gitignore
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,8 +1,10 @@
build
dist
*.egg-info
*.pyc
*.swp
.coverage.*
.*cache
.coverage
.coverage.*
.tox
build
dist
htmlcov
63 changes: 63 additions & 0 deletions prometheus_client/context_managers.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,63 @@
from timeit import default_timer

from .decorator import decorate


class _ExceptionCounter(object):
def __init__(self, counter, exception):
self._counter = counter
self._exception = exception

def __enter__(self):
pass

def __exit__(self, typ, value, traceback):
if isinstance(value, self._exception):
self._counter.inc()

def __call__(self, f):
def wrapped(func, *args, **kwargs):
with self:
return func(*args, **kwargs)
return decorate(f, wrapped)


class _InprogressTracker(object):
def __init__(self, gauge):
self._gauge = gauge

def __enter__(self):
self._gauge.inc()

def __exit__(self, typ, value, traceback):
self._gauge.dec()

def __call__(self, f):
def wrapped(func, *args, **kwargs):
with self:
return func(*args, **kwargs)
return decorate(f, wrapped)


class _Timer(object):
def __init__(self, callback):
self._callback = callback

def _new_timer(self):
return self.__class__(self._callback)

def __enter__(self):
self._start = default_timer()

def __exit__(self, typ, value, traceback):
# Time can go backwards.
duration = max(default_timer() - self._start, 0)
self._callback(duration)

def __call__(self, f):
def wrapped(func, *args, **kwargs):
# Obtaining new instance of timer every time
# ensures thread safety and reentrancy.
with self._new_timer():
return func(*args, **kwargs)
return decorate(f, wrapped)
Loading