diff options
author | ko1 <ko1@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2015-04-10 10:13:00 +0000 |
---|---|---|
committer | ko1 <ko1@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2015-04-10 10:13:00 +0000 |
commit | 1a688f001c5f7ba76cba57e6433977c69f4ad620 (patch) | |
tree | bbbc3564a93531367a2ec870a3b701986eb64026 /ext/objspace | |
parent | ff381ca19bde87981e2c53cab618ae1e771869bf (diff) | |
download | ruby-1a688f001c5f7ba76cba57e6433977c69f4ad620.tar.gz |
* ext/objspace/objspace.c (setup_hash): unify common routine.
git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@50208 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
Diffstat (limited to 'ext/objspace')
-rw-r--r-- | ext/objspace/objspace.c | 50 |
1 files changed, 23 insertions, 27 deletions
diff --git a/ext/objspace/objspace.c b/ext/objspace/objspace.c index ab3240b057..ec2840feb5 100644 --- a/ext/objspace/objspace.c +++ b/ext/objspace/objspace.c @@ -125,6 +125,26 @@ set_zero_i(st_data_t key, st_data_t val, st_data_t arg) return ST_CONTINUE; } +static VALUE +setup_hash(int argc, VALUE *argv) +{ + VALUE hash; + + if (rb_scan_args(argc, argv, "01", &hash) == 1) { + if (!RB_TYPE_P(hash, T_HASH)) + rb_raise(rb_eTypeError, "non-hash given"); + } + + if (hash == Qnil) { + hash = rb_hash_new(); + } + else if (!RHASH_EMPTY_P(hash)) { + st_foreach(RHASH_TBL(hash), set_zero_i, hash); + } + + return hash; +} + static int cos_i(void *vstart, void *vend, size_t stride, void *data) { @@ -206,12 +226,7 @@ count_objects_size(int argc, VALUE *argv, VALUE os) size_t counts[T_MASK+1]; size_t total = 0; enum ruby_value_type i; - VALUE hash; - - if (rb_scan_args(argc, argv, "01", &hash) == 1) { - if (!RB_TYPE_P(hash, T_HASH)) - rb_raise(rb_eTypeError, "non-hash given"); - } + VALUE hash = setup_hash(argc, argv); for (i = 0; i <= T_MASK; i++) { counts[i] = 0; @@ -281,12 +296,7 @@ count_nodes(int argc, VALUE *argv, VALUE os) { size_t nodes[NODE_LAST+1]; size_t i; - VALUE hash; - - if (rb_scan_args(argc, argv, "01", &hash) == 1) { - if (!RB_TYPE_P(hash, T_HASH)) - rb_raise(rb_eTypeError, "non-hash given"); - } + VALUE hash = setup_hash(argc, argv); for (i = 0; i <= NODE_LAST; i++) { nodes[i] = 0; @@ -485,22 +495,8 @@ cto_i(void *vstart, void *vend, size_t stride, void *data) static VALUE count_tdata_objects(int argc, VALUE *argv, VALUE self) { - VALUE hash; - - if (rb_scan_args(argc, argv, "01", &hash) == 1) { - if (!RB_TYPE_P(hash, T_HASH)) - rb_raise(rb_eTypeError, "non-hash given"); - } - - if (hash == Qnil) { - hash = rb_hash_new(); - } - else if (!RHASH_EMPTY_P(hash)) { - st_foreach(RHASH_TBL(hash), set_zero_i, hash); - } - + VALUE hash = setup_hash(argc, argv); rb_objspace_each_objects(cto_i, (void *)hash); - return hash; } |