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.40 by root, Tue Aug 30 15:45:10 2005 UTC vs.
Revision 1.105 by root, Sun Mar 25 00:20:27 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
16 aio_read $fh, 30000, 1024, $buffer, 0, sub { 17 aio_read $fh, 30000, 1024, $buffer, 0, sub {
17 $_[0] > 0 or die "read error: $!"; 18 $_[0] > 0 or die "read error: $!";
18 }; 19 };
19 20
20 # Event 21 # version 2+ has request and group objects
22 use IO::AIO 2;
23
24 aioreq_pri 4; # give next request a very high priority
25 my $req = aio_unlink "/tmp/file", sub { };
26 $req->cancel; # cancel request if still in queue
27
28 my $grp = aio_group sub { print "all stats done\n" };
29 add $grp aio_stat "..." for ...;
30
31 # AnyEvent integration
32 open my $fh, "<&=" . IO::AIO::poll_fileno or die "$!";
33 my $w = AnyEvent->io (fh => $fh, poll => 'r', cb => sub { IO::AIO::poll_cb });
34
35 # Event integration
21 Event->io (fd => IO::AIO::poll_fileno, 36 Event->io (fd => IO::AIO::poll_fileno,
22 poll => 'r', 37 poll => 'r',
23 cb => \&IO::AIO::poll_cb); 38 cb => \&IO::AIO::poll_cb);
24 39
25 # Glib/Gtk2 40 # Glib/Gtk2 integration
26 add_watch Glib::IO IO::AIO::poll_fileno, 41 add_watch Glib::IO IO::AIO::poll_fileno,
27 in => sub { IO::AIO::poll_cb; 1 }; 42 in => sub { IO::AIO::poll_cb; 1 };
28 43
29 # Tk 44 # Tk integration
30 Tk::Event::IO->fileevent (IO::AIO::poll_fileno, "", 45 Tk::Event::IO->fileevent (IO::AIO::poll_fileno, "",
31 readable => \&IO::AIO::poll_cb); 46 readable => \&IO::AIO::poll_cb);
32 47
33 # Danga::Socket 48 # Danga::Socket integration
34 Danga::Socket->AddOtherFds (IO::AIO::poll_fileno => 49 Danga::Socket->AddOtherFds (IO::AIO::poll_fileno =>
35 \&IO::AIO::poll_cb); 50 \&IO::AIO::poll_cb);
36 51
37
38=head1 DESCRIPTION 52=head1 DESCRIPTION
39 53
40This module implements asynchronous I/O using whatever means your 54This module implements asynchronous I/O using whatever means your
41operating system supports. 55operating system supports.
42 56
57Asynchronous means that operations that can normally block your program
58(e.g. reading from disk) will be done asynchronously: the operation
59will still block, but you can do something else in the meantime. This
60is extremely useful for programs that need to stay interactive even
61when doing heavy I/O (GUI programs, high performance network servers
62etc.), but can also be used to easily do operations in parallel that are
63normally done sequentially, e.g. stat'ing many files, which is much faster
64on a RAID volume or over NFS when you do a number of stat operations
65concurrently.
66
67While most of this works on all types of file descriptors (for example
68sockets), using these functions on file descriptors that support
69nonblocking operation (again, sockets, pipes etc.) is very inefficient or
70might not work (aio_read fails on sockets/pipes/fifos). Use an event loop
71for that (such as the L<Event|Event> module): IO::AIO will naturally fit
72into such an event loop itself.
73
43Currently, a number of threads are started that execute your read/writes 74In this version, a number of threads are started that execute your
44and signal their completion. You don't need thread support in your libc or 75requests and signal their completion. You don't need thread support
45perl, and the threads created by this module will not be visible to the 76in perl, and the threads created by this module will not be visible
46pthreads library. In the future, this module might make use of the native 77to perl. In the future, this module might make use of the native aio
47aio functions available on many operating systems. However, they are often 78functions available on many operating systems. However, they are often
48not well-supported (Linux doesn't allow them on normal files currently, 79not well-supported or restricted (GNU/Linux doesn't allow them on normal
49for example), and they would only support aio_read and aio_write, so the 80files currently, for example), and they would only support aio_read and
50remaining functionality would have to be implemented using threads anyway. 81aio_write, so the remaining functionality would have to be implemented
82using threads anyway.
51 83
52Although the module will work with in the presence of other threads, it is 84Although the module will work with in the presence of other (Perl-)
53currently not reentrant, so use appropriate locking yourself, always call 85threads, it is currently not reentrant in any way, so use appropriate
54C<poll_cb> from within the same thread, or never call C<poll_cb> (or other 86locking yourself, always call C<poll_cb> from within the same thread, or
55C<aio_> functions) recursively. 87never call C<poll_cb> (or other C<aio_> functions) recursively.
88
89=head2 EXAMPLE
90
91This is a simple example that uses the Event module and loads
92F</etc/passwd> asynchronously:
93
94 use Fcntl;
95 use Event;
96 use IO::AIO;
97
98 # register the IO::AIO callback with Event
99 Event->io (fd => IO::AIO::poll_fileno,
100 poll => 'r',
101 cb => \&IO::AIO::poll_cb);
102
103 # queue the request to open /etc/passwd
104 aio_open "/etc/passwd", O_RDONLY, 0, sub {
105 my $fh = shift
106 or die "error while opening: $!";
107
108 # stat'ing filehandles is generally non-blocking
109 my $size = -s $fh;
110
111 # queue a request to read the file
112 my $contents;
113 aio_read $fh, 0, $size, $contents, 0, sub {
114 $_[0] == $size
115 or die "short read: $!";
116
117 close $fh;
118
119 # file contents now in $contents
120 print $contents;
121
122 # exit event loop and program
123 Event::unloop;
124 };
125 };
126
127 # possibly queue up other requests, or open GUI windows,
128 # check for sockets etc. etc.
129
130 # process events as long as there are some:
131 Event::loop;
132
133=head1 REQUEST ANATOMY AND LIFETIME
134
135Every C<aio_*> function creates a request. which is a C data structure not
136directly visible to Perl.
137
138If called in non-void context, every request function returns a Perl
139object representing the request. In void context, nothing is returned,
140which saves a bit of memory.
141
142The perl object is a fairly standard ref-to-hash object. The hash contents
143are not used by IO::AIO so you are free to store anything you like in it.
144
145During their existance, aio requests travel through the following states,
146in order:
147
148=over 4
149
150=item ready
151
152Immediately after a request is created it is put into the ready state,
153waiting for a thread to execute it.
154
155=item execute
156
157A thread has accepted the request for processing and is currently
158executing it (e.g. blocking in read).
159
160=item pending
161
162The request has been executed and is waiting for result processing.
163
164While request submission and execution is fully asynchronous, result
165processing is not and relies on the perl interpreter calling C<poll_cb>
166(or another function with the same effect).
167
168=item result
169
170The request results are processed synchronously by C<poll_cb>.
171
172The C<poll_cb> function will process all outstanding aio requests by
173calling their callbacks, freeing memory associated with them and managing
174any groups they are contained in.
175
176=item done
177
178Request has reached the end of its lifetime and holds no resources anymore
179(except possibly for the Perl object, but its connection to the actual
180aio request is severed and calling its methods will either do nothing or
181result in a runtime error).
182
183=back
56 184
57=cut 185=cut
58 186
59package IO::AIO; 187package IO::AIO;
60 188
61no warnings; 189no warnings;
190use strict 'vars';
62 191
63use base 'Exporter'; 192use base 'Exporter';
64 193
65use Fcntl ();
66
67BEGIN { 194BEGIN {
68 $VERSION = 1.6; 195 our $VERSION = '2.33';
69 196
70 @EXPORT = qw(aio_sendfile aio_read aio_write aio_open aio_close aio_stat 197 our @AIO_REQ = qw(aio_sendfile aio_read aio_write aio_open aio_close aio_stat
71 aio_lstat aio_unlink aio_rmdir aio_readdir aio_scandir aio_symlink 198 aio_lstat aio_unlink aio_rmdir aio_readdir aio_scandir aio_symlink
72 aio_fsync aio_fdatasync aio_readahead); 199 aio_readlink aio_fsync aio_fdatasync aio_readahead aio_rename aio_link
73 @EXPORT_OK = qw(poll_fileno poll_cb min_parallel max_parallel 200 aio_move aio_copy aio_group aio_nop aio_mknod aio_load aio_rmtree aio_mkdir);
74 max_outstanding nreqs); 201 our @EXPORT = (@AIO_REQ, qw(aioreq_pri aioreq_nice aio_block));
202 our @EXPORT_OK = qw(poll_fileno poll_cb poll_wait flush
203 min_parallel max_parallel max_idle
204 nreqs nready npending nthreads
205 max_poll_time max_poll_reqs);
206
207 @IO::AIO::GRP::ISA = 'IO::AIO::REQ';
75 208
76 require XSLoader; 209 require XSLoader;
77 XSLoader::load IO::AIO, $VERSION; 210 XSLoader::load ("IO::AIO", $VERSION);
78} 211}
79 212
80=head1 FUNCTIONS 213=head1 FUNCTIONS
81 214
82=head2 AIO FUNCTIONS 215=head2 AIO REQUEST FUNCTIONS
83 216
84All 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
85with 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,
86and they all accept an additional (and optional) C<$callback> argument 219and they all accept an additional (and optional) C<$callback> argument
87which 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
90syscall has been executed asynchronously. 223syscall has been executed asynchronously.
91 224
92All functions expecting a filehandle keep a copy of the filehandle 225All functions expecting a filehandle keep a copy of the filehandle
93internally until the request has finished. 226internally until the request has finished.
94 227
228All functions return request objects of type L<IO::AIO::REQ> that allow
229further manipulation of those requests while they are in-flight.
230
95The pathnames you pass to these routines I<must> be absolute and 231The pathnames you pass to these routines I<must> be absolute and
96encoded 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
97request is being executed, the current working directory could have 233request is being executed, the current working directory could have
98changed. Alternatively, you can make sure that you never change the 234changed. Alternatively, you can make sure that you never change the
99current working directory. 235current working directory anywhere in the program and then use relative
236paths.
100 237
101To encode pathnames to byte form, either make sure you either: a) 238To encode pathnames as octets, either make sure you either: a) always pass
102always pass in filenames you got from outside (command line, readdir 239in filenames you got from outside (command line, readdir etc.) without
103etc.), 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
104your 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
105environment, d) use Glib::filename_from_unicode on unicode filenames or e) 242environment, d) use Glib::filename_from_unicode on unicode filenames or e)
106use 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.
107 247
108=over 4 248=over 4
249
250=item $prev_pri = aioreq_pri [$pri]
251
252Returns the priority value that would be used for the next request and, if
253C<$pri> is given, sets the priority for the next aio request.
254
255The default priority is C<0>, the minimum and maximum priorities are C<-4>
256and C<4>, respectively. Requests with higher priority will be serviced
257first.
258
259The priority will be reset to C<0> after each call to one of the C<aio_*>
260functions.
261
262Example: open a file with low priority, then read something from it with
263higher priority so the read request is serviced before other low priority
264open requests (potentially spamming the cache):
265
266 aioreq_pri -3;
267 aio_open ..., sub {
268 return unless $_[0];
269
270 aioreq_pri -2;
271 aio_read $_[0], ..., sub {
272 ...
273 };
274 };
275
276=item aioreq_nice $pri_adjust
277
278Similar to C<aioreq_pri>, but subtracts the given value from the current
279priority, so the effect is cumulative.
109 280
110=item aio_open $pathname, $flags, $mode, $callback->($fh) 281=item aio_open $pathname, $flags, $mode, $callback->($fh)
111 282
112Asynchronously open or create a file and call the callback with a newly 283Asynchronously open or create a file and call the callback with a newly
113created filehandle for the file. 284created filehandle for the file.
119list. They are the same as used by C<sysopen>. 290list. They are the same as used by C<sysopen>.
120 291
121Likewise, C<$mode> specifies the mode of the newly created file, if it 292Likewise, C<$mode> specifies the mode of the newly created file, if it
122didn't exist and C<O_CREAT> has been given, just like perl's C<sysopen>, 293didn't exist and C<O_CREAT> has been given, just like perl's C<sysopen>,
123except that it is mandatory (i.e. use C<0> if you don't create new files, 294except that it is mandatory (i.e. use C<0> if you don't create new files,
124and C<0666> or C<0777> if you do). 295and C<0666> or C<0777> if you do). Note that the C<$mode> will be modified
296by the umask in effect then the request is being executed, so better never
297change the umask.
125 298
126Example: 299Example:
127 300
128 aio_open "/etc/passwd", O_RDONLY, 0, sub { 301 aio_open "/etc/passwd", O_RDONLY, 0, sub {
129 if ($_[0]) { 302 if ($_[0]) {
228=item aio_unlink $pathname, $callback->($status) 401=item aio_unlink $pathname, $callback->($status)
229 402
230Asynchronously unlink (delete) a file and call the callback with the 403Asynchronously unlink (delete) a file and call the callback with the
231result code. 404result code.
232 405
406=item aio_mknod $path, $mode, $dev, $callback->($status)
407
408[EXPERIMENTAL]
409
410Asynchronously create a device node (or fifo). See mknod(2).
411
412The only (POSIX-) portable way of calling this function is:
413
414 aio_mknod $path, IO::AIO::S_IFIFO | $mode, 0, sub { ...
415
416=item aio_link $srcpath, $dstpath, $callback->($status)
417
418Asynchronously create a new link to the existing object at C<$srcpath> at
419the path C<$dstpath> and call the callback with the result code.
420
421=item aio_symlink $srcpath, $dstpath, $callback->($status)
422
423Asynchronously create a new symbolic link to the existing object at C<$srcpath> at
424the path C<$dstpath> and call the callback with the result code.
425
426=item aio_readlink $path, $callback->($link)
427
428Asynchronously read the symlink specified by C<$path> and pass it to
429the callback. If an error occurs, nothing or undef gets passed to the
430callback.
431
432=item aio_rename $srcpath, $dstpath, $callback->($status)
433
434Asynchronously rename the object at C<$srcpath> to C<$dstpath>, just as
435rename(2) and call the callback with the result code.
436
437=item aio_mkdir $pathname, $mode, $callback->($status)
438
439Asynchronously mkdir (create) a directory and call the callback with
440the result code. C<$mode> will be modified by the umask at the time the
441request is executed, so do not change your umask.
442
233=item aio_rmdir $pathname, $callback->($status) 443=item aio_rmdir $pathname, $callback->($status)
234 444
235Asynchronously rmdir (delete) a directory and call the callback with the 445Asynchronously rmdir (delete) a directory and call the callback with the
236result code. 446result code.
237 447
238=item aio_readdir $pathname $callback->($entries) 448=item aio_readdir $pathname, $callback->($entries)
239 449
240Unlike the POSIX call of the same name, C<aio_readdir> reads an entire 450Unlike the POSIX call of the same name, C<aio_readdir> reads an entire
241directory (i.e. opendir + readdir + closedir). The entries will not be 451directory (i.e. opendir + readdir + closedir). The entries will not be
242sorted, and will B<NOT> include the C<.> and C<..> entries. 452sorted, and will B<NOT> include the C<.> and C<..> entries.
243 453
244The callback a single argument which is either C<undef> or an array-ref 454The callback a single argument which is either C<undef> or an array-ref
245with the filenames. 455with the filenames.
246 456
457=item aio_load $path, $data, $callback->($status)
458
459This is a composite request that tries to fully load the given file into
460memory. Status is the same as with aio_read.
461
462=cut
463
464sub aio_load($$;$) {
465 aio_block {
466 my ($path, undef, $cb) = @_;
467 my $data = \$_[1];
468
469 my $pri = aioreq_pri;
470 my $grp = aio_group $cb;
471
472 aioreq_pri $pri;
473 add $grp aio_open $path, O_RDONLY, 0, sub {
474 my $fh = shift
475 or return $grp->result (-1);
476
477 aioreq_pri $pri;
478 add $grp aio_read $fh, 0, (-s $fh), $$data, 0, sub {
479 $grp->result ($_[0]);
480 };
481 };
482
483 $grp
484 }
485}
486
487=item aio_copy $srcpath, $dstpath, $callback->($status)
488
489Try to copy the I<file> (directories not supported as either source or
490destination) from C<$srcpath> to C<$dstpath> and call the callback with
491the C<0> (error) or C<-1> ok.
492
493This is a composite request that it creates the destination file with
494mode 0200 and copies the contents of the source file into it using
495C<aio_sendfile>, followed by restoring atime, mtime, access mode and
496uid/gid, in that order.
497
498If an error occurs, the partial destination file will be unlinked, if
499possible, except when setting atime, mtime, access mode and uid/gid, where
500errors are being ignored.
501
502=cut
503
504sub aio_copy($$;$) {
505 aio_block {
506 my ($src, $dst, $cb) = @_;
507
508 my $pri = aioreq_pri;
509 my $grp = aio_group $cb;
510
511 aioreq_pri $pri;
512 add $grp aio_open $src, O_RDONLY, 0, sub {
513 if (my $src_fh = $_[0]) {
514 my @stat = stat $src_fh;
515
516 aioreq_pri $pri;
517 add $grp aio_open $dst, O_CREAT | O_WRONLY | O_TRUNC, 0200, sub {
518 if (my $dst_fh = $_[0]) {
519 aioreq_pri $pri;
520 add $grp aio_sendfile $dst_fh, $src_fh, 0, $stat[7], sub {
521 if ($_[0] == $stat[7]) {
522 $grp->result (0);
523 close $src_fh;
524
525 # those should not normally block. should. should.
526 utime $stat[8], $stat[9], $dst;
527 chmod $stat[2] & 07777, $dst_fh;
528 chown $stat[4], $stat[5], $dst_fh;
529 close $dst_fh;
530 } else {
531 $grp->result (-1);
532 close $src_fh;
533 close $dst_fh;
534
535 aioreq $pri;
536 add $grp aio_unlink $dst;
537 }
538 };
539 } else {
540 $grp->result (-1);
541 }
542 },
543
544 } else {
545 $grp->result (-1);
546 }
547 };
548
549 $grp
550 }
551}
552
553=item aio_move $srcpath, $dstpath, $callback->($status)
554
555Try to move the I<file> (directories not supported as either source or
556destination) from C<$srcpath> to C<$dstpath> and call the callback with
557the C<0> (error) or C<-1> ok.
558
559This is a composite request that tries to rename(2) the file first. If
560rename files with C<EXDEV>, it copies the file with C<aio_copy> and, if
561that is successful, unlinking the C<$srcpath>.
562
563=cut
564
565sub aio_move($$;$) {
566 aio_block {
567 my ($src, $dst, $cb) = @_;
568
569 my $pri = aioreq_pri;
570 my $grp = aio_group $cb;
571
572 aioreq_pri $pri;
573 add $grp aio_rename $src, $dst, sub {
574 if ($_[0] && $! == EXDEV) {
575 aioreq_pri $pri;
576 add $grp aio_copy $src, $dst, sub {
577 $grp->result ($_[0]);
578
579 if (!$_[0]) {
580 aioreq_pri $pri;
581 add $grp aio_unlink $src;
582 }
583 };
584 } else {
585 $grp->result ($_[0]);
586 }
587 };
588
589 $grp
590 }
591}
592
247=item aio_scandir $path, $maxreq, $callback->($dirs, $nondirs) 593=item aio_scandir $path, $maxreq, $callback->($dirs, $nondirs)
248 594
249Scans a directory (similar to C<aio_readdir>) and tries to separate the 595Scans a directory (similar to C<aio_readdir>) but additionally tries to
250entries of directory C<$path> into two sets of names, ones you can recurse 596efficiently separate the entries of directory C<$path> into two sets of
251into (directories), and ones you cannot recurse into (everything else). 597names, directories you can recurse into (directories), and ones you cannot
598recurse into (everything else, including symlinks to directories).
252 599
253C<aio_scandir> is a composite request that consists of many 600C<aio_scandir> is a composite request that creates of many sub requests_
254aio-primitives. C<$maxreq> specifies the maximum number of outstanding 601C<$maxreq> specifies the maximum number of outstanding aio requests that
255aio requests that this function generates. If it is C<< <= 0 >>, then a 602this function generates. If it is C<< <= 0 >>, then a suitable default
256suitable default will be chosen (currently 8). 603will be chosen (currently 4).
257 604
258On error, the callback is called without arguments, otherwise it receives 605On error, the callback is called without arguments, otherwise it receives
259two array-refs with path-relative entry names. 606two array-refs with path-relative entry names.
260 607
261Example: 608Example:
269Implementation notes. 616Implementation notes.
270 617
271The C<aio_readdir> cannot be avoided, but C<stat()>'ing every entry can. 618The C<aio_readdir> cannot be avoided, but C<stat()>'ing every entry can.
272 619
273After reading the directory, the modification time, size etc. of the 620After reading the directory, the modification time, size etc. of the
274directory before and after the readdir is checked, and if they match, the 621directory before and after the readdir is checked, and if they match (and
275link count will be used to decide how many entries are directories (if 622isn't the current time), the link count will be used to decide how many
276>= 2). Otherwise, no knowledge of the number of subdirectories will be 623entries are directories (if >= 2). Otherwise, no knowledge of the number
277assumed. 624of subdirectories will be assumed.
278 625
279Then entires will be sorted into likely directories (everything without a 626Then entries will be sorted into likely directories (everything without
280non-initial dot) and likely non-directories (everything else). Then every 627a non-initial dot currently) and likely non-directories (everything
281entry + C</.> will be C<stat>'ed, likely directories first. This is often 628else). Then every entry plus an appended C</.> will be C<stat>'ed,
629likely directories first. If that succeeds, it assumes that the entry
630is a directory or a symlink to directory (which will be checked
631seperately). This is often faster than stat'ing the entry itself because
282faster because filesystems might detect the type of the entry without 632filesystems might detect the type of the entry without reading the inode
283reading the inode data (e.g. ext2s filetype feature). If that succeeds, 633data (e.g. ext2fs filetype feature).
284it assumes that the entry is a directory or a symlink to directory (which
285will be checked seperately).
286 634
287If the known number of directories has been reached, the rest of the 635If the known number of directories (link count - 2) has been reached, the
288entries is assumed to be non-directories. 636rest of the entries is assumed to be non-directories.
637
638This only works with certainty on POSIX (= UNIX) filesystems, which
639fortunately are the vast majority of filesystems around.
640
641It will also likely work on non-POSIX filesystems with reduced efficiency
642as those tend to return 0 or 1 as link counts, which disables the
643directory counting heuristic.
289 644
290=cut 645=cut
291 646
292sub aio_scandir($$$) { 647sub aio_scandir($$;$) {
648 aio_block {
293 my ($path, $maxreq, $cb) = @_; 649 my ($path, $maxreq, $cb) = @_;
294 650
651 my $pri = aioreq_pri;
652
653 my $grp = aio_group $cb;
654
295 $maxreq = 8 if $maxreq <= 0; 655 $maxreq = 4 if $maxreq <= 0;
296 656
297 # stat once 657 # stat once
658 aioreq_pri $pri;
298 aio_stat $path, sub { 659 add $grp aio_stat $path, sub {
299 $cb->() if $_[0]; 660 return $grp->result () if $_[0];
661 my $now = time;
300 my $hash1 = join ":", (stat _)[0,1,3,7,9]; 662 my $hash1 = join ":", (stat _)[0,1,3,7,9];
301 663
302 # read the directory entries 664 # read the directory entries
665 aioreq_pri $pri;
303 aio_readdir $path, sub { 666 add $grp aio_readdir $path, sub {
304 my $entries = shift 667 my $entries = shift
305 or return $cb->(); 668 or return $grp->result ();
306 669
307 # stat the dir another time 670 # stat the dir another time
671 aioreq_pri $pri;
308 aio_stat $path, sub { 672 add $grp aio_stat $path, sub {
309 my $hash2 = join ":", (stat _)[0,1,3,7,9]; 673 my $hash2 = join ":", (stat _)[0,1,3,7,9];
310 674
311 my $ndirs; 675 my $ndirs;
312 676
313 # take the slow route if anything looks fishy 677 # take the slow route if anything looks fishy
314 if ($hash1 ne $hash2) { 678 if ($hash1 ne $hash2 or (stat _)[9] == $now) {
315 $ndirs = -1; 679 $ndirs = -1;
316 } else { 680 } else {
317 # if nlink == 2, we are finished 681 # if nlink == 2, we are finished
318 # on non-posix-fs's, we rely on nlink < 2 682 # on non-posix-fs's, we rely on nlink < 2
319 $ndirs = (stat _)[3] - 2 683 $ndirs = (stat _)[3] - 2
320 or $cb->([], $entries); 684 or return $grp->result ([], $entries);
321 }
322
323 # sort into likely dirs and likely nondirs
324 # dirs == files without ".", short entries first
325 $entries = [map $_->[0],
326 sort { $b->[1] cmp $a->[1] }
327 map [$_, sprintf "%s%04d", (/.\./ ? "1" : "0"), length],
328 @$entries];
329
330 my (@dirs, @nondirs);
331
332 my ($statcb, $schedcb);
333 my $nreq = 0;
334
335 $schedcb = sub {
336 if (@$entries) {
337 if ($nreq < $maxreq) {
338 my $ent = pop @$entries;
339 $nreq++;
340 aio_stat "$path/$ent/.", sub { $statcb->($_[0], $ent) };
341 }
342 } elsif (!$nreq) {
343 # finished
344 undef $statcb;
345 undef $schedcb;
346 $cb->(\@dirs, \@nondirs);
347 undef $cb;
348 } 685 }
686
687 # sort into likely dirs and likely nondirs
688 # dirs == files without ".", short entries first
689 $entries = [map $_->[0],
690 sort { $b->[1] cmp $a->[1] }
691 map [$_, sprintf "%s%04d", (/.\./ ? "1" : "0"), length],
692 @$entries];
693
694 my (@dirs, @nondirs);
695
696 my $statgrp = add $grp aio_group sub {
697 $grp->result (\@dirs, \@nondirs);
698 };
699
700 limit $statgrp $maxreq;
701 feed $statgrp sub {
702 return unless @$entries;
703 my $entry = pop @$entries;
704
705 aioreq_pri $pri;
706 add $statgrp aio_stat "$path/$entry/.", sub {
707 if ($_[0] < 0) {
708 push @nondirs, $entry;
709 } else {
710 # need to check for real directory
711 aioreq_pri $pri;
712 add $statgrp aio_lstat "$path/$entry", sub {
713 if (-d _) {
714 push @dirs, $entry;
715
716 unless (--$ndirs) {
717 push @nondirs, @$entries;
718 feed $statgrp;
719 }
720 } else {
721 push @nondirs, $entry;
722 }
723 }
724 }
725 };
726 };
349 }; 727 };
350 $statcb = sub {
351 my ($status, $entry) = @_;
352
353 if ($status < 0) {
354 $nreq--;
355 push @nondirs, $entry;
356 &$schedcb;
357 } else {
358 # need to check for real directory
359 aio_lstat "$path/$entry", sub {
360 $nreq--;
361
362 if (-d _) {
363 push @dirs, $entry;
364
365 if (!--$ndirs) {
366 push @nondirs, @$entries;
367 $entries = [];
368 }
369 } else {
370 push @nondirs, $entry;
371 }
372
373 &$schedcb;
374 }
375 }
376 };
377
378 &$schedcb while @$entries && $nreq < $maxreq;
379 }; 728 };
380 }; 729 };
730
731 $grp
381 }; 732 }
733}
734
735=item aio_rmtree $path, $callback->($status)
736
737Delete a directory tree starting (and including) C<$path>, return the
738status of the final C<rmdir> only. This is a composite request that
739uses C<aio_scandir> to recurse into and rmdir directories, and unlink
740everything else.
741
742=cut
743
744sub aio_rmtree;
745sub aio_rmtree($;$) {
746 aio_block {
747 my ($path, $cb) = @_;
748
749 my $pri = aioreq_pri;
750 my $grp = aio_group $cb;
751
752 aioreq_pri $pri;
753 add $grp aio_scandir $path, 0, sub {
754 my ($dirs, $nondirs) = @_;
755
756 my $dirgrp = aio_group sub {
757 add $grp aio_rmdir $path, sub {
758 $grp->result ($_[0]);
759 };
760 };
761
762 (aioreq_pri $pri), add $dirgrp aio_rmtree "$path/$_" for @$dirs;
763 (aioreq_pri $pri), add $dirgrp aio_unlink "$path/$_" for @$nondirs;
764
765 add $grp $dirgrp;
766 };
767
768 $grp
769 }
382} 770}
383 771
384=item aio_fsync $fh, $callback->($status) 772=item aio_fsync $fh, $callback->($status)
385 773
386Asynchronously call fsync on the given filehandle and call the callback 774Asynchronously call fsync on the given filehandle and call the callback
392callback with the fdatasync result code. 780callback with the fdatasync result code.
393 781
394If this call isn't available because your OS lacks it or it couldn't be 782If this call isn't available because your OS lacks it or it couldn't be
395detected, it will be emulated by calling C<fsync> instead. 783detected, it will be emulated by calling C<fsync> instead.
396 784
785=item aio_group $callback->(...)
786
787This is a very special aio request: Instead of doing something, it is a
788container for other aio requests, which is useful if you want to bundle
789many requests into a single, composite, request with a definite callback
790and the ability to cancel the whole request with its subrequests.
791
792Returns an object of class L<IO::AIO::GRP>. See its documentation below
793for more info.
794
795Example:
796
797 my $grp = aio_group sub {
798 print "all stats done\n";
799 };
800
801 add $grp
802 (aio_stat ...),
803 (aio_stat ...),
804 ...;
805
806=item aio_nop $callback->()
807
808This is a special request - it does nothing in itself and is only used for
809side effects, such as when you want to add a dummy request to a group so
810that finishing the requests in the group depends on executing the given
811code.
812
813While this request does nothing, it still goes through the execution
814phase and still requires a worker thread. Thus, the callback will not
815be executed immediately but only after other requests in the queue have
816entered their execution phase. This can be used to measure request
817latency.
818
819=item IO::AIO::aio_busy $fractional_seconds, $callback->() *NOT EXPORTED*
820
821Mainly used for debugging and benchmarking, this aio request puts one of
822the request workers to sleep for the given time.
823
824While it is theoretically handy to have simple I/O scheduling requests
825like sleep and file handle readable/writable, the overhead this creates is
826immense (it blocks a thread for a long time) so do not use this function
827except to put your application under artificial I/O pressure.
828
397=back 829=back
398 830
831=head2 IO::AIO::REQ CLASS
832
833All non-aggregate C<aio_*> functions return an object of this class when
834called in non-void context.
835
836=over 4
837
838=item cancel $req
839
840Cancels the request, if possible. Has the effect of skipping execution
841when entering the B<execute> state and skipping calling the callback when
842entering the the B<result> state, but will leave the request otherwise
843untouched. That means that requests that currently execute will not be
844stopped and resources held by the request will not be freed prematurely.
845
846=item cb $req $callback->(...)
847
848Replace (or simply set) the callback registered to the request.
849
850=back
851
852=head2 IO::AIO::GRP CLASS
853
854This class is a subclass of L<IO::AIO::REQ>, so all its methods apply to
855objects of this class, too.
856
857A IO::AIO::GRP object is a special request that can contain multiple other
858aio requests.
859
860You create one by calling the C<aio_group> constructing function with a
861callback that will be called when all contained requests have entered the
862C<done> state:
863
864 my $grp = aio_group sub {
865 print "all requests are done\n";
866 };
867
868You add requests by calling the C<add> method with one or more
869C<IO::AIO::REQ> objects:
870
871 $grp->add (aio_unlink "...");
872
873 add $grp aio_stat "...", sub {
874 $_[0] or return $grp->result ("error");
875
876 # add another request dynamically, if first succeeded
877 add $grp aio_open "...", sub {
878 $grp->result ("ok");
879 };
880 };
881
882This makes it very easy to create composite requests (see the source of
883C<aio_move> for an application) that work and feel like simple requests.
884
885=over 4
886
887=item * The IO::AIO::GRP objects will be cleaned up during calls to
888C<IO::AIO::poll_cb>, just like any other request.
889
890=item * They can be canceled like any other request. Canceling will cancel not
891only the request itself, but also all requests it contains.
892
893=item * They can also can also be added to other IO::AIO::GRP objects.
894
895=item * You must not add requests to a group from within the group callback (or
896any later time).
897
898=back
899
900Their lifetime, simplified, looks like this: when they are empty, they
901will finish very quickly. If they contain only requests that are in the
902C<done> state, they will also finish. Otherwise they will continue to
903exist.
904
905That means after creating a group you have some time to add requests. And
906in the callbacks of those requests, you can add further requests to the
907group. And only when all those requests have finished will the the group
908itself finish.
909
910=over 4
911
912=item add $grp ...
913
914=item $grp->add (...)
915
916Add one or more requests to the group. Any type of L<IO::AIO::REQ> can
917be added, including other groups, as long as you do not create circular
918dependencies.
919
920Returns all its arguments.
921
922=item $grp->cancel_subs
923
924Cancel all subrequests and clears any feeder, but not the group request
925itself. Useful when you queued a lot of events but got a result early.
926
927=item $grp->result (...)
928
929Set the result value(s) that will be passed to the group callback when all
930subrequests have finished and set thre groups errno to the current value
931of errno (just like calling C<errno> without an error number). By default,
932no argument will be passed and errno is zero.
933
934=item $grp->errno ([$errno])
935
936Sets the group errno value to C<$errno>, or the current value of errno
937when the argument is missing.
938
939Every aio request has an associated errno value that is restored when
940the callback is invoked. This method lets you change this value from its
941default (0).
942
943Calling C<result> will also set errno, so make sure you either set C<$!>
944before the call to C<result>, or call c<errno> after it.
945
946=item feed $grp $callback->($grp)
947
948Sets a feeder/generator on this group: every group can have an attached
949generator that generates requests if idle. The idea behind this is that,
950although you could just queue as many requests as you want in a group,
951this might starve other requests for a potentially long time. For
952example, C<aio_scandir> might generate hundreds of thousands C<aio_stat>
953requests, delaying any later requests for a long time.
954
955To avoid this, and allow incremental generation of requests, you can
956instead a group and set a feeder on it that generates those requests. The
957feed callback will be called whenever there are few enough (see C<limit>,
958below) requests active in the group itself and is expected to queue more
959requests.
960
961The feed callback can queue as many requests as it likes (i.e. C<add> does
962not impose any limits).
963
964If the feed does not queue more requests when called, it will be
965automatically removed from the group.
966
967If the feed limit is C<0>, it will be set to C<2> automatically.
968
969Example:
970
971 # stat all files in @files, but only ever use four aio requests concurrently:
972
973 my $grp = aio_group sub { print "finished\n" };
974 limit $grp 4;
975 feed $grp sub {
976 my $file = pop @files
977 or return;
978
979 add $grp aio_stat $file, sub { ... };
980 };
981
982=item limit $grp $num
983
984Sets the feeder limit for the group: The feeder will be called whenever
985the group contains less than this many requests.
986
987Setting the limit to C<0> will pause the feeding process.
988
989=back
990
399=head2 SUPPORT FUNCTIONS 991=head2 SUPPORT FUNCTIONS
992
993=head3 EVENT PROCESSING AND EVENT LOOP INTEGRATION
400 994
401=over 4 995=over 4
402 996
403=item $fileno = IO::AIO::poll_fileno 997=item $fileno = IO::AIO::poll_fileno
404 998
409 1003
410See C<poll_cb> for an example. 1004See C<poll_cb> for an example.
411 1005
412=item IO::AIO::poll_cb 1006=item IO::AIO::poll_cb
413 1007
414Process all outstanding events on the result pipe. You have to call this 1008Process some outstanding events on the result pipe. You have to call this
415regularly. Returns the number of events processed. Returns immediately 1009regularly. Returns the number of events processed. Returns immediately
416when no events are outstanding. 1010when no events are outstanding. The amount of events processed depends on
1011the settings of C<IO::AIO::max_poll_req> and C<IO::AIO::max_poll_time>.
1012
1013If not all requests were processed for whatever reason, the filehandle
1014will still be ready when C<poll_cb> returns.
417 1015
418Example: Install an Event watcher that automatically calls 1016Example: Install an Event watcher that automatically calls
419IO::AIO::poll_cb with high priority: 1017IO::AIO::poll_cb with high priority:
420 1018
421 Event->io (fd => IO::AIO::poll_fileno, 1019 Event->io (fd => IO::AIO::poll_fileno,
422 poll => 'r', async => 1, 1020 poll => 'r', async => 1,
423 cb => \&IO::AIO::poll_cb); 1021 cb => \&IO::AIO::poll_cb);
424 1022
1023=item IO::AIO::max_poll_reqs $nreqs
1024
1025=item IO::AIO::max_poll_time $seconds
1026
1027These set the maximum number of requests (default C<0>, meaning infinity)
1028that are being processed by C<IO::AIO::poll_cb> in one call, respectively
1029the maximum amount of time (default C<0>, meaning infinity) spent in
1030C<IO::AIO::poll_cb> to process requests (more correctly the mininum amount
1031of time C<poll_cb> is allowed to use).
1032
1033Setting C<max_poll_time> to a non-zero value creates an overhead of one
1034syscall per request processed, which is not normally a problem unless your
1035callbacks are really really fast or your OS is really really slow (I am
1036not mentioning Solaris here). Using C<max_poll_reqs> incurs no overhead.
1037
1038Setting these is useful if you want to ensure some level of
1039interactiveness when perl is not fast enough to process all requests in
1040time.
1041
1042For interactive programs, values such as C<0.01> to C<0.1> should be fine.
1043
1044Example: Install an Event watcher that automatically calls
1045IO::AIO::poll_cb with low priority, to ensure that other parts of the
1046program get the CPU sometimes even under high AIO load.
1047
1048 # try not to spend much more than 0.1s in poll_cb
1049 IO::AIO::max_poll_time 0.1;
1050
1051 # use a low priority so other tasks have priority
1052 Event->io (fd => IO::AIO::poll_fileno,
1053 poll => 'r', nice => 1,
1054 cb => &IO::AIO::poll_cb);
1055
425=item IO::AIO::poll_wait 1056=item IO::AIO::poll_wait
426 1057
1058If there are any outstanding requests and none of them in the result
427Wait till the result filehandle becomes ready for reading (simply does a 1059phase, wait till the result filehandle becomes ready for reading (simply
428C<select> on the filehandle. This is useful if you want to synchronously wait 1060does a C<select> on the filehandle. This is useful if you want to
429for some requests to finish). 1061synchronously wait for some requests to finish).
430 1062
431See C<nreqs> for an example. 1063See C<nreqs> for an example.
432 1064
1065=item IO::AIO::poll
1066
1067Waits until some requests have been handled.
1068
1069Returns the number of requests processed, but is otherwise strictly
1070equivalent to:
1071
1072 IO::AIO::poll_wait, IO::AIO::poll_cb
1073
433=item IO::AIO::nreqs 1074=item IO::AIO::flush
434 1075
435Returns the number of requests currently outstanding (i.e. for which their 1076Wait till all outstanding AIO requests have been handled.
436callback has not been invoked yet).
437 1077
438Example: wait till there are no outstanding requests anymore: 1078Strictly equivalent to:
439 1079
440 IO::AIO::poll_wait, IO::AIO::poll_cb 1080 IO::AIO::poll_wait, IO::AIO::poll_cb
441 while IO::AIO::nreqs; 1081 while IO::AIO::nreqs;
442 1082
443=item IO::AIO::flush 1083=back
444 1084
445Wait till all outstanding AIO requests have been handled. 1085=head3 CONTROLLING THE NUMBER OF THREADS
446 1086
447Strictly equivalent to: 1087=over
448
449 IO::AIO::poll_wait, IO::AIO::poll_cb
450 while IO::AIO::nreqs;
451
452=item IO::AIO::poll
453
454Waits until some requests have been handled.
455
456Strictly equivalent to:
457
458 IO::AIO::poll_wait, IO::AIO::poll_cb
459 if IO::AIO::nreqs;
460 1088
461=item IO::AIO::min_parallel $nthreads 1089=item IO::AIO::min_parallel $nthreads
462 1090
463Set the minimum number of AIO threads to C<$nthreads>. The current default 1091Set the minimum number of AIO threads to C<$nthreads>. The current
464is C<4>, which means four asynchronous operations can be done at one time 1092default is C<8>, which means eight asynchronous operations can execute
465(the number of outstanding operations, however, is unlimited). 1093concurrently at any one time (the number of outstanding requests,
1094however, is unlimited).
466 1095
467IO::AIO starts threads only on demand, when an AIO request is queued and 1096IO::AIO starts threads only on demand, when an AIO request is queued and
468no free thread exists. 1097no free thread exists. Please note that queueing up a hundred requests can
1098create demand for a hundred threads, even if it turns out that everything
1099is in the cache and could have been processed faster by a single thread.
469 1100
470It is recommended to keep the number of threads low, as some Linux 1101It is recommended to keep the number of threads relatively low, as some
471kernel versions will scale negatively with the number of threads (higher 1102Linux kernel versions will scale negatively with the number of threads
472parallelity => MUCH higher latency). With current Linux 2.6 versions, 4-32 1103(higher parallelity => MUCH higher latency). With current Linux 2.6
473threads should be fine. 1104versions, 4-32 threads should be fine.
474 1105
475Under most circumstances you don't need to call this function, as the 1106Under most circumstances you don't need to call this function, as the
476module selects a default that is suitable for low to moderate load. 1107module selects a default that is suitable for low to moderate load.
477 1108
478=item IO::AIO::max_parallel $nthreads 1109=item IO::AIO::max_parallel $nthreads
487This module automatically runs C<max_parallel 0> at program end, to ensure 1118This module automatically runs C<max_parallel 0> at program end, to ensure
488that all threads are killed and that there are no outstanding requests. 1119that all threads are killed and that there are no outstanding requests.
489 1120
490Under normal circumstances you don't need to call this function. 1121Under normal circumstances you don't need to call this function.
491 1122
1123=item IO::AIO::max_idle $nthreads
1124
1125Limit the number of threads (default: 4) that are allowed to idle (i.e.,
1126threads that did not get a request to process within 10 seconds). That
1127means if a thread becomes idle while C<$nthreads> other threads are also
1128idle, it will free its resources and exit.
1129
1130This is useful when you allow a large number of threads (e.g. 100 or 1000)
1131to allow for extremely high load situations, but want to free resources
1132under normal circumstances (1000 threads can easily consume 30MB of RAM).
1133
1134The default is probably ok in most situations, especially if thread
1135creation is fast. If thread creation is very slow on your system you might
1136want to use larger values.
1137
492=item $oldnreqs = IO::AIO::max_outstanding $nreqs 1138=item $oldmaxreqs = IO::AIO::max_outstanding $maxreqs
1139
1140This is a very bad function to use in interactive programs because it
1141blocks, and a bad way to reduce concurrency because it is inexact: Better
1142use an C<aio_group> together with a feed callback.
493 1143
494Sets the maximum number of outstanding requests to C<$nreqs>. If you 1144Sets the maximum number of outstanding requests to C<$nreqs>. If you
495try to queue up more than this number of requests, the caller will block until 1145to queue up more than this number of requests, the next call to the
496some requests have been handled. 1146C<poll_cb> (and C<poll_some> and other functions calling C<poll_cb>)
1147function will block until the limit is no longer exceeded.
497 1148
498The default is very large, so normally there is no practical limit. If you 1149The default value is very large, so there is no practical limit on the
499queue up many requests in a loop it often improves speed if you set 1150number of outstanding requests.
500this to a relatively low number, such as C<100>.
501 1151
502Under normal circumstances you don't need to call this function. 1152You can still queue as many requests as you want. Therefore,
1153C<max_oustsanding> is mainly useful in simple scripts (with low values) or
1154as a stop gap to shield against fatal memory overflow (with large values).
1155
1156=back
1157
1158=head3 STATISTICAL INFORMATION
1159
1160=over
1161
1162=item IO::AIO::nreqs
1163
1164Returns the number of requests currently in the ready, execute or pending
1165states (i.e. for which their callback has not been invoked yet).
1166
1167Example: wait till there are no outstanding requests anymore:
1168
1169 IO::AIO::poll_wait, IO::AIO::poll_cb
1170 while IO::AIO::nreqs;
1171
1172=item IO::AIO::nready
1173
1174Returns the number of requests currently in the ready state (not yet
1175executed).
1176
1177=item IO::AIO::npending
1178
1179Returns the number of requests currently in the pending state (executed,
1180but not yet processed by poll_cb).
503 1181
504=back 1182=back
505 1183
506=cut 1184=cut
507 1185
519 or return undef; 1197 or return undef;
520 1198
521 *$sym 1199 *$sym
522} 1200}
523 1201
524min_parallel 4; 1202min_parallel 8;
525 1203
526END { 1204END { flush }
527 max_parallel 0;
528}
529 1205
5301; 12061;
531 1207
532=head2 FORK BEHAVIOUR 1208=head2 FORK BEHAVIOUR
1209
1210This module should do "the right thing" when the process using it forks:
533 1211
534Before the fork, IO::AIO enters a quiescent state where no requests 1212Before the fork, IO::AIO enters a quiescent state where no requests
535can be added in other threads and no results will be processed. After 1213can be added in other threads and no results will be processed. After
536the fork the parent simply leaves the quiescent state and continues 1214the fork the parent simply leaves the quiescent state and continues
537request/result processing, while the child clears the request/result 1215request/result processing, while the child frees the request/result queue
538queue (so the requests started before the fork will only be handled in 1216(so that the requests started before the fork will only be handled in the
539the parent). Threats will be started on demand until the limit ste in the 1217parent). Threads will be started on demand until the limit set in the
540parent process has been reached again. 1218parent process has been reached again.
541 1219
1220In short: the parent will, after a short pause, continue as if fork had
1221not been called, while the child will act as if IO::AIO has not been used
1222yet.
1223
1224=head2 MEMORY USAGE
1225
1226Per-request usage:
1227
1228Each aio request uses - depending on your architecture - around 100-200
1229bytes of memory. In addition, stat requests need a stat buffer (possibly
1230a few hundred bytes), readdir requires a result buffer and so on. Perl
1231scalars and other data passed into aio requests will also be locked and
1232will consume memory till the request has entered the done state.
1233
1234This is now awfully much, so queuing lots of requests is not usually a
1235problem.
1236
1237Per-thread usage:
1238
1239In the execution phase, some aio requests require more memory for
1240temporary buffers, and each thread requires a stack and other data
1241structures (usually around 16k-128k, depending on the OS).
1242
1243=head1 KNOWN BUGS
1244
1245Known bugs will be fixed in the next release.
1246
542=head1 SEE ALSO 1247=head1 SEE ALSO
543 1248
544L<Coro>, L<Linux::AIO>. 1249L<Coro::AIO>.
545 1250
546=head1 AUTHOR 1251=head1 AUTHOR
547 1252
548 Marc Lehmann <schmorp@schmorp.de> 1253 Marc Lehmann <schmorp@schmorp.de>
549 http://home.schmorp.de/ 1254 http://home.schmorp.de/

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines