mirror of
https://github.com/quay/quay.git
synced 2026-01-26 06:21:37 +03:00
5f63b3a7bb
* chore: drop deprecated tables and remove unused code * isort imports * migration: check for table existence before drop
66 lines
1.7 KiB
Python
66 lines
1.7 KiB
Python
from time import time
|
|
|
|
from gevent.hub import get_hub
|
|
from greenlet import settrace
|
|
from prometheus_client import Counter, Histogram
|
|
|
|
greenlet_switch = Counter("greenlet_switch_total", "number of greenlet context switches")
|
|
greenlet_throw = Counter("greenlet_throw_total", "number of greenlet throws")
|
|
greenlet_duration = Histogram(
|
|
"greenlet_duration_seconds",
|
|
"seconds in which a particular greenlet is executing",
|
|
)
|
|
|
|
_latest_switch = None
|
|
|
|
|
|
def enable_tracing():
|
|
settrace(greenlet_callback)
|
|
|
|
|
|
def greenlet_callback(event, args):
|
|
"""
|
|
This is a callback that is executed greenlet on all events.
|
|
"""
|
|
if event in ("switch", "throw"):
|
|
# It's only safe to unpack args under these two events.
|
|
(origin, _target) = args
|
|
|
|
if origin is get_hub():
|
|
# This greenlet is the one that manages the loop itself, thus noop.
|
|
return
|
|
|
|
if event == "switch":
|
|
switch_callback(args)
|
|
return
|
|
if event == "throw":
|
|
throw_callback(args)
|
|
return
|
|
|
|
|
|
def switch_callback(_args):
|
|
"""
|
|
This is a callback that is executed specifically on greenlet switches.
|
|
"""
|
|
global _latest_switch
|
|
greenlet_switch.inc()
|
|
|
|
if _latest_switch is None:
|
|
# This is the first switch.
|
|
_latest_switch = time()
|
|
return
|
|
|
|
now = time()
|
|
greenlet_duration.observe(now - _latest_switch)
|
|
_latest_switch = now
|
|
|
|
|
|
def throw_callback(_args):
|
|
"""
|
|
This is a callback that is executed on execeptions from origin to target.
|
|
|
|
This callback is running in the context of the target greenlet and any exceptions will replace
|
|
the original, as if target.throw() was used replacing the exception.
|
|
"""
|
|
greenlet_throw.inc()
|