[Truffle] Add Hash#{<,<=,>,>=}

jruby · Jun 28, 2016 · 81fdf12 · pitr-ch · Jun 29, 2016 · eregon
1 parent c83120b
commit 81fdf12
Showing 5 changed files with 35 additions and 40 deletions.
diff --git a/spec/truffle/tags/core/hash/gt_tags.txt b/spec/truffle/tags/core/hash/gt_tags.txt
diff --git a/spec/truffle/tags/core/hash/gte_tags.txt b/spec/truffle/tags/core/hash/gte_tags.txt
diff --git a/spec/truffle/tags/core/hash/lt_tags.txt b/spec/truffle/tags/core/hash/lt_tags.txt
diff --git a/spec/truffle/tags/core/hash/lte_tags.txt b/spec/truffle/tags/core/hash/lte_tags.txt
diff --git a/truffle/src/main/ruby/core/hash.rb b/truffle/src/main/ruby/core/hash.rb
@@ -36,6 +36,16 @@
 class Hash
   include Enumerable
 
+  def self.contains_all_internal(one, two)
+    one.all? do |key, value|
+      begin
+        value === two.fetch(key)
+      rescue KeyError
+        false
+      end
+    end
+  end
+
   def self.new_from_associate_array(associate_array)
     hash = new
     associate_array.each do |array|
@@ -95,6 +105,19 @@ def self._constructor_fallback(*args)
   # Used internally to get around subclasses redefining #[]=
   alias_method :__store__, :[]=
 
+  def <(other)
+     other = Rubinius::Type.coerce_to(other, Hash, :to_hash)
+     return false if self.size >= other.size
+     self.class.contains_all_internal(self, other)
+  end
+
+  def <=(other)
+    other = Rubinius::Type.coerce_to(other, Hash, :to_hash)
+    return false if self.size > other.size
+    self.class.contains_all_internal(self, other)
+  end
+
+
   def ==(other)
     return true if self.equal? other
     unless other.kind_of? Hash
@@ -122,6 +145,18 @@ def ==(other)
     true
   end
 
+  def >(other)
+    other = Rubinius::Type.coerce_to(other, Hash, :to_hash)
+    return false if self.size <= other.size
+    self.class.contains_all_internal(other, self)
+  end
+
+  def >=(other)
+    other = Rubinius::Type.coerce_to(other, Hash, :to_hash)
+    return false if self.size < other.size
+    self.class.contains_all_internal(other, self)
+  end
+
   def assoc(key)
     each_item { |e| return e.key, e.value if key == e.key }
   end