technicie/MarietjeDjango
1
0
Fork 0
mirror of https://gitlab.science.ru.nl/technicie/MarietjeDjango.git synced 2025-12-09 18:52:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

metrics: Add queue length in seconds

Browse Source
This commit is contained in:
Daan Sprenkels
2018-04-09 19:19:22 +02:00
parent 126372a758
commit 644e654fef
1 changed files with 16 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
marietje/metrics/views.py
View File

@ -6,16 +6,25 @@ from queues.models import Queue, PlaylistSong
# Export queue length
queue_length_gauge = Gauge('marietje_queue_length', 'Marietje queue length', ['name'])
# Export the queue duration
queue_duration_gauge = Gauge('marietje_queue_duration', 'Marietje queue duration in seconds', ['name'])
# Export the amount of different users in the queue
queue_users_gauge = Gauge('marietje_queue_users', 'Users holding a queue at some point', ['name'])
for queue in Queue.objects.all():
def _get_queue_length():
return (PlaylistSong.objects
.filter(Q(playlist=queue.playlist_id), Q(state=0) | Q(state=1))
.count())
queue_length_gauge.labels(name=queue).set_function(_get_queue_length)
# Export the amount of different users in the queue
queue_users_gauge = Gauge('marietje_queue_users', 'Users holding a queue at some point', ['name'])
for queue in Queue.objects.all():
def _get_queue_duration():
playlist_songs = (PlaylistSong.objects
.filter(Q(playlist=queue.playlist_id), Q(state=0) | Q(state=1))
.select_related('song')
.all())
return sum(ps.song.duration for ps in playlist_songs)
def _get_queue_distinct_users():
# Ideally, we would use .distinct() here, but it is not supported on
# sqlite3. This makes testing a bit tedious. Also, it does not give me
@ -24,6 +33,9 @@ for queue in Queue.objects.all():
return len(set(PlaylistSong.objects
.filter(Q(playlist=queue.playlist_id), Q(state=0) | Q(state=1))
.values_list('user')))
queue_length_gauge.labels(name=queue).set_function(_get_queue_length)
queue_duration_gauge.labels(name=queue).set_function(_get_queue_duration)
queue_users_gauge.labels(name=queue).set_function(_get_queue_distinct_users)