| … | |
… | |
| 52 | our $main; # main coroutine |
52 | our $main; # main coroutine |
| 53 | our $current; # current coroutine |
53 | our $current; # current coroutine |
| 54 | |
54 | |
| 55 | our $VERSION = '3.3'; |
55 | our $VERSION = '3.3'; |
| 56 | |
56 | |
| 57 | our @EXPORT = qw(async cede schedule terminate current unblock_sub); |
57 | our @EXPORT = qw(async async_pool cede schedule terminate current unblock_sub); |
| 58 | our %EXPORT_TAGS = ( |
58 | our %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 | ); |
| 61 | our @EXPORT_OK = (@{$EXPORT_TAGS{prio}}, qw(nready)); |
61 | our @EXPORT_OK = (@{$EXPORT_TAGS{prio}}, qw(nready)); |
| 62 | |
62 | |
| … | |
… | |
| 203 | my $coro = new Coro @_; |
203 | my $coro = new Coro @_; |
| 204 | $coro->ready; |
204 | $coro->ready; |
| 205 | $coro |
205 | $coro |
| 206 | } |
206 | } |
| 207 | |
207 | |
|
|
208 | =item async_pool { ... } [@args...] |
|
|
209 | |
|
|
210 | Similar to C<async>, but uses a coroutine pool, so you should not call |
|
|
211 | terminate or join (although you are allowed to), and you get a coroutine |
|
|
212 | that might have executed other code already (which can be good or bad :). |
|
|
213 | |
|
|
214 | Also, the block is executed in an C<eval> context and a warning will be |
|
|
215 | issued in case of an exception instead of terminating the program, as C<async> does. |
|
|
216 | |
|
|
217 | The priority will be reset to C<0> after each job, otherwise the coroutine |
|
|
218 | will be re-used "as-is". |
|
|
219 | |
|
|
220 | The pool size is limited to 8 idle coroutines (this can be adjusted by |
|
|
221 | changing $Coro::POOL_SIZE), and there can be as many non-idle coros as |
|
|
222 | required. |
|
|
223 | |
|
|
224 | If you are concerned about pooled coroutines growing a lot because a |
|
|
225 | single C<async_pool> used a lot of stackspace you can e.g. C<async_pool { |
|
|
226 | terminate }> once per second or so to slowly replenish the pool. |
|
|
227 | |
|
|
228 | =cut |
|
|
229 | |
|
|
230 | our $POOL_SIZE = 8; |
|
|
231 | our @pool; |
|
|
232 | |
|
|
233 | sub pool_handler { |
|
|
234 | while () { |
|
|
235 | my ($cb, @arg) = @{ delete $current->{_invoke} }; |
|
|
236 | |
|
|
237 | eval { |
|
|
238 | $cb->(@arg); |
|
|
239 | }; |
|
|
240 | warn $@ if $@; |
|
|
241 | |
|
|
242 | last if @pool >= $POOL_SIZE; |
|
|
243 | push @pool, $current; |
|
|
244 | |
|
|
245 | $current->prio (0); |
|
|
246 | schedule; |
|
|
247 | } |
|
|
248 | } |
|
|
249 | |
|
|
250 | sub async_pool(&@) { |
|
|
251 | # this is also inlined into the unlock_scheduler |
|
|
252 | my $coro = (pop @pool or new Coro \&pool_handler); |
|
|
253 | |
|
|
254 | $coro->{_invoke} = [@_]; |
|
|
255 | $coro->ready; |
|
|
256 | |
|
|
257 | $coro |
|
|
258 | } |
|
|
259 | |
| 208 | =item schedule |
260 | =item schedule |
| 209 | |
261 | |
| 210 | Calls the scheduler. Please note that the current coroutine will not be put |
262 | Calls the scheduler. Please note that the current coroutine will not be put |
| 211 | into the ready queue, so calling this function usually means you will |
263 | into the ready queue, so calling this function usually means you will |
| 212 | never be called again unless something else (e.g. an event handler) calls |
264 | never be called again unless something else (e.g. an event handler) calls |
| … | |
… | |
| 461 | In short: simply use C<unblock_sub { ... }> instead of C<sub { ... }> when |
513 | In short: simply use C<unblock_sub { ... }> instead of C<sub { ... }> when |
| 462 | creating event callbacks that want to block. |
514 | creating event callbacks that want to block. |
| 463 | |
515 | |
| 464 | =cut |
516 | =cut |
| 465 | |
517 | |
| 466 | our @unblock_pool; |
|
|
| 467 | our @unblock_queue; |
518 | our @unblock_queue; |
| 468 | our $UNBLOCK_POOL_SIZE = 2; |
|
|
| 469 | |
519 | |
| 470 | sub unblock_handler_ { |
520 | # we create a special coro because we want to cede, |
| 471 | while () { |
521 | # to reduce pressure on the coro pool (because most callbacks |
| 472 | my ($cb, @arg) = @{ delete $Coro::current->{arg} }; |
522 | # return immediately and can be reused) and because we cannot cede |
| 473 | $cb->(@arg); |
523 | # inside an event callback. |
| 474 | |
|
|
| 475 | last if @unblock_pool >= $UNBLOCK_POOL_SIZE; |
|
|
| 476 | push @unblock_pool, $Coro::current; |
|
|
| 477 | schedule; |
|
|
| 478 | } |
|
|
| 479 | } |
|
|
| 480 | |
|
|
| 481 | our $unblock_scheduler = async { |
524 | our $unblock_scheduler = async { |
| 482 | while () { |
525 | while () { |
| 483 | while (my $cb = pop @unblock_queue) { |
526 | while (my $cb = pop @unblock_queue) { |
|
|
527 | # this is an inlined copy of async_pool |
| 484 | my $handler = (pop @unblock_pool or new Coro \&unblock_handler_); |
528 | my $coro = (pop @pool or new Coro \&pool_handler); |
| 485 | $handler->{arg} = $cb; |
529 | |
|
|
530 | $coro->{_invoke} = $cb; |
| 486 | $handler->ready; |
531 | $coro->ready; |
| 487 | cede; |
532 | cede; # for short-lived callbacks, this reduces pressure on the coro pool |
| 488 | } |
533 | } |
| 489 | |
534 | schedule; # sleep well |
| 490 | schedule; |
|
|
| 491 | } |
535 | } |
| 492 | }; |
536 | }; |
| 493 | |
537 | |
| 494 | sub unblock_sub(&) { |
538 | sub unblock_sub(&) { |
| 495 | my $cb = shift; |
539 | my $cb = shift; |
| 496 | |
540 | |
| 497 | sub { |
541 | sub { |
| 498 | push @unblock_queue, [$cb, @_]; |
542 | unshift @unblock_queue, [$cb, @_]; |
| 499 | $unblock_scheduler->ready; |
543 | $unblock_scheduler->ready; |
| 500 | } |
544 | } |
| 501 | } |
545 | } |
| 502 | |
546 | |
| 503 | =back |
547 | =back |
