diff options
author | nobu <nobu@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2010-12-04 02:21:53 +0000 |
---|---|---|
committer | nobu <nobu@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2010-12-04 02:21:53 +0000 |
commit | 2d4b0d626170c2dada457ea3d11d1d6732775db4 (patch) | |
tree | 7eef99474c246405218d57d69e1f5041870a8cf2 /hash.c | |
parent | 5c6b43d91eadb3e9aa96ef8b2ff5a244081e30f0 (diff) | |
download | ruby-2d4b0d626170c2dada457ea3d11d1d6732775db4.tar.gz |
* hash.c (rb_hash_update_by): new API for Hash#update.
git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@30078 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
Diffstat (limited to 'hash.c')
-rw-r--r-- | hash.c | 37 |
1 files changed, 37 insertions, 0 deletions
@@ -1785,6 +1785,43 @@ rb_hash_update(VALUE hash1, VALUE hash2) return hash1; } +struct update_arg { + VALUE hash; + rb_hash_update_func *func; +}; + +static int +rb_hash_update_func_i(VALUE key, VALUE value, VALUE arg0) +{ + struct update_arg *arg = (struct update_arg *)arg0; + VALUE hash = arg->hash; + + if (key == Qundef) return ST_CONTINUE; + if (rb_hash_has_key(hash, key)) { + value = (*arg->func)(key, rb_hash_aref(hash, key), value); + } + hash_update(hash, key); + st_insert(RHASH(hash)->ntbl, key, value); + return ST_CONTINUE; +} + +VALUE +rb_hash_update_by(VALUE hash1, VALUE hash2, rb_hash_update_func *func) +{ + rb_hash_modify(hash1); + hash2 = to_hash(hash2); + if (func) { + struct update_arg arg; + arg.hash = hash1; + arg.func = func; + rb_hash_foreach(hash2, rb_hash_update_func_i, (VALUE)&arg); + } + else { + rb_hash_foreach(hash2, rb_hash_update_i, hash1); + } + return hash1; +} + /* * call-seq: * hsh.merge(other_hash) -> new_hash |