From 703e529751deb086a46a9c029ce38d4fff14e84c Mon Sep 17 00:00:00 2001
From: Samuel Williams <samuel.williams@oriontransfer.co.nz>
Date: Thu, 20 Aug 2020 13:47:55 +1200
Subject: [PATCH] Add `rb_thread_current_scheduler()`.

---
 internal/thread.h | 6 ++++--
 thread.c          | 6 ++++++
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/internal/thread.h b/internal/thread.h
index 91626b7e76..e7a9019b88 100644
--- a/internal/thread.h
+++ b/internal/thread.h
@@ -37,9 +37,11 @@ void rb_mutex_allow_trap(VALUE self, int val);
 VALUE rb_uninterruptible(VALUE (*b_proc)(VALUE), VALUE data);
 VALUE rb_mutex_owned_p(VALUE self);
 
-VALUE rb_thread_scheduler_get(VALUE);
-VALUE rb_thread_scheduler_set(VALUE, VALUE);
+VALUE rb_thread_scheduler_get(VALUE thread);
+VALUE rb_thread_scheduler_set(VALUE thread, VALUE scheduler);
+
 VALUE rb_thread_scheduler_if_nonblocking(VALUE thread);
+VALUE rb_thread_current_scheduler();
 
 RUBY_SYMBOL_EXPORT_BEGIN
 /* Temporary.  This API will be removed (renamed). */
diff --git a/thread.c b/thread.c
index 2f265c5cfc..d10d4118c3 100644
--- a/thread.c
+++ b/thread.c
@@ -3749,6 +3749,12 @@ rb_thread_scheduler(VALUE klass)
     return rb_thread_scheduler_if_nonblocking(rb_thread_current());
 }
 
+static VALUE
+rb_thread_current_scheduler()
+{
+    return rb_thread_scheduler_if_nonblocking(rb_thread_current());
+}
+
 VALUE
 rb_thread_scheduler_if_nonblocking(VALUE thread)
 {