[ruby-dev:28171] Re: thread based generator.rb

なかだです。

At Thu, 29 Dec 2005 16:12:38 +0900,
Tanaka Akira wrote in [ruby-dev:28166]:
> そして、Thread.list は呼び出されるたびに毎回生きているスレッ
> ドを集めて array を生成していますから、その結果に回収済みの
> オブジェクトを含むことはありません。

そういやそうでした。しかし、ちょっと面倒な感じ。


Index: eval.c
===================================================================
RCS file: /src/ruby/eval.c,v
retrieving revision 1.859
diff -U2 -p -r1.859 eval.c
--- eval.c	27 Dec 2005 16:35:57 -0000	1.859
+++ eval.c	29 Dec 2005 09:03:23 -0000
@@ -9756,4 +9756,17 @@ struct thread {
 #define END_FOREACH(x)    END_FOREACH_FROM(curr_thread,x)
 
+#define FOREACH_LIVE_THREAD_FROM(f, x)	\
+    FOREACH_THREAD_FROM(f, x)		\
+	switch ((x)->status) {		\
+	  case THREAD_TO_KILL:		\
+	  case THREAD_RUNNABLE:		\
+	    break;			\
+	  case THREAD_STOPPED:		\
+	    if ((x)->wait_for) break;	\
+	  default:			\
+	    continue;			\
+	}
+#define FOREACH_LIVE_THREAD(x)	FOREACH_LIVE_THREAD_FROM(curr_thread,x)
+
 struct thread_status_t {
     NODE *node;
@@ -9985,4 +9998,19 @@ static struct {
 } new_thread;
 
+static int
+mark_loading_thread(ID key, VALUE value, int lev)
+{
+    rb_thread_t th = (rb_thread_t)value;
+    rb_gc_mark(th->thread);
+    return ST_CONTINUE;
+}
+
+static void
+mark_loading_tbl()
+{
+    if (!loading_tbl) return;
+    st_foreach(loading_tbl, mark_loading_thread, 0);
+}
+
 void
 rb_gc_mark_threads(void)
@@ -9994,5 +10022,7 @@ rb_gc_mark_threads(void)
 
     if (!curr_thread) return;
-    FOREACH_THREAD_FROM(main_thread, th) {
+    rb_gc_mark(main_thread->thread);
+    rb_gc_mark(curr_thread->thread);
+    FOREACH_LIVE_THREAD_FROM(main_thread, th) {
 	rb_gc_mark(th->thread);
     } END_FOREACH_FROM(main_thread, th);
@@ -10002,4 +10032,5 @@ rb_gc_mark_threads(void)
 	rb_gc_mark(new_thread.arg);
     }
+    mark_loading_tbl();
 }
 
@@ -12528,4 +12559,5 @@ rb_callcc(VALUE self)
 
     THREAD_ALLOC(th);
+    th->status = THREAD_STOPPED;
     cont = Data_Wrap_Struct(rb_cCont, thread_mark, thread_free, th);
 
@@ -12645,5 +12677,5 @@ thgroup_list(VALUE group)
     ary = rb_ary_new();
 
-    FOREACH_THREAD(th) {
+    FOREACH_LIVE_THREAD(th) {
 	if (th->thgroup == data->group) {
 	    rb_ary_push(ary, th->thread);


-- 
--- 僕の前にBugはない。
--- 僕の後ろにBugはできる。
    中田 伸悦