kotovalexarian-likes-github/ruby--ruby
1
0
Fork 0
mirror of https://github.com/ruby/ruby.git synced 2022-11-09 12:17:21 -05:00
Code Releases Activity

Allow non-finalizable objects in ObjectSpace::WeakMap

Browse source 
[feature #16035]

This goes one step farther than what nobu did in [feature #13498]

With this patch, special objects such as static symbols, integers, etc can be used as either key or values inside WeakMap. They simply don't have a finalizer defined on them.

This is useful if you need to deduplicate value objects
This commit is contained in:
Jean Boussier 2019-08-01 14:41:21 -04:00 committed by Nobuyoshi Nakada
parent e4be2fda3d
commit a4a19b114b
Notes: git 2019-08-31 04:40:14 +09:00 
Merged: https://github.com/ruby/ruby/pull/2313
2 changed files with 30 additions and 21 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
gc.c
View file

@ -2986,18 +2986,12 @@ should_be_callable(VALUE block)
}
static void
should_be_finalizable_internal(VALUE obj)
should_be_finalizable(VALUE obj)
{
if (!FL_ABLE(obj)) {
rb_raise(rb_eArgError, "cannot define finalizer for %s",
rb_obj_classname(obj));
}
}
static void
should_be_finalizable(VALUE obj)
{
should_be_finalizable_internal(obj);
rb_check_frozen(obj);
}
@ -10253,6 +10247,7 @@ wmap_allocate(VALUE klass)
static int
wmap_live_p(rb_objspace_t *objspace, VALUE obj)
{
if (!FL_ABLE(obj)) return TRUE;
if (!is_id_value(objspace, obj)) return FALSE;
if (!is_live_object(objspace, obj)) return FALSE;
return TRUE;
@ -10510,10 +10505,13 @@ wmap_aset(VALUE self, VALUE wmap, VALUE orig)
struct weakmap *w;
TypedData_Get_Struct(self, struct weakmap, &weakmap_type, w);
should_be_finalizable_internal(orig);
should_be_finalizable_internal(wmap);
define_final0(orig, w->final);
define_final0(wmap, w->final);
if (FL_ABLE(orig)) {
define_final0(orig, w->final);
}
if (FL_ABLE(wmap)) {
define_final0(wmap, w->final);
}
st_update(w->obj2wmap, (st_data_t)orig, wmap_aset_update, wmap);
st_insert(w->wmap2obj, (st_data_t)wmap, (st_data_t)orig);
return nonspecial_obj_id(orig);

31
test/ruby/test_weakmap.rb
View file

@ -16,16 +16,27 @@ class TestWeakMap < Test::Unit::TestCase
def test_aset_const
x = Object.new
assert_raise(ArgumentError) {@wm[true] = x}
assert_raise(ArgumentError) {@wm[false] = x}
assert_raise(ArgumentError) {@wm[nil] = x}
assert_raise(ArgumentError) {@wm[42] = x}
assert_raise(ArgumentError) {@wm[:foo] = x}
assert_raise(ArgumentError) {@wm[x] = true}
assert_raise(ArgumentError) {@wm[x] = false}
assert_raise(ArgumentError) {@wm[x] = nil}
assert_raise(ArgumentError) {@wm[x] = 42}
assert_raise(ArgumentError) {@wm[x] = :foo}
@wm[true] = x
assert_same(x, @wm[true])
@wm[false] = x
assert_same(x, @wm[false])
@wm[nil] = x
assert_same(x, @wm[nil])
@wm[42] = x
assert_same(x, @wm[42])
@wm[:foo] = x
assert_same(x, @wm[:foo])
@wm[x] = true
assert_same(true, @wm[x])
@wm[x] = false
assert_same(false, @wm[x])
@wm[x] = nil
assert_same(nil, @wm[x])
@wm[x] = 42
assert_same(42, @wm[x])
@wm[x] = :foo
assert_same(:foo, @wm[x])
end
def assert_weak_include(m, k, n = 100)