lib/PublicInbox/Listener.pm

   1 # Copyright (C) 2015-2019 all contributors <meta@public-inbox.org>
   2 # License: AGPL-3.0+ <https://www.gnu.org/licenses/agpl-3.0.txt>
   3 #
   4 # Used by -nntpd for listen sockets
   5 package PublicInbox::Listener;
   6 use strict;
   7 use warnings;
   8 use base 'PublicInbox::DS';
   9 use Socket qw(SOL_SOCKET SO_KEEPALIVE IPPROTO_TCP TCP_NODELAY);
  10 use fields qw(post_accept);
  11 require IO::Handle;
  12 use PublicInbox::Syscall qw(EPOLLIN EPOLLEXCLUSIVE EPOLLET);
  13
  14 sub new ($$$) {
  15         my ($class, $s, $cb) = @_;
  16         setsockopt($s, SOL_SOCKET, SO_KEEPALIVE, 1);
  17         setsockopt($s, IPPROTO_TCP, TCP_NODELAY, 1); # ignore errors on non-TCP
  18         listen($s, 1024);
  19         my $self = fields::new($class);
  20         $self->SUPER::new($s, EPOLLIN|EPOLLET|EPOLLEXCLUSIVE);
  21         $self->{post_accept} = $cb;
  22         $self
  23 }
  24
  25 sub event_step {
  26         my ($self) = @_;
  27         my $sock = $self->{sock} or return;
  28
  29         # no loop here, we want to fairly distribute clients
  30         # between multiple processes sharing the same socket
  31         # XXX our event loop needs better granularity for
  32         # a single accept() here to be, umm..., acceptable
  33         # on high-traffic sites.
  34         if (my $addr = accept(my $c, $sock)) {
  35                 IO::Handle::blocking($c, 0); # no accept4 :<
  36                 $self->{post_accept}->($c, $addr, $sock);
  37                 $self->requeue;
  38         }
  39 }
  40
  41 1;