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.86 by root, Sat Oct 28 23:32:29 2006 UTC vs.
Revision 1.116 by root, Wed Oct 3 21:27:51 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
61etc.), but can also be used to easily do operations in parallel that are 62etc.), 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 63normally 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 64on a RAID volume or over NFS when you do a number of stat operations
64concurrently. 65concurrently.
65 66
66While this works on all types of file descriptors (for example sockets), 67While most of this works on all types of file descriptors (for
67using these functions on file descriptors that support nonblocking 68example sockets), using these functions on file descriptors that
68operation (again, sockets, pipes etc.) is very inefficient. Use an event 69support nonblocking operation (again, sockets, pipes etc.) is very
69loop for that (such as the L<Event|Event> module): IO::AIO will naturally 70inefficient. Use an event loop for that (such as the L<Event|Event>
70fit into such an event loop itself. 71module): IO::AIO will naturally fit into such an event loop itself.
71 72
72In this version, a number of threads are started that execute your 73In this version, a number of threads are started that execute your
73requests and signal their completion. You don't need thread support 74requests and signal their completion. You don't need thread support
74in perl, and the threads created by this module will not be visible 75in 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 76to 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 78not well-supported or restricted (GNU/Linux doesn't allow them on normal
78files currently, for example), and they would only support aio_read and 79files currently, for example), and they would only support aio_read and
79aio_write, so the remaining functionality would have to be implemented 80aio_write, so the remaining functionality would have to be implemented
80using threads anyway. 81using threads anyway.
81 82
82Although the module will work with in the presence of other (Perl-) 83Although the module will work in the presence of other (Perl-) threads,
83threads, it is currently not reentrant in any way, so use appropriate 84it is currently not reentrant in any way, so use appropriate locking
84locking yourself, always call C<poll_cb> from within the same thread, or 85yourself, always call C<poll_cb> from within the same thread, or never
85never call C<poll_cb> (or other C<aio_> functions) recursively. 86call C<poll_cb> (or other C<aio_> functions) recursively.
86 87
87=head2 EXAMPLE 88=head2 EXAMPLE
88 89
89This is a simple example that uses the Event module and loads 90This is a simple example that uses the Event module and loads
90F</etc/passwd> asynchronously: 91F</etc/passwd> asynchronously:
98 poll => 'r', 99 poll => 'r',
99 cb => \&IO::AIO::poll_cb); 100 cb => \&IO::AIO::poll_cb);
100 101
101 # queue the request to open /etc/passwd 102 # queue the request to open /etc/passwd
102 aio_open "/etc/passwd", O_RDONLY, 0, sub { 103 aio_open "/etc/passwd", O_RDONLY, 0, sub {
103 my $fh = $_[0] 104 my $fh = shift
104 or die "error while opening: $!"; 105 or die "error while opening: $!";
105 106
106 # stat'ing filehandles is generally non-blocking 107 # stat'ing filehandles is generally non-blocking
107 my $size = -s $fh; 108 my $size = -s $fh;
108 109
176Request has reached the end of its lifetime and holds no resources anymore 177Request 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 178(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 179aio request is severed and calling its methods will either do nothing or
179result in a runtime error). 180result in a runtime error).
180 181
182=back
183
181=cut 184=cut
182 185
183package IO::AIO; 186package IO::AIO;
184 187
185no warnings; 188no warnings;
186use strict 'vars'; 189use strict 'vars';
187 190
188use base 'Exporter'; 191use base 'Exporter';
189 192
190BEGIN { 193BEGIN {
191 our $VERSION = '2.1'; 194 our $VERSION = '2.5';
192 195
193 our @AIO_REQ = qw(aio_sendfile aio_read aio_write aio_open aio_close aio_stat 196 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 197 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 198 aio_readlink aio_fsync aio_fdatasync aio_readahead aio_rename aio_link
196 aio_copy aio_group aio_nop aio_mknod); 199 aio_move aio_copy aio_group aio_nop aio_mknod aio_load aio_rmtree aio_mkdir
200 aio_chown aio_chmod aio_utime aio_truncate);
197 our @EXPORT = (@AIO_REQ, qw(aioreq_pri aioreq_nice)); 201 our @EXPORT = (@AIO_REQ, qw(aioreq_pri aioreq_nice aio_block));
198 our @EXPORT_OK = qw(poll_fileno poll_cb poll_wait flush 202 our @EXPORT_OK = qw(poll_fileno poll_cb poll_wait flush
199 min_parallel max_parallel max_idle 203 min_parallel max_parallel max_idle
200 nreqs nready npending nthreads 204 nreqs nready npending nthreads
201 max_poll_time max_poll_reqs); 205 max_poll_time max_poll_reqs);
202 206
206 XSLoader::load ("IO::AIO", $VERSION); 210 XSLoader::load ("IO::AIO", $VERSION);
207} 211}
208 212
209=head1 FUNCTIONS 213=head1 FUNCTIONS
210 214
211=head2 AIO FUNCTIONS 215=head2 AIO REQUEST FUNCTIONS
212 216
213All the C<aio_*> calls are more or less thin wrappers around the syscall 217All the C<aio_*> calls are more or less thin wrappers around the syscall
214with the same name (sans C<aio_>). The arguments are similar or identical, 218with the same name (sans C<aio_>). The arguments are similar or identical,
215and they all accept an additional (and optional) C<$callback> argument 219and they all accept an additional (and optional) C<$callback> argument
216which must be a code reference. This code reference will get called with 220which must be a code reference. This code reference will get called with
219syscall has been executed asynchronously. 223syscall has been executed asynchronously.
220 224
221All functions expecting a filehandle keep a copy of the filehandle 225All functions expecting a filehandle keep a copy of the filehandle
222internally until the request has finished. 226internally until the request has finished.
223 227
224All requests return objects of type L<IO::AIO::REQ> that allow further 228All functions return request objects of type L<IO::AIO::REQ> that allow
225manipulation of those requests while they are in-flight. 229further manipulation of those requests while they are in-flight.
226 230
227The pathnames you pass to these routines I<must> be absolute and 231The pathnames you pass to these routines I<must> be absolute and
228encoded in byte form. The reason for the former is that at the time the 232encoded as octets. The reason for the former is that at the time the
229request is being executed, the current working directory could have 233request is being executed, the current working directory could have
230changed. Alternatively, you can make sure that you never change the 234changed. Alternatively, you can make sure that you never change the
231current working directory. 235current working directory anywhere in the program and then use relative
236paths.
232 237
233To encode pathnames to byte form, either make sure you either: a) 238To encode pathnames as octets, either make sure you either: a) always pass
234always pass in filenames you got from outside (command line, readdir 239in filenames you got from outside (command line, readdir etc.) without
235etc.), b) are ASCII or ISO 8859-1, c) use the Encode module and encode 240tinkering, b) are ASCII or ISO 8859-1, c) use the Encode module and encode
236your pathnames to the locale (or other) encoding in effect in the user 241your pathnames to the locale (or other) encoding in effect in the user
237environment, d) use Glib::filename_from_unicode on unicode filenames or e) 242environment, d) use Glib::filename_from_unicode on unicode filenames or e)
238use something else. 243use something else to ensure your scalar has the correct contents.
244
245This works, btw. independent of the internal UTF-8 bit, which IO::AIO
246handles correctly wether it is set or not.
239 247
240=over 4 248=over 4
241 249
242=item $prev_pri = aioreq_pri [$pri] 250=item $prev_pri = aioreq_pri [$pri]
243 251
263 aio_read $_[0], ..., sub { 271 aio_read $_[0], ..., sub {
264 ... 272 ...
265 }; 273 };
266 }; 274 };
267 275
276
268=item aioreq_nice $pri_adjust 277=item aioreq_nice $pri_adjust
269 278
270Similar to C<aioreq_pri>, but subtracts the given value from the current 279Similar to C<aioreq_pri>, but subtracts the given value from the current
271priority, so effects are cumulative. 280priority, so the effect is cumulative.
281
272 282
273=item aio_open $pathname, $flags, $mode, $callback->($fh) 283=item aio_open $pathname, $flags, $mode, $callback->($fh)
274 284
275Asynchronously open or create a file and call the callback with a newly 285Asynchronously open or create a file and call the callback with a newly
276created filehandle for the file. 286created filehandle for the file.
282list. They are the same as used by C<sysopen>. 292list. They are the same as used by C<sysopen>.
283 293
284Likewise, C<$mode> specifies the mode of the newly created file, if it 294Likewise, C<$mode> specifies the mode of the newly created file, if it
285didn't exist and C<O_CREAT> has been given, just like perl's C<sysopen>, 295didn't exist and C<O_CREAT> has been given, just like perl's C<sysopen>,
286except that it is mandatory (i.e. use C<0> if you don't create new files, 296except that it is mandatory (i.e. use C<0> if you don't create new files,
287and C<0666> or C<0777> if you do). 297and C<0666> or C<0777> if you do). Note that the C<$mode> will be modified
298by the umask in effect then the request is being executed, so better never
299change the umask.
288 300
289Example: 301Example:
290 302
291 aio_open "/etc/passwd", O_RDONLY, 0, sub { 303 aio_open "/etc/passwd", O_RDONLY, 0, sub {
292 if ($_[0]) { 304 if ($_[0]) {
295 } else { 307 } else {
296 die "open failed: $!\n"; 308 die "open failed: $!\n";
297 } 309 }
298 }; 310 };
299 311
312
300=item aio_close $fh, $callback->($status) 313=item aio_close $fh, $callback->($status)
301 314
302Asynchronously close a file and call the callback with the result 315Asynchronously close a file and call the callback with the result
303code. I<WARNING:> although accepted, you should not pass in a perl 316code.
304filehandle here, as perl will likely close the file descriptor another
305time when the filehandle is destroyed. Normally, you can safely call perls
306C<close> or just let filehandles go out of scope.
307 317
308This is supposed to be a bug in the API, so that might change. It's 318Unlike the other functions operating on files, this function uses the
309therefore best to avoid this function. 319PerlIO layer to close the filehandle. The reason is that the PerlIO API
320insists on closing the underlying fd itself, no matter what, and doesn't
321allow modifications to the fd. Unfortunately, it is not clear that you can
322call PerlIO from different threads (actually, its quite clear that this
323won't work in some cases), so while it likely works perfectly with simple
324file handles (such as the ones created by C<aio_open>) it might fail in
325interesting ways for others.
326
327Having said that, aio_close tries to clean up the filehandle as much as
328possible before handing it to an io thread, and generally does work.
329
310 330
311=item aio_read $fh,$offset,$length, $data,$dataoffset, $callback->($retval) 331=item aio_read $fh,$offset,$length, $data,$dataoffset, $callback->($retval)
312 332
313=item aio_write $fh,$offset,$length, $data,$dataoffset, $callback->($retval) 333=item aio_write $fh,$offset,$length, $data,$dataoffset, $callback->($retval)
314 334
315Reads or writes C<length> bytes from the specified C<fh> and C<offset> 335Reads or writes C<$length> bytes from the specified C<$fh> and C<$offset>
316into the scalar given by C<data> and offset C<dataoffset> and calls the 336into the scalar given by C<$data> and offset C<$dataoffset> and calls the
317callback without the actual number of bytes read (or -1 on error, just 337callback without the actual number of bytes read (or -1 on error, just
318like the syscall). 338like the syscall).
319 339
340If C<$offset> is undefined, then the current file descriptor offset will
341be used (and updated), otherwise the file descriptor offset will not be
342changed by these calls.
343
344If C<$length> is undefined in C<aio_write>, use the remaining length of C<$data>.
345
346If C<$dataoffset> is less than zero, it will be counted from the end of
347C<$data>.
348
320The C<$data> scalar I<MUST NOT> be modified in any way while the request 349The C<$data> scalar I<MUST NOT> be modified in any way while the request
321is outstanding. Modifying it can result in segfaults or WW3 (if the 350is outstanding. Modifying it can result in segfaults or World War III (if
322necessary/optional hardware is installed). 351the necessary/optional hardware is installed).
323 352
324Example: Read 15 bytes at offset 7 into scalar C<$buffer>, starting at 353Example: Read 15 bytes at offset 7 into scalar C<$buffer>, starting at
325offset C<0> within the scalar: 354offset C<0> within the scalar:
326 355
327 aio_read $fh, 7, 15, $buffer, 0, sub { 356 aio_read $fh, 7, 15, $buffer, 0, sub {
328 $_[0] > 0 or die "read error: $!"; 357 $_[0] > 0 or die "read error: $!";
329 print "read $_[0] bytes: <$buffer>\n"; 358 print "read $_[0] bytes: <$buffer>\n";
330 }; 359 };
360
331 361
332=item aio_sendfile $out_fh, $in_fh, $in_offset, $length, $callback->($retval) 362=item aio_sendfile $out_fh, $in_fh, $in_offset, $length, $callback->($retval)
333 363
334Tries to copy C<$length> bytes from C<$in_fh> to C<$out_fh>. It starts 364Tries to copy C<$length> bytes from C<$in_fh> to C<$out_fh>. It starts
335reading at byte offset C<$in_offset>, and starts writing at the current 365reading at byte offset C<$in_offset>, and starts writing at the current
349C<$in_fh> than are written, and there is no way to find out how many 379C<$in_fh> than are written, and there is no way to find out how many
350bytes have been read from C<aio_sendfile> alone, as C<aio_sendfile> only 380bytes have been read from C<aio_sendfile> alone, as C<aio_sendfile> only
351provides the number of bytes written to C<$out_fh>. Only if the result 381provides the number of bytes written to C<$out_fh>. Only if the result
352value equals C<$length> one can assume that C<$length> bytes have been 382value equals C<$length> one can assume that C<$length> bytes have been
353read. 383read.
384
354 385
355=item aio_readahead $fh,$offset,$length, $callback->($retval) 386=item aio_readahead $fh,$offset,$length, $callback->($retval)
356 387
357C<aio_readahead> populates the page cache with data from a file so that 388C<aio_readahead> populates the page cache with data from a file so that
358subsequent reads from that file will not block on disk I/O. The C<$offset> 389subsequent reads from that file will not block on disk I/O. The C<$offset>
364file. The current file offset of the file is left unchanged. 395file. The current file offset of the file is left unchanged.
365 396
366If that syscall doesn't exist (likely if your OS isn't Linux) it will be 397If that syscall doesn't exist (likely if your OS isn't Linux) it will be
367emulated by simply reading the data, which would have a similar effect. 398emulated by simply reading the data, which would have a similar effect.
368 399
400
369=item aio_stat $fh_or_path, $callback->($status) 401=item aio_stat $fh_or_path, $callback->($status)
370 402
371=item aio_lstat $fh, $callback->($status) 403=item aio_lstat $fh, $callback->($status)
372 404
373Works like perl's C<stat> or C<lstat> in void context. The callback will 405Works like perl's C<stat> or C<lstat> in void context. The callback will
386 aio_stat "/etc/passwd", sub { 418 aio_stat "/etc/passwd", sub {
387 $_[0] and die "stat failed: $!"; 419 $_[0] and die "stat failed: $!";
388 print "size is ", -s _, "\n"; 420 print "size is ", -s _, "\n";
389 }; 421 };
390 422
423
424=item aio_utime $fh_or_path, $atime, $mtime, $callback->($status)
425
426Works like perl's C<utime> function (including the special case of $atime
427and $mtime being undef). Fractional times are supported if the underlying
428syscalls support them.
429
430When called with a pathname, uses utimes(2) if available, otherwise
431utime(2). If called on a file descriptor, uses futimes(2) if available,
432otherwise returns ENOSYS, so this is not portable.
433
434Examples:
435
436 # set atime and mtime to current time (basically touch(1)):
437 aio_utime "path", undef, undef;
438 # set atime to current time and mtime to beginning of the epoch:
439 aio_utime "path", time, undef; # undef==0
440
441
442=item aio_chown $fh_or_path, $uid, $gid, $callback->($status)
443
444Works like perl's C<chown> function, except that C<undef> for either $uid
445or $gid is being interpreted as "do not change" (but -1 can also be used).
446
447Examples:
448
449 # same as "chown root path" in the shell:
450 aio_chown "path", 0, -1;
451 # same as above:
452 aio_chown "path", 0, undef;
453
454
455=item aio_truncate $fh_or_path, $offset, $callback->($status)
456
457Works like truncate(2) or ftruncate(2).
458
459
460=item aio_chmod $fh_or_path, $mode, $callback->($status)
461
462Works like perl's C<chmod> function.
463
464
391=item aio_unlink $pathname, $callback->($status) 465=item aio_unlink $pathname, $callback->($status)
392 466
393Asynchronously unlink (delete) a file and call the callback with the 467Asynchronously unlink (delete) a file and call the callback with the
394result code. 468result code.
395 469
470
396=item aio_mknod $path, $mode, $dev, $callback->($status) 471=item aio_mknod $path, $mode, $dev, $callback->($status)
397 472
398[EXPERIMENTAL] 473[EXPERIMENTAL]
399 474
400Asynchronously create a device node (or fifo). See mknod(2). 475Asynchronously create a device node (or fifo). See mknod(2).
401 476
402The only (POSIX-) portable way of calling this function is: 477The only (POSIX-) portable way of calling this function is:
403 478
404 aio_mknod $path, IO::AIO::S_IFIFO | $mode, 0, sub { ... 479 aio_mknod $path, IO::AIO::S_IFIFO | $mode, 0, sub { ...
480
405 481
406=item aio_link $srcpath, $dstpath, $callback->($status) 482=item aio_link $srcpath, $dstpath, $callback->($status)
407 483
408Asynchronously create a new link to the existing object at C<$srcpath> at 484Asynchronously create a new link to the existing object at C<$srcpath> at
409the path C<$dstpath> and call the callback with the result code. 485the path C<$dstpath> and call the callback with the result code.
410 486
487
411=item aio_symlink $srcpath, $dstpath, $callback->($status) 488=item aio_symlink $srcpath, $dstpath, $callback->($status)
412 489
413Asynchronously create a new symbolic link to the existing object at C<$srcpath> at 490Asynchronously create a new symbolic link to the existing object at C<$srcpath> at
414the path C<$dstpath> and call the callback with the result code. 491the path C<$dstpath> and call the callback with the result code.
415 492
493
494=item aio_readlink $path, $callback->($link)
495
496Asynchronously read the symlink specified by C<$path> and pass it to
497the callback. If an error occurs, nothing or undef gets passed to the
498callback.
499
500
416=item aio_rename $srcpath, $dstpath, $callback->($status) 501=item aio_rename $srcpath, $dstpath, $callback->($status)
417 502
418Asynchronously rename the object at C<$srcpath> to C<$dstpath>, just as 503Asynchronously rename the object at C<$srcpath> to C<$dstpath>, just as
419rename(2) and call the callback with the result code. 504rename(2) and call the callback with the result code.
420 505
506
507=item aio_mkdir $pathname, $mode, $callback->($status)
508
509Asynchronously mkdir (create) a directory and call the callback with
510the result code. C<$mode> will be modified by the umask at the time the
511request is executed, so do not change your umask.
512
513
421=item aio_rmdir $pathname, $callback->($status) 514=item aio_rmdir $pathname, $callback->($status)
422 515
423Asynchronously rmdir (delete) a directory and call the callback with the 516Asynchronously rmdir (delete) a directory and call the callback with the
424result code. 517result code.
518
425 519
426=item aio_readdir $pathname, $callback->($entries) 520=item aio_readdir $pathname, $callback->($entries)
427 521
428Unlike the POSIX call of the same name, C<aio_readdir> reads an entire 522Unlike the POSIX call of the same name, C<aio_readdir> reads an entire
429directory (i.e. opendir + readdir + closedir). The entries will not be 523directory (i.e. opendir + readdir + closedir). The entries will not be
430sorted, and will B<NOT> include the C<.> and C<..> entries. 524sorted, and will B<NOT> include the C<.> and C<..> entries.
431 525
432The callback a single argument which is either C<undef> or an array-ref 526The callback a single argument which is either C<undef> or an array-ref
433with the filenames. 527with the filenames.
528
529
530=item aio_load $path, $data, $callback->($status)
531
532This is a composite request that tries to fully load the given file into
533memory. Status is the same as with aio_read.
534
535=cut
536
537sub aio_load($$;$) {
538 aio_block {
539 my ($path, undef, $cb) = @_;
540 my $data = \$_[1];
541
542 my $pri = aioreq_pri;
543 my $grp = aio_group $cb;
544
545 aioreq_pri $pri;
546 add $grp aio_open $path, O_RDONLY, 0, sub {
547 my $fh = shift
548 or return $grp->result (-1);
549
550 aioreq_pri $pri;
551 add $grp aio_read $fh, 0, (-s $fh), $$data, 0, sub {
552 $grp->result ($_[0]);
553 };
554 };
555
556 $grp
557 }
558}
434 559
435=item aio_copy $srcpath, $dstpath, $callback->($status) 560=item aio_copy $srcpath, $dstpath, $callback->($status)
436 561
437Try to copy the I<file> (directories not supported as either source or 562Try to copy the I<file> (directories not supported as either source or
438destination) from C<$srcpath> to C<$dstpath> and call the callback with 563destination) from C<$srcpath> to C<$dstpath> and call the callback with
448errors are being ignored. 573errors are being ignored.
449 574
450=cut 575=cut
451 576
452sub aio_copy($$;$) { 577sub aio_copy($$;$) {
578 aio_block {
453 my ($src, $dst, $cb) = @_; 579 my ($src, $dst, $cb) = @_;
454 580
455 my $pri = aioreq_pri; 581 my $pri = aioreq_pri;
456 my $grp = aio_group $cb; 582 my $grp = aio_group $cb;
457 583
458 aioreq_pri $pri; 584 aioreq_pri $pri;
459 add $grp aio_open $src, O_RDONLY, 0, sub { 585 add $grp aio_open $src, O_RDONLY, 0, sub {
460 if (my $src_fh = $_[0]) { 586 if (my $src_fh = $_[0]) {
461 my @stat = stat $src_fh; 587 my @stat = stat $src_fh;
462 588
463 aioreq_pri $pri; 589 aioreq_pri $pri;
464 add $grp aio_open $dst, O_CREAT | O_WRONLY | O_TRUNC, 0200, sub { 590 add $grp aio_open $dst, O_CREAT | O_WRONLY | O_TRUNC, 0200, sub {
465 if (my $dst_fh = $_[0]) { 591 if (my $dst_fh = $_[0]) {
466 aioreq_pri $pri; 592 aioreq_pri $pri;
467 add $grp aio_sendfile $dst_fh, $src_fh, 0, $stat[7], sub { 593 add $grp aio_sendfile $dst_fh, $src_fh, 0, $stat[7], sub {
468 if ($_[0] == $stat[7]) { 594 if ($_[0] == $stat[7]) {
469 $grp->result (0); 595 $grp->result (0);
470 close $src_fh; 596 close $src_fh;
471 597
472 # those should not normally block. should. should. 598 # those should not normally block. should. should.
473 utime $stat[8], $stat[9], $dst; 599 utime $stat[8], $stat[9], $dst;
474 chmod $stat[2] & 07777, $dst_fh; 600 chmod $stat[2] & 07777, $dst_fh;
475 chown $stat[4], $stat[5], $dst_fh; 601 chown $stat[4], $stat[5], $dst_fh;
476 close $dst_fh; 602 close $dst_fh;
477 } else { 603 } else {
478 $grp->result (-1); 604 $grp->result (-1);
479 close $src_fh; 605 close $src_fh;
480 close $dst_fh; 606 close $dst_fh;
481 607
482 aioreq $pri; 608 aioreq $pri;
483 add $grp aio_unlink $dst; 609 add $grp aio_unlink $dst;
610 }
484 } 611 };
612 } else {
613 $grp->result (-1);
485 }; 614 }
486 } else {
487 $grp->result (-1);
488 } 615 },
616
617 } else {
618 $grp->result (-1);
489 }, 619 }
490
491 } else {
492 $grp->result (-1);
493 } 620 };
621
622 $grp
494 }; 623 }
495
496 $grp
497} 624}
498 625
499=item aio_move $srcpath, $dstpath, $callback->($status) 626=item aio_move $srcpath, $dstpath, $callback->($status)
500 627
501Try to move the I<file> (directories not supported as either source or 628Try to move the I<file> (directories not supported as either source or
507that is successful, unlinking the C<$srcpath>. 634that is successful, unlinking the C<$srcpath>.
508 635
509=cut 636=cut
510 637
511sub aio_move($$;$) { 638sub aio_move($$;$) {
639 aio_block {
512 my ($src, $dst, $cb) = @_; 640 my ($src, $dst, $cb) = @_;
513 641
514 my $pri = aioreq_pri; 642 my $pri = aioreq_pri;
515 my $grp = aio_group $cb; 643 my $grp = aio_group $cb;
516 644
517 aioreq_pri $pri; 645 aioreq_pri $pri;
518 add $grp aio_rename $src, $dst, sub { 646 add $grp aio_rename $src, $dst, sub {
519 if ($_[0] && $! == EXDEV) { 647 if ($_[0] && $! == EXDEV) {
520 aioreq_pri $pri; 648 aioreq_pri $pri;
521 add $grp aio_copy $src, $dst, sub { 649 add $grp aio_copy $src, $dst, sub {
650 $grp->result ($_[0]);
651
652 if (!$_[0]) {
653 aioreq_pri $pri;
654 add $grp aio_unlink $src;
655 }
656 };
657 } else {
522 $grp->result ($_[0]); 658 $grp->result ($_[0]);
523
524 if (!$_[0]) {
525 aioreq_pri $pri;
526 add $grp aio_unlink $src;
527 }
528 }; 659 }
529 } else {
530 $grp->result ($_[0]);
531 } 660 };
661
662 $grp
532 }; 663 }
533
534 $grp
535} 664}
536 665
537=item aio_scandir $path, $maxreq, $callback->($dirs, $nondirs) 666=item aio_scandir $path, $maxreq, $callback->($dirs, $nondirs)
538 667
539Scans a directory (similar to C<aio_readdir>) but additionally tries to 668Scans a directory (similar to C<aio_readdir>) but additionally tries to
586as those tend to return 0 or 1 as link counts, which disables the 715as those tend to return 0 or 1 as link counts, which disables the
587directory counting heuristic. 716directory counting heuristic.
588 717
589=cut 718=cut
590 719
591sub aio_scandir($$$) { 720sub aio_scandir($$;$) {
721 aio_block {
592 my ($path, $maxreq, $cb) = @_; 722 my ($path, $maxreq, $cb) = @_;
593 723
594 my $pri = aioreq_pri; 724 my $pri = aioreq_pri;
595 725
596 my $grp = aio_group $cb; 726 my $grp = aio_group $cb;
597 727
598 $maxreq = 4 if $maxreq <= 0; 728 $maxreq = 4 if $maxreq <= 0;
599 729
600 # stat once 730 # stat once
601 aioreq_pri $pri;
602 add $grp aio_stat $path, sub {
603 return $grp->result () if $_[0];
604 my $now = time;
605 my $hash1 = join ":", (stat _)[0,1,3,7,9];
606
607 # read the directory entries
608 aioreq_pri $pri; 731 aioreq_pri $pri;
609 add $grp aio_readdir $path, sub { 732 add $grp aio_stat $path, sub {
610 my $entries = shift
611 or return $grp->result (); 733 return $grp->result () if $_[0];
734 my $now = time;
735 my $hash1 = join ":", (stat _)[0,1,3,7,9];
612 736
613 # stat the dir another time 737 # read the directory entries
614 aioreq_pri $pri; 738 aioreq_pri $pri;
739 add $grp aio_readdir $path, sub {
740 my $entries = shift
741 or return $grp->result ();
742
743 # stat the dir another time
744 aioreq_pri $pri;
615 add $grp aio_stat $path, sub { 745 add $grp aio_stat $path, sub {
616 my $hash2 = join ":", (stat _)[0,1,3,7,9]; 746 my $hash2 = join ":", (stat _)[0,1,3,7,9];
617 747
618 my $ndirs; 748 my $ndirs;
619 749
620 # take the slow route if anything looks fishy 750 # take the slow route if anything looks fishy
621 if ($hash1 ne $hash2 or (stat _)[9] == $now) { 751 if ($hash1 ne $hash2 or (stat _)[9] == $now) {
622 $ndirs = -1; 752 $ndirs = -1;
623 } else { 753 } else {
624 # if nlink == 2, we are finished 754 # if nlink == 2, we are finished
625 # on non-posix-fs's, we rely on nlink < 2 755 # on non-posix-fs's, we rely on nlink < 2
626 $ndirs = (stat _)[3] - 2 756 $ndirs = (stat _)[3] - 2
627 or return $grp->result ([], $entries); 757 or return $grp->result ([], $entries);
628 } 758 }
629 759
630 # sort into likely dirs and likely nondirs 760 # sort into likely dirs and likely nondirs
631 # dirs == files without ".", short entries first 761 # dirs == files without ".", short entries first
632 $entries = [map $_->[0], 762 $entries = [map $_->[0],
633 sort { $b->[1] cmp $a->[1] } 763 sort { $b->[1] cmp $a->[1] }
634 map [$_, sprintf "%s%04d", (/.\./ ? "1" : "0"), length], 764 map [$_, sprintf "%s%04d", (/.\./ ? "1" : "0"), length],
635 @$entries]; 765 @$entries];
636 766
637 my (@dirs, @nondirs); 767 my (@dirs, @nondirs);
638 768
639 my $statgrp = add $grp aio_group sub { 769 my $statgrp = add $grp aio_group sub {
640 $grp->result (\@dirs, \@nondirs); 770 $grp->result (\@dirs, \@nondirs);
641 }; 771 };
642 772
643 limit $statgrp $maxreq; 773 limit $statgrp $maxreq;
644 feed $statgrp sub { 774 feed $statgrp sub {
645 return unless @$entries; 775 return unless @$entries;
646 my $entry = pop @$entries; 776 my $entry = pop @$entries;
647 777
648 aioreq_pri $pri; 778 aioreq_pri $pri;
649 add $statgrp aio_stat "$path/$entry/.", sub { 779 add $statgrp aio_stat "$path/$entry/.", sub {
650 if ($_[0] < 0) { 780 if ($_[0] < 0) {
651 push @nondirs, $entry; 781 push @nondirs, $entry;
652 } else { 782 } else {
653 # need to check for real directory 783 # need to check for real directory
654 aioreq_pri $pri; 784 aioreq_pri $pri;
655 add $statgrp aio_lstat "$path/$entry", sub { 785 add $statgrp aio_lstat "$path/$entry", sub {
656 if (-d _) { 786 if (-d _) {
657 push @dirs, $entry; 787 push @dirs, $entry;
658 788
659 unless (--$ndirs) { 789 unless (--$ndirs) {
660 push @nondirs, @$entries; 790 push @nondirs, @$entries;
661 feed $statgrp; 791 feed $statgrp;
792 }
793 } else {
794 push @nondirs, $entry;
662 } 795 }
663 } else {
664 push @nondirs, $entry;
665 } 796 }
666 } 797 }
667 } 798 };
668 }; 799 };
669 }; 800 };
670 }; 801 };
671 }; 802 };
803
804 $grp
672 }; 805 }
806}
673 807
808=item aio_rmtree $path, $callback->($status)
809
810Delete a directory tree starting (and including) C<$path>, return the
811status of the final C<rmdir> only. This is a composite request that
812uses C<aio_scandir> to recurse into and rmdir directories, and unlink
813everything else.
814
815=cut
816
817sub aio_rmtree;
818sub aio_rmtree($;$) {
819 aio_block {
820 my ($path, $cb) = @_;
821
822 my $pri = aioreq_pri;
823 my $grp = aio_group $cb;
824
825 aioreq_pri $pri;
826 add $grp aio_scandir $path, 0, sub {
827 my ($dirs, $nondirs) = @_;
828
829 my $dirgrp = aio_group sub {
830 add $grp aio_rmdir $path, sub {
831 $grp->result ($_[0]);
832 };
833 };
834
835 (aioreq_pri $pri), add $dirgrp aio_rmtree "$path/$_" for @$dirs;
836 (aioreq_pri $pri), add $dirgrp aio_unlink "$path/$_" for @$nondirs;
837
838 add $grp $dirgrp;
839 };
840
674 $grp 841 $grp
842 }
675} 843}
676 844
677=item aio_fsync $fh, $callback->($status) 845=item aio_fsync $fh, $callback->($status)
678 846
679Asynchronously call fsync on the given filehandle and call the callback 847Asynchronously call fsync on the given filehandle and call the callback
933that are being processed by C<IO::AIO::poll_cb> in one call, respectively 1101that are being processed by C<IO::AIO::poll_cb> in one call, respectively
934the maximum amount of time (default C<0>, meaning infinity) spent in 1102the maximum amount of time (default C<0>, meaning infinity) spent in
935C<IO::AIO::poll_cb> to process requests (more correctly the mininum amount 1103C<IO::AIO::poll_cb> to process requests (more correctly the mininum amount
936of time C<poll_cb> is allowed to use). 1104of time C<poll_cb> is allowed to use).
937 1105
1106Setting C<max_poll_time> to a non-zero value creates an overhead of one
1107syscall per request processed, which is not normally a problem unless your
1108callbacks are really really fast or your OS is really really slow (I am
1109not mentioning Solaris here). Using C<max_poll_reqs> incurs no overhead.
1110
938Setting these is useful if you want to ensure some level of 1111Setting these is useful if you want to ensure some level of
939interactiveness when perl is not fast enough to process all requests in 1112interactiveness when perl is not fast enough to process all requests in
940time. 1113time.
941 1114
942For interactive programs, values such as C<0.01> to C<0.1> should be fine. 1115For interactive programs, values such as C<0.01> to C<0.1> should be fine.
943 1116
944Example: Install an Event watcher that automatically calls 1117Example: Install an Event watcher that automatically calls
945IO::AIO::poll_some with low priority, to ensure that other parts of the 1118IO::AIO::poll_cb with low priority, to ensure that other parts of the
946program get the CPU sometimes even under high AIO load. 1119program get the CPU sometimes even under high AIO load.
947 1120
948 # try not to spend much more than 0.1s in poll_cb 1121 # try not to spend much more than 0.1s in poll_cb
949 IO::AIO::max_poll_time 0.1; 1122 IO::AIO::max_poll_time 0.1;
950 1123
953 poll => 'r', nice => 1, 1126 poll => 'r', nice => 1,
954 cb => &IO::AIO::poll_cb); 1127 cb => &IO::AIO::poll_cb);
955 1128
956=item IO::AIO::poll_wait 1129=item IO::AIO::poll_wait
957 1130
1131If there are any outstanding requests and none of them in the result
958Wait till the result filehandle becomes ready for reading (simply does a 1132phase, wait till the result filehandle becomes ready for reading (simply
959C<select> on the filehandle. This is useful if you want to synchronously 1133does a C<select> on the filehandle. This is useful if you want to
960wait for some requests to finish). 1134synchronously wait for some requests to finish).
961 1135
962See C<nreqs> for an example. 1136See C<nreqs> for an example.
963 1137
964=item IO::AIO::poll 1138=item IO::AIO::poll
965 1139
966Waits until some requests have been handled. 1140Waits until some requests have been handled.
967 1141
1142Returns the number of requests processed, but is otherwise strictly
968Strictly equivalent to: 1143equivalent to:
969 1144
970 IO::AIO::poll_wait, IO::AIO::poll_cb 1145 IO::AIO::poll_wait, IO::AIO::poll_cb
971 if IO::AIO::nreqs;
972 1146
973=item IO::AIO::flush 1147=item IO::AIO::flush
974 1148
975Wait till all outstanding AIO requests have been handled. 1149Wait till all outstanding AIO requests have been handled.
976 1150
977Strictly equivalent to: 1151Strictly equivalent to:
978 1152
979 IO::AIO::poll_wait, IO::AIO::poll_cb 1153 IO::AIO::poll_wait, IO::AIO::poll_cb
980 while IO::AIO::nreqs; 1154 while IO::AIO::nreqs;
981 1155
1156=back
1157
982=head3 CONTROLLING THE NUMBER OF THREADS 1158=head3 CONTROLLING THE NUMBER OF THREADS
1159
1160=over
983 1161
984=item IO::AIO::min_parallel $nthreads 1162=item IO::AIO::min_parallel $nthreads
985 1163
986Set the minimum number of AIO threads to C<$nthreads>. The current 1164Set the minimum number of AIO threads to C<$nthreads>. The current
987default is C<8>, which means eight asynchronous operations can execute 1165default is C<8>, which means eight asynchronous operations can execute
1035This is a very bad function to use in interactive programs because it 1213This is a very bad function to use in interactive programs because it
1036blocks, and a bad way to reduce concurrency because it is inexact: Better 1214blocks, and a bad way to reduce concurrency because it is inexact: Better
1037use an C<aio_group> together with a feed callback. 1215use an C<aio_group> together with a feed callback.
1038 1216
1039Sets the maximum number of outstanding requests to C<$nreqs>. If you 1217Sets the maximum number of outstanding requests to C<$nreqs>. If you
1040to queue up more than this number of requests, the next call to the 1218do queue up more than this number of requests, the next call to the
1041C<poll_cb> (and C<poll_some> and other functions calling C<poll_cb>) 1219C<poll_cb> (and C<poll_some> and other functions calling C<poll_cb>)
1042function will block until the limit is no longer exceeded. 1220function will block until the limit is no longer exceeded.
1043 1221
1044The default value is very large, so there is no practical limit on the 1222The default value is very large, so there is no practical limit on the
1045number of outstanding requests. 1223number of outstanding requests.
1046 1224
1047You can still queue as many requests as you want. Therefore, 1225You can still queue as many requests as you want. Therefore,
1048C<max_oustsanding> is mainly useful in simple scripts (with low values) or 1226C<max_oustsanding> is mainly useful in simple scripts (with low values) or
1049as a stop gap to shield against fatal memory overflow (with large values). 1227as a stop gap to shield against fatal memory overflow (with large values).
1050 1228
1229=back
1230
1051=head3 STATISTICAL INFORMATION 1231=head3 STATISTICAL INFORMATION
1232
1233=over
1052 1234
1053=item IO::AIO::nreqs 1235=item IO::AIO::nreqs
1054 1236
1055Returns the number of requests currently in the ready, execute or pending 1237Returns the number of requests currently in the ready, execute or pending
1056states (i.e. for which their callback has not been invoked yet). 1238states (i.e. for which their callback has not been invoked yet).
1072 1254
1073=back 1255=back
1074 1256
1075=cut 1257=cut
1076 1258
1077# support function to convert a fd into a perl filehandle
1078sub _fd2fh {
1079 return undef if $_[0] < 0;
1080
1081 # try to generate nice filehandles
1082 my $sym = "IO::AIO::fd#$_[0]";
1083 local *$sym;
1084
1085 open *$sym, "+<&=$_[0]" # usually works under any unix
1086 or open *$sym, "<&=$_[0]" # cygwin needs this
1087 or open *$sym, ">&=$_[0]" # or this
1088 or return undef;
1089
1090 *$sym
1091}
1092
1093min_parallel 8; 1259min_parallel 8;
1094 1260
1095END { 1261END { flush }
1096 min_parallel 1;
1097 flush;
1098};
1099 1262
11001; 12631;
1101 1264
1102=head2 FORK BEHAVIOUR 1265=head2 FORK BEHAVIOUR
1103 1266
1123bytes of memory. In addition, stat requests need a stat buffer (possibly 1286bytes of memory. In addition, stat requests need a stat buffer (possibly
1124a few hundred bytes), readdir requires a result buffer and so on. Perl 1287a few hundred bytes), readdir requires a result buffer and so on. Perl
1125scalars and other data passed into aio requests will also be locked and 1288scalars and other data passed into aio requests will also be locked and
1126will consume memory till the request has entered the done state. 1289will consume memory till the request has entered the done state.
1127 1290
1128This is now awfully much, so queuing lots of requests is not usually a 1291This is not awfully much, so queuing lots of requests is not usually a
1129problem. 1292problem.
1130 1293
1131Per-thread usage: 1294Per-thread usage:
1132 1295
1133In the execution phase, some aio requests require more memory for 1296In the execution phase, some aio requests require more memory for

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines