Cache SafeRe patterns

yupengyan · Jul 14, 2017 · 0e930ef · 0e930ef
1 parent d281f11
commit 0e930ef
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/src/util/SafeRe.py b/src/util/SafeRe.py
@@ -4,6 +4,8 @@
 class UnsafePatternError(Exception):
     pass
 
+cached_patterns = {}
+
 
 def isSafePattern(pattern):
     if len(pattern) > 255:
@@ -16,5 +18,10 @@ def isSafePattern(pattern):
 
 
 def match(pattern, *args, **kwargs):
-    if isSafePattern(pattern):
-        return re.match(pattern, *args, **kwargs)
+    cached_pattern = cached_patterns.get(pattern)
+    if cached_pattern:
+        return cached_pattern.match(*args, **kwargs)
+    else:
+        if isSafePattern(pattern):
+            cached_patterns[pattern] = re.compile(pattern)
+            return cached_patterns[pattern].match(*args, **kwargs)