Compare commits
No commits in common. "7208aede8d1a99decab7fa151509d9df06238e2d" and "a2f7682c6fbb09c7828d0ebc77a48416aa1887df" have entirely different histories.
7208aede8d
...
a2f7682c6f
1 changed files with 0 additions and 24 deletions
|
@ -2,8 +2,6 @@ from django.apps import apps
|
||||||
from prometheus_client.core import CounterMetricFamily, REGISTRY
|
from prometheus_client.core import CounterMetricFamily, REGISTRY
|
||||||
from django.db.models import Case, Value, When, BooleanField, Count
|
from django.db.models import Case, Value, When, BooleanField, Count
|
||||||
from inventory.models import Item
|
from inventory.models import Item
|
||||||
from tickets.models import IssueThread
|
|
||||||
|
|
||||||
|
|
||||||
class ItemCountCollector(object):
|
class ItemCountCollector(object):
|
||||||
|
|
||||||
|
@ -34,29 +32,7 @@ class ItemCountCollector(object):
|
||||||
|
|
||||||
yield counter
|
yield counter
|
||||||
|
|
||||||
class TicketCountCollector(object):
|
|
||||||
|
|
||||||
def collect(self):
|
|
||||||
counter = CounterMetricFamily("c3lf_ticket_count", "Current number of tickets", labels=['event', 'event_id'])
|
|
||||||
|
|
||||||
yield counter
|
|
||||||
|
|
||||||
if not apps.models_ready or not apps.apps_ready:
|
|
||||||
return
|
|
||||||
|
|
||||||
queryset = (
|
|
||||||
IssueThread.objects
|
|
||||||
.values('event__slug', 'event_id')
|
|
||||||
.annotate(amount=Count('id'))
|
|
||||||
)
|
|
||||||
|
|
||||||
for e in queryset:
|
|
||||||
counter.add_metric([e["event__slug"].lower()], e["amount"])
|
|
||||||
|
|
||||||
yield counter
|
|
||||||
|
|
||||||
try:
|
try:
|
||||||
REGISTRY.register(ItemCountCollector())
|
REGISTRY.register(ItemCountCollector())
|
||||||
REGISTRY.register(TicketCountCollector())
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
print(e)
|
print(e)
|
||||||
|
|
Loading…
Add table
Reference in a new issue