Caching Type obj

gogotanaka · gogotanaka · commit eba6ccc7eb15 · 2014-12-03T11:38:08.000-08:00
diff --git a/lib/java.rb b/lib/java.rb
@@ -51,18 +51,31 @@ module Boolean; end
 FalseClass.send(:include, Boolean)
 
 class Type
+  @@types = []
+
   attr_accessor :sym, :klass, :condition
 
   def initialize(sym, klass, &condition)
     @sym, @klass, @condition = sym, klass, condition
+    @@types << self
   end
 
-  def self.define_new(sym, klass, &condition)
-    Module.class_eval do
-      define_method(sym) do |meth|
-        define_typed_method(meth, Type.new(sym, klass, &condition))
+  class << self
+    def define_new(sym, klass, &condition)
+      Module.class_eval do
+        define_method(sym) do |meth|
+          define_typed_method(meth, Type.find_or_create(sym, klass, &condition))
+        end
+        private sym
       end
-      private sym
+    end
+
+    def find_type(sym)
+      @@types.find { |type| type.sym == sym }
+    end
+
+    def find_or_create(sym, klass, &condition)
+      find_type(sym) || new(sym, klass, &condition)
     end
   end
 end
