* hash.c: use RHASH_ITER_LEV(), not iter_leve directly.

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@35309 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
This commit is contained in:
nobu 2012-04-12 06:10:31 +00:00
parent 1d6d4d2461
commit e2aaf978b5

18
hash.c
View File

@ -171,9 +171,7 @@ hash_foreach_iter(st_data_t key, st_data_t value, st_data_t argp)
static VALUE static VALUE
hash_foreach_ensure(VALUE hash) hash_foreach_ensure(VALUE hash)
{ {
RHASH(hash)->iter_lev--; if (--RHASH_ITER_LEV(hash) == 0) {
if (RHASH(hash)->iter_lev == 0) {
if (FL_TEST(hash, HASH_DELETED)) { if (FL_TEST(hash, HASH_DELETED)) {
st_cleanup_safe(RHASH(hash)->ntbl, (st_data_t)Qundef); st_cleanup_safe(RHASH(hash)->ntbl, (st_data_t)Qundef);
FL_UNSET(hash, HASH_DELETED); FL_UNSET(hash, HASH_DELETED);
@ -199,7 +197,7 @@ rb_hash_foreach(VALUE hash, int (*func)(ANYARGS), VALUE farg)
if (!RHASH(hash)->ntbl) if (!RHASH(hash)->ntbl)
return; return;
RHASH(hash)->iter_lev++; RHASH_ITER_LEV(hash)++;
arg.hash = hash; arg.hash = hash;
arg.func = (rb_foreach_func *)func; arg.func = (rb_foreach_func *)func;
arg.arg = farg; arg.arg = farg;
@ -265,7 +263,7 @@ rb_hash_modify(VALUE hash)
static void static void
hash_update(VALUE hash, VALUE key) hash_update(VALUE hash, VALUE key)
{ {
if (RHASH(hash)->iter_lev > 0 && !st_lookup(RHASH(hash)->ntbl, key, 0)) { if (RHASH_ITER_LEV(hash) > 0 && !st_lookup(RHASH(hash)->ntbl, key, 0)) {
rb_raise(rb_eRuntimeError, "can't add a new key into hash during iteration"); rb_raise(rb_eRuntimeError, "can't add a new key into hash during iteration");
} }
} }
@ -466,7 +464,7 @@ rb_hash_rehash(VALUE hash)
{ {
st_table *tbl; st_table *tbl;
if (RHASH(hash)->iter_lev > 0) { if (RHASH_ITER_LEV(hash) > 0) {
rb_raise(rb_eRuntimeError, "rehash during iteration"); rb_raise(rb_eRuntimeError, "rehash during iteration");
} }
rb_hash_modify_check(hash); rb_hash_modify_check(hash);
@ -778,7 +776,7 @@ rb_hash_delete_key(VALUE hash, VALUE key)
if (!RHASH(hash)->ntbl) if (!RHASH(hash)->ntbl)
return Qundef; return Qundef;
if (RHASH(hash)->iter_lev > 0) { if (RHASH_ITER_LEV(hash) > 0) {
if (st_delete_safe(RHASH(hash)->ntbl, &ktmp, &val, (st_data_t)Qundef)) { if (st_delete_safe(RHASH(hash)->ntbl, &ktmp, &val, (st_data_t)Qundef)) {
FL_SET(hash, HASH_DELETED); FL_SET(hash, HASH_DELETED);
return (VALUE)val; return (VALUE)val;
@ -868,11 +866,11 @@ rb_hash_shift(VALUE hash)
rb_hash_modify_check(hash); rb_hash_modify_check(hash);
if (RHASH(hash)->ntbl) { if (RHASH(hash)->ntbl) {
var.key = Qundef; var.key = Qundef;
rb_hash_foreach(hash, RHASH(hash)->iter_lev > 0 ? shift_i_safe : shift_i, rb_hash_foreach(hash, RHASH_ITER_LEV(hash) > 0 ? shift_i_safe : shift_i,
(VALUE)&var); (VALUE)&var);
if (var.key != Qundef) { if (var.key != Qundef) {
if (RHASH(hash)->iter_lev > 0) { if (RHASH_ITER_LEV(hash) > 0) {
rb_hash_delete_key(hash, var.key); rb_hash_delete_key(hash, var.key);
} }
return rb_assoc_new(var.key, var.val); return rb_assoc_new(var.key, var.val);
@ -1091,7 +1089,7 @@ rb_hash_clear(VALUE hash)
if (!RHASH(hash)->ntbl) if (!RHASH(hash)->ntbl)
return hash; return hash;
if (RHASH(hash)->ntbl->num_entries > 0) { if (RHASH(hash)->ntbl->num_entries > 0) {
if (RHASH(hash)->iter_lev > 0) if (RHASH_ITER_LEV(hash) > 0)
rb_hash_foreach(hash, clear_i, 0); rb_hash_foreach(hash, clear_i, 0);
else else
st_clear(RHASH(hash)->ntbl); st_clear(RHASH(hash)->ntbl);