File Coverage

blib/lib/Net/Async/EmptyPort.pm

Criterion	Covered	Total	%
statement	29	29	100.0
branch	5	8	62.5
condition	9	14	64.2
subroutine	8	8	100.0
pod	2	2	100.0
total	53	61	86.8

package Net::Async::EmptyPort;
$Net::Async::EmptyPort::VERSION = '0.001000';
# ABSTRACT: Asynchronously wait for a port to open

use Moo;
use Future::Utils 'try_repeat_until_success', 'try_repeat';

has _loop => (
   is       => 'ro',
   init_arg => 'loop',
   required => 1,
   handles => {
      _connect => 'connect',
      _listen => 'listen',
      _delay => 'delay_future',
      _timeout => 'timeout_future',
   },
);

my %family_map = (
   tcp => 'stream',
   udp => 'dgram',
);
sub empty_port {
   my ($self, $args) = @_;

   $args          //= {};
   $args->{host}  //= '127.0.0.1';
   $args->{port}  //= 0;
   $args->{proto} //= 'tcp';

   if ($args->{port} == 0) {
      $self->_listen(
         on_socket => sub {},
         host => $args->{host},
         socktype => $family_map{$args->{proto}},
         service => $args->{port},
      )
   } else {
      my $port = $args->{port};

      try_repeat {
         $self->_listen(
            on_socket => sub {},
            host => $args->{host},
            socktype => $family_map{$args->{proto}},
            service => $port++,
         )
      } while => sub {
         !shift->is_done && $port < 65000
      },
   }
}

sub wait_port {
   my ($self, $args) = @_;

   die 'port is a required argument'
      unless $args->{port};

   $args->{host}     //= '127.0.0.1';
   $args->{proto}    //= 'tcp';
   $args->{max_wait} //= 10;

   my $amount  = 2;
   my $attempt = 0;

   my $f = try_repeat_until_success {
      $self->_delay(
         after => $amount * (2 ** $attempt++) - $amount,
      )->then(sub {
         $self->_connect(
            host => $args->{host},
            socktype => $family_map{$args->{proto}},
            service  => $args->{port},
         )
      })
   };

   $f = Future->wait_any(
      $f,
      $self->_timeout( after => $args->{max_wait} )
   ) if $args->{max_wait} > 0;

   $f
}

1;

__END__