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

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines