From 1bc32adad4a6980ebd592948e0981e448af5253b Mon Sep 17 00:00:00 2001
From: Alex Willmer <alex@moreati.org.uk>
Date: Sat, 7 Apr 2018 14:34:41 +0100
Subject: [PATCH] Issue #160: Cache minimize_source()

Before

```
python tests/bench/local.py
0 120.245933533
1 119.300842285
2 118.355035782
3 123.839855194
```

After

```
python tests/bench/local.py
0 66.3640499115
1 65.9508705139
2 72.4799633026
3 65.7958984375
```
---
 mitogen/parent.py           | 8 ++++++++
 tests/module_finder_test.py | 3 +++
 2 files changed, 11 insertions(+)

diff --git a/mitogen/parent.py b/mitogen/parent.py
index 40f808c5..4ec2ce5c 100644
--- a/mitogen/parent.py
+++ b/mitogen/parent.py
@@ -44,6 +44,11 @@ import time
 import types
 import zlib
 
+try:
+    from functools import lru_cache
+except ImportError:
+    from mitogen.compat.functools import lru_cache
+
 import mitogen.core
 from mitogen.core import LOG
 from mitogen.core import IOLOG
@@ -84,6 +89,7 @@ def is_immediate_child(msg, stream):
     return msg.src_id == stream.remote_id
 
 
+@lru_cache()
 def minimize_source(source):
     """Remove most comments and docstrings from Python source code.
     """
@@ -152,6 +158,8 @@ def strip_docstrings(tokens):
 
 
 def reindent(tokens, indent=' '):
+    """Replace existing indentation in a token steam, with `indent`.
+    """
     old_levels = []
     old_level = 0
     new_level = 0
diff --git a/tests/module_finder_test.py b/tests/module_finder_test.py
index 0affa27d..41ebd9ba 100644
--- a/tests/module_finder_test.py
+++ b/tests/module_finder_test.py
@@ -198,6 +198,9 @@ class FindRelatedTest(testlib.TestCase):
         related = self.call('mitogen.fakessh')
         self.assertEquals(related, [
             'mitogen',
+            'mitogen.compat',
+            'mitogen.compat.collections',
+            'mitogen.compat.functools',
             'mitogen.core',
             'mitogen.master',
             'mitogen.parent',