NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-08-08 08:38:16 -06:00
Code Issues Actions 15 Packages Projects Releases Wiki Activity

Simplify event queue handling logic

Browse Source
This commit is contained in:
Jeremy Stretch 2024-07-18 16:41:23 -04:00
parent b4e8a3101a
commit 00dc9c464b
1 changed files with 6 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
netbox/extras/events.py
View File

@ -1,3 +1,4 @@
from collections import defaultdict
import logging import logging
from django.conf import settings from django.conf import settings
@ -152,35 +153,20 @@ def process_event_queue(events):
""" """
Flush a list of object representation to RQ for EventRule processing. Flush a list of object representation to RQ for EventRule processing.
""" """
events_cache = { events_cache = defaultdict(dict)
'type_create': {},
'type_update': {},
'type_delete': {},
}
event_actions = {
# TODO: Add EventRule support for dynamically registered event types
OBJECT_CREATED: 'type_create',
OBJECT_UPDATED: 'type_update',
OBJECT_DELETED: 'type_delete',
JOB_STARTED: 'type_job_start',
JOB_COMPLETED: 'type_job_end',
# Map failed & errored jobs to type_job_end
JOB_FAILED: 'type_job_end',
JOB_ERRORED: 'type_job_end',
}
for event in events: for event in events:
action_flag = event_actions[event['event_type']] event_type = event['event_type']
object_type = event['object_type'] object_type = event['object_type']
# Cache applicable Event Rules # Cache applicable Event Rules
if object_type not in events_cache[action_flag]: if object_type not in events_cache[event_type]:
events_cache[action_flag][object_type] = EventRule.objects.filter( events_cache[event_type][object_type] = EventRule.objects.filter(
event_types__contains=[event['event_type']], event_types__contains=[event['event_type']],
object_types=object_type, object_types=object_type,
enabled=True enabled=True
) )
event_rules = events_cache[action_flag][object_type] event_rules = events_cache[event_type][object_type]
process_event_rules( process_event_rules(
event_rules=event_rules, event_rules=event_rules,