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, $sock) = @_;
  15         $sock->blocking(0);
  16         my $self = bless { sock => $sock, wq => $wq }, $cls;
  17         $self->SUPER::new($sock, EPOLLEXCLUSIVE|EPOLLIN|EPOLLET);
  18         $self;
  19 }
  20
  21 sub event_step {
  22         my ($self) = @_;
  23         my $n;
  24         do {
  25                 $n = $self->{wq}->recv_and_run($self->{sock});
  26         } while ($n);
  27         return if !defined($n) && $! == EAGAIN; # likely
  28         warn "wq worker error: $!\n" if !defined($n) && $! != ECONNRESET;
  29         $self->{wq}->wq_atexit_child if $self->{sock} == $self->{wq}->{-wq_s2};
  30         $self->close; # PublicInbox::DS::close
  31 }
  32
  33 1;