File: 06-wait-all-children-with-timeout.t

package info (click to toggle)
libparallel-prefork-perl 0.18-1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm, forky, sid, trixie
  • size: 204 kB
  • sloc: perl: 1,976; makefile: 2
file content (60 lines) | stat: -rw-r--r-- 1,312 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
#! /usr/bin/perl

use strict;
use warnings;

use Fcntl qw/:flock/;
use File::Temp qw/tempfile/;
use Test::More tests => 4;

use Parallel::Prefork;

my $reaped = 0;
my $pm = Parallel::Prefork->new({
    max_workers   => 30,
    fork_delay    => 0,
    on_child_reap => sub {
        $reaped++;
    }
});

my ($fh, $filename) = tempfile;
syswrite $fh, '0', 1;
close $fh;

my $manager_pid = $$;

until ($pm->signal_received) {
    $pm->start and next;

    open my $fh, '+<', $filename
        or die "failed to open temporary file: $filename: ";
    flock $fh, LOCK_EX;
    sysread $fh, my $worker_count, 10;
    $worker_count++;
    seek $fh, 0, 0;
    syswrite $fh, $worker_count, length($worker_count);
    flock $fh, LOCK_UN;
    close $fh;

    my $rcv = 0;
    local $SIG{TERM} = sub { $rcv++ };

    if ($worker_count == $pm->max_workers) {
        kill 'TERM', $manager_pid;
    }

    sleep(100) while $rcv * 10 < $worker_count;

    $pm->finish;
}
is $pm->wait_all_children(1), 20, 'should reap one worker.';
$pm->signal_all_children('TERM');
is $pm->wait_all_children(1), 10, 'should reap one worker.';
$pm->signal_all_children('TERM');
$pm->wait_all_children();
is $pm->num_workers, 0, 'all workers reaped.';

is($reaped, $pm->max_workers, "properly called on_child_reap callback");

unlink $filename;