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

Comparing Coro/Coro.pm (file contents):
Revision 1.129 by root, Wed Sep 19 22:33:08 2007 UTC vs.
Revision 1.145 by root, Wed Oct 3 16:03:17 2007 UTC

50 50
51our $idle; # idle handler 51our $idle; # idle handler
52our $main; # main coroutine 52our $main; # main coroutine
53our $current; # current coroutine 53our $current; # current coroutine
54 54
55our $VERSION = '3.7'; 55our $VERSION = '4.0';
56 56
57our @EXPORT = qw(async async_pool cede schedule terminate current unblock_sub); 57our @EXPORT = qw(async async_pool cede schedule terminate current unblock_sub);
58our %EXPORT_TAGS = ( 58our %EXPORT_TAGS = (
59 prio => [qw(PRIO_MAX PRIO_HIGH PRIO_NORMAL PRIO_LOW PRIO_IDLE PRIO_MIN)], 59 prio => [qw(PRIO_MAX PRIO_HIGH PRIO_NORMAL PRIO_LOW PRIO_IDLE PRIO_MIN)],
60); 60);
113reasons. If performance is not essential you are encouraged to use the 113reasons. If performance is not essential you are encouraged to use the
114C<Coro::current> function instead. 114C<Coro::current> function instead.
115 115
116=cut 116=cut
117 117
118$main->{desc} = "[main::]";
119
118# maybe some other module used Coro::Specific before... 120# maybe some other module used Coro::Specific before...
119$main->{specific} = $current->{specific} 121$main->{_specific} = $current->{_specific}
120 if $current; 122 if $current;
121 123
122_set_current $main; 124_set_current $main;
123 125
124sub current() { $current } 126sub current() { $current }
149 # free coroutine data and mark as destructed 151 # free coroutine data and mark as destructed
150 $self->_destroy 152 $self->_destroy
151 or return; 153 or return;
152 154
153 # call all destruction callbacks 155 # call all destruction callbacks
154 $_->(@{$self->{status}}) 156 $_->(@{$self->{_status}})
155 for @{(delete $self->{destroy_cb}) || []}; 157 for @{(delete $self->{_on_destroy}) || []};
156} 158}
157 159
158# this coroutine is necessary because a coroutine 160# this coroutine is necessary because a coroutine
159# cannot destroy itself. 161# cannot destroy itself.
160my @destroy; 162my @destroy;
161my $manager; 163my $manager;
162 164
163$manager = new Coro sub { 165$manager = new Coro sub {
164 $current->desc ("[coro manager]");
165
166 while () { 166 while () {
167 (shift @destroy)->_cancel 167 (shift @destroy)->_cancel
168 while @destroy; 168 while @destroy;
169 169
170 &schedule; 170 &schedule;
171 } 171 }
172}; 172};
173 173$manager->desc ("[coro manager]");
174$manager->prio (PRIO_MAX); 174$manager->prio (PRIO_MAX);
175 175
176# static methods. not really. 176# static methods. not really.
177 177
178=back 178=back
186=item async { ... } [@args...] 186=item async { ... } [@args...]
187 187
188Create a new asynchronous coroutine and return it's coroutine object 188Create a new asynchronous coroutine and return it's coroutine object
189(usually unused). When the sub returns the new coroutine is automatically 189(usually unused). When the sub returns the new coroutine is automatically
190terminated. 190terminated.
191
192See the C<Coro::State::new> constructor for info about the coroutine
193environment.
191 194
192Calling C<exit> in a coroutine will do the same as calling exit outside 195Calling C<exit> in a coroutine will do the same as calling exit outside
193the coroutine. Likewise, when the coroutine dies, the program will exit, 196the coroutine. Likewise, when the coroutine dies, the program will exit,
194just as it would in the main program. 197just as it would in the main program.
195 198
224The pool size is limited to 8 idle coroutines (this can be adjusted by 227The pool size is limited to 8 idle coroutines (this can be adjusted by
225changing $Coro::POOL_SIZE), and there can be as many non-idle coros as 228changing $Coro::POOL_SIZE), and there can be as many non-idle coros as
226required. 229required.
227 230
228If you are concerned about pooled coroutines growing a lot because a 231If you are concerned about pooled coroutines growing a lot because a
229single C<async_pool> used a lot of stackspace you can e.g. C<async_pool { 232single C<async_pool> used a lot of stackspace you can e.g. C<async_pool
230terminate }> once per second or so to slowly replenish the pool. 233{ terminate }> once per second or so to slowly replenish the pool. In
234addition to that, when the stacks used by a handler grows larger than 16kb
235(adjustable with $Coro::POOL_RSS) it will also exit.
231 236
232=cut 237=cut
233 238
234our $POOL_SIZE = 8; 239our $POOL_SIZE = 8;
240our $POOL_RSS = 16 * 1024;
235our @pool; 241our @async_pool;
236 242
237sub pool_handler { 243sub pool_handler {
244 my $cb;
245
238 while () { 246 while () {
239 $current->{desc} = "[async_pool]";
240
241 eval { 247 eval {
242 my ($cb, @arg) = @{ delete $current->{_invoke} or return }; 248 while () {
243 $cb->(@arg); 249 _pool_1 $cb;
250 &$cb;
251 _pool_2 $cb;
252 &schedule;
253 }
244 }; 254 };
255
256 last if $@ eq "\3terminate\2\n";
245 warn $@ if $@; 257 warn $@ if $@;
246
247 last if @pool >= $POOL_SIZE;
248
249 push @pool, $current;
250 $current->{desc} = "[async_pool idle]";
251 $current->save (Coro::State::SAVE_DEF);
252 $current->prio (0);
253 schedule;
254 } 258 }
255} 259}
256 260
257sub async_pool(&@) { 261sub async_pool(&@) {
258 # this is also inlined into the unlock_scheduler 262 # this is also inlined into the unlock_scheduler
259 my $coro = (pop @pool) || new Coro \&pool_handler;; 263 my $coro = (pop @async_pool) || new Coro \&pool_handler;
260 264
261 $coro->{_invoke} = [@_]; 265 $coro->{_invoke} = [@_];
262 $coro->ready; 266 $coro->ready;
263 267
264 $coro 268 $coro
307 311
308=item terminate [arg...] 312=item terminate [arg...]
309 313
310Terminates the current coroutine with the given status values (see L<cancel>). 314Terminates the current coroutine with the given status values (see L<cancel>).
311 315
316=item killall
317
318Kills/terminates/cancels all coroutines except the currently running
319one. This is useful after a fork, either in the child or the parent, as
320usually only one of them should inherit the running coroutines.
321
312=cut 322=cut
313 323
314sub terminate { 324sub terminate {
315 $current->cancel (@_); 325 $current->cancel (@_);
326}
327
328sub killall {
329 for (Coro::State::list) {
330 $_->cancel
331 if $_ != $current && UNIVERSAL::isa $_, "Coro";
332 }
316} 333}
317 334
318=back 335=back
319 336
320# dynamic methods 337# dynamic methods
330Create a new coroutine and return it. When the sub returns the coroutine 347Create a new coroutine and return it. When the sub returns the coroutine
331automatically terminates as if C<terminate> with the returned values were 348automatically terminates as if C<terminate> with the returned values were
332called. To make the coroutine run you must first put it into the ready queue 349called. To make the coroutine run you must first put it into the ready queue
333by calling the ready method. 350by calling the ready method.
334 351
335See C<async> for additional discussion. 352See C<async> and C<Coro::State::new> for additional info about the
353coroutine environment.
336 354
337=cut 355=cut
338 356
339sub _run_coro { 357sub _run_coro {
340 terminate &{+shift}; 358 terminate &{+shift};
364 382
365=cut 383=cut
366 384
367sub cancel { 385sub cancel {
368 my $self = shift; 386 my $self = shift;
369 $self->{status} = [@_]; 387 $self->{_status} = [@_];
370 388
371 if ($current == $self) { 389 if ($current == $self) {
372 push @destroy, $self; 390 push @destroy, $self;
373 $manager->ready; 391 $manager->ready;
374 &schedule while 1; 392 &schedule while 1;
378} 396}
379 397
380=item $coroutine->join 398=item $coroutine->join
381 399
382Wait until the coroutine terminates and return any values given to the 400Wait until the coroutine terminates and return any values given to the
383C<terminate> or C<cancel> functions. C<join> can be called multiple times 401C<terminate> or C<cancel> functions. C<join> can be called concurrently
384from multiple coroutine. 402from multiple coroutines.
385 403
386=cut 404=cut
387 405
388sub join { 406sub join {
389 my $self = shift; 407 my $self = shift;
390 408
391 unless ($self->{status}) { 409 unless ($self->{_status}) {
392 my $current = $current; 410 my $current = $current;
393 411
394 push @{$self->{destroy_cb}}, sub { 412 push @{$self->{_on_destroy}}, sub {
395 $current->ready; 413 $current->ready;
396 undef $current; 414 undef $current;
397 }; 415 };
398 416
399 &schedule while $current; 417 &schedule while $current;
400 } 418 }
401 419
402 wantarray ? @{$self->{status}} : $self->{status}[0]; 420 wantarray ? @{$self->{_status}} : $self->{_status}[0];
403} 421}
404 422
405=item $coroutine->on_destroy (\&cb) 423=item $coroutine->on_destroy (\&cb)
406 424
407Registers a callback that is called when this coroutine gets destroyed, 425Registers a callback that is called when this coroutine gets destroyed,
411=cut 429=cut
412 430
413sub on_destroy { 431sub on_destroy {
414 my ($self, $cb) = @_; 432 my ($self, $cb) = @_;
415 433
416 push @{ $self->{destroy_cb} }, $cb; 434 push @{ $self->{_on_destroy} }, $cb;
417} 435}
418 436
419=item $oldprio = $coroutine->prio ($newprio) 437=item $oldprio = $coroutine->prio ($newprio)
420 438
421Sets (or gets, if the argument is missing) the priority of the 439Sets (or gets, if the argument is missing) the priority of the
445 463
446=item $olddesc = $coroutine->desc ($newdesc) 464=item $olddesc = $coroutine->desc ($newdesc)
447 465
448Sets (or gets in case the argument is missing) the description for this 466Sets (or gets in case the argument is missing) the description for this
449coroutine. This is just a free-form string you can associate with a coroutine. 467coroutine. This is just a free-form string you can associate with a coroutine.
468
469This method simply sets the C<< $coroutine->{desc} >> member to the given string. You
470can modify this member directly if you wish.
450 471
451=cut 472=cut
452 473
453sub desc { 474sub desc {
454 my $old = $_[0]{desc}; 475 my $old = $_[0]{desc};
530 551
531# we create a special coro because we want to cede, 552# we create a special coro because we want to cede,
532# to reduce pressure on the coro pool (because most callbacks 553# to reduce pressure on the coro pool (because most callbacks
533# return immediately and can be reused) and because we cannot cede 554# return immediately and can be reused) and because we cannot cede
534# inside an event callback. 555# inside an event callback.
535our $unblock_scheduler = async { 556our $unblock_scheduler = new Coro sub {
536 $current->desc ("[unblock_sub scheduler]");
537 while () { 557 while () {
538 while (my $cb = pop @unblock_queue) { 558 while (my $cb = pop @unblock_queue) {
539 # this is an inlined copy of async_pool 559 # this is an inlined copy of async_pool
540 my $coro = (pop @pool or new Coro \&pool_handler); 560 my $coro = (pop @async_pool) || new Coro \&pool_handler;
541 561
542 $coro->{_invoke} = $cb; 562 $coro->{_invoke} = $cb;
543 $coro->ready; 563 $coro->ready;
544 cede; # for short-lived callbacks, this reduces pressure on the coro pool 564 cede; # for short-lived callbacks, this reduces pressure on the coro pool
545 } 565 }
546 schedule; # sleep well 566 schedule; # sleep well
547 } 567 }
548}; 568};
569$unblock_scheduler->desc ("[unblock_sub scheduler]");
549 570
550sub unblock_sub(&) { 571sub unblock_sub(&) {
551 my $cb = shift; 572 my $cb = shift;
552 573
553 sub { 574 sub {

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines