From 0d63b9b4a49d7477b5201f5d40ecd64916cd0e43 Mon Sep 17 00:00:00 2001
From: Koichi Sasada <ko1@atdot.net>
Date: Tue, 5 Nov 2024 11:49:21 +0900
Subject: [PATCH] check closing flag

`Ractor.receive` and `Ractor.yield` should stop when the
incoming/outgoing port is closed.
---
 ractor.c | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/ractor.c b/ractor.c
index 17a2248350..0b33340f11 100644
--- a/ractor.c
+++ b/ractor.c
@@ -753,7 +753,7 @@ ractor_wait_receive(rb_execution_context_t *ec, rb_ractor_t *cr, struct rb_racto
 
     RACTOR_LOCK(cr);
     {
-        while (ractor_queue_empty_p(cr, rq)) {
+        while (ractor_queue_empty_p(cr, rq) && !cr->sync.incoming_port_closed) {
             ractor_sleep(ec, cr, wait_receiving);
         }
     }
@@ -1364,6 +1364,9 @@ ractor_try_yield(rb_execution_context_t *ec, rb_ractor_t *cr, struct rb_ractor_q
 
         return true;
     }
+    else if (cr->sync.outgoing_port_closed) {
+        rb_raise(rb_eRactorClosedError, "The outgoing-port is already closed");
+    }
     else {
         RUBY_DEBUG_LOG("no take basket");
         return false;
@@ -1375,7 +1378,7 @@ ractor_wait_yield(rb_execution_context_t *ec, rb_ractor_t *cr, struct rb_ractor_
 {
     RACTOR_LOCK_SELF(cr);
     {
-        while (!ractor_check_take_basket(cr, ts)) {
+        while (!ractor_check_take_basket(cr, ts) && !cr->sync.outgoing_port_closed) {
             ractor_sleep(ec, cr, wait_yielding);
         }
     }