our @EXPORT = qw(future await_future await_all);
-sub future (&) {
+sub future (&;$) {
my $f = $_[0]->(CPS::Future->new);
- return $f if ((caller(1)||'') eq 'start');
+ return $f if ((caller(1+($_[1]||0))||'') eq 'start');
await_future($f);
}
+our @await;
+
sub await_future {
my $f = shift;
return $f if $f->is_ready;
require Object::Remote;
my $loop = Object::Remote->current_loop;
- $f->on_ready(sub { $loop->stop });
- $loop->run;
+ {
+ local @await = (@await, $f);
+ $f->on_ready(sub {
+ $loop->stop if $f == $await[-1]
+ });
+ $loop->run;
+ }
+ if (@await and $await[-1]->is_ready) {
+ $loop->stop;
+ }
return wantarray ? $f->get : ($f->get)[0];
}
sub await_all {
- await_future(CPS::Future->needs_all(@_));
+ await_future(CPS::Future->wait_all(@_));
map $_->get, @_;
}
package start;
+our $start = sub { my ($obj, $call) = (shift, shift); $obj->$call(@_); };
+
sub AUTOLOAD {
my $invocant = shift;
my ($method) = our $AUTOLOAD =~ /^start::(.+)$/;
- if (ref($invocant) eq 'ARRAY') {
- return [ map $_->${\"start::${method}"}, @$invocant ];
- }
my $res;
unless (eval { $res = $invocant->$method(@_); 1 }) {
my $f = CPS::Future->new;
return $res;
}
+package maybe;
+
+sub start {
+ my ($obj, $call) = (shift, shift);
+ if ((caller(1)||'') eq 'start') {
+ $obj->$start::start($call => @_);
+ } else {
+ $obj->$call(@_);
+ }
+}
+
+package maybe::start;
+
+sub AUTOLOAD {
+ my $invocant = shift;
+ my ($method) = our $AUTOLOAD =~ /^maybe::start::(.+)$/;
+ $method = "start::${method}" if ((caller(1)||'') eq 'start');
+ $invocant->$method(@_);
+}
+
+package then;
+
+sub AUTOLOAD {
+ my $invocant = shift;
+ my ($method) = our $AUTOLOAD =~ /^then::(.+)$/;
+ my @args = @_;
+ # Need two copies since if we're called on an already complete future
+ # $f will be freed immediately
+ my $ret = my $f = CPS::Future->new;
+ $invocant->on_fail(sub { $f->fail(@_); undef($f); });
+ $invocant->on_done(sub {
+ my ($obj) = @_;
+ my $next = $obj->${\"start::${method}"}(@args);
+ $next->on_done(sub { $f->done(@_); undef($f); });
+ $next->on_fail(sub { $f->fail(@_); undef($f); });
+ });
+ return $ret;
+}
+
1;
=head1 NAME