feat(ume): keyword key-alert rules with label and case-insensitive match

Add description keyword matching alongside notificationId rules, require
per-rule labels, and improve the AI monitor form layout and clear-on-push UX.

Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
oliver 2026-06-22 16:48:01 +08:00
parent 99e7b1557f
commit 34db62c8d2
11 changed files with 449 additions and 134 deletions

View file

@ -1,5 +1,6 @@
from __future__ import annotations
import re
import threading
import time
from typing import Any
@ -7,13 +8,72 @@ from typing import Any
from sqlalchemy.orm import Session
from .models import UmeKeyAlertRule
from .ume_sync_service import _is_alarm_cleared, notification_id_from_norm
from .ume_sync_service import _is_alarm_cleared, _pick, _s, notification_id_from_norm
_RULE_CACHE_LOCK = threading.Lock()
_RULE_CACHE: dict[str, UmeKeyAlertRule] = {}
_RULE_CACHE: list[UmeKeyAlertRule] = []
_RULE_CACHE_LOADED_AT = 0.0
_RULE_CACHE_TTL_S = 30.0
_ITEM_SEP_RE = re.compile(r"[,,;;|\n]+")
def _fold(text: str) -> str:
"""Case-insensitive compare key for keyword matching."""
return str(text or "").strip().casefold()
def parse_rule_items(text: str) -> list[str]:
"""Split batch rule input (comma/semicolon/newline separated)."""
raw = str(text or "").strip()
if not raw:
return []
parts = [p.strip() for p in _ITEM_SEP_RE.split(raw) if p.strip()]
seen: set[str] = set()
out: list[str] = []
for part in parts:
key = _fold(part)
if key in seen:
continue
seen.add(key)
out.append(part)
return out
def normalize_match_type(value: str) -> str:
mt = str(value or "").strip().lower()
if mt in {"keyword", "keywords", "desc", "description", "cause"}:
return "keyword"
return "notification_id"
def rule_storage_key(*, match_type: str, value: str) -> str:
mt = normalize_match_type(match_type)
v = str(value or "").strip()
if not v:
raise ValueError("match value is required")
if mt == "keyword":
return f"kw:{_fold(v)[:120]}"
return v[:128]
def rule_match_value(row: UmeKeyAlertRule) -> str:
mv = str(getattr(row, "match_value", "") or "").strip()
if mv:
return mv
pk = str(row.notification_id or "").strip()
if pk.startswith("kw:"):
return pk[3:]
return pk
def rule_match_type(row: UmeKeyAlertRule) -> str:
mt = str(getattr(row, "match_type", "") or "").strip().lower()
if mt in {"keyword", "notification_id"}:
return mt
pk = str(row.notification_id or "").strip()
return "keyword" if pk.startswith("kw:") else "notification_id"
def invalidate_key_alert_rule_cache() -> None:
global _RULE_CACHE_LOADED_AT
@ -22,23 +82,48 @@ def invalidate_key_alert_rule_cache() -> None:
_RULE_CACHE_LOADED_AT = 0.0
def _load_enabled_rules(db: Session) -> dict[str, UmeKeyAlertRule]:
def _load_enabled_rules(db: Session) -> list[UmeKeyAlertRule]:
global _RULE_CACHE_LOADED_AT
now = time.time()
with _RULE_CACHE_LOCK:
if _RULE_CACHE and (now - _RULE_CACHE_LOADED_AT) < _RULE_CACHE_TTL_S:
return dict(_RULE_CACHE)
return list(_RULE_CACHE)
rows = (
db.query(UmeKeyAlertRule)
.filter(UmeKeyAlertRule.enabled == 1)
.all()
)
loaded = {str(row.notification_id or "").strip(): row for row in rows if str(row.notification_id or "").strip()}
with _RULE_CACHE_LOCK:
_RULE_CACHE.clear()
_RULE_CACHE.update(loaded)
_RULE_CACHE.extend(rows)
_RULE_CACHE_LOADED_AT = now
return dict(loaded)
return list(rows)
def _alarm_search_text(norm: dict[str, Any]) -> str:
parts = [
notification_id_from_norm(norm),
_s(_pick(norm, "nativeProbableCause", "native-probable-cause")),
_s(_pick(norm, "objectName", "object-name")),
_s(_pick(norm, "eventType", "event-type")),
]
return " ".join(p for p in parts if p).casefold()
def _keyword_matches(norm: dict[str, Any], keyword: str) -> bool:
kw = _fold(keyword)
if not kw:
return False
return kw in _alarm_search_text(norm)
def _rule_matches_norm(rule: UmeKeyAlertRule, norm: dict[str, Any]) -> bool:
mt = rule_match_type(rule)
mv = rule_match_value(rule)
if mt == "keyword":
return _keyword_matches(norm, mv)
nid = notification_id_from_norm(norm)
return bool(nid) and nid == mv
def match_key_alert_rule(
@ -47,19 +132,15 @@ def match_key_alert_rule(
norm: dict[str, Any],
action: str,
) -> UmeKeyAlertRule | None:
notification_id = notification_id_from_norm(norm)
if not notification_id:
return None
rules = _load_enabled_rules(db)
rule = rules.get(notification_id)
if rule is None:
return None
act = str(action or "").strip().lower()
if act in {"inserted", "updated"}:
if _is_alarm_cleared(norm):
return None
return rule
if act == "deleted":
if int(getattr(rule, "forward_on_clear", 0) or 0) == 1:
for rule in _load_enabled_rules(db):
if not _rule_matches_norm(rule, norm):
continue
if act in {"inserted", "updated"}:
if _is_alarm_cleared(norm):
continue
return rule
if act == "deleted":
if int(getattr(rule, "forward_on_clear", 0) or 0) == 1:
return rule
return None