1
0
Fork 0
mirror of https://github.com/ruby/ruby.git synced 2022-11-09 12:17:21 -05:00

Call scheduler.block instead of scheduler.kernel_sleep for blocking Queue/SizedQueue operations

* scheduler.unblock was already already called before but with no corresponding scheduler.block
* add test that Queue#pop makes the scheduler wait until it gets an element.
This commit is contained in:
Benoit Daloze 2020-09-17 14:45:44 +02:00
parent 738a089b3a
commit 9472d16061
4 changed files with 35 additions and 9 deletions

View file

@ -93,6 +93,7 @@ class Scheduler
end
ensure
@urgent.each(&:close)
@urgent = nil
end
def current_time
@ -139,7 +140,7 @@ class Scheduler
end
if io = @urgent&.last
@urgent.last.write_nonblock('.')
io.write_nonblock('.')
end
end

View file

@ -115,6 +115,31 @@ class TestFiberMutex < Test::Unit::TestCase
assert processed == 3
end
def test_queue_pop_waits
queue = Queue.new
running = false
thread = Thread.new do
scheduler = Scheduler.new
Thread.current.scheduler = scheduler
result = nil
Fiber.schedule do
result = queue.pop
end
running = true
scheduler.run
result
end
Thread.pass until running
sleep 0.1
queue << :done
assert_equal :done, thread.value
end
def test_mutex_deadlock
err = /No live threads left. Deadlock\?/
assert_in_out_err %W[-I#{__dir__} -], <<-RUBY, ['in synchronize'], err, success: false

View file

@ -134,7 +134,7 @@ rb_thread_local_storage(VALUE thread)
static void sleep_hrtime(rb_thread_t *, rb_hrtime_t, unsigned int fl);
static void sleep_forever(rb_thread_t *th, unsigned int fl);
static void rb_thread_sleep_deadly_allow_spurious_wakeup(void);
static void rb_thread_sleep_deadly_allow_spurious_wakeup(VALUE blocker);
static int rb_threadptr_dead(rb_thread_t *th);
static void rb_check_deadlock(rb_ractor_t *r);
static int rb_threadptr_pending_interrupt_empty_p(const rb_thread_t *th);
@ -1485,11 +1485,11 @@ rb_thread_sleep_interruptible(void)
}
static void
rb_thread_sleep_deadly_allow_spurious_wakeup(void)
rb_thread_sleep_deadly_allow_spurious_wakeup(VALUE blocker)
{
VALUE scheduler = rb_thread_current_scheduler();
if (scheduler != Qnil) {
rb_scheduler_kernel_sleepv(scheduler, 0, NULL);
rb_scheduler_block(scheduler, blocker);
} else {
thread_debug("rb_thread_sleep_deadly_allow_spurious_wakeup\n");
sleep_forever(GET_THREAD(), SLEEP_DEADLOCKABLE);

View file

@ -483,9 +483,9 @@ rb_mutex_abandon_all(rb_mutex_t *mutexes)
#endif
static VALUE
rb_mutex_sleep_forever(VALUE time)
rb_mutex_sleep_forever(VALUE self)
{
rb_thread_sleep_deadly_allow_spurious_wakeup();
rb_thread_sleep_deadly_allow_spurious_wakeup(self);
return Qnil;
}
@ -516,7 +516,7 @@ rb_mutex_sleep(VALUE self, VALUE timeout)
mutex_lock_uninterruptible(self);
} else {
if (NIL_P(timeout)) {
rb_ensure(rb_mutex_sleep_forever, Qnil, mutex_lock_uninterruptible, self);
rb_ensure(rb_mutex_sleep_forever, self, mutex_lock_uninterruptible, self);
} else {
rb_hrtime_t rel = rb_timeval2hrtime(&t);
rb_ensure(rb_mutex_wait_for, (VALUE)&rel, mutex_lock_uninterruptible, self);
@ -904,9 +904,9 @@ rb_queue_push(VALUE self, VALUE obj)
}
static VALUE
queue_sleep(VALUE arg)
queue_sleep(VALUE self)
{
rb_thread_sleep_deadly_allow_spurious_wakeup();
rb_thread_sleep_deadly_allow_spurious_wakeup(self);
return Qnil;
}