* lib/generator.rb: should work with another thread. (more robust code)

[ruby-dev:28177]


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

ChangeLog

@@ -1,3 +1,8 @@
+Fri Dec 30 01:04:52 2005  Hirokazu Yamamoto  <ocean@m2.ccsnet.ne.jp>
+
+	* lib/generator.rb: should work with another thread. (more robust code)
+	  [ruby-dev:28177]
+
 Thu Dec 29 23:59:37 2005  Nobuyoshi Nakada  <nobu@ruby-lang.org>
 
 	* eval.c (rb_gc_mark_threads): keep unmarked threads which won't wake

lib/generator.rb

@@ -73,14 +73,12 @@ class Generator
     @queue = []
     @loop_thread = Thread.new do
       Thread.stop
-      Thread.critical = true
       begin
         @block.call(self) # exception safe?
       rescue
         @main_thread.raise $!
       ensure
         @main_thread.wakeup
-        Thread.critical = false
       end
     end
     self
@@ -89,21 +87,28 @@ class Generator
   # Yields an element to the generator.
   def yield(value)
     if Thread.current != @loop_thread
-      raise RuntimeError.new("Generator#yield must be called in Generator.new{|g| ... }")
+      raise "should be called in Generator.new{|g| ... }"
     end
-    @queue << value
-    @main_thread.wakeup
-    Thread.stop
     Thread.critical = true
+    begin
+      @queue << value
+      @main_thread.wakeup
+      Thread.stop
+    ensure
+      Thread.critical = false
+    end
     self
   end
 
   # Returns true if the generator has reached the end.
   def end?
     if @queue.empty?
+      if @main_thread
+        raise "should not be called in Generator.new{|g| ... }"
+      end
       Thread.critical = true
-      @main_thread = Thread.current
       begin
+        @main_thread = Thread.current
         @loop_thread.wakeup
         Thread.stop
       rescue ThreadError