switch to local-izing on_result in await_all

diff --git a/lib/Tak.pm b/lib/Tak.pm
index 1e0de48..dfa1930 100644 (file)
--- a/lib/Tak.pm
+++ b/lib/Tak.pm
@@ -20,15 +20,18 @@ sub await_all {
   return unless @requests;
   my %req = map +("$_" => "$_"), @requests;
   my $done;
-  my %orig;
   my %on_r = map {
-    my $orig = $orig{$_} = $_->{on_result};
+    my $orig = $_->{on_result};
     my $tag = $req{$_};
     ($_ => sub { delete $req{$tag}; $orig->(@_); $done = 1 unless keys %req; })
   } @requests;
-  $_->{on_result} = $on_r{$_} for @requests;
-  $class->loop_until($done);
-  $_->{on_result} = $orig{$_} for @requests;
+  my $call = sub { $class->loop_until($done) };
+  foreach (@requests) {
+    my $req = $_;
+    my $inner = $call;
+    $call = sub { local $req->{on_result} = $on_r{$req}; $inner->() };
+  }
+  $call->();
   return;
 }