ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/Coro/Coro.pm
(Generate patch)

Comparing Coro/Coro.pm (file contents):
Revision 1.186 by root, Sun May 25 01:32:36 2008 UTC vs.
Revision 1.229 by root, Thu Nov 20 06:32:55 2008 UTC

16 cede; # yield to coroutine 16 cede; # yield to coroutine
17 print "3\n"; 17 print "3\n";
18 cede; # and again 18 cede; # and again
19 19
20 # use locking 20 # use locking
21 use Coro::Semaphore;
21 my $lock = new Coro::Semaphore; 22 my $lock = new Coro::Semaphore;
22 my $locked; 23 my $locked;
23 24
24 $lock->down; 25 $lock->down;
25 $locked = 1; 26 $locked = 1;
55 56
56=cut 57=cut
57 58
58package Coro; 59package Coro;
59 60
60use strict; 61use strict qw(vars subs);
61no warnings "uninitialized"; 62no warnings "uninitialized";
62 63
63use Coro::State; 64use Coro::State;
64 65
65use base qw(Coro::State Exporter); 66use base qw(Coro::State Exporter);
66 67
67our $idle; # idle handler 68our $idle; # idle handler
68our $main; # main coroutine 69our $main; # main coroutine
69our $current; # current coroutine 70our $current; # current coroutine
70 71
71our $VERSION = '4.72'; 72our $VERSION = 5.0;
72 73
73our @EXPORT = qw(async async_pool cede schedule terminate current unblock_sub); 74our @EXPORT = qw(async async_pool cede schedule terminate current unblock_sub);
74our %EXPORT_TAGS = ( 75our %EXPORT_TAGS = (
75 prio => [qw(PRIO_MAX PRIO_HIGH PRIO_NORMAL PRIO_LOW PRIO_IDLE PRIO_MIN)], 76 prio => [qw(PRIO_MAX PRIO_HIGH PRIO_NORMAL PRIO_LOW PRIO_IDLE PRIO_MIN)],
76); 77);
81=item $Coro::main 82=item $Coro::main
82 83
83This variable stores the coroutine object that represents the main 84This variable stores the coroutine object that represents the main
84program. While you cna C<ready> it and do most other things you can do to 85program. While you cna C<ready> it and do most other things you can do to
85coroutines, it is mainly useful to compare again C<$Coro::current>, to see 86coroutines, it is mainly useful to compare again C<$Coro::current>, to see
86wether you are running in the main program or not. 87whether you are running in the main program or not.
87 88
88=cut 89=cut
89 90
90$main = new Coro; 91# $main is now being initialised by Coro::State
91 92
92=item $Coro::current 93=item $Coro::current
93 94
94The coroutine object representing the current coroutine (the last 95The coroutine object representing the current coroutine (the last
95coroutine that the Coro scheduler switched to). The initial value is 96coroutine that the Coro scheduler switched to). The initial value is
96C<$main> (of course). 97C<$Coro::main> (of course).
97 98
98This variable is B<strictly> I<read-only>. You can take copies of the 99This variable is B<strictly> I<read-only>. You can take copies of the
99value stored in it and use it as any other coroutine object, but you must 100value stored in it and use it as any other coroutine object, but you must
100not otherwise modify the variable itself. 101not otherwise modify the variable itself.
101 102
102=cut 103=cut
103
104$main->{desc} = "[main::]";
105
106# maybe some other module used Coro::Specific before...
107$main->{_specific} = $current->{_specific}
108 if $current;
109
110_set_current $main;
111 104
112sub current() { $current } # [DEPRECATED] 105sub current() { $current } # [DEPRECATED]
113 106
114=item $Coro::idle 107=item $Coro::idle
115 108
151 $self->_destroy 144 $self->_destroy
152 or return; 145 or return;
153 146
154 # call all destruction callbacks 147 # call all destruction callbacks
155 $_->(@{$self->{_status}}) 148 $_->(@{$self->{_status}})
156 for @{(delete $self->{_on_destroy}) || []}; 149 for @{ delete $self->{_on_destroy} || [] };
157} 150}
158 151
159# this coroutine is necessary because a coroutine 152# this coroutine is necessary because a coroutine
160# cannot destroy itself. 153# cannot destroy itself.
161my @destroy; 154our @destroy;
162my $manager; 155our $manager;
163 156
164$manager = new Coro sub { 157$manager = new Coro sub {
165 while () { 158 while () {
166 (shift @destroy)->_cancel 159 (shift @destroy)->_cancel
167 while @destroy; 160 while @destroy;
168 161
169 &schedule; 162 &schedule;
170 } 163 }
171}; 164};
172$manager->desc ("[coro manager]"); 165$manager->{desc} = "[coro manager]";
173$manager->prio (PRIO_MAX); 166$manager->prio (PRIO_MAX);
174 167
175=back 168=back
176 169
177=head2 SIMPLE COROUTINE CREATION 170=head2 SIMPLE COROUTINE CREATION
219Similar to C<async>, but uses a coroutine pool, so you should not call 212Similar to C<async>, but uses a coroutine pool, so you should not call
220terminate or join on it (although you are allowed to), and you get a 213terminate or join on it (although you are allowed to), and you get a
221coroutine that might have executed other code already (which can be good 214coroutine that might have executed other code already (which can be good
222or bad :). 215or bad :).
223 216
224On the plus side, this function is faster than creating (and destroying) 217On the plus side, this function is about twice as fast as creating (and
225a completely new coroutine, so if you need a lot of generic coroutines in 218destroying) a completely new coroutine, so if you need a lot of generic
226quick successsion, use C<async_pool>, not C<async>. 219coroutines in quick successsion, use C<async_pool>, not C<async>.
227 220
228The code block is executed in an C<eval> context and a warning will be 221The code block is executed in an C<eval> context and a warning will be
229issued in case of an exception instead of terminating the program, as 222issued in case of an exception instead of terminating the program, as
230C<async> does. As the coroutine is being reused, stuff like C<on_destroy> 223C<async> does. As the coroutine is being reused, stuff like C<on_destroy>
231will not work in the expected way, unless you call terminate or cancel, 224will not work in the expected way, unless you call terminate or cancel,
234 227
235The priority will be reset to C<0> after each run, tracing will be 228The priority will be reset to C<0> after each run, tracing will be
236disabled, the description will be reset and the default output filehandle 229disabled, the description will be reset and the default output filehandle
237gets restored, so you can change all these. Otherwise the coroutine will 230gets restored, so you can change all these. Otherwise the coroutine will
238be re-used "as-is": most notably if you change other per-coroutine global 231be re-used "as-is": most notably if you change other per-coroutine global
239stuff such as C<$/> you I<must needs> to revert that change, which is most 232stuff such as C<$/> you I<must needs> revert that change, which is most
240simply done by using local as in: C< local $/ >. 233simply done by using local as in: C<< local $/ >>.
241 234
242The pool size is limited to C<8> idle coroutines (this can be adjusted by 235The idle pool size is limited to C<8> idle coroutines (this can be
243changing $Coro::POOL_SIZE), and there can be as many non-idle coros as 236adjusted by changing $Coro::POOL_SIZE), but there can be as many non-idle
244required. 237coros as required.
245 238
246If you are concerned about pooled coroutines growing a lot because a 239If you are concerned about pooled coroutines growing a lot because a
247single C<async_pool> used a lot of stackspace you can e.g. C<async_pool 240single C<async_pool> used a lot of stackspace you can e.g. C<async_pool
248{ terminate }> once per second or so to slowly replenish the pool. In 241{ terminate }> once per second or so to slowly replenish the pool. In
249addition to that, when the stacks used by a handler grows larger than 16kb 242addition to that, when the stacks used by a handler grows larger than 16kb
254our $POOL_SIZE = 8; 247our $POOL_SIZE = 8;
255our $POOL_RSS = 16 * 1024; 248our $POOL_RSS = 16 * 1024;
256our @async_pool; 249our @async_pool;
257 250
258sub pool_handler { 251sub pool_handler {
259 my $cb;
260
261 while () { 252 while () {
262 eval { 253 eval {
263 while () { 254 &{&_pool_handler} while 1;
264 _pool_1 $cb;
265 &$cb;
266 _pool_2 $cb;
267 &schedule;
268 }
269 }; 255 };
270 256
271 last if $@ eq "\3async_pool terminate\2\n";
272 warn $@ if $@; 257 warn $@ if $@;
273 } 258 }
274}
275
276sub async_pool(&@) {
277 # this is also inlined into the unlock_scheduler
278 my $coro = (pop @async_pool) || new Coro \&pool_handler;
279
280 $coro->{_invoke} = [@_];
281 $coro->ready;
282
283 $coro
284} 259}
285 260
286=back 261=back
287 262
288=head2 STATIC METHODS 263=head2 STATIC METHODS
307This makes C<schedule> I<the> generic method to use to block the current 282This makes C<schedule> I<the> generic method to use to block the current
308coroutine and wait for events: first you remember the current coroutine in 283coroutine and wait for events: first you remember the current coroutine in
309a variable, then arrange for some callback of yours to call C<< ->ready 284a variable, then arrange for some callback of yours to call C<< ->ready
310>> on that once some event happens, and last you call C<schedule> to put 285>> on that once some event happens, and last you call C<schedule> to put
311yourself to sleep. Note that a lot of things can wake your coroutine up, 286yourself to sleep. Note that a lot of things can wake your coroutine up,
312so you need to check wether the event indeed happened, e.g. by storing the 287so you need to check whether the event indeed happened, e.g. by storing the
313status in a variable. 288status in a variable.
314 289
315The canonical way to wait on external events is this: 290See B<HOW TO WAIT FOR A CALLBACK>, below, for some ways to wait for callbacks.
316
317 {
318 # remember current coroutine
319 my $current = $Coro::current;
320
321 # register a hypothetical event handler
322 on_event_invoke sub {
323 # wake up sleeping coroutine
324 $current->ready;
325 undef $current;
326 };
327
328 # call schedule until event occurred.
329 # in case we are woken up for other reasons
330 # (current still defined), loop.
331 Coro::schedule while $current;
332 }
333 291
334=item cede 292=item cede
335 293
336"Cede" to other coroutines. This function puts the current coroutine into 294"Cede" to other coroutines. This function puts the current coroutine into
337the ready queue and calls C<schedule>, which has the effect of giving 295the ready queue and calls C<schedule>, which has the effect of giving
356Kills/terminates/cancels all coroutines except the currently running 314Kills/terminates/cancels all coroutines except the currently running
357one. This is useful after a fork, either in the child or the parent, as 315one. This is useful after a fork, either in the child or the parent, as
358usually only one of them should inherit the running coroutines. 316usually only one of them should inherit the running coroutines.
359 317
360Note that while this will try to free some of the main programs resources, 318Note that while this will try to free some of the main programs resources,
361you cnanot free all of them, so if a coroutine that is not the main 319you cannot free all of them, so if a coroutine that is not the main
362program calls this function, there will be some one-time resource leak. 320program calls this function, there will be some one-time resource leak.
363 321
364=cut 322=cut
365 323
366sub terminate { 324sub terminate {
367 $current->cancel (@_); 325 $current->{_status} = [@_];
326 push @destroy, $current;
327 $manager->ready;
328 do { &schedule } while 1;
368} 329}
369 330
370sub killall { 331sub killall {
371 for (Coro::State::list) { 332 for (Coro::State::list) {
372 $_->cancel 333 $_->cancel
393See C<async> and C<Coro::State::new> for additional info about the 354See C<async> and C<Coro::State::new> for additional info about the
394coroutine environment. 355coroutine environment.
395 356
396=cut 357=cut
397 358
398sub _run_coro { 359sub _terminate {
399 terminate &{+shift}; 360 terminate &{+shift};
400}
401
402sub new {
403 my $class = shift;
404
405 $class->SUPER::new (\&_run_coro, @_)
406} 361}
407 362
408=item $success = $coroutine->ready 363=item $success = $coroutine->ready
409 364
410Put the given coroutine into the end of its ready queue (there is one 365Put the given coroutine into the end of its ready queue (there is one
415once all the coroutines of higher priority and all coroutines of the same 370once all the coroutines of higher priority and all coroutines of the same
416priority that were put into the ready queue earlier have been resumed. 371priority that were put into the ready queue earlier have been resumed.
417 372
418=item $is_ready = $coroutine->is_ready 373=item $is_ready = $coroutine->is_ready
419 374
420Return wether the coroutine is currently the ready queue or not, 375Return whether the coroutine is currently the ready queue or not,
421 376
422=item $coroutine->cancel (arg...) 377=item $coroutine->cancel (arg...)
423 378
424Terminates the given coroutine and makes it return the given arguments as 379Terminates the given coroutine and makes it return the given arguments as
425status (default: the empty list). Never returns if the coroutine is the 380status (default: the empty list). Never returns if the coroutine is the
427 382
428=cut 383=cut
429 384
430sub cancel { 385sub cancel {
431 my $self = shift; 386 my $self = shift;
432 $self->{_status} = [@_];
433 387
434 if ($current == $self) { 388 if ($current == $self) {
435 push @destroy, $self; 389 terminate @_;
436 $manager->ready;
437 &schedule while 1;
438 } else { 390 } else {
391 $self->{_status} = [@_];
439 $self->_cancel; 392 $self->_cancel;
440 } 393 }
441} 394}
395
396=item $coroutine->schedule_to
397
398Puts the current coroutine to sleep (like C<Coro::schedule>), but instead
399of continuing with the next coro from the ready queue, always switch to
400the given coroutine object (regardless of priority etc.). The readyness
401state of that coroutine isn't changed.
402
403This is an advanced method for special cases - I'd love to hear about any
404uses for this one.
405
406=item $coroutine->cede_to
407
408Like C<schedule_to>, but puts the current coroutine into the ready
409queue. This has the effect of temporarily switching to the given
410coroutine, and continuing some time later.
411
412This is an advanced method for special cases - I'd love to hear about any
413uses for this one.
414
415=item $coroutine->throw ([$scalar])
416
417If C<$throw> is specified and defined, it will be thrown as an exception
418inside the coroutine at the next convenient point in time. Otherwise
419clears the exception object.
420
421Coro will check for the exception each time a schedule-like-function
422returns, i.e. after each C<schedule>, C<cede>, C<< Coro::Semaphore->down
423>>, C<< Coro::Handle->readable >> and so on. Most of these functions
424detect this case and return early in case an exception is pending.
425
426The exception object will be thrown "as is" with the specified scalar in
427C<$@>, i.e. if it is a string, no line number or newline will be appended
428(unlike with C<die>).
429
430This can be used as a softer means than C<cancel> to ask a coroutine to
431end itself, although there is no guarantee that the exception will lead to
432termination, and if the exception isn't caught it might well end the whole
433program.
434
435You might also think of C<throw> as being the moral equivalent of
436C<kill>ing a coroutine with a signal (in this case, a scalar).
442 437
443=item $coroutine->join 438=item $coroutine->join
444 439
445Wait until the coroutine terminates and return any values given to the 440Wait until the coroutine terminates and return any values given to the
446C<terminate> or C<cancel> functions. C<join> can be called concurrently 441C<terminate> or C<cancel> functions. C<join> can be called concurrently
508higher values mean lower priority, just as in unix). 503higher values mean lower priority, just as in unix).
509 504
510=item $olddesc = $coroutine->desc ($newdesc) 505=item $olddesc = $coroutine->desc ($newdesc)
511 506
512Sets (or gets in case the argument is missing) the description for this 507Sets (or gets in case the argument is missing) the description for this
513coroutine. This is just a free-form string you can associate with a coroutine. 508coroutine. This is just a free-form string you can associate with a
509coroutine.
514 510
515This method simply sets the C<< $coroutine->{desc} >> member to the given string. You 511This method simply sets the C<< $coroutine->{desc} >> member to the given
516can modify this member directly if you wish. 512string. You can modify this member directly if you wish.
517
518=item $coroutine->throw ([$scalar])
519
520If C<$throw> is specified and defined, it will be thrown as an exception
521inside the coroutine at the next convinient point in time (usually after
522it gains control at the next schedule/transfer/cede). Otherwise clears the
523exception object.
524
525The exception object will be thrown "as is" with the specified scalar in
526C<$@>, i.e. if it is a string, no line number or newline will be appended
527(unlike with C<die>).
528
529This can be used as a softer means than C<cancel> to ask a coroutine to
530end itself, although there is no guarentee that the exception will lead to
531termination, and if the exception isn't caught it might well end the whole
532program.
533 513
534=cut 514=cut
535 515
536sub desc { 516sub desc {
537 my $old = $_[0]{desc}; 517 my $old = $_[0]{desc};
629# return immediately and can be reused) and because we cannot cede 609# return immediately and can be reused) and because we cannot cede
630# inside an event callback. 610# inside an event callback.
631our $unblock_scheduler = new Coro sub { 611our $unblock_scheduler = new Coro sub {
632 while () { 612 while () {
633 while (my $cb = pop @unblock_queue) { 613 while (my $cb = pop @unblock_queue) {
634 # this is an inlined copy of async_pool 614 &async_pool (@$cb);
635 my $coro = (pop @async_pool) || new Coro \&pool_handler;
636 615
637 $coro->{_invoke} = $cb;
638 $coro->ready;
639 cede; # for short-lived callbacks, this reduces pressure on the coro pool 616 # for short-lived callbacks, this reduces pressure on the coro pool
617 # as the chance is very high that the async_poll coro will be back
618 # in the idle state when cede returns
619 cede;
640 } 620 }
641 schedule; # sleep well 621 schedule; # sleep well
642 } 622 }
643}; 623};
644$unblock_scheduler->desc ("[unblock_sub scheduler]"); 624$unblock_scheduler->{desc} = "[unblock_sub scheduler]";
645 625
646sub unblock_sub(&) { 626sub unblock_sub(&) {
647 my $cb = shift; 627 my $cb = shift;
648 628
649 sub { 629 sub {
650 unshift @unblock_queue, [$cb, @_]; 630 unshift @unblock_queue, [$cb, @_];
651 $unblock_scheduler->ready; 631 $unblock_scheduler->ready;
652 } 632 }
653} 633}
654 634
635=item $cb = Coro::rouse_cb
636
637Create and return a "rouse callback". That's a code reference that, when
638called, will save its arguments and notify the owner coroutine of the
639callback.
640
641See the next function.
642
643=item @args = Coro::rouse_wait [$cb]
644
645Wait for the specified rouse callback (or the last one tht was created in
646this coroutine).
647
648As soon as the callback is invoked (or when the calback was invoked before
649C<rouse_wait>), it will return a copy of the arguments originally passed
650to the rouse callback.
651
652See the section B<HOW TO WAIT FOR A CALLBACK> for an actual usage example.
653
655=back 654=back
656 655
657=cut 656=cut
658 657
6591; 6581;
660 659
660=head1 HOW TO WAIT FOR A CALLBACK
661
662It is very common for a coroutine to wait for some callback to be
663called. This occurs naturally when you use coroutines in an otherwise
664event-based program, or when you use event-based libraries.
665
666These typically register a callback for some event, and call that callback
667when the event occured. In a coroutine, however, you typically want to
668just wait for the event, simplyifying things.
669
670For example C<< AnyEvent->child >> registers a callback to be called when
671a specific child has exited:
672
673 my $child_watcher = AnyEvent->child (pid => $pid, cb => sub { ... });
674
675But from withina coroutine, you often just want to write this:
676
677 my $status = wait_for_child $pid;
678
679Coro offers two functions specifically designed to make this easy,
680C<Coro::rouse_cb> and C<Coro::rouse_wait>.
681
682The first function, C<rouse_cb>, generates and returns a callback that,
683when invoked, will save it's arguments and notify the coroutine that
684created the callback.
685
686The second function, C<rouse_wait>, waits for the callback to be called
687(by calling C<schedule> to go to sleep) and returns the arguments
688originally passed to the callback.
689
690Using these functions, it becomes easy to write the C<wait_for_child>
691function mentioned above:
692
693 sub wait_for_child($) {
694 my ($pid) = @_;
695
696 my $watcher = AnyEvent->child (pid => $pid, cb => Coro::rouse_cb);
697
698 my ($rpid, $rstatus) = Coro::rouse_wait;
699 $rstatus
700 }
701
702In the case where C<rouse_cb> and C<rouse_wait> are not flexible enough,
703you can roll your own, using C<schedule>:
704
705 sub wait_for_child($) {
706 my ($pid) = @_;
707
708 # store the current coroutine in $current,
709 # and provide result variables for the closure passed to ->child
710 my $current = $Coro::current;
711 my ($done, $rstatus);
712
713 # pass a closure to ->child
714 my $watcher = AnyEvent->child (pid => $pid, cb => sub {
715 $rstatus = $_[1]; # remember rstatus
716 $done = 1; # mark $rstatus as valud
717 });
718
719 # wait until the closure has been called
720 schedule while !$done;
721
722 $rstatus
723 }
724
725
661=head1 BUGS/LIMITATIONS 726=head1 BUGS/LIMITATIONS
727
728=over 4
729
730=item fork with pthread backend
731
732When Coro is compiled using the pthread backend (which isn't recommended
733but required on many BSDs as their libcs are completely broken), then
734coroutines will not survive a fork. There is no known workaround except to
735fix your libc and use a saner backend.
736
737=item perl process emulation ("threads")
662 738
663This module is not perl-pseudo-thread-safe. You should only ever use this 739This module is not perl-pseudo-thread-safe. You should only ever use this
664module from the same thread (this requirement might be removed in the 740module from the same thread (this requirement might be removed in the
665future to allow per-thread schedulers, but Coro::State does not yet allow 741future to allow per-thread schedulers, but Coro::State does not yet allow
666this). I recommend disabling thread support and using processes, as this 742this). I recommend disabling thread support and using processes, as having
667is much faster and uses less memory. 743the windows process emulation enabled under unix roughly halves perl
744performance, even when not used.
745
746=item coroutine switching not signal safe
747
748You must not switch to another coroutine from within a signal handler
749(only relevant with %SIG - most event libraries provide safe signals).
750
751That means you I<MUST NOT> call any function that might "block" the
752current coroutine - C<cede>, C<schedule> C<< Coro::Semaphore->down >> or
753anything that calls those. Everything else, including calling C<ready>,
754works.
755
756=back
757
668 758
669=head1 SEE ALSO 759=head1 SEE ALSO
670 760
671Event-Loop integration: L<Coro::AnyEvent>, L<Coro::EV>, L<Coro::Event>. 761Event-Loop integration: L<Coro::AnyEvent>, L<Coro::EV>, L<Coro::Event>.
672 762

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines