lib/PublicInbox/WQWorker.pm

   1 # Copyright (C) 2021 all contributors <meta@public-inbox.org>
   2 # License: AGPL-3.0+ <https://www.gnu.org/licenses/agpl-3.0.txt>
   3
   4 # for PublicInbox::IPC wq_* (work queue) workers
   5 package PublicInbox::WQWorker;
   6 use strict;
   7 use v5.10.1;
   8 use parent qw(PublicInbox::DS);
   9 use PublicInbox::Syscall qw(EPOLLIN EPOLLEXCLUSIVE EPOLLET);
  10 use Errno qw(EAGAIN ECONNRESET);
  11 use IO::Handle (); # blocking
  12
  13 sub new {
  14         my ($cls, $wq, $field) = @_;
  15         my $s2 = $wq->{$field // '-wq_s2'} // die "BUG: no {$field}";
  16         $s2->blocking(0);
  17         my $self = bless { sock => $s2, wq => $wq }, $cls;
  18         $self->SUPER::new($s2, EPOLLEXCLUSIVE|EPOLLIN|EPOLLET);
  19         $self;
  20 }
  21
  22 sub event_step {
  23         my ($self) = @_;
  24         my $n;
  25         do {
  26                 $n = $self->{wq}->recv_and_run($self->{sock});
  27         } while ($n);
  28         return if !defined($n) && $! == EAGAIN; # likely
  29         warn "wq worker error: $!\n" if !defined($n) && $! != ECONNRESET;
  30         $self->{wq}->wq_atexit_child if $self->{sock} == $self->{wq}->{-wq_s2};
  31         $self->close; # PublicInbox::DS::close
  32 }
  33
  34 1;