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

Comparing IO-AIO/AIO.pm (file contents):
Revision 1.87 by root, Sun Oct 29 00:52:02 2006 UTC vs.
Revision 1.119 by root, Sun Dec 2 20:54:33 2007 UTC

5=head1 SYNOPSIS 5=head1 SYNOPSIS
6 6
7 use IO::AIO; 7 use IO::AIO;
8 8
9 aio_open "/etc/passwd", O_RDONLY, 0, sub { 9 aio_open "/etc/passwd", O_RDONLY, 0, sub {
10 my ($fh) = @_; 10 my $fh = shift
11 or die "/etc/passwd: $!";
11 ... 12 ...
12 }; 13 };
13 14
14 aio_unlink "/tmp/file", sub { }; 15 aio_unlink "/tmp/file", sub { };
15 16
25 $req->cancel; # cancel request if still in queue 26 $req->cancel; # cancel request if still in queue
26 27
27 my $grp = aio_group sub { print "all stats done\n" }; 28 my $grp = aio_group sub { print "all stats done\n" };
28 add $grp aio_stat "..." for ...; 29 add $grp aio_stat "..." for ...;
29 30
30 # AnyEvent integration 31 # AnyEvent integration (EV, Event, Glib, Tk, urxvt, pureperl...)
31 open my $fh, "<&=" . IO::AIO::poll_fileno or die "$!"; 32 open my $fh, "<&=" . IO::AIO::poll_fileno or die "$!";
32 my $w = AnyEvent->io (fh => $fh, poll => 'r', cb => sub { IO::AIO::poll_cb }); 33 my $w = AnyEvent->io (fh => $fh, poll => 'r', cb => sub { IO::AIO::poll_cb });
34
35 # EV integration
36 my $w = EV::io IO::AIO::poll_fileno, EV::READ, \&IO::AIO::poll_cb;
33 37
34 # Event integration 38 # Event integration
35 Event->io (fd => IO::AIO::poll_fileno, 39 Event->io (fd => IO::AIO::poll_fileno,
36 poll => 'r', 40 poll => 'r',
37 cb => \&IO::AIO::poll_cb); 41 cb => \&IO::AIO::poll_cb);
61etc.), but can also be used to easily do operations in parallel that are 65etc.), but can also be used to easily do operations in parallel that are
62normally done sequentially, e.g. stat'ing many files, which is much faster 66normally done sequentially, e.g. stat'ing many files, which is much faster
63on a RAID volume or over NFS when you do a number of stat operations 67on a RAID volume or over NFS when you do a number of stat operations
64concurrently. 68concurrently.
65 69
66While this works on all types of file descriptors (for example sockets), 70While most of this works on all types of file descriptors (for
67using these functions on file descriptors that support nonblocking 71example sockets), using these functions on file descriptors that
68operation (again, sockets, pipes etc.) is very inefficient. Use an event 72support nonblocking operation (again, sockets, pipes etc.) is very
69loop for that (such as the L<Event|Event> module): IO::AIO will naturally 73inefficient. Use an event loop for that (such as the L<Event|Event>
70fit into such an event loop itself. 74module): IO::AIO will naturally fit into such an event loop itself.
71 75
72In this version, a number of threads are started that execute your 76In this version, a number of threads are started that execute your
73requests and signal their completion. You don't need thread support 77requests and signal their completion. You don't need thread support
74in perl, and the threads created by this module will not be visible 78in perl, and the threads created by this module will not be visible
75to perl. In the future, this module might make use of the native aio 79to perl. In the future, this module might make use of the native aio
77not well-supported or restricted (GNU/Linux doesn't allow them on normal 81not well-supported or restricted (GNU/Linux doesn't allow them on normal
78files currently, for example), and they would only support aio_read and 82files currently, for example), and they would only support aio_read and
79aio_write, so the remaining functionality would have to be implemented 83aio_write, so the remaining functionality would have to be implemented
80using threads anyway. 84using threads anyway.
81 85
82Although the module will work with in the presence of other (Perl-) 86Although the module will work in the presence of other (Perl-) threads,
83threads, it is currently not reentrant in any way, so use appropriate 87it is currently not reentrant in any way, so use appropriate locking
84locking yourself, always call C<poll_cb> from within the same thread, or 88yourself, always call C<poll_cb> from within the same thread, or never
85never call C<poll_cb> (or other C<aio_> functions) recursively. 89call C<poll_cb> (or other C<aio_> functions) recursively.
86 90
87=head2 EXAMPLE 91=head2 EXAMPLE
88 92
89This is a simple example that uses the Event module and loads 93This is a simple example that uses the Event module and loads
90F</etc/passwd> asynchronously: 94F</etc/passwd> asynchronously:
98 poll => 'r', 102 poll => 'r',
99 cb => \&IO::AIO::poll_cb); 103 cb => \&IO::AIO::poll_cb);
100 104
101 # queue the request to open /etc/passwd 105 # queue the request to open /etc/passwd
102 aio_open "/etc/passwd", O_RDONLY, 0, sub { 106 aio_open "/etc/passwd", O_RDONLY, 0, sub {
103 my $fh = $_[0] 107 my $fh = shift
104 or die "error while opening: $!"; 108 or die "error while opening: $!";
105 109
106 # stat'ing filehandles is generally non-blocking 110 # stat'ing filehandles is generally non-blocking
107 my $size = -s $fh; 111 my $size = -s $fh;
108 112
176Request has reached the end of its lifetime and holds no resources anymore 180Request has reached the end of its lifetime and holds no resources anymore
177(except possibly for the Perl object, but its connection to the actual 181(except possibly for the Perl object, but its connection to the actual
178aio request is severed and calling its methods will either do nothing or 182aio request is severed and calling its methods will either do nothing or
179result in a runtime error). 183result in a runtime error).
180 184
185=back
186
181=cut 187=cut
182 188
183package IO::AIO; 189package IO::AIO;
190
191use Carp ();
184 192
185no warnings; 193no warnings;
186use strict 'vars'; 194use strict 'vars';
187 195
188use base 'Exporter'; 196use base 'Exporter';
189 197
190BEGIN { 198BEGIN {
191 our $VERSION = '2.1'; 199 our $VERSION = '2.6';
192 200
193 our @AIO_REQ = qw(aio_sendfile aio_read aio_write aio_open aio_close aio_stat 201 our @AIO_REQ = qw(aio_sendfile aio_read aio_write aio_open aio_close aio_stat
194 aio_lstat aio_unlink aio_rmdir aio_readdir aio_scandir aio_symlink 202 aio_lstat aio_unlink aio_rmdir aio_readdir aio_scandir aio_symlink
195 aio_fsync aio_fdatasync aio_readahead aio_rename aio_link aio_move 203 aio_readlink aio_sync aio_fsync aio_fdatasync aio_readahead aio_rename aio_link
196 aio_copy aio_group aio_nop aio_mknod); 204 aio_move aio_copy aio_group aio_nop aio_mknod aio_load aio_rmtree aio_mkdir
205 aio_chown aio_chmod aio_utime aio_truncate);
197 our @EXPORT = (@AIO_REQ, qw(aioreq_pri aioreq_nice)); 206 our @EXPORT = (@AIO_REQ, qw(aioreq_pri aioreq_nice aio_block));
198 our @EXPORT_OK = qw(poll_fileno poll_cb poll_wait flush 207 our @EXPORT_OK = qw(poll_fileno poll_cb poll_wait flush
199 min_parallel max_parallel max_idle 208 min_parallel max_parallel max_idle
200 nreqs nready npending nthreads 209 nreqs nready npending nthreads
201 max_poll_time max_poll_reqs); 210 max_poll_time max_poll_reqs);
202 211
267 aio_read $_[0], ..., sub { 276 aio_read $_[0], ..., sub {
268 ... 277 ...
269 }; 278 };
270 }; 279 };
271 280
281
272=item aioreq_nice $pri_adjust 282=item aioreq_nice $pri_adjust
273 283
274Similar to C<aioreq_pri>, but subtracts the given value from the current 284Similar to C<aioreq_pri>, but subtracts the given value from the current
275priority, so the effect is cumulative. 285priority, so the effect is cumulative.
286
276 287
277=item aio_open $pathname, $flags, $mode, $callback->($fh) 288=item aio_open $pathname, $flags, $mode, $callback->($fh)
278 289
279Asynchronously open or create a file and call the callback with a newly 290Asynchronously open or create a file and call the callback with a newly
280created filehandle for the file. 291created filehandle for the file.
286list. They are the same as used by C<sysopen>. 297list. They are the same as used by C<sysopen>.
287 298
288Likewise, C<$mode> specifies the mode of the newly created file, if it 299Likewise, C<$mode> specifies the mode of the newly created file, if it
289didn't exist and C<O_CREAT> has been given, just like perl's C<sysopen>, 300didn't exist and C<O_CREAT> has been given, just like perl's C<sysopen>,
290except that it is mandatory (i.e. use C<0> if you don't create new files, 301except that it is mandatory (i.e. use C<0> if you don't create new files,
291and C<0666> or C<0777> if you do). 302and C<0666> or C<0777> if you do). Note that the C<$mode> will be modified
303by the umask in effect then the request is being executed, so better never
304change the umask.
292 305
293Example: 306Example:
294 307
295 aio_open "/etc/passwd", O_RDONLY, 0, sub { 308 aio_open "/etc/passwd", O_RDONLY, 0, sub {
296 if ($_[0]) { 309 if ($_[0]) {
299 } else { 312 } else {
300 die "open failed: $!\n"; 313 die "open failed: $!\n";
301 } 314 }
302 }; 315 };
303 316
317
304=item aio_close $fh, $callback->($status) 318=item aio_close $fh, $callback->($status)
305 319
306Asynchronously close a file and call the callback with the result 320Asynchronously close a file and call the callback with the result
307code. I<WARNING:> although accepted, you should not pass in a perl 321code.
308filehandle here, as perl will likely close the file descriptor another
309time when the filehandle is destroyed. Normally, you can safely call perls
310C<close> or just let filehandles go out of scope.
311 322
312This is supposed to be a bug in the API, so that might change. It's 323Unfortunately, you can't do this to perl. Perl I<insists> very strongly on
313therefore best to avoid this function. 324closing the file descriptor associated with the filehandle itself. Here is
325what aio_close will try:
326
327 1. dup()licate the fd
328 2. asynchronously close() the duplicated fd
329 3. dup()licate the fd once more
330 4. let perl close() the filehandle
331 5. asynchronously close the duplicated fd
332
333The idea is that the first close() flushes stuff to disk that closing an
334fd will flush, so when perl closes the fd, nothing much will need to be
335flushed. The second async. close() will then flush stuff to disk that
336closing the last fd to the file will flush.
337
338Just FYI, SuSv3 has this to say on close:
339
340 All outstanding record locks owned by the process on the file
341 associated with the file descriptor shall be removed.
342
343 If fildes refers to a socket, close() shall cause the socket to be
344 destroyed. ... close() shall block for up to the current linger
345 interval until all data is transmitted.
346 [this actually sounds like a specification bug, but who knows]
347
348And at least Linux additionally actually flushes stuff on every close,
349even when the file itself is still open.
350
351Sounds enourmously inefficient and complicated? Yes... please show me how
352to nuke perl's fd out of existence...
353
354=cut
355
356sub aio_close($;$) {
357 aio_block {
358 my ($fh, $cb) = @_;
359
360 my $pri = aioreq_pri;
361 my $grp = aio_group $cb;
362
363 my $fd = fileno $fh;
364
365 defined $fd or Carp::croak "aio_close called with fd-less filehandle";
366
367 # if the dups fail we will simply get EBADF
368 my $fd2 = _dup $fd;
369 aioreq_pri $pri;
370 add $grp _aio_close $fd2, sub {
371 my $fd2 = _dup $fd;
372 close $fh;
373 aioreq_pri $pri;
374 add $grp _aio_close $fd2, sub {
375 $grp->result ($_[0]);
376 };
377 };
378
379 $grp
380 }
381}
382
314 383
315=item aio_read $fh,$offset,$length, $data,$dataoffset, $callback->($retval) 384=item aio_read $fh,$offset,$length, $data,$dataoffset, $callback->($retval)
316 385
317=item aio_write $fh,$offset,$length, $data,$dataoffset, $callback->($retval) 386=item aio_write $fh,$offset,$length, $data,$dataoffset, $callback->($retval)
318 387
319Reads or writes C<length> bytes from the specified C<fh> and C<offset> 388Reads or writes C<$length> bytes from the specified C<$fh> and C<$offset>
320into the scalar given by C<data> and offset C<dataoffset> and calls the 389into the scalar given by C<$data> and offset C<$dataoffset> and calls the
321callback without the actual number of bytes read (or -1 on error, just 390callback without the actual number of bytes read (or -1 on error, just
322like the syscall). 391like the syscall).
323 392
393If C<$offset> is undefined, then the current file descriptor offset will
394be used (and updated), otherwise the file descriptor offset will not be
395changed by these calls.
396
397If C<$length> is undefined in C<aio_write>, use the remaining length of C<$data>.
398
399If C<$dataoffset> is less than zero, it will be counted from the end of
400C<$data>.
401
324The C<$data> scalar I<MUST NOT> be modified in any way while the request 402The C<$data> scalar I<MUST NOT> be modified in any way while the request
325is outstanding. Modifying it can result in segfaults or WW3 (if the 403is outstanding. Modifying it can result in segfaults or World War III (if
326necessary/optional hardware is installed). 404the necessary/optional hardware is installed).
327 405
328Example: Read 15 bytes at offset 7 into scalar C<$buffer>, starting at 406Example: Read 15 bytes at offset 7 into scalar C<$buffer>, starting at
329offset C<0> within the scalar: 407offset C<0> within the scalar:
330 408
331 aio_read $fh, 7, 15, $buffer, 0, sub { 409 aio_read $fh, 7, 15, $buffer, 0, sub {
332 $_[0] > 0 or die "read error: $!"; 410 $_[0] > 0 or die "read error: $!";
333 print "read $_[0] bytes: <$buffer>\n"; 411 print "read $_[0] bytes: <$buffer>\n";
334 }; 412 };
413
335 414
336=item aio_sendfile $out_fh, $in_fh, $in_offset, $length, $callback->($retval) 415=item aio_sendfile $out_fh, $in_fh, $in_offset, $length, $callback->($retval)
337 416
338Tries to copy C<$length> bytes from C<$in_fh> to C<$out_fh>. It starts 417Tries to copy C<$length> bytes from C<$in_fh> to C<$out_fh>. It starts
339reading at byte offset C<$in_offset>, and starts writing at the current 418reading at byte offset C<$in_offset>, and starts writing at the current
353C<$in_fh> than are written, and there is no way to find out how many 432C<$in_fh> than are written, and there is no way to find out how many
354bytes have been read from C<aio_sendfile> alone, as C<aio_sendfile> only 433bytes have been read from C<aio_sendfile> alone, as C<aio_sendfile> only
355provides the number of bytes written to C<$out_fh>. Only if the result 434provides the number of bytes written to C<$out_fh>. Only if the result
356value equals C<$length> one can assume that C<$length> bytes have been 435value equals C<$length> one can assume that C<$length> bytes have been
357read. 436read.
437
358 438
359=item aio_readahead $fh,$offset,$length, $callback->($retval) 439=item aio_readahead $fh,$offset,$length, $callback->($retval)
360 440
361C<aio_readahead> populates the page cache with data from a file so that 441C<aio_readahead> populates the page cache with data from a file so that
362subsequent reads from that file will not block on disk I/O. The C<$offset> 442subsequent reads from that file will not block on disk I/O. The C<$offset>
368file. The current file offset of the file is left unchanged. 448file. The current file offset of the file is left unchanged.
369 449
370If that syscall doesn't exist (likely if your OS isn't Linux) it will be 450If that syscall doesn't exist (likely if your OS isn't Linux) it will be
371emulated by simply reading the data, which would have a similar effect. 451emulated by simply reading the data, which would have a similar effect.
372 452
453
373=item aio_stat $fh_or_path, $callback->($status) 454=item aio_stat $fh_or_path, $callback->($status)
374 455
375=item aio_lstat $fh, $callback->($status) 456=item aio_lstat $fh, $callback->($status)
376 457
377Works like perl's C<stat> or C<lstat> in void context. The callback will 458Works like perl's C<stat> or C<lstat> in void context. The callback will
390 aio_stat "/etc/passwd", sub { 471 aio_stat "/etc/passwd", sub {
391 $_[0] and die "stat failed: $!"; 472 $_[0] and die "stat failed: $!";
392 print "size is ", -s _, "\n"; 473 print "size is ", -s _, "\n";
393 }; 474 };
394 475
476
477=item aio_utime $fh_or_path, $atime, $mtime, $callback->($status)
478
479Works like perl's C<utime> function (including the special case of $atime
480and $mtime being undef). Fractional times are supported if the underlying
481syscalls support them.
482
483When called with a pathname, uses utimes(2) if available, otherwise
484utime(2). If called on a file descriptor, uses futimes(2) if available,
485otherwise returns ENOSYS, so this is not portable.
486
487Examples:
488
489 # set atime and mtime to current time (basically touch(1)):
490 aio_utime "path", undef, undef;
491 # set atime to current time and mtime to beginning of the epoch:
492 aio_utime "path", time, undef; # undef==0
493
494
495=item aio_chown $fh_or_path, $uid, $gid, $callback->($status)
496
497Works like perl's C<chown> function, except that C<undef> for either $uid
498or $gid is being interpreted as "do not change" (but -1 can also be used).
499
500Examples:
501
502 # same as "chown root path" in the shell:
503 aio_chown "path", 0, -1;
504 # same as above:
505 aio_chown "path", 0, undef;
506
507
508=item aio_truncate $fh_or_path, $offset, $callback->($status)
509
510Works like truncate(2) or ftruncate(2).
511
512
513=item aio_chmod $fh_or_path, $mode, $callback->($status)
514
515Works like perl's C<chmod> function.
516
517
395=item aio_unlink $pathname, $callback->($status) 518=item aio_unlink $pathname, $callback->($status)
396 519
397Asynchronously unlink (delete) a file and call the callback with the 520Asynchronously unlink (delete) a file and call the callback with the
398result code. 521result code.
399 522
523
400=item aio_mknod $path, $mode, $dev, $callback->($status) 524=item aio_mknod $path, $mode, $dev, $callback->($status)
401 525
402[EXPERIMENTAL] 526[EXPERIMENTAL]
403 527
404Asynchronously create a device node (or fifo). See mknod(2). 528Asynchronously create a device node (or fifo). See mknod(2).
405 529
406The only (POSIX-) portable way of calling this function is: 530The only (POSIX-) portable way of calling this function is:
407 531
408 aio_mknod $path, IO::AIO::S_IFIFO | $mode, 0, sub { ... 532 aio_mknod $path, IO::AIO::S_IFIFO | $mode, 0, sub { ...
533
409 534
410=item aio_link $srcpath, $dstpath, $callback->($status) 535=item aio_link $srcpath, $dstpath, $callback->($status)
411 536
412Asynchronously create a new link to the existing object at C<$srcpath> at 537Asynchronously create a new link to the existing object at C<$srcpath> at
413the path C<$dstpath> and call the callback with the result code. 538the path C<$dstpath> and call the callback with the result code.
414 539
540
415=item aio_symlink $srcpath, $dstpath, $callback->($status) 541=item aio_symlink $srcpath, $dstpath, $callback->($status)
416 542
417Asynchronously create a new symbolic link to the existing object at C<$srcpath> at 543Asynchronously create a new symbolic link to the existing object at C<$srcpath> at
418the path C<$dstpath> and call the callback with the result code. 544the path C<$dstpath> and call the callback with the result code.
419 545
546
547=item aio_readlink $path, $callback->($link)
548
549Asynchronously read the symlink specified by C<$path> and pass it to
550the callback. If an error occurs, nothing or undef gets passed to the
551callback.
552
553
420=item aio_rename $srcpath, $dstpath, $callback->($status) 554=item aio_rename $srcpath, $dstpath, $callback->($status)
421 555
422Asynchronously rename the object at C<$srcpath> to C<$dstpath>, just as 556Asynchronously rename the object at C<$srcpath> to C<$dstpath>, just as
423rename(2) and call the callback with the result code. 557rename(2) and call the callback with the result code.
424 558
559
560=item aio_mkdir $pathname, $mode, $callback->($status)
561
562Asynchronously mkdir (create) a directory and call the callback with
563the result code. C<$mode> will be modified by the umask at the time the
564request is executed, so do not change your umask.
565
566
425=item aio_rmdir $pathname, $callback->($status) 567=item aio_rmdir $pathname, $callback->($status)
426 568
427Asynchronously rmdir (delete) a directory and call the callback with the 569Asynchronously rmdir (delete) a directory and call the callback with the
428result code. 570result code.
571
429 572
430=item aio_readdir $pathname, $callback->($entries) 573=item aio_readdir $pathname, $callback->($entries)
431 574
432Unlike the POSIX call of the same name, C<aio_readdir> reads an entire 575Unlike the POSIX call of the same name, C<aio_readdir> reads an entire
433directory (i.e. opendir + readdir + closedir). The entries will not be 576directory (i.e. opendir + readdir + closedir). The entries will not be
434sorted, and will B<NOT> include the C<.> and C<..> entries. 577sorted, and will B<NOT> include the C<.> and C<..> entries.
435 578
436The callback a single argument which is either C<undef> or an array-ref 579The callback a single argument which is either C<undef> or an array-ref
437with the filenames. 580with the filenames.
581
582
583=item aio_load $path, $data, $callback->($status)
584
585This is a composite request that tries to fully load the given file into
586memory. Status is the same as with aio_read.
587
588=cut
589
590sub aio_load($$;$) {
591 aio_block {
592 my ($path, undef, $cb) = @_;
593 my $data = \$_[1];
594
595 my $pri = aioreq_pri;
596 my $grp = aio_group $cb;
597
598 aioreq_pri $pri;
599 add $grp aio_open $path, O_RDONLY, 0, sub {
600 my $fh = shift
601 or return $grp->result (-1);
602
603 aioreq_pri $pri;
604 add $grp aio_read $fh, 0, (-s $fh), $$data, 0, sub {
605 $grp->result ($_[0]);
606 };
607 };
608
609 $grp
610 }
611}
438 612
439=item aio_copy $srcpath, $dstpath, $callback->($status) 613=item aio_copy $srcpath, $dstpath, $callback->($status)
440 614
441Try to copy the I<file> (directories not supported as either source or 615Try to copy the I<file> (directories not supported as either source or
442destination) from C<$srcpath> to C<$dstpath> and call the callback with 616destination) from C<$srcpath> to C<$dstpath> and call the callback with
452errors are being ignored. 626errors are being ignored.
453 627
454=cut 628=cut
455 629
456sub aio_copy($$;$) { 630sub aio_copy($$;$) {
631 aio_block {
457 my ($src, $dst, $cb) = @_; 632 my ($src, $dst, $cb) = @_;
458 633
459 my $pri = aioreq_pri; 634 my $pri = aioreq_pri;
460 my $grp = aio_group $cb; 635 my $grp = aio_group $cb;
461 636
462 aioreq_pri $pri; 637 aioreq_pri $pri;
463 add $grp aio_open $src, O_RDONLY, 0, sub { 638 add $grp aio_open $src, O_RDONLY, 0, sub {
464 if (my $src_fh = $_[0]) { 639 if (my $src_fh = $_[0]) {
465 my @stat = stat $src_fh; 640 my @stat = stat $src_fh;
466 641
467 aioreq_pri $pri; 642 aioreq_pri $pri;
468 add $grp aio_open $dst, O_CREAT | O_WRONLY | O_TRUNC, 0200, sub { 643 add $grp aio_open $dst, O_CREAT | O_WRONLY | O_TRUNC, 0200, sub {
469 if (my $dst_fh = $_[0]) { 644 if (my $dst_fh = $_[0]) {
470 aioreq_pri $pri; 645 aioreq_pri $pri;
471 add $grp aio_sendfile $dst_fh, $src_fh, 0, $stat[7], sub { 646 add $grp aio_sendfile $dst_fh, $src_fh, 0, $stat[7], sub {
472 if ($_[0] == $stat[7]) { 647 if ($_[0] == $stat[7]) {
473 $grp->result (0); 648 $grp->result (0);
474 close $src_fh; 649 close $src_fh;
475 650
476 # those should not normally block. should. should. 651 # those should not normally block. should. should.
477 utime $stat[8], $stat[9], $dst; 652 utime $stat[8], $stat[9], $dst;
478 chmod $stat[2] & 07777, $dst_fh; 653 chmod $stat[2] & 07777, $dst_fh;
479 chown $stat[4], $stat[5], $dst_fh; 654 chown $stat[4], $stat[5], $dst_fh;
480 close $dst_fh; 655 close $dst_fh;
481 } else { 656 } else {
482 $grp->result (-1); 657 $grp->result (-1);
483 close $src_fh; 658 close $src_fh;
484 close $dst_fh; 659 close $dst_fh;
485 660
486 aioreq $pri; 661 aioreq $pri;
487 add $grp aio_unlink $dst; 662 add $grp aio_unlink $dst;
663 }
488 } 664 };
665 } else {
666 $grp->result (-1);
489 }; 667 }
490 } else {
491 $grp->result (-1);
492 } 668 },
669
670 } else {
671 $grp->result (-1);
493 }, 672 }
494
495 } else {
496 $grp->result (-1);
497 } 673 };
674
675 $grp
498 }; 676 }
499
500 $grp
501} 677}
502 678
503=item aio_move $srcpath, $dstpath, $callback->($status) 679=item aio_move $srcpath, $dstpath, $callback->($status)
504 680
505Try to move the I<file> (directories not supported as either source or 681Try to move the I<file> (directories not supported as either source or
511that is successful, unlinking the C<$srcpath>. 687that is successful, unlinking the C<$srcpath>.
512 688
513=cut 689=cut
514 690
515sub aio_move($$;$) { 691sub aio_move($$;$) {
692 aio_block {
516 my ($src, $dst, $cb) = @_; 693 my ($src, $dst, $cb) = @_;
517 694
518 my $pri = aioreq_pri; 695 my $pri = aioreq_pri;
519 my $grp = aio_group $cb; 696 my $grp = aio_group $cb;
520 697
521 aioreq_pri $pri; 698 aioreq_pri $pri;
522 add $grp aio_rename $src, $dst, sub { 699 add $grp aio_rename $src, $dst, sub {
523 if ($_[0] && $! == EXDEV) { 700 if ($_[0] && $! == EXDEV) {
524 aioreq_pri $pri; 701 aioreq_pri $pri;
525 add $grp aio_copy $src, $dst, sub { 702 add $grp aio_copy $src, $dst, sub {
703 $grp->result ($_[0]);
704
705 if (!$_[0]) {
706 aioreq_pri $pri;
707 add $grp aio_unlink $src;
708 }
709 };
710 } else {
526 $grp->result ($_[0]); 711 $grp->result ($_[0]);
527
528 if (!$_[0]) {
529 aioreq_pri $pri;
530 add $grp aio_unlink $src;
531 }
532 }; 712 }
533 } else {
534 $grp->result ($_[0]);
535 } 713 };
714
715 $grp
536 }; 716 }
537
538 $grp
539} 717}
540 718
541=item aio_scandir $path, $maxreq, $callback->($dirs, $nondirs) 719=item aio_scandir $path, $maxreq, $callback->($dirs, $nondirs)
542 720
543Scans a directory (similar to C<aio_readdir>) but additionally tries to 721Scans a directory (similar to C<aio_readdir>) but additionally tries to
590as those tend to return 0 or 1 as link counts, which disables the 768as those tend to return 0 or 1 as link counts, which disables the
591directory counting heuristic. 769directory counting heuristic.
592 770
593=cut 771=cut
594 772
595sub aio_scandir($$$) { 773sub aio_scandir($$;$) {
774 aio_block {
596 my ($path, $maxreq, $cb) = @_; 775 my ($path, $maxreq, $cb) = @_;
597 776
598 my $pri = aioreq_pri; 777 my $pri = aioreq_pri;
599 778
600 my $grp = aio_group $cb; 779 my $grp = aio_group $cb;
601 780
602 $maxreq = 4 if $maxreq <= 0; 781 $maxreq = 4 if $maxreq <= 0;
603 782
604 # stat once 783 # stat once
605 aioreq_pri $pri;
606 add $grp aio_stat $path, sub {
607 return $grp->result () if $_[0];
608 my $now = time;
609 my $hash1 = join ":", (stat _)[0,1,3,7,9];
610
611 # read the directory entries
612 aioreq_pri $pri; 784 aioreq_pri $pri;
613 add $grp aio_readdir $path, sub { 785 add $grp aio_stat $path, sub {
614 my $entries = shift
615 or return $grp->result (); 786 return $grp->result () if $_[0];
787 my $now = time;
788 my $hash1 = join ":", (stat _)[0,1,3,7,9];
616 789
617 # stat the dir another time 790 # read the directory entries
618 aioreq_pri $pri; 791 aioreq_pri $pri;
792 add $grp aio_readdir $path, sub {
793 my $entries = shift
794 or return $grp->result ();
795
796 # stat the dir another time
797 aioreq_pri $pri;
619 add $grp aio_stat $path, sub { 798 add $grp aio_stat $path, sub {
620 my $hash2 = join ":", (stat _)[0,1,3,7,9]; 799 my $hash2 = join ":", (stat _)[0,1,3,7,9];
621 800
622 my $ndirs; 801 my $ndirs;
623 802
624 # take the slow route if anything looks fishy 803 # take the slow route if anything looks fishy
625 if ($hash1 ne $hash2 or (stat _)[9] == $now) { 804 if ($hash1 ne $hash2 or (stat _)[9] == $now) {
626 $ndirs = -1; 805 $ndirs = -1;
627 } else { 806 } else {
628 # if nlink == 2, we are finished 807 # if nlink == 2, we are finished
629 # on non-posix-fs's, we rely on nlink < 2 808 # on non-posix-fs's, we rely on nlink < 2
630 $ndirs = (stat _)[3] - 2 809 $ndirs = (stat _)[3] - 2
631 or return $grp->result ([], $entries); 810 or return $grp->result ([], $entries);
632 } 811 }
633 812
634 # sort into likely dirs and likely nondirs 813 # sort into likely dirs and likely nondirs
635 # dirs == files without ".", short entries first 814 # dirs == files without ".", short entries first
636 $entries = [map $_->[0], 815 $entries = [map $_->[0],
637 sort { $b->[1] cmp $a->[1] } 816 sort { $b->[1] cmp $a->[1] }
638 map [$_, sprintf "%s%04d", (/.\./ ? "1" : "0"), length], 817 map [$_, sprintf "%s%04d", (/.\./ ? "1" : "0"), length],
639 @$entries]; 818 @$entries];
640 819
641 my (@dirs, @nondirs); 820 my (@dirs, @nondirs);
642 821
643 my $statgrp = add $grp aio_group sub { 822 my $statgrp = add $grp aio_group sub {
644 $grp->result (\@dirs, \@nondirs); 823 $grp->result (\@dirs, \@nondirs);
645 }; 824 };
646 825
647 limit $statgrp $maxreq; 826 limit $statgrp $maxreq;
648 feed $statgrp sub { 827 feed $statgrp sub {
649 return unless @$entries; 828 return unless @$entries;
650 my $entry = pop @$entries; 829 my $entry = pop @$entries;
651 830
652 aioreq_pri $pri; 831 aioreq_pri $pri;
653 add $statgrp aio_stat "$path/$entry/.", sub { 832 add $statgrp aio_stat "$path/$entry/.", sub {
654 if ($_[0] < 0) { 833 if ($_[0] < 0) {
655 push @nondirs, $entry; 834 push @nondirs, $entry;
656 } else { 835 } else {
657 # need to check for real directory 836 # need to check for real directory
658 aioreq_pri $pri; 837 aioreq_pri $pri;
659 add $statgrp aio_lstat "$path/$entry", sub { 838 add $statgrp aio_lstat "$path/$entry", sub {
660 if (-d _) { 839 if (-d _) {
661 push @dirs, $entry; 840 push @dirs, $entry;
662 841
663 unless (--$ndirs) { 842 unless (--$ndirs) {
664 push @nondirs, @$entries; 843 push @nondirs, @$entries;
665 feed $statgrp; 844 feed $statgrp;
845 }
846 } else {
847 push @nondirs, $entry;
666 } 848 }
667 } else {
668 push @nondirs, $entry;
669 } 849 }
670 } 850 }
671 } 851 };
672 }; 852 };
673 }; 853 };
674 }; 854 };
675 }; 855 };
856
857 $grp
676 }; 858 }
677
678 $grp
679} 859}
860
861=item aio_rmtree $path, $callback->($status)
862
863Delete a directory tree starting (and including) C<$path>, return the
864status of the final C<rmdir> only. This is a composite request that
865uses C<aio_scandir> to recurse into and rmdir directories, and unlink
866everything else.
867
868=cut
869
870sub aio_rmtree;
871sub aio_rmtree($;$) {
872 aio_block {
873 my ($path, $cb) = @_;
874
875 my $pri = aioreq_pri;
876 my $grp = aio_group $cb;
877
878 aioreq_pri $pri;
879 add $grp aio_scandir $path, 0, sub {
880 my ($dirs, $nondirs) = @_;
881
882 my $dirgrp = aio_group sub {
883 add $grp aio_rmdir $path, sub {
884 $grp->result ($_[0]);
885 };
886 };
887
888 (aioreq_pri $pri), add $dirgrp aio_rmtree "$path/$_" for @$dirs;
889 (aioreq_pri $pri), add $dirgrp aio_unlink "$path/$_" for @$nondirs;
890
891 add $grp $dirgrp;
892 };
893
894 $grp
895 }
896}
897
898=item aio_sync $callback->($status)
899
900Asynchronously call sync and call the callback when finished.
680 901
681=item aio_fsync $fh, $callback->($status) 902=item aio_fsync $fh, $callback->($status)
682 903
683Asynchronously call fsync on the given filehandle and call the callback 904Asynchronously call fsync on the given filehandle and call the callback
684with the fsync result code. 905with the fsync result code.
937that are being processed by C<IO::AIO::poll_cb> in one call, respectively 1158that are being processed by C<IO::AIO::poll_cb> in one call, respectively
938the maximum amount of time (default C<0>, meaning infinity) spent in 1159the maximum amount of time (default C<0>, meaning infinity) spent in
939C<IO::AIO::poll_cb> to process requests (more correctly the mininum amount 1160C<IO::AIO::poll_cb> to process requests (more correctly the mininum amount
940of time C<poll_cb> is allowed to use). 1161of time C<poll_cb> is allowed to use).
941 1162
1163Setting C<max_poll_time> to a non-zero value creates an overhead of one
1164syscall per request processed, which is not normally a problem unless your
1165callbacks are really really fast or your OS is really really slow (I am
1166not mentioning Solaris here). Using C<max_poll_reqs> incurs no overhead.
1167
942Setting these is useful if you want to ensure some level of 1168Setting these is useful if you want to ensure some level of
943interactiveness when perl is not fast enough to process all requests in 1169interactiveness when perl is not fast enough to process all requests in
944time. 1170time.
945 1171
946For interactive programs, values such as C<0.01> to C<0.1> should be fine. 1172For interactive programs, values such as C<0.01> to C<0.1> should be fine.
947 1173
948Example: Install an Event watcher that automatically calls 1174Example: Install an Event watcher that automatically calls
949IO::AIO::poll_some with low priority, to ensure that other parts of the 1175IO::AIO::poll_cb with low priority, to ensure that other parts of the
950program get the CPU sometimes even under high AIO load. 1176program get the CPU sometimes even under high AIO load.
951 1177
952 # try not to spend much more than 0.1s in poll_cb 1178 # try not to spend much more than 0.1s in poll_cb
953 IO::AIO::max_poll_time 0.1; 1179 IO::AIO::max_poll_time 0.1;
954 1180
957 poll => 'r', nice => 1, 1183 poll => 'r', nice => 1,
958 cb => &IO::AIO::poll_cb); 1184 cb => &IO::AIO::poll_cb);
959 1185
960=item IO::AIO::poll_wait 1186=item IO::AIO::poll_wait
961 1187
1188If there are any outstanding requests and none of them in the result
962Wait till the result filehandle becomes ready for reading (simply does a 1189phase, wait till the result filehandle becomes ready for reading (simply
963C<select> on the filehandle. This is useful if you want to synchronously 1190does a C<select> on the filehandle. This is useful if you want to
964wait for some requests to finish). 1191synchronously wait for some requests to finish).
965 1192
966See C<nreqs> for an example. 1193See C<nreqs> for an example.
967 1194
968=item IO::AIO::poll 1195=item IO::AIO::poll
969 1196
970Waits until some requests have been handled. 1197Waits until some requests have been handled.
971 1198
1199Returns the number of requests processed, but is otherwise strictly
972Strictly equivalent to: 1200equivalent to:
973 1201
974 IO::AIO::poll_wait, IO::AIO::poll_cb 1202 IO::AIO::poll_wait, IO::AIO::poll_cb
975 if IO::AIO::nreqs;
976 1203
977=item IO::AIO::flush 1204=item IO::AIO::flush
978 1205
979Wait till all outstanding AIO requests have been handled. 1206Wait till all outstanding AIO requests have been handled.
980 1207
981Strictly equivalent to: 1208Strictly equivalent to:
982 1209
983 IO::AIO::poll_wait, IO::AIO::poll_cb 1210 IO::AIO::poll_wait, IO::AIO::poll_cb
984 while IO::AIO::nreqs; 1211 while IO::AIO::nreqs;
985 1212
1213=back
1214
986=head3 CONTROLLING THE NUMBER OF THREADS 1215=head3 CONTROLLING THE NUMBER OF THREADS
1216
1217=over
987 1218
988=item IO::AIO::min_parallel $nthreads 1219=item IO::AIO::min_parallel $nthreads
989 1220
990Set the minimum number of AIO threads to C<$nthreads>. The current 1221Set the minimum number of AIO threads to C<$nthreads>. The current
991default is C<8>, which means eight asynchronous operations can execute 1222default is C<8>, which means eight asynchronous operations can execute
1039This is a very bad function to use in interactive programs because it 1270This is a very bad function to use in interactive programs because it
1040blocks, and a bad way to reduce concurrency because it is inexact: Better 1271blocks, and a bad way to reduce concurrency because it is inexact: Better
1041use an C<aio_group> together with a feed callback. 1272use an C<aio_group> together with a feed callback.
1042 1273
1043Sets the maximum number of outstanding requests to C<$nreqs>. If you 1274Sets the maximum number of outstanding requests to C<$nreqs>. If you
1044to queue up more than this number of requests, the next call to the 1275do queue up more than this number of requests, the next call to the
1045C<poll_cb> (and C<poll_some> and other functions calling C<poll_cb>) 1276C<poll_cb> (and C<poll_some> and other functions calling C<poll_cb>)
1046function will block until the limit is no longer exceeded. 1277function will block until the limit is no longer exceeded.
1047 1278
1048The default value is very large, so there is no practical limit on the 1279The default value is very large, so there is no practical limit on the
1049number of outstanding requests. 1280number of outstanding requests.
1050 1281
1051You can still queue as many requests as you want. Therefore, 1282You can still queue as many requests as you want. Therefore,
1052C<max_oustsanding> is mainly useful in simple scripts (with low values) or 1283C<max_oustsanding> is mainly useful in simple scripts (with low values) or
1053as a stop gap to shield against fatal memory overflow (with large values). 1284as a stop gap to shield against fatal memory overflow (with large values).
1054 1285
1286=back
1287
1055=head3 STATISTICAL INFORMATION 1288=head3 STATISTICAL INFORMATION
1289
1290=over
1056 1291
1057=item IO::AIO::nreqs 1292=item IO::AIO::nreqs
1058 1293
1059Returns the number of requests currently in the ready, execute or pending 1294Returns the number of requests currently in the ready, execute or pending
1060states (i.e. for which their callback has not been invoked yet). 1295states (i.e. for which their callback has not been invoked yet).
1076 1311
1077=back 1312=back
1078 1313
1079=cut 1314=cut
1080 1315
1081# support function to convert a fd into a perl filehandle
1082sub _fd2fh {
1083 return undef if $_[0] < 0;
1084
1085 # try to generate nice filehandles
1086 my $sym = "IO::AIO::fd#$_[0]";
1087 local *$sym;
1088
1089 open *$sym, "+<&=$_[0]" # usually works under any unix
1090 or open *$sym, "<&=$_[0]" # cygwin needs this
1091 or open *$sym, ">&=$_[0]" # or this
1092 or return undef;
1093
1094 *$sym
1095}
1096
1097min_parallel 8; 1316min_parallel 8;
1098 1317
1099END { 1318END { flush }
1100 min_parallel 1;
1101 flush;
1102};
1103 1319
11041; 13201;
1105 1321
1106=head2 FORK BEHAVIOUR 1322=head2 FORK BEHAVIOUR
1107 1323
1127bytes of memory. In addition, stat requests need a stat buffer (possibly 1343bytes of memory. In addition, stat requests need a stat buffer (possibly
1128a few hundred bytes), readdir requires a result buffer and so on. Perl 1344a few hundred bytes), readdir requires a result buffer and so on. Perl
1129scalars and other data passed into aio requests will also be locked and 1345scalars and other data passed into aio requests will also be locked and
1130will consume memory till the request has entered the done state. 1346will consume memory till the request has entered the done state.
1131 1347
1132This is now awfully much, so queuing lots of requests is not usually a 1348This is not awfully much, so queuing lots of requests is not usually a
1133problem. 1349problem.
1134 1350
1135Per-thread usage: 1351Per-thread usage:
1136 1352
1137In the execution phase, some aio requests require more memory for 1353In the execution phase, some aio requests require more memory for

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines