ZhAgent/apps/application/flow/compare/regex_compare.py

# coding=utf-8
"""
    @project: maxkb
    @Author：wangliang181230
    @file： regex_compare.py
    @date：2026/3/30 12:11
    @desc:
"""
import re
from typing import List

from application.flow.compare import Compare
from common.cache.mem_cache import MemCache


match_cache = MemCache('regex', {
    'TIMEOUT': 3600, # 缓存有效期为 1 小时
    'OPTIONS': {
        'MAX_ENTRIES': 500, # 最多缓存 500 个条目
        'CULL_FREQUENCY': 10, # 达到上限时，删除约 1/10 的缓存
    },
})


def compile_and_cache(regex):
    match = match_cache.get(regex)
    if not match:
        match = re.compile(regex).match
        match_cache.set(regex, match)
    return match

class RegexCompare(Compare):

    def support(self, node_id, fields: List[str], source_value, compare, target_value):
        if compare == 'regex':
            return True

    def compare(self, source_value, compare, target_value):
        match = compile_and_cache(str(target_value))
        return bool(match(str(source_value)))