lib/PublicInbox/Listener.pm

   1 # Copyright (C) 2015-2018 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
  13 sub new ($$$) {
  14         my ($class, $s, $cb) = @_;
  15         setsockopt($s, SOL_SOCKET, SO_KEEPALIVE, 1);
  16         setsockopt($s, IPPROTO_TCP, TCP_NODELAY, 1); # ignore errors on non-TCP
  17         listen($s, 1024);
  18         IO::Handle::blocking($s, 0);
  19         my $self = fields::new($class);
  20         $self->SUPER::new($s); # calls epoll_create for the first socket
  21         $self->watch_read(1);
  22         $self->{post_accept} = $cb;
  23         $self
  24 }
  25
  26 sub event_read {
  27         my ($self) = @_;
  28         my $sock = $self->{sock};
  29
  30         # no loop here, we want to fairly distribute clients
  31         # between multiple processes sharing the same socket
  32         # XXX our event loop needs better granularity for
  33         # a single accept() here to be, umm..., acceptable
  34         # on high-traffic sites.
  35         if (my $addr = accept(my $c, $sock)) {
  36                 IO::Handle::blocking($c, 0); # no accept4 :<
  37                 $self->{post_accept}->($c, $addr, $sock);
  38         }
  39 }
  40
  41 1;