ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/IO-AIO/AIO.pm
Revision: 1.91
Committed: Mon Oct 30 23:30:29 2006 UTC (17 years, 6 months ago) by root
Branch: MAIN
CVS Tags: rel-2_2
Changes since 1.90: +1 -1 lines
Log Message:
*** empty log message ***

File Contents

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