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.50 by root, Sat Jun 24 16:27:02 2006 UTC vs.
Revision 1.86 by root, Sat Oct 28 23:32:29 2006 UTC

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

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines