[ruby-dev:13094] Re: patch for ext/gdbm

新井です。

ruby 1.7.0 (2001-05-02) [i586-linux]

ですが、パッチのようにした方が GDBM#clear は速いようです。

    file = "/tmp/foo.gdbm"
    File.delete(file) rescue nil
    gdbm = GDBM.new(file)

    10000.times {|i|
      gdbm[i.to_s] = i.to_s
    }
    start = Time.now
    gdbm.clear
    p Time.now - start

    => (パッチ前) 18.167391
    => (パッチ後) 1.122306

あと、GDBM#delete を Hash のインタフェースにあわせてみました。

パッチ前
  o 削除した後 self を返す
  o 値がなければ nil を返す。ブロックがあればそれを評価した後 nil を返す

パッチ後
  o 削除した値を返す
  o 値がなければnilを返す。ただしブロックが指定されていればその結果を返す。

# test が同梱されてない...


Index: ext/gdbm/gdbm.c
===================================================================
RCS file: /src/ruby/ext/gdbm/gdbm.c,v
retrieving revision 1.15
diff -u -p -u -r1.15 gdbm.c
--- ext/gdbm/gdbm.c	2001/05/02 04:22:18	1.15
+++ ext/gdbm/gdbm.c	2001/05/03 07:12:11
@@ -354,31 +354,11 @@ static VALUE
 fgdbm_delete(obj, keystr)
     VALUE obj, keystr;
 {
-    datum key;
-    struct dbmdata *dbmp;
-    GDBM_FILE dbm;
-
-    rb_secure(4);
-    StringValue(keystr);
-    key.dptr = RSTRING(keystr)->ptr;
-    key.dsize = RSTRING(keystr)->len;
+    VALUE valstr;
 
-    GetDBM(obj, dbmp);
-    dbm = dbmp->di_dbm;
-
-    if (!gdbm_exists(dbm, key)) {
-	if (rb_block_given_p()) rb_yield(keystr);
-	return Qnil;
-    }
-
-    if (gdbm_delete(dbm, key)) {
-	dbmp->di_size = -1;
-	rb_raise(rb_eGDBMError, "%s", gdbm_strerror(gdbm_errno));
-    }
-    else if (dbmp->di_size >= 0) {
-	dbmp->di_size--;
-    }
-    return obj;
+    valstr = fgdbm_fetch(obj, keystr, Qnil);
+    rb_gdbm_delete(obj, keystr);
+    return valstr;
 }
 
 static VALUE
@@ -438,7 +418,7 @@ static VALUE
 fgdbm_clear(obj)
     VALUE obj;
 {
-    datum key;
+    datum key, nextkey;
     struct dbmdata *dbmp;
     GDBM_FILE dbm;
 
@@ -447,6 +427,7 @@ fgdbm_clear(obj)
     dbm = dbmp->di_dbm;
     dbmp->di_size = -1;
 
+#if 0
     while (key = gdbm_firstkey(dbm), key.dptr) {
 	if (gdbm_delete(dbm, key)) {
 	    free(key.dptr);
@@ -454,6 +435,19 @@ fgdbm_clear(obj)
 	}
 	free(key.dptr); 
     }
+#else
+    while (key = gdbm_firstkey(dbm), key.dptr) {
+        for (; key.dptr; key = nextkey) {
+            nextkey = gdbm_nextkey(dbm, key);
+            if (gdbm_delete(dbm, key)) {
+                free(key.dptr);
+                if (nextkey.dptr) free(nextkey.dptr);
+                rb_raise(rb_eGDBMError, "%s", gdbm_strerror(gdbm_errno));
+            }
+            free(key.dptr);
+        }
+    }
+#endif
     dbmp->di_size = 0;
 
     return obj;

--
新井康司 (Koji Arai)