X-Git-Url: http://www.git.stargrave.org/?a=blobdiff_plain;f=script%2Fpublic-inbox-watch;h=20534bf2a3ec196e36cdd3c9d2bb0c6136fba849;hb=1e3c53a422b8d23cff961e43f77ea0a835cdef78;hp=2057066a2a9038589120d7ed424550aef44b1444;hpb=b2b1006759730507731fcd3fc3e0de68239e3b92;p=public-inbox.git diff --git a/script/public-inbox-watch b/script/public-inbox-watch index 2057066a..20534bf2 100755 --- a/script/public-inbox-watch +++ b/script/public-inbox-watch @@ -2,9 +2,15 @@ # Copyright (C) 2016-2020 all contributors # License: AGPL-3.0+ use strict; -use warnings; +use IO::Handle; use PublicInbox::WatchMaildir; use PublicInbox::Config; +use PublicInbox::DS; +use PublicInbox::Sigfd; +use PublicInbox::Syscall qw($SFD_NONBLOCK); +my $oldset = PublicInbox::Sigfd::block_signals(); +STDOUT->autoflush(1); +STDERR->autoflush(1); my ($config, $watch_md); my $reload = sub { $config = PublicInbox::Config->new; @@ -14,14 +20,26 @@ my $reload = sub { $reload->(); if ($watch_md) { my $scan = sub { $watch_md->trigger_scan('full') if $watch_md }; - $SIG{HUP} = $reload; - $SIG{USR1} = $scan; - $SIG{ALRM} = sub { $SIG{ALRM} = 'DEFAULT'; $scan->() }; - $SIG{QUIT} = $SIG{TERM} = $SIG{INT} = sub { + my $quit = sub { $watch_md->quit if $watch_md; $watch_md = undef; }; + my $sig = { + HUP => $reload, + USR1 => $scan, + CHLD => \&PublicInbox::DS::enqueue_reap, + }; + $sig->{QUIT} = $sig->{TERM} = $sig->{INT} = $quit; + # --no-scan is only intended for testing atm, undocumented. - alarm(1) unless (grep(/\A--no-scan\z/, @ARGV)); - $watch_md->watch while ($watch_md); + unless (grep(/\A--no-scan\z/, @ARGV)) { + PublicInbox::DS::requeue($scan); + } + my $sigfd = PublicInbox::Sigfd->new($sig, $SFD_NONBLOCK); + local %SIG = (%SIG, %$sig) if !$sigfd; + if (!$sigfd) { + PublicInbox::Sigfd::set_sigmask($oldset); + PublicInbox::DS->SetLoopTimeout(1000); + } + $watch_md->watch($sig, $oldset) while ($watch_md); }