new remote code

[scpubgit/Tak.git] / lib / Tak / Loop.pm

diff --git a/lib/Tak/Loop.pm b/lib/Tak/Loop.pm
index fe9dff9..888136c 100644 (file)
--- a/lib/Tak/Loop.pm
+++ b/lib/Tak/Loop.pm
@@ -1,5 +1,46 @@
 package Tak::Loop;
 
+use IO::Select;
 use Moo;
 
+has is_running => (is => 'rw', clearer => 'loop_stop');
+
+has _read_watches => (is => 'ro', default => sub { {} });
+has _read_select => (is => 'ro', default => sub { IO::Select->new });
+
+sub watch_io {
+  my ($self, %watch) = @_;
+  my $fh = $watch{handle};
+  if (my $cb = $watch{on_read_ready}) {
+    $self->_read_select->add($fh);
+    $self->_read_watches->{$fh} = $cb;
+  }
+}
+
+sub unwatch_io {
+  my ($self, %watch) = @_;
+  my $fh = $watch{handle};
+  if ($watch{on_read_ready}) {
+    $self->_read_select->remove($fh);
+    delete $self->_read_watches->{$fh};
+  }
+}
+
+sub loop_once {
+  my ($self) = @_;
+  my $read = $self->_read_watches;
+  my ($readable) = IO::Select->select($self->_read_select);
+  foreach my $fh (@$readable) {
+    $read->{$fh}();
+  }
+}
+
+sub loop_forever {
+  my ($self) = @_;
+  $self->is_running(1);
+  while ($self->is_running) {
+    $self->loop_once;
+  }
+}
+
 1;