[ruby-dev:42430] Re: [BUG: trunk] Lazy sweep and ObjectSpace.each_object

nariです。

> 　これだと、例えば例外が出た場合、dont_lazy_sweep が TRUE のままになって
> しまうような気がします。ensure で囲まないと。
>

ぉぉ、そうですね…。ご指摘ありがとうございます。
そのようなパッチを作ってみました。本メールの一番最後に貼り付けておきます。

> # 余談
>
> ちょっと関係ないですが、gc_lazy_sweep() という関数は、下記の挙動だと思
> うのですが；
>
> ・もし、遅延した slot があれば、sweep をする
> これで、freelist が NULL でなければ、ここで終わり
> ・GC を開始（mark）
> ・遅延 sweep をして、結果を返す
>
> sweep という関数で mark までしちゃうのはいいでしょうか。
>

なるほど。
関数名にgc_というプレフィクスが付いているので、マークしてもいいのかなと
か、そういう言い訳を思いついたのですがやっぱりダメですね。

> あまり、対案はないのですが、なんとなく、garbage_collect() を 今の
> gc_lazy_sweep() として、今の garbage_collect() を、
> garbage_collect_eager() みたいにしてしまうのはどうだろう、と思いました。

garbage_collect_eager() よりも garbage_collect_full() の方が馴染み
があるのかなと思いました。こちらの方も暇を見つけて直しておきます。

パッチ:
diff --git a/gc.c b/gc.c
index b011f4a..ad49fd5 100644
--- a/gc.c
+++ b/gc.c
@@ -332,6 +332,7 @@ typedef struct rb_objspace {
     } heap;
     struct {
 	int dont_gc;
+	int dont_lazy_sweep;
 	int during_gc;
     } flags;
     struct {
@@ -2040,6 +2041,17 @@ lazy_sweep(rb_objspace_t *objspace)
     return FALSE;
 }

+static void
+rest_sweep(rb_objspace_t *objspace)
+{
+    if (objspace->heap.sweep_slots) {
+       while (objspace->heap.sweep_slots) {
+           lazy_sweep(objspace);
+       }
+       after_gc_sweep(objspace);
+    }
+}
+
 static void gc_marks(rb_objspace_t *objspace);

 static int
@@ -2047,6 +2059,9 @@ gc_lazy_sweep(rb_objspace_t *objspace)
 {
     int res;

+    if (objspace->flags.dont_lazy_sweep)
+        return garbage_collect(objspace);
+
     INIT_GC_PROF_PARAMS;

     if (!ready_to_gc(objspace)) return TRUE;
@@ -2489,6 +2504,55 @@ Init_heap(void)
     init_heap(&rb_objspace);
 }

+
+static VALUE
+lazy_sweep_enable(void)
+{
+    rb_objspace_t *objspace = &rb_objspace;
+
+    objspace->flags.dont_lazy_sweep = FALSE;
+    return Qnil;
+}
+
+static VALUE
+objspace_each_objects(VALUE arg)
+{
+    size_t i;
+    RVALUE *membase = 0;
+    RVALUE *pstart, *pend;
+    rb_objspace_t *objspace = &rb_objspace;
+    VALUE *args = (VALUE *)arg;
+    volatile VALUE v;
+
+    i = 0;
+    while (i < heaps_used) {
+	while (0 < i && (uintptr_t)membase <
(uintptr_t)objspace->heap.sorted[i-1].slot->membase)
+	    i--;
+	while (i < heaps_used &&
(uintptr_t)objspace->heap.sorted[i].slot->membase <=
(uintptr_t)membase )
+	    i++;
+	if (heaps_used <= i)
+	  break;
+	membase = objspace->heap.sorted[i].slot->membase;
+
+	pstart = objspace->heap.sorted[i].slot->slot;
+	pend = pstart + objspace->heap.sorted[i].slot->limit;
+
+	for (; pstart != pend; pstart++) {
+	    if (pstart->as.basic.flags) {
+		v = (VALUE)pstart; /* acquire to save this object */
+		break;
+	    }
+	}
+	if (pstart != pend) {
+	    if ((*(int (*)(void *, void *, size_t, void *))args[0])(pstart,
pend, sizeof(RVALUE), (void *)args[1])) {
+		return;
+	    }
+	}
+    }
+
+    return Qnil;
+}
+
 /*
  * rb_objspace_each_objects() is special C API to walk through
  * Ruby object space.  This C API is too difficult to use it.
@@ -2530,39 +2594,15 @@ rb_objspace_each_objects(int (*callback)(void
*vstart, void *vend,
 					 size_t stride, void *d),
 			 void *data)
 {
-    size_t i;
-    RVALUE *membase = 0;
-    RVALUE *pstart, *pend;
+    VALUE args[2];
     rb_objspace_t *objspace = &rb_objspace;
-    volatile VALUE v;
-
-    i = 0;
-    while (i < heaps_used) {
-	while (0 < i && (uintptr_t)membase <
(uintptr_t)objspace->heap.sorted[i-1].slot->membase)
-	    i--;
-	while (i < heaps_used &&
(uintptr_t)objspace->heap.sorted[i].slot->membase <=
(uintptr_t)membase )
-	    i++;
-	if (heaps_used <= i)
-	  break;
-	membase = objspace->heap.sorted[i].slot->membase;

-	pstart = objspace->heap.sorted[i].slot->slot;
-	pend = pstart + objspace->heap.sorted[i].slot->limit;
-
-	for (; pstart != pend; pstart++) {
-	    if (pstart->as.basic.flags) {
-		v = (VALUE)pstart; /* acquire to save this object */
-		break;
-	    }
-	}
-	if (pstart != pend) {
-	    if ((*callback)(pstart, pend, sizeof(RVALUE), data)) {
-		return;
-	    }
-	}
-    }
+    rest_sweep(objspace);
+    objspace->flags.dont_lazy_sweep = TRUE;

-    return;
+    args[0] = (VALUE)callback;
+    args[1] = (VALUE)data;
+    rb_ensure(objspace_each_objects, (VALUE)args, lazy_sweep_enable, Qnil);
 }

 struct os_each_struct {


-- 
Narihiro Nakamura (nari)