XF/neptune
2
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Create sync versions of pathway to ingest messages to work around sync-only Django management commands

Browse Source
This commit is contained in:
Mark Veidemanis
2023-02-09 07:20:45 +00:00
parent 1b1dbbc76c
commit f0455984ef
3 changed files with 57 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
core/lib/rules.py
View File

@@ -275,12 +275,7 @@ class NotificationRuleData(object):
return
async def ingest_matches(self, index, matches, meta, mode):
"""
Store all matches for an index.
:param index: the index to store the matches for
:param matches: the matches to store
"""
def reform_matches(self, index, matches, meta, mode):
if not isinstance(matches, list):
matches = [matches]
matches_copy = matches.copy()
@@ -291,7 +286,25 @@ class NotificationRuleData(object):
matches_copy[match_index]["meta"] = meta
matches_copy[match_index]["match_ts"] = match_ts
matches_copy[match_index]["mode"] = mode
await self.db.async_store_matches(matches_copy)
return matches_copy
async def ingest_matches(self, index, matches, meta, mode):
"""
Store all matches for an index.
:param index: the index to store the matches for
:param matches: the matches to store
"""
new_matches = self.reform_matches(index, matches, meta, mode)
await self.db.async_store_matches(new_matches)
def ingest_matches_sync(self, index, matches, meta, mode):
"""
Store all matches for an index.
:param index: the index to store the matches for
:param matches: the matches to store
"""
new_matches = self.reform_matches(index, matches, meta, mode)
self.db.store_matches(new_matches)
async def rule_matched(self, index, message, meta, mode):
"""
@@ -312,6 +325,25 @@ class NotificationRuleData(object):
self.store_match(index, message)
await self.ingest_matches(index, message, meta, mode)
def rule_matched_sync(self, index, message, meta, mode):
"""
A rule has matched.
If the previous run did not match, send a notification after formatting
the aggregations.
:param index: the index the rule matched on
:param message: the message object that matched
:param aggs: the aggregations that matched
"""
current_match = self.get_match(index, message)
log.debug(f"Rule matched: {index} - current match: {current_match}")
if current_match is False:
# Matched now, but not before
if "matched" not in meta:
meta["matched"] = self.format_aggs(meta["aggs"])
rule_notify(self.object, index, message, meta)
self.store_match(index, message)
self.ingest_matches_sync(index, message, meta, mode)
async def rule_no_match(self, index=None):
"""
A rule has not matched.