Add more performance optimizations + benchmarks

Author: eliotwrobson

src/mystace/mustache_tree.py

@@ -57,12 +57,12 @@ def get(self, key: str) -> t.Any:
         if key == ".":
             return self.context
 
-        # Split only once and cache
-        keys = key.split(".")
+        # Use iterator to avoid copying keys
+        chain_iter = iter(key.split("."))
         curr_ctx = self.context
         parent_node = self.parent_context_node
 
-        first_key = keys[0]
+        first_key = next(chain_iter)
 
         # TODO I think this is where changes need to be made if we want to
         # support indexing and lambdas.
@@ -83,7 +83,7 @@ def get(self, key: str) -> t.Any:
             return None
 
         # Loop through the rest
-        for key in keys[1:]:
+        for key in chain_iter:
             if isinstance(outer_context, list):
                 try:
                     int_key = int(key)

tests/test_speed.py

@@ -39,6 +39,24 @@ def generate_test_case_long(n: int) -> TestCaseT:
     return template, names
 
 
+def generate_test_case_simple_variables(n: int) -> TestCaseT:
+    """Simple test case with just variable substitutions."""
+    template = " ".join(f"{{{{var{i}}}}}" for i in range(n))
+    data = {f"var{i}": f"value{i}" for i in range(n)}
+    return template, data
+
+
+def generate_test_case_simple_sections(n: int) -> TestCaseT:
+    """Simple test case with basic sections."""
+    template = ""
+    data = {}
+    for i in range(n):
+        section_name = f"section{i}"
+        template += f"{{{{#{section_name}}}}}content{{{{/{section_name}}}}}"
+        data[section_name] = True
+    return template, data
+
+
 def generate_test_case_nested(n: int) -> TestCaseT:
     fake = Faker()
 
@@ -333,6 +351,8 @@ def render_function(_, obj):
     elif render_function_name == "mstache":
         import mstache
 
+        # mstache.render() uses a default LRU cache for tokenization
+        # so repeated calls benefit from caching automatically
         render_function = mstache.render
     else:
         assert_never(render_function_name)
@@ -364,6 +384,8 @@ def render_function(_, obj):
         generate_test_case_nested,
         generate_test_case_long,
         generate_test_case_random(1),
+        generate_test_case_simple_variables,
+        generate_test_case_simple_sections,
     ],
 )
 @pytest.mark.parametrize("render_function_name", t.get_args(RenderFunctionT))