# Copyright (C) all contributors
# License: AGPL-3.0+
package PublicInbox::Limiter;
use v5.12;
use PublicInbox::Spawn;
sub new {
my ($class, $max) = @_;
bless {
# 32 is same as the git-daemon connection limit
max => $max || 32,
running => 0,
run_queue => [],
# RLIMIT_CPU => undef,
# RLIMIT_DATA => undef,
# RLIMIT_CORE => undef,
}, $class;
}
sub setup_rlimit {
my ($self, $name, $cfg) = @_;
for my $rlim (@PublicInbox::Spawn::RLIMITS) {
my $k = lc($rlim);
$k =~ tr/_//d;
$k = "publicinboxlimiter.$name.$k";
my $v = $cfg->{$k} // next;
my @rlimit = split(/\s*,\s*/, $v);
if (scalar(@rlimit) == 1) {
push @rlimit, $rlimit[0];
} elsif (scalar(@rlimit) != 2) {
warn "could not parse $k: $v\n";
}
my $inf = $v =~ /\binfinity\b/i ?
$PublicInbox::Spawn::RLIMITS{RLIM_INFINITY} // eval {
require BSD::Resource;
BSD::Resource::RLIM_INFINITY();
} // do {
warn "BSD::Resource missing for $rlim";
next;
} : undef;
for my $i (0..$#rlimit) {
next if $rlimit[$i] ne 'INFINITY';
$rlimit[$i] = $inf;
}
$self->{$rlim} = \@rlimit;
}
}
1;