about summary refs log tree commit homepage
path: root/lib/PublicInbox/OnDestroy.pm
diff options
context:
space:
mode:
Diffstat (limited to 'lib/PublicInbox/OnDestroy.pm')
-rw-r--r--lib/PublicInbox/OnDestroy.pm32
1 files changed, 21 insertions, 11 deletions
diff --git a/lib/PublicInbox/OnDestroy.pm b/lib/PublicInbox/OnDestroy.pm
index 615bc450..4301edff 100644
--- a/lib/PublicInbox/OnDestroy.pm
+++ b/lib/PublicInbox/OnDestroy.pm
@@ -1,21 +1,31 @@
-# Copyright (C) 2020-2021 all contributors <meta@public-inbox.org>
+# Copyright (C) all contributors <meta@public-inbox.org>
 # License: AGPL-3.0+ <https://www.gnu.org/licenses/agpl-3.0.txt>
 
 package PublicInbox::OnDestroy;
+use v5.12;
+use parent qw(Exporter);
+use autodie qw(fork);
+our @EXPORT = qw(on_destroy);
+our $fork_gen = 0;
 
-sub new {
-        shift; # ($class, $cb, @args)
-        bless [ @_ ], __PACKAGE__;
+# either parent or child is expected to exit or exec shortly after this:
+sub fork_tmp () {
+        my $pid = fork;
+        ++$fork_gen if $pid == 0;
+        $pid;
 }
 
+# all children
+sub all (@) { bless [ undef, @_ ], __PACKAGE__ }
+
+# same process
+sub on_destroy (@) { bless [ $fork_gen, @_ ], __PACKAGE__ }
+
+sub cancel { @{$_[0]} = () }
+
 sub DESTROY {
-        my ($cb, @args) = @{$_[0]};
-        if (!ref($cb) && $cb) {
-                my $pid = $cb;
-                return if $pid != $$;
-                $cb = shift @args;
-        }
-        $cb->(@args) if $cb;
+        my ($fgen, $cb, @args) = @{$_[0]};
+        $cb->(@args) if ($cb && ($fgen // $fork_gen) == $fork_gen);
 }
 
 1;