epoll: add cache/mark array debugging methods

Occasionally users may want a way to check if they
stored an IO object or not, this makes it easier.

2 files changed, 54 insertions, 0 deletions

diff --git a/ext/sleepy_penguin/epoll.c b/ext/sleepy_penguin/epoll.c
index 2361e18..4db2179 100644
--- a/ext/sleepy_penguin/epoll.c
+++ b/ext/sleepy_penguin/epoll.c
@@ -220,6 +220,7 @@ static VALUE ctl(VALUE self, VALUE io, VALUE flags, int op)
                 break;
         case EPOLL_CTL_DEL:
                 rb_ary_store(ep->marks, fd, Qnil);
+                rb_ary_store(ep->flag_cache, fd, Qnil);
         }
 
         return INT2NUM(rv);
@@ -306,6 +307,7 @@ static VALUE delete(VALUE self, VALUE io)
                 }
         }
         rb_ary_store(ep->marks, fd, Qnil);
+        rb_ary_store(ep->flag_cache, fd, Qnil);
 
         return io;
 }
@@ -588,6 +590,27 @@ static VALUE init_copy(VALUE copy, VALUE orig)
         return copy;
 }
 
+static VALUE io_for(VALUE self, VALUE obj)
+{
+        struct rb_epoll *ep = ep_get(self);
+
+        return rb_ary_entry(ep->marks, my_fileno(obj));
+}
+
+static VALUE flags_for(VALUE self, VALUE obj)
+{
+        struct rb_epoll *ep = ep_get(self);
+
+        return rb_ary_entry(ep->flag_cache, my_fileno(obj));
+}
+
+static VALUE include_p(VALUE self, VALUE obj)
+{
+        struct rb_epoll *ep = ep_get(self);
+
+        return NIL_P(rb_ary_entry(ep->marks, my_fileno(obj))) ? Qfalse : Qtrue;
+}
+
 /*
  * we close (or lose to GC) epoll descriptors at fork to avoid leakage
  * and invalid objects being referenced later in the child
@@ -633,6 +656,9 @@ void sleepy_penguin_init_epoll(void)
         rb_define_method(cEpoll, "mod", mod, 2);
         rb_define_method(cEpoll, "del", del, 1);
         rb_define_method(cEpoll, "delete", delete, 1);
+        rb_define_method(cEpoll, "io_for", io_for, 1);
+        rb_define_method(cEpoll, "flags_for", flags_for, 1);
+        rb_define_method(cEpoll, "include?", include_p, 1);
         rb_define_method(cEpoll, "set", set, 2);
         rb_define_method(cEpoll, "wait", epwait, -1);
         rb_define_const(cEpoll, "CLOEXEC", INT2NUM(EPOLL_CLOEXEC));
diff --git a/test/test_epoll.rb b/test/test_epoll.rb
index b73cb02..6e1b768 100644
--- a/test/test_epoll.rb
+++ b/test/test_epoll.rb
@@ -323,4 +323,32 @@ class TestEpoll < Test::Unit::TestCase
     assert_equal @rd, @ep.delete(@rd)
     assert_nil @ep.delete(@rd)
   end
+
+  def test_io_for
+    @ep.add @rd, Epoll::IN
+    assert_equal @rd, @ep.io_for(@rd.fileno)
+    assert_equal @rd, @ep.io_for(@rd)
+    @ep.del @rd
+    assert_nil @ep.io_for(@rd.fileno)
+    assert_nil @ep.io_for(@rd)
+  end
+
+  def test_flags_for
+    @ep.add @rd, Epoll::IN
+    assert_equal Epoll::IN, @ep.flags_for(@rd.fileno)
+    assert_equal Epoll::IN, @ep.flags_for(@rd)
+
+    @ep.del @rd
+    assert_nil @ep.flags_for(@rd.fileno)
+    assert_nil @ep.flags_for(@rd)
+  end
+
+  def test_include?
+    assert ! @ep.include?(@rd)
+    @ep.add @rd, Epoll::IN
+    assert @ep.include?(@rd)
+    assert @ep.include?(@rd.fileno)
+    assert ! @ep.include?(@wr)
+    assert ! @ep.include?(@wr.fileno)
+  end
 end