ensure primitive map keys are promoted to the widest possible type to avoid

problems with equals

Author: Nicolas Laurent

src/main/java/org/truffleruby/core/objectspace/WeakMapNodes.java

@@ -55,6 +55,16 @@ protected int size(RubyWeakMap map) {
     @CoreMethod(names = { "key?", "member?", "include?" }, required = 1)
     public abstract static class MemberNode extends CoreMethodArrayArgumentsNode {
 
+        @Specialization // ensures all integer keys are promoted to long
+        protected Object isMember(RubyWeakMap map, long key) {
+            return isMember(map, (Object) key);
+        }
+
+        @Specialization // ensures all floating keys are promoted to double
+        protected Object isMember(RubyWeakMap map, double key) {
+            return isMember(map, (Object) key);
+        }
+
         @Specialization
         protected boolean isMember(RubyWeakMap map, Object key) {
             return map.storage.get(key) != null;
@@ -64,6 +74,16 @@ protected boolean isMember(RubyWeakMap map, Object key) {
     @CoreMethod(names = "[]", required = 1)
     public abstract static class GetIndexNode extends CoreMethodArrayArgumentsNode {
 
+        @Specialization // ensures all integer keys are promoted to long
+        protected Object get(RubyWeakMap map, long key) {
+            return get(map, (Object) key);
+        }
+
+        @Specialization // ensures all floating keys are promoted to double
+        protected Object get(RubyWeakMap map, double key) {
+            return get(map, (Object) key);
+        }
+
         @Specialization
         protected Object get(RubyWeakMap map, Object key) {
             Object value = map.storage.get(key);
@@ -74,6 +94,16 @@ protected Object get(RubyWeakMap map, Object key) {
     @Primitive(name = "weakmap_aset")
     public abstract static class SetIndexNode extends CoreMethodArrayArgumentsNode {
 
+        @Specialization // ensures all integer keys are promoted to long
+        protected Object set(RubyWeakMap map, long key, Object value) {
+            return set(map, (Object) key, value);
+        }
+
+        @Specialization // ensures all floating keys are promoted to double
+        protected Object set(RubyWeakMap map, double key, Object value) {
+            return set(map, (Object) key, value);
+        }
+
         @Specialization
         protected Object set(RubyWeakMap map, Object key, Object value) {
             map.storage.put(key, value);