From d387029f39d976565c955377117103499d47ff09 Mon Sep 17 00:00:00 2001
From: Samuel Williams <samuel.williams@oriontransfer.co.nz>
Date: Thu, 20 Aug 2020 13:51:45 +1200
Subject: Standardised scheduler interface.

---
 scheduler.c | 71 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 71 insertions(+)
 create mode 100644 scheduler.c

(limited to 'scheduler.c')

diff --git a/scheduler.c b/scheduler.c
new file mode 100644
index 0000000000..4eaf12b333
--- /dev/null
+++ b/scheduler.c
@@ -0,0 +1,71 @@
+/**********************************************************************
+
+  scheduler.c
+
+  $Author$
+
+  Copyright (C) 2020 Samuel Grant Dawson Williams
+
+**********************************************************************/
+
+#include "internal/scheduler.h"
+#include "ruby/io.h"
+
+static ID id_kernel_sleep;
+static ID id_io_read;
+static ID id_io_write;
+static ID id_io_wait;
+
+void
+Init_Scheduler(void)
+{
+    id_kernel_sleep = rb_intern_const("kernel_sleep");
+    id_io_read = rb_intern_const("io_read");
+    id_io_write = rb_intern_const("io_write");
+    id_io_wait = rb_intern_const("io_wait");
+}
+
+VALUE
+rb_scheduler_timeout(struct timeval *timeout) {
+    if (timeout) {
+        return rb_float_new((double)timeout->tv_sec + (0.000001f * timeout->tv_usec));
+    }
+
+    return Qnil;
+}
+
+VALUE rb_scheduler_kernel_sleep(VALUE scheduler, VALUE timeout)
+{
+    return rb_funcall(scheduler, id_kernel_sleep, 1, timeout);
+}
+
+VALUE rb_scheduler_kernel_sleepv(VALUE scheduler, int argc, VALUE * argv)
+{
+    return rb_funcallv(scheduler, id_kernel_sleep, argc, argv);
+}
+
+VALUE rb_scheduler_io_wait(VALUE scheduler, VALUE io, VALUE events, VALUE timeout)
+{
+    return rb_funcall(scheduler, id_io_wait, 3, io, events, timeout);
+}
+
+VALUE rb_scheduler_io_wait_readable(VALUE scheduler, VALUE io)
+{
+    return rb_scheduler_io_wait(scheduler, io, RB_UINT2NUM(RUBY_IO_READABLE), Qnil);
+}
+
+VALUE rb_scheduler_io_wait_writable(VALUE scheduler, VALUE io)
+{
+    return rb_scheduler_io_wait(scheduler, io, RB_UINT2NUM(RUBY_IO_WRITABLE), Qnil);
+}
+
+VALUE rb_scheduler_io_read(VALUE scheduler, VALUE io, VALUE buffer, VALUE offset, VALUE length)
+{
+    return rb_funcall(scheduler, id_io_read, 4, io, buffer, offset, length);
+}
+
+VALUE rb_scheduler_io_write(VALUE scheduler, VALUE io, VALUE buffer, VALUE offset, VALUE length)
+{
+    // We should ensure string has capacity to receive data, and then resize it afterwards.
+    return rb_funcall(scheduler, id_io_write, 4, io, buffer, offset, length);
+}
-- 
cgit v1.2.3