Merge pull request #1587 from codeflash-ai/codeflash/optimize-pr1199-2026-02-20T06.40.01

⚡️ Speed up function `_should_include_method` by 22% in PR #1199 (`omni-java`)

Author: claude[bot]

codeflash/languages/base.py

@@ -7,6 +7,8 @@
 
 from __future__ import annotations
 
+import fnmatch
+import re
 from dataclasses import dataclass, field
 from typing import TYPE_CHECKING, Any, Protocol, runtime_checkable
 
@@ -172,6 +174,23 @@ class FunctionFilterCriteria:
     min_lines: int | None = None
     max_lines: int | None = None
 
+    def __post_init__(self) -> None:
+        """Pre-compile regex patterns from glob patterns for faster matching."""
+        self._include_regexes = [re.compile(fnmatch.translate(p)) for p in self.include_patterns]
+        self._exclude_regexes = [re.compile(fnmatch.translate(p)) for p in self.exclude_patterns]
+
+    def matches_include_patterns(self, name: str) -> bool:
+        """Check if name matches any include pattern."""
+        if not self._include_regexes:
+            return True
+        return any(regex.match(name) for regex in self._include_regexes)
+
+    def matches_exclude_patterns(self, name: str) -> bool:
+        """Check if name matches any exclude pattern."""
+        if not self._exclude_regexes:
+            return False
+        return any(regex.match(name) for regex in self._exclude_regexes)
+
 
 @dataclass
 class ReferenceInfo:

codeflash/languages/java/discovery.py

@@ -16,6 +16,8 @@
 from codeflash.models.function_types import FunctionParent
 
 if TYPE_CHECKING:
+    from tree_sitter import Node
+
     from codeflash.languages.java.parser import JavaAnalyzer, JavaMethodNode
 
 logger = logging.getLogger(__name__)
@@ -136,18 +138,14 @@ def _should_include_method(
         return False
 
     # Check include patterns
-    if criteria.include_patterns:
-        import fnmatch
-
-        if not any(fnmatch.fnmatch(method.name, pattern) for pattern in criteria.include_patterns):
-            return False
+    if not criteria.matches_include_patterns(method.name):
+        return False
 
     # Check exclude patterns
-    if criteria.exclude_patterns:
-        import fnmatch
+    if criteria.matches_exclude_patterns(method.name):
+        return False
 
-        if any(fnmatch.fnmatch(method.name, pattern) for pattern in criteria.exclude_patterns):
-            return False
+    # Check require_return - void methods don't have return values
 
     # Check require_return - void methods don't have return values
     if criteria.require_return:
@@ -203,7 +201,7 @@ def discover_test_methods(file_path: Path, analyzer: JavaAnalyzer | None = None)
 
 
 def _walk_tree_for_test_methods(
-    node,
+    node: Node,
     source_bytes: bytes,
     file_path: Path,
     test_methods: list[FunctionToOptimize],