memoist 0.1.0 → 0.2.0

data/lib/memoist.rb

@@ -23,36 +23,41 @@ module Memoist
     end
 
     def memoize_all
-      prime_cache ".*"
+      prime_cache
     end
 
     def unmemoize_all
-      flush_cache ".*"
+      flush_cache
     end
 
     def prime_cache(*syms)
+      if syms.empty?
+        syms = methods.collect do |m|
+          m[12..-1] if m.to_s.start_with?("_unmemoized_")
+        end.compact
+      end
+
       syms.each do |sym|
-        methods.each do |m|
-          if m.to_s =~ /^_unmemoized_(#{sym})/
-            if method(m).arity == 0
-              __send__($1)
-            else
-              ivar = Memoist.memoized_ivar_for($1)
-              instance_variable_set(ivar, {})
-            end
-          end
+        m = method(:"_unmemoized_#{sym}") rescue next
+        if m.arity == 0
+          __send__(sym)
+        else
+          ivar = Memoist.memoized_ivar_for(sym)
+          instance_variable_set(ivar, {})
         end
       end
     end
 
     def flush_cache(*syms)
+      if syms.empty?
+        syms = (methods + private_methods + protected_methods).collect do |m|
+          m[12..-1] if m.to_s.start_with?("_unmemoized_")
+        end.compact
+      end
+
       syms.each do |sym|
-        (methods + private_methods + protected_methods).each do |m|
-          if m.to_s =~ /^_unmemoized_(#{sym.to_s.gsub(/\?\Z/, '\?')})/
-            ivar = Memoist.memoized_ivar_for($1)
-            instance_variable_get(ivar).clear if instance_variable_defined?(ivar)
-          end
-        end
+        ivar = Memoist.memoized_ivar_for(sym)
+        instance_variable_get(ivar).clear if instance_variable_defined?(ivar)
       end
     end
   end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: memoist
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
   prerelease: 
 platform: ruby
 authors:
@@ -19,7 +19,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-01-24 00:00:00.000000000 Z
+date: 2012-08-15 00:00:00.000000000 Z
 dependencies: []
 description: 
 email:
@@ -66,7 +66,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 1.8.10
+rubygems_version: 1.8.24
 signing_key: 
 specification_version: 3
 summary: memoize methods invocation