diff options
author | naruse <naruse@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2018-03-22 07:00:17 +0000 |
---|---|---|
committer | naruse <naruse@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2018-03-22 07:00:17 +0000 |
commit | 547176267f8378a193774c4ca529e854e811c928 (patch) | |
tree | d9d792c7d4cd23045d2daef4d9476da5e1940ddf /hash.c | |
parent | f4aea9108d19b304e8a7f9d2ec4763eadf4a64ba (diff) |
merge revision(s) 62042,62044: [Backport #14380]
hash.c: support key swapping in Hash#transform_keys!
* hash.c (rb_hash_transform_keys_bang): support key swapping in
Hash#transform_keys!
[Bug #14380] [ruby-core:84951]
* test/ruby/test_hash.rb (test_transform_keys_bang):
add assertions for this change
Fix rubyspec against the change in Hash#transform_keys!
[Bug #14380] [ruby-core:84951]
git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/branches/ruby_2_5@62889 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
Diffstat (limited to 'hash.c')
-rw-r--r-- | hash.c | 16 |
1 files changed, 10 insertions, 6 deletions
@@ -1910,6 +1910,8 @@ rb_hash_transform_keys(VALUE hash) return result; } +static VALUE rb_hash_flatten(int argc, VALUE *argv, VALUE hash); + /* * call-seq: * hsh.transform_keys! {|key| block } -> hsh @@ -1933,12 +1935,14 @@ rb_hash_transform_keys_bang(VALUE hash) RETURN_SIZED_ENUMERATOR(hash, 0, 0, hash_enum_size); rb_hash_modify_check(hash); if (RHASH(hash)->ntbl) { - long i; - VALUE keys = rb_hash_keys(hash); - for (i = 0; i < RARRAY_LEN(keys); ++i) { - VALUE key = RARRAY_AREF(keys, i), new_key = rb_yield(key); - rb_hash_aset(hash, new_key, rb_hash_delete(hash, key)); - } + long i; + VALUE pairs = rb_hash_flatten(0, NULL, hash); + rb_hash_clear(hash); + for (i = 0; i < RARRAY_LEN(pairs); i += 2) { + VALUE key = RARRAY_AREF(pairs, i), new_key = rb_yield(key), + val = RARRAY_AREF(pairs, i+1); + rb_hash_aset(hash, new_key, val); + } } return hash; } |