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.91 by root, Mon Oct 30 23:30:29 2006 UTC vs.
Revision 1.120 by root, Sun Dec 2 21:51:36 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 most of this works on all types of file descriptors (for example 70While most of this works on all types of file descriptors (for
67sockets), using these functions on file descriptors that support 71example sockets), using these functions on file descriptors that
68nonblocking operation (again, sockets, pipes etc.) is very inefficient or 72support nonblocking operation (again, sockets, pipes etc.) is very
69might not work (aio_read fails on sockets/pipes/fifos). Use an event loop 73inefficient. Use an event loop for that (such as the L<Event|Event>
70for that (such as the L<Event|Event> module): IO::AIO will naturally fit 74module): IO::AIO will naturally fit into such an event loop itself.
71into such an event loop itself.
72 75
73In this version, a number of threads are started that execute your 76In this version, a number of threads are started that execute your
74requests and signal their completion. You don't need thread support 77requests and signal their completion. You don't need thread support
75in 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
76to 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
78not 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
79files currently, for example), and they would only support aio_read and 82files currently, for example), and they would only support aio_read and
80aio_write, so the remaining functionality would have to be implemented 83aio_write, so the remaining functionality would have to be implemented
81using threads anyway. 84using threads anyway.
82 85
83Although the module will work with in the presence of other (Perl-) 86Although the module will work in the presence of other (Perl-) threads,
84threads, it is currently not reentrant in any way, so use appropriate 87it is currently not reentrant in any way, so use appropriate locking
85locking 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
86never call C<poll_cb> (or other C<aio_> functions) recursively. 89call C<poll_cb> (or other C<aio_> functions) recursively.
87 90
88=head2 EXAMPLE 91=head2 EXAMPLE
89 92
90This is a simple example that uses the Event module and loads 93This is a simple example that uses the Event module and loads
91F</etc/passwd> asynchronously: 94F</etc/passwd> asynchronously:
99 poll => 'r', 102 poll => 'r',
100 cb => \&IO::AIO::poll_cb); 103 cb => \&IO::AIO::poll_cb);
101 104
102 # queue the request to open /etc/passwd 105 # queue the request to open /etc/passwd
103 aio_open "/etc/passwd", O_RDONLY, 0, sub { 106 aio_open "/etc/passwd", O_RDONLY, 0, sub {
104 my $fh = $_[0] 107 my $fh = shift
105 or die "error while opening: $!"; 108 or die "error while opening: $!";
106 109
107 # stat'ing filehandles is generally non-blocking 110 # stat'ing filehandles is generally non-blocking
108 my $size = -s $fh; 111 my $size = -s $fh;
109 112
183 186
184=cut 187=cut
185 188
186package IO::AIO; 189package IO::AIO;
187 190
191use Carp ();
192
188no warnings; 193no warnings;
189use strict 'vars'; 194use strict 'vars';
190 195
191use base 'Exporter'; 196use base 'Exporter';
192 197
193BEGIN { 198BEGIN {
194 our $VERSION = '2.2'; 199 our $VERSION = '2.6';
195 200
196 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
197 aio_lstat aio_unlink aio_rmdir aio_readdir aio_scandir aio_symlink 202 aio_stat aio_lstat aio_unlink aio_rmdir aio_readdir
198 aio_readlink aio_fsync aio_fdatasync aio_readahead aio_rename aio_link 203 aio_scandir aio_symlink aio_readlink aio_sync aio_fsync
199 aio_move aio_copy aio_group aio_nop aio_mknod); 204 aio_fdatasync aio_pathsync aio_readahead
205 aio_rename aio_link aio_move aio_copy aio_group
206 aio_nop aio_mknod aio_load aio_rmtree aio_mkdir aio_chown
207 aio_chmod aio_utime aio_truncate);
208
200 our @EXPORT = (@AIO_REQ, qw(aioreq_pri aioreq_nice)); 209 our @EXPORT = (@AIO_REQ, qw(aioreq_pri aioreq_nice aio_block));
201 our @EXPORT_OK = qw(poll_fileno poll_cb poll_wait flush 210 our @EXPORT_OK = qw(poll_fileno poll_cb poll_wait flush
202 min_parallel max_parallel max_idle 211 min_parallel max_parallel max_idle
203 nreqs nready npending nthreads 212 nreqs nready npending nthreads
204 max_poll_time max_poll_reqs); 213 max_poll_time max_poll_reqs);
205 214
270 aio_read $_[0], ..., sub { 279 aio_read $_[0], ..., sub {
271 ... 280 ...
272 }; 281 };
273 }; 282 };
274 283
284
275=item aioreq_nice $pri_adjust 285=item aioreq_nice $pri_adjust
276 286
277Similar to C<aioreq_pri>, but subtracts the given value from the current 287Similar to C<aioreq_pri>, but subtracts the given value from the current
278priority, so the effect is cumulative. 288priority, so the effect is cumulative.
289
279 290
280=item aio_open $pathname, $flags, $mode, $callback->($fh) 291=item aio_open $pathname, $flags, $mode, $callback->($fh)
281 292
282Asynchronously open or create a file and call the callback with a newly 293Asynchronously open or create a file and call the callback with a newly
283created filehandle for the file. 294created filehandle for the file.
289list. They are the same as used by C<sysopen>. 300list. They are the same as used by C<sysopen>.
290 301
291Likewise, C<$mode> specifies the mode of the newly created file, if it 302Likewise, C<$mode> specifies the mode of the newly created file, if it
292didn't exist and C<O_CREAT> has been given, just like perl's C<sysopen>, 303didn't exist and C<O_CREAT> has been given, just like perl's C<sysopen>,
293except that it is mandatory (i.e. use C<0> if you don't create new files, 304except that it is mandatory (i.e. use C<0> if you don't create new files,
294and C<0666> or C<0777> if you do). 305and C<0666> or C<0777> if you do). Note that the C<$mode> will be modified
306by the umask in effect then the request is being executed, so better never
307change the umask.
295 308
296Example: 309Example:
297 310
298 aio_open "/etc/passwd", O_RDONLY, 0, sub { 311 aio_open "/etc/passwd", O_RDONLY, 0, sub {
299 if ($_[0]) { 312 if ($_[0]) {
302 } else { 315 } else {
303 die "open failed: $!\n"; 316 die "open failed: $!\n";
304 } 317 }
305 }; 318 };
306 319
320
307=item aio_close $fh, $callback->($status) 321=item aio_close $fh, $callback->($status)
308 322
309Asynchronously close a file and call the callback with the result 323Asynchronously close a file and call the callback with the result
310code. I<WARNING:> although accepted, you should not pass in a perl 324code.
311filehandle here, as perl will likely close the file descriptor another
312time when the filehandle is destroyed. Normally, you can safely call perls
313C<close> or just let filehandles go out of scope.
314 325
315This is supposed to be a bug in the API, so that might change. It's 326Unfortunately, you can't do this to perl. Perl I<insists> very strongly on
316therefore best to avoid this function. 327closing the file descriptor associated with the filehandle itself. Here is
328what aio_close will try:
329
330 1. dup()licate the fd
331 2. asynchronously close() the duplicated fd
332 3. dup()licate the fd once more
333 4. let perl close() the filehandle
334 5. asynchronously close the duplicated fd
335
336The idea is that the first close() flushes stuff to disk that closing an
337fd will flush, so when perl closes the fd, nothing much will need to be
338flushed. The second async. close() will then flush stuff to disk that
339closing the last fd to the file will flush.
340
341Just FYI, SuSv3 has this to say on close:
342
343 All outstanding record locks owned by the process on the file
344 associated with the file descriptor shall be removed.
345
346 If fildes refers to a socket, close() shall cause the socket to be
347 destroyed. ... close() shall block for up to the current linger
348 interval until all data is transmitted.
349 [this actually sounds like a specification bug, but who knows]
350
351And at least Linux additionally actually flushes stuff on every close,
352even when the file itself is still open.
353
354Sounds enourmously inefficient and complicated? Yes... please show me how
355to nuke perl's fd out of existence...
356
357=cut
358
359sub aio_close($;$) {
360 aio_block {
361 my ($fh, $cb) = @_;
362
363 my $pri = aioreq_pri;
364 my $grp = aio_group $cb;
365
366 my $fd = fileno $fh;
367
368 defined $fd or Carp::croak "aio_close called with fd-less filehandle";
369
370 # if the dups fail we will simply get EBADF
371 my $fd2 = _dup $fd;
372 aioreq_pri $pri;
373 add $grp _aio_close $fd2, sub {
374 my $fd2 = _dup $fd;
375 close $fh;
376 aioreq_pri $pri;
377 add $grp _aio_close $fd2, sub {
378 $grp->result ($_[0]);
379 };
380 };
381
382 $grp
383 }
384}
385
317 386
318=item aio_read $fh,$offset,$length, $data,$dataoffset, $callback->($retval) 387=item aio_read $fh,$offset,$length, $data,$dataoffset, $callback->($retval)
319 388
320=item aio_write $fh,$offset,$length, $data,$dataoffset, $callback->($retval) 389=item aio_write $fh,$offset,$length, $data,$dataoffset, $callback->($retval)
321 390
322Reads or writes C<length> bytes from the specified C<fh> and C<offset> 391Reads or writes C<$length> bytes from the specified C<$fh> and C<$offset>
323into the scalar given by C<data> and offset C<dataoffset> and calls the 392into the scalar given by C<$data> and offset C<$dataoffset> and calls the
324callback without the actual number of bytes read (or -1 on error, just 393callback without the actual number of bytes read (or -1 on error, just
325like the syscall). 394like the syscall).
326 395
396If C<$offset> is undefined, then the current file descriptor offset will
397be used (and updated), otherwise the file descriptor offset will not be
398changed by these calls.
399
400If C<$length> is undefined in C<aio_write>, use the remaining length of C<$data>.
401
402If C<$dataoffset> is less than zero, it will be counted from the end of
403C<$data>.
404
327The C<$data> scalar I<MUST NOT> be modified in any way while the request 405The C<$data> scalar I<MUST NOT> be modified in any way while the request
328is outstanding. Modifying it can result in segfaults or WW3 (if the 406is outstanding. Modifying it can result in segfaults or World War III (if
329necessary/optional hardware is installed). 407the necessary/optional hardware is installed).
330 408
331Example: Read 15 bytes at offset 7 into scalar C<$buffer>, starting at 409Example: Read 15 bytes at offset 7 into scalar C<$buffer>, starting at
332offset C<0> within the scalar: 410offset C<0> within the scalar:
333 411
334 aio_read $fh, 7, 15, $buffer, 0, sub { 412 aio_read $fh, 7, 15, $buffer, 0, sub {
335 $_[0] > 0 or die "read error: $!"; 413 $_[0] > 0 or die "read error: $!";
336 print "read $_[0] bytes: <$buffer>\n"; 414 print "read $_[0] bytes: <$buffer>\n";
337 }; 415 };
416
338 417
339=item aio_sendfile $out_fh, $in_fh, $in_offset, $length, $callback->($retval) 418=item aio_sendfile $out_fh, $in_fh, $in_offset, $length, $callback->($retval)
340 419
341Tries to copy C<$length> bytes from C<$in_fh> to C<$out_fh>. It starts 420Tries to copy C<$length> bytes from C<$in_fh> to C<$out_fh>. It starts
342reading at byte offset C<$in_offset>, and starts writing at the current 421reading at byte offset C<$in_offset>, and starts writing at the current
356C<$in_fh> than are written, and there is no way to find out how many 435C<$in_fh> than are written, and there is no way to find out how many
357bytes have been read from C<aio_sendfile> alone, as C<aio_sendfile> only 436bytes have been read from C<aio_sendfile> alone, as C<aio_sendfile> only
358provides the number of bytes written to C<$out_fh>. Only if the result 437provides the number of bytes written to C<$out_fh>. Only if the result
359value equals C<$length> one can assume that C<$length> bytes have been 438value equals C<$length> one can assume that C<$length> bytes have been
360read. 439read.
440
361 441
362=item aio_readahead $fh,$offset,$length, $callback->($retval) 442=item aio_readahead $fh,$offset,$length, $callback->($retval)
363 443
364C<aio_readahead> populates the page cache with data from a file so that 444C<aio_readahead> populates the page cache with data from a file so that
365subsequent reads from that file will not block on disk I/O. The C<$offset> 445subsequent reads from that file will not block on disk I/O. The C<$offset>
371file. The current file offset of the file is left unchanged. 451file. The current file offset of the file is left unchanged.
372 452
373If that syscall doesn't exist (likely if your OS isn't Linux) it will be 453If that syscall doesn't exist (likely if your OS isn't Linux) it will be
374emulated by simply reading the data, which would have a similar effect. 454emulated by simply reading the data, which would have a similar effect.
375 455
456
376=item aio_stat $fh_or_path, $callback->($status) 457=item aio_stat $fh_or_path, $callback->($status)
377 458
378=item aio_lstat $fh, $callback->($status) 459=item aio_lstat $fh, $callback->($status)
379 460
380Works like perl's C<stat> or C<lstat> in void context. The callback will 461Works like perl's C<stat> or C<lstat> in void context. The callback will
393 aio_stat "/etc/passwd", sub { 474 aio_stat "/etc/passwd", sub {
394 $_[0] and die "stat failed: $!"; 475 $_[0] and die "stat failed: $!";
395 print "size is ", -s _, "\n"; 476 print "size is ", -s _, "\n";
396 }; 477 };
397 478
479
480=item aio_utime $fh_or_path, $atime, $mtime, $callback->($status)
481
482Works like perl's C<utime> function (including the special case of $atime
483and $mtime being undef). Fractional times are supported if the underlying
484syscalls support them.
485
486When called with a pathname, uses utimes(2) if available, otherwise
487utime(2). If called on a file descriptor, uses futimes(2) if available,
488otherwise returns ENOSYS, so this is not portable.
489
490Examples:
491
492 # set atime and mtime to current time (basically touch(1)):
493 aio_utime "path", undef, undef;
494 # set atime to current time and mtime to beginning of the epoch:
495 aio_utime "path", time, undef; # undef==0
496
497
498=item aio_chown $fh_or_path, $uid, $gid, $callback->($status)
499
500Works like perl's C<chown> function, except that C<undef> for either $uid
501or $gid is being interpreted as "do not change" (but -1 can also be used).
502
503Examples:
504
505 # same as "chown root path" in the shell:
506 aio_chown "path", 0, -1;
507 # same as above:
508 aio_chown "path", 0, undef;
509
510
511=item aio_truncate $fh_or_path, $offset, $callback->($status)
512
513Works like truncate(2) or ftruncate(2).
514
515
516=item aio_chmod $fh_or_path, $mode, $callback->($status)
517
518Works like perl's C<chmod> function.
519
520
398=item aio_unlink $pathname, $callback->($status) 521=item aio_unlink $pathname, $callback->($status)
399 522
400Asynchronously unlink (delete) a file and call the callback with the 523Asynchronously unlink (delete) a file and call the callback with the
401result code. 524result code.
402 525
526
403=item aio_mknod $path, $mode, $dev, $callback->($status) 527=item aio_mknod $path, $mode, $dev, $callback->($status)
404 528
405[EXPERIMENTAL] 529[EXPERIMENTAL]
406 530
407Asynchronously create a device node (or fifo). See mknod(2). 531Asynchronously create a device node (or fifo). See mknod(2).
408 532
409The only (POSIX-) portable way of calling this function is: 533The only (POSIX-) portable way of calling this function is:
410 534
411 aio_mknod $path, IO::AIO::S_IFIFO | $mode, 0, sub { ... 535 aio_mknod $path, IO::AIO::S_IFIFO | $mode, 0, sub { ...
536
412 537
413=item aio_link $srcpath, $dstpath, $callback->($status) 538=item aio_link $srcpath, $dstpath, $callback->($status)
414 539
415Asynchronously create a new link to the existing object at C<$srcpath> at 540Asynchronously create a new link to the existing object at C<$srcpath> at
416the path C<$dstpath> and call the callback with the result code. 541the path C<$dstpath> and call the callback with the result code.
417 542
543
418=item aio_symlink $srcpath, $dstpath, $callback->($status) 544=item aio_symlink $srcpath, $dstpath, $callback->($status)
419 545
420Asynchronously create a new symbolic link to the existing object at C<$srcpath> at 546Asynchronously create a new symbolic link to the existing object at C<$srcpath> at
421the path C<$dstpath> and call the callback with the result code. 547the path C<$dstpath> and call the callback with the result code.
548
422 549
423=item aio_readlink $path, $callback->($link) 550=item aio_readlink $path, $callback->($link)
424 551
425Asynchronously read the symlink specified by C<$path> and pass it to 552Asynchronously read the symlink specified by C<$path> and pass it to
426the callback. If an error occurs, nothing or undef gets passed to the 553the callback. If an error occurs, nothing or undef gets passed to the
427callback. 554callback.
428 555
556
429=item aio_rename $srcpath, $dstpath, $callback->($status) 557=item aio_rename $srcpath, $dstpath, $callback->($status)
430 558
431Asynchronously rename the object at C<$srcpath> to C<$dstpath>, just as 559Asynchronously rename the object at C<$srcpath> to C<$dstpath>, just as
432rename(2) and call the callback with the result code. 560rename(2) and call the callback with the result code.
433 561
562
563=item aio_mkdir $pathname, $mode, $callback->($status)
564
565Asynchronously mkdir (create) a directory and call the callback with
566the result code. C<$mode> will be modified by the umask at the time the
567request is executed, so do not change your umask.
568
569
434=item aio_rmdir $pathname, $callback->($status) 570=item aio_rmdir $pathname, $callback->($status)
435 571
436Asynchronously rmdir (delete) a directory and call the callback with the 572Asynchronously rmdir (delete) a directory and call the callback with the
437result code. 573result code.
574
438 575
439=item aio_readdir $pathname, $callback->($entries) 576=item aio_readdir $pathname, $callback->($entries)
440 577
441Unlike the POSIX call of the same name, C<aio_readdir> reads an entire 578Unlike the POSIX call of the same name, C<aio_readdir> reads an entire
442directory (i.e. opendir + readdir + closedir). The entries will not be 579directory (i.e. opendir + readdir + closedir). The entries will not be
443sorted, and will B<NOT> include the C<.> and C<..> entries. 580sorted, and will B<NOT> include the C<.> and C<..> entries.
444 581
445The callback a single argument which is either C<undef> or an array-ref 582The callback a single argument which is either C<undef> or an array-ref
446with the filenames. 583with the filenames.
584
585
586=item aio_load $path, $data, $callback->($status)
587
588This is a composite request that tries to fully load the given file into
589memory. Status is the same as with aio_read.
590
591=cut
592
593sub aio_load($$;$) {
594 aio_block {
595 my ($path, undef, $cb) = @_;
596 my $data = \$_[1];
597
598 my $pri = aioreq_pri;
599 my $grp = aio_group $cb;
600
601 aioreq_pri $pri;
602 add $grp aio_open $path, O_RDONLY, 0, sub {
603 my $fh = shift
604 or return $grp->result (-1);
605
606 aioreq_pri $pri;
607 add $grp aio_read $fh, 0, (-s $fh), $$data, 0, sub {
608 $grp->result ($_[0]);
609 };
610 };
611
612 $grp
613 }
614}
447 615
448=item aio_copy $srcpath, $dstpath, $callback->($status) 616=item aio_copy $srcpath, $dstpath, $callback->($status)
449 617
450Try to copy the I<file> (directories not supported as either source or 618Try to copy the I<file> (directories not supported as either source or
451destination) from C<$srcpath> to C<$dstpath> and call the callback with 619destination) from C<$srcpath> to C<$dstpath> and call the callback with
461errors are being ignored. 629errors are being ignored.
462 630
463=cut 631=cut
464 632
465sub aio_copy($$;$) { 633sub aio_copy($$;$) {
634 aio_block {
466 my ($src, $dst, $cb) = @_; 635 my ($src, $dst, $cb) = @_;
467 636
468 my $pri = aioreq_pri; 637 my $pri = aioreq_pri;
469 my $grp = aio_group $cb; 638 my $grp = aio_group $cb;
470 639
471 aioreq_pri $pri; 640 aioreq_pri $pri;
472 add $grp aio_open $src, O_RDONLY, 0, sub { 641 add $grp aio_open $src, O_RDONLY, 0, sub {
473 if (my $src_fh = $_[0]) { 642 if (my $src_fh = $_[0]) {
474 my @stat = stat $src_fh; 643 my @stat = stat $src_fh;
475 644
476 aioreq_pri $pri; 645 aioreq_pri $pri;
477 add $grp aio_open $dst, O_CREAT | O_WRONLY | O_TRUNC, 0200, sub { 646 add $grp aio_open $dst, O_CREAT | O_WRONLY | O_TRUNC, 0200, sub {
478 if (my $dst_fh = $_[0]) { 647 if (my $dst_fh = $_[0]) {
479 aioreq_pri $pri; 648 aioreq_pri $pri;
480 add $grp aio_sendfile $dst_fh, $src_fh, 0, $stat[7], sub { 649 add $grp aio_sendfile $dst_fh, $src_fh, 0, $stat[7], sub {
481 if ($_[0] == $stat[7]) { 650 if ($_[0] == $stat[7]) {
482 $grp->result (0); 651 $grp->result (0);
483 close $src_fh; 652 close $src_fh;
484 653
485 # those should not normally block. should. should. 654 # those should not normally block. should. should.
486 utime $stat[8], $stat[9], $dst; 655 utime $stat[8], $stat[9], $dst;
487 chmod $stat[2] & 07777, $dst_fh; 656 chmod $stat[2] & 07777, $dst_fh;
488 chown $stat[4], $stat[5], $dst_fh; 657 chown $stat[4], $stat[5], $dst_fh;
658
659 aioreq_pri $pri;
489 close $dst_fh; 660 add $grp aio_close $dst_fh;
490 } else { 661 } else {
491 $grp->result (-1); 662 $grp->result (-1);
492 close $src_fh; 663 close $src_fh;
493 close $dst_fh; 664 close $dst_fh;
494 665
495 aioreq $pri; 666 aioreq $pri;
496 add $grp aio_unlink $dst; 667 add $grp aio_unlink $dst;
668 }
497 } 669 };
670 } else {
671 $grp->result (-1);
498 }; 672 }
499 } else {
500 $grp->result (-1);
501 } 673 },
674
675 } else {
676 $grp->result (-1);
502 }, 677 }
503
504 } else {
505 $grp->result (-1);
506 } 678 };
679
680 $grp
507 }; 681 }
508
509 $grp
510} 682}
511 683
512=item aio_move $srcpath, $dstpath, $callback->($status) 684=item aio_move $srcpath, $dstpath, $callback->($status)
513 685
514Try to move the I<file> (directories not supported as either source or 686Try to move the I<file> (directories not supported as either source or
520that is successful, unlinking the C<$srcpath>. 692that is successful, unlinking the C<$srcpath>.
521 693
522=cut 694=cut
523 695
524sub aio_move($$;$) { 696sub aio_move($$;$) {
697 aio_block {
525 my ($src, $dst, $cb) = @_; 698 my ($src, $dst, $cb) = @_;
526 699
527 my $pri = aioreq_pri; 700 my $pri = aioreq_pri;
528 my $grp = aio_group $cb; 701 my $grp = aio_group $cb;
529 702
530 aioreq_pri $pri; 703 aioreq_pri $pri;
531 add $grp aio_rename $src, $dst, sub { 704 add $grp aio_rename $src, $dst, sub {
532 if ($_[0] && $! == EXDEV) { 705 if ($_[0] && $! == EXDEV) {
533 aioreq_pri $pri; 706 aioreq_pri $pri;
534 add $grp aio_copy $src, $dst, sub { 707 add $grp aio_copy $src, $dst, sub {
708 $grp->result ($_[0]);
709
710 if (!$_[0]) {
711 aioreq_pri $pri;
712 add $grp aio_unlink $src;
713 }
714 };
715 } else {
535 $grp->result ($_[0]); 716 $grp->result ($_[0]);
536
537 if (!$_[0]) {
538 aioreq_pri $pri;
539 add $grp aio_unlink $src;
540 }
541 }; 717 }
542 } else {
543 $grp->result ($_[0]);
544 } 718 };
719
720 $grp
545 }; 721 }
546
547 $grp
548} 722}
549 723
550=item aio_scandir $path, $maxreq, $callback->($dirs, $nondirs) 724=item aio_scandir $path, $maxreq, $callback->($dirs, $nondirs)
551 725
552Scans a directory (similar to C<aio_readdir>) but additionally tries to 726Scans a directory (similar to C<aio_readdir>) but additionally tries to
599as those tend to return 0 or 1 as link counts, which disables the 773as those tend to return 0 or 1 as link counts, which disables the
600directory counting heuristic. 774directory counting heuristic.
601 775
602=cut 776=cut
603 777
604sub aio_scandir($$$) { 778sub aio_scandir($$;$) {
779 aio_block {
605 my ($path, $maxreq, $cb) = @_; 780 my ($path, $maxreq, $cb) = @_;
606 781
607 my $pri = aioreq_pri; 782 my $pri = aioreq_pri;
608 783
609 my $grp = aio_group $cb; 784 my $grp = aio_group $cb;
610 785
611 $maxreq = 4 if $maxreq <= 0; 786 $maxreq = 4 if $maxreq <= 0;
612 787
613 # stat once 788 # stat once
614 aioreq_pri $pri;
615 add $grp aio_stat $path, sub {
616 return $grp->result () if $_[0];
617 my $now = time;
618 my $hash1 = join ":", (stat _)[0,1,3,7,9];
619
620 # read the directory entries
621 aioreq_pri $pri; 789 aioreq_pri $pri;
622 add $grp aio_readdir $path, sub { 790 add $grp aio_stat $path, sub {
623 my $entries = shift
624 or return $grp->result (); 791 return $grp->result () if $_[0];
792 my $now = time;
793 my $hash1 = join ":", (stat _)[0,1,3,7,9];
625 794
626 # stat the dir another time 795 # read the directory entries
627 aioreq_pri $pri; 796 aioreq_pri $pri;
797 add $grp aio_readdir $path, sub {
798 my $entries = shift
799 or return $grp->result ();
800
801 # stat the dir another time
802 aioreq_pri $pri;
628 add $grp aio_stat $path, sub { 803 add $grp aio_stat $path, sub {
629 my $hash2 = join ":", (stat _)[0,1,3,7,9]; 804 my $hash2 = join ":", (stat _)[0,1,3,7,9];
630 805
631 my $ndirs; 806 my $ndirs;
632 807
633 # take the slow route if anything looks fishy 808 # take the slow route if anything looks fishy
634 if ($hash1 ne $hash2 or (stat _)[9] == $now) { 809 if ($hash1 ne $hash2 or (stat _)[9] == $now) {
635 $ndirs = -1; 810 $ndirs = -1;
636 } else { 811 } else {
637 # if nlink == 2, we are finished 812 # if nlink == 2, we are finished
638 # on non-posix-fs's, we rely on nlink < 2 813 # on non-posix-fs's, we rely on nlink < 2
639 $ndirs = (stat _)[3] - 2 814 $ndirs = (stat _)[3] - 2
640 or return $grp->result ([], $entries); 815 or return $grp->result ([], $entries);
641 } 816 }
642 817
643 # sort into likely dirs and likely nondirs 818 # sort into likely dirs and likely nondirs
644 # dirs == files without ".", short entries first 819 # dirs == files without ".", short entries first
645 $entries = [map $_->[0], 820 $entries = [map $_->[0],
646 sort { $b->[1] cmp $a->[1] } 821 sort { $b->[1] cmp $a->[1] }
647 map [$_, sprintf "%s%04d", (/.\./ ? "1" : "0"), length], 822 map [$_, sprintf "%s%04d", (/.\./ ? "1" : "0"), length],
648 @$entries]; 823 @$entries];
649 824
650 my (@dirs, @nondirs); 825 my (@dirs, @nondirs);
651 826
652 my $statgrp = add $grp aio_group sub { 827 my $statgrp = add $grp aio_group sub {
653 $grp->result (\@dirs, \@nondirs); 828 $grp->result (\@dirs, \@nondirs);
654 }; 829 };
655 830
656 limit $statgrp $maxreq; 831 limit $statgrp $maxreq;
657 feed $statgrp sub { 832 feed $statgrp sub {
658 return unless @$entries; 833 return unless @$entries;
659 my $entry = pop @$entries; 834 my $entry = pop @$entries;
660 835
661 aioreq_pri $pri; 836 aioreq_pri $pri;
662 add $statgrp aio_stat "$path/$entry/.", sub { 837 add $statgrp aio_stat "$path/$entry/.", sub {
663 if ($_[0] < 0) { 838 if ($_[0] < 0) {
664 push @nondirs, $entry; 839 push @nondirs, $entry;
665 } else { 840 } else {
666 # need to check for real directory 841 # need to check for real directory
667 aioreq_pri $pri; 842 aioreq_pri $pri;
668 add $statgrp aio_lstat "$path/$entry", sub { 843 add $statgrp aio_lstat "$path/$entry", sub {
669 if (-d _) { 844 if (-d _) {
670 push @dirs, $entry; 845 push @dirs, $entry;
671 846
672 unless (--$ndirs) { 847 unless (--$ndirs) {
673 push @nondirs, @$entries; 848 push @nondirs, @$entries;
674 feed $statgrp; 849 feed $statgrp;
850 }
851 } else {
852 push @nondirs, $entry;
675 } 853 }
676 } else {
677 push @nondirs, $entry;
678 } 854 }
679 } 855 }
680 } 856 };
681 }; 857 };
682 }; 858 };
683 }; 859 };
684 }; 860 };
861
862 $grp
685 }; 863 }
686
687 $grp
688} 864}
865
866=item aio_rmtree $path, $callback->($status)
867
868Delete a directory tree starting (and including) C<$path>, return the
869status of the final C<rmdir> only. This is a composite request that
870uses C<aio_scandir> to recurse into and rmdir directories, and unlink
871everything else.
872
873=cut
874
875sub aio_rmtree;
876sub aio_rmtree($;$) {
877 aio_block {
878 my ($path, $cb) = @_;
879
880 my $pri = aioreq_pri;
881 my $grp = aio_group $cb;
882
883 aioreq_pri $pri;
884 add $grp aio_scandir $path, 0, sub {
885 my ($dirs, $nondirs) = @_;
886
887 my $dirgrp = aio_group sub {
888 add $grp aio_rmdir $path, sub {
889 $grp->result ($_[0]);
890 };
891 };
892
893 (aioreq_pri $pri), add $dirgrp aio_rmtree "$path/$_" for @$dirs;
894 (aioreq_pri $pri), add $dirgrp aio_unlink "$path/$_" for @$nondirs;
895
896 add $grp $dirgrp;
897 };
898
899 $grp
900 }
901}
902
903=item aio_sync $callback->($status)
904
905Asynchronously call sync and call the callback when finished.
689 906
690=item aio_fsync $fh, $callback->($status) 907=item aio_fsync $fh, $callback->($status)
691 908
692Asynchronously call fsync on the given filehandle and call the callback 909Asynchronously call fsync on the given filehandle and call the callback
693with the fsync result code. 910with the fsync result code.
697Asynchronously call fdatasync on the given filehandle and call the 914Asynchronously call fdatasync on the given filehandle and call the
698callback with the fdatasync result code. 915callback with the fdatasync result code.
699 916
700If this call isn't available because your OS lacks it or it couldn't be 917If this call isn't available because your OS lacks it or it couldn't be
701detected, it will be emulated by calling C<fsync> instead. 918detected, it will be emulated by calling C<fsync> instead.
919
920=item aio_pathsync $path, $callback->($status)
921
922This request tries to open, fsync and close the given path. This is a
923composite request intended tosync directories after directory operations
924(E.g. rename). This might not work on all operating systems or have any
925specific effect, but usually it makes sure that directory changes get
926written to disc. It works for anything that can be opened for read-only,
927not just directories.
928
929Passes C<0> when everything went ok, and C<-1> on error.
930
931=cut
932
933sub aio_pathsync($;$) {
934 aio_block {
935 my ($path, $cb) = @_;
936
937 my $pri = aioreq_pri;
938 my $grp = aio_group $cb;
939
940 aioreq_pri $pri;
941 add $grp aio_open $path, O_RDONLY, 0, sub {
942 my ($fh) = @_;
943 if ($fh) {
944 aioreq_pri $pri;
945 add $grp aio_fsync $fh, sub {
946 $grp->result ($_[0]);
947
948 aioreq_pri $pri;
949 add $grp aio_close $fh;
950 };
951 } else {
952 $grp->result (-1);
953 }
954 };
955
956 $grp
957 }
958}
702 959
703=item aio_group $callback->(...) 960=item aio_group $callback->(...)
704 961
705This is a very special aio request: Instead of doing something, it is a 962This is a very special aio request: Instead of doing something, it is a
706container for other aio requests, which is useful if you want to bundle 963container for other aio requests, which is useful if you want to bundle
843itself. Useful when you queued a lot of events but got a result early. 1100itself. Useful when you queued a lot of events but got a result early.
844 1101
845=item $grp->result (...) 1102=item $grp->result (...)
846 1103
847Set the result value(s) that will be passed to the group callback when all 1104Set the result value(s) that will be passed to the group callback when all
848subrequests have finished and set thre groups errno to the current value 1105subrequests have finished and set the groups errno to the current value
849of errno (just like calling C<errno> without an error number). By default, 1106of errno (just like calling C<errno> without an error number). By default,
850no argument will be passed and errno is zero. 1107no argument will be passed and errno is zero.
851 1108
852=item $grp->errno ([$errno]) 1109=item $grp->errno ([$errno])
853 1110
971 poll => 'r', nice => 1, 1228 poll => 'r', nice => 1,
972 cb => &IO::AIO::poll_cb); 1229 cb => &IO::AIO::poll_cb);
973 1230
974=item IO::AIO::poll_wait 1231=item IO::AIO::poll_wait
975 1232
1233If there are any outstanding requests and none of them in the result
976Wait till the result filehandle becomes ready for reading (simply does a 1234phase, wait till the result filehandle becomes ready for reading (simply
977C<select> on the filehandle. This is useful if you want to synchronously 1235does a C<select> on the filehandle. This is useful if you want to
978wait for some requests to finish). 1236synchronously wait for some requests to finish).
979 1237
980See C<nreqs> for an example. 1238See C<nreqs> for an example.
981 1239
982=item IO::AIO::poll 1240=item IO::AIO::poll
983 1241
984Waits until some requests have been handled. 1242Waits until some requests have been handled.
985 1243
1244Returns the number of requests processed, but is otherwise strictly
986Strictly equivalent to: 1245equivalent to:
987 1246
988 IO::AIO::poll_wait, IO::AIO::poll_cb 1247 IO::AIO::poll_wait, IO::AIO::poll_cb
989 if IO::AIO::nreqs;
990 1248
991=item IO::AIO::flush 1249=item IO::AIO::flush
992 1250
993Wait till all outstanding AIO requests have been handled. 1251Wait till all outstanding AIO requests have been handled.
994 1252
995Strictly equivalent to: 1253Strictly equivalent to:
996 1254
997 IO::AIO::poll_wait, IO::AIO::poll_cb 1255 IO::AIO::poll_wait, IO::AIO::poll_cb
998 while IO::AIO::nreqs; 1256 while IO::AIO::nreqs;
999 1257
1258=back
1259
1000=head3 CONTROLLING THE NUMBER OF THREADS 1260=head3 CONTROLLING THE NUMBER OF THREADS
1261
1262=over
1001 1263
1002=item IO::AIO::min_parallel $nthreads 1264=item IO::AIO::min_parallel $nthreads
1003 1265
1004Set the minimum number of AIO threads to C<$nthreads>. The current 1266Set the minimum number of AIO threads to C<$nthreads>. The current
1005default is C<8>, which means eight asynchronous operations can execute 1267default is C<8>, which means eight asynchronous operations can execute
1053This is a very bad function to use in interactive programs because it 1315This is a very bad function to use in interactive programs because it
1054blocks, and a bad way to reduce concurrency because it is inexact: Better 1316blocks, and a bad way to reduce concurrency because it is inexact: Better
1055use an C<aio_group> together with a feed callback. 1317use an C<aio_group> together with a feed callback.
1056 1318
1057Sets the maximum number of outstanding requests to C<$nreqs>. If you 1319Sets the maximum number of outstanding requests to C<$nreqs>. If you
1058to queue up more than this number of requests, the next call to the 1320do queue up more than this number of requests, the next call to the
1059C<poll_cb> (and C<poll_some> and other functions calling C<poll_cb>) 1321C<poll_cb> (and C<poll_some> and other functions calling C<poll_cb>)
1060function will block until the limit is no longer exceeded. 1322function will block until the limit is no longer exceeded.
1061 1323
1062The default value is very large, so there is no practical limit on the 1324The default value is very large, so there is no practical limit on the
1063number of outstanding requests. 1325number of outstanding requests.
1064 1326
1065You can still queue as many requests as you want. Therefore, 1327You can still queue as many requests as you want. Therefore,
1066C<max_oustsanding> is mainly useful in simple scripts (with low values) or 1328C<max_oustsanding> is mainly useful in simple scripts (with low values) or
1067as a stop gap to shield against fatal memory overflow (with large values). 1329as a stop gap to shield against fatal memory overflow (with large values).
1068 1330
1331=back
1332
1069=head3 STATISTICAL INFORMATION 1333=head3 STATISTICAL INFORMATION
1334
1335=over
1070 1336
1071=item IO::AIO::nreqs 1337=item IO::AIO::nreqs
1072 1338
1073Returns the number of requests currently in the ready, execute or pending 1339Returns the number of requests currently in the ready, execute or pending
1074states (i.e. for which their callback has not been invoked yet). 1340states (i.e. for which their callback has not been invoked yet).
1090 1356
1091=back 1357=back
1092 1358
1093=cut 1359=cut
1094 1360
1095# support function to convert a fd into a perl filehandle
1096sub _fd2fh {
1097 return undef if $_[0] < 0;
1098
1099 # try to generate nice filehandles
1100 my $sym = "IO::AIO::fd#$_[0]";
1101 local *$sym;
1102
1103 open *$sym, "+<&=$_[0]" # usually works under any unix
1104 or open *$sym, "<&=$_[0]" # cygwin needs this
1105 or open *$sym, ">&=$_[0]" # or this
1106 or return undef;
1107
1108 *$sym
1109}
1110
1111min_parallel 8; 1361min_parallel 8;
1112 1362
1113END { 1363END { flush }
1114 min_parallel 1;
1115 flush;
1116};
1117 1364
11181; 13651;
1119 1366
1120=head2 FORK BEHAVIOUR 1367=head2 FORK BEHAVIOUR
1121 1368
1141bytes of memory. In addition, stat requests need a stat buffer (possibly 1388bytes of memory. In addition, stat requests need a stat buffer (possibly
1142a few hundred bytes), readdir requires a result buffer and so on. Perl 1389a few hundred bytes), readdir requires a result buffer and so on. Perl
1143scalars and other data passed into aio requests will also be locked and 1390scalars and other data passed into aio requests will also be locked and
1144will consume memory till the request has entered the done state. 1391will consume memory till the request has entered the done state.
1145 1392
1146This is now awfully much, so queuing lots of requests is not usually a 1393This is not awfully much, so queuing lots of requests is not usually a
1147problem. 1394problem.
1148 1395
1149Per-thread usage: 1396Per-thread usage:
1150 1397
1151In the execution phase, some aio requests require more memory for 1398In the execution phase, some aio requests require more memory for

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines