Don't call Scheduler#close if it doesn't exist.

ruby · Oct 1, 2020 · 1366010 · 1366010
1 parent bc23216
commit 1366010
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/scheduler.c b/scheduler.c
@@ -39,7 +39,9 @@ Init_Scheduler(void)
 
 VALUE rb_scheduler_close(VALUE scheduler)
 {
-    return rb_funcall(scheduler, id_close, 0);
+    if (rb_respond_to(scheduler, id_close)) {
+        return rb_funcall(scheduler, id_close, 0);
+    }
 }
 
 VALUE

diff --git a/test/fiber/test_scheduler.rb b/test/fiber/test_scheduler.rb
@@ -49,4 +49,12 @@ def test_close_at_exit
     end
     RUBY
   end
+
+  def test_optional_close
+    thread = Thread.new do
+      Thread.current.scheduler = Object.new
+    end
+
+    thread.join
+  end
 end