Memoize #find_all

nanoc · Apr 1, 2017 · 9e33620 · 9e33620
1 parent 698eb90
commit 9e33620
Showing 1 changed file with 17 additions and 2 deletions.
diff --git a/lib/nanoc/base/entities/identifiable_collection.rb b/lib/nanoc/base/entities/identifiable_collection.rb
@@ -39,8 +39,11 @@ def [](arg)
 
     contract C::Any => C::IterOf[C::RespondTo[:identifier]]
     def find_all(arg)
-      pat = Nanoc::Int::Pattern.from(arg)
-      select { |i| pat.match?(i.identifier) }
+      if frozen?
+        find_all_memoized(arg)
+      else
+        find_all_unmemoized(arg)
+      end
     end
 
     contract C::None => C::ArrayOf[C::RespondTo[:identifier]]
@@ -83,6 +86,18 @@ def get_memoized(arg)
     end
     memoize :get_memoized
 
+    contract C::Any => C::IterOf[C::RespondTo[:identifier]]
+    def find_all_unmemoized(arg)
+      pat = Nanoc::Int::Pattern.from(arg)
+      select { |i| pat.match?(i.identifier) }
+    end
+
+    contract C::Any => C::IterOf[C::RespondTo[:identifier]]
+    def find_all_memoized(arg)
+      find_all_unmemoized(arg)
+    end
+    memoize :find_all_memoized
+
     def object_with_identifier(identifier)
       if frozen?
         @mapping[identifier.to_s]