* string.c (rb_fstring): Use st_update instead of st_lookup +

  st_insert.
* string.c (fstr_update_callback): New callback for st_update.

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@43968 b2dd03c8-39d4-4d8f-98ff-823fe69b080e

1 files changed, 23 insertions, 15 deletions

diff --git a/string.c b/string.c
index 231bb2f4fd..0f05656c66 100644
--- a/string.c
+++ b/string.c
@@ -132,22 +132,16 @@ static const struct st_hash_type fstring_hash_type = {
     rb_str_hash,
 };
 
-VALUE
-rb_fstring(VALUE str)
+static int
+fstr_update_callback(st_data_t *key, st_data_t *value, st_data_t arg, int existing)
 {
-    st_data_t fstr;
-    Check_Type(str, T_STRING);
-
-    if (FL_TEST(str, RSTRING_FSTR))
-	return str;
-
-    if (st_lookup(frozen_strings, (st_data_t)str, &fstr)) {
-	str = (VALUE)fstr;
+    VALUE *fstr = (VALUE *)arg;
+    if (existing) {
 	/* because of lazy sweep, str may be unmaked already and swept
 	 * at next time */
-	rb_gc_resurrect(str);
-    }
-    else {
+	rb_gc_resurrect(*fstr = *key);
+    } else {
+	VALUE str = *key;
 	if (STR_SHARED_P(str)) {
 	    /* str should not be shared */
 	    str = rb_enc_str_new(RSTRING_PTR(str), RSTRING_LEN(str), STR_ENC_GET(str));
@@ -157,9 +151,23 @@ rb_fstring(VALUE str)
 	    str = rb_str_new_frozen(str);
 	}
 	RBASIC(str)->flags |= RSTRING_FSTR;
-	st_insert(frozen_strings, str, str);
+	*fstr = *key = str;
     }
-    return str;
+
+    return ST_CONTINUE;
+}
+
+VALUE
+rb_fstring(VALUE str)
+{
+    VALUE fstr = Qnil;
+    Check_Type(str, T_STRING);
+
+    if (FL_TEST(str, RSTRING_FSTR))
+	return str;
+
+    st_update(frozen_strings, (st_data_t)str, fstr_update_callback, (st_data_t)&fstr);
+    return fstr;
 }
 
 static int